vtkPLagrangianParticleTracker.h

Go to the documentation of this file.

// SPDX-FileCopyrightText: Copyright (c) Ken Martin, Will Schroeder, Bill Lorensen
// SPDX-License-Identifier: BSD-3-Clause
 
#ifndef vtkPLagrangianParticleTracker_h
#define vtkPLagrangianParticleTracker_h
 
#include "vtkFiltersParallelFlowPathsModule.h" // For export macro
#include "vtkLagrangianParticleTracker.h"
#include "vtkNew.h" // for ivars
 
#include <map> // for std::map
 
VTK_ABI_NAMESPACE_BEGIN
class ParticleFeedManager;
class ParticleIdManager;
class ParticleStreamManager;
class vtkMPIController;
class vtkMultiBlockDataSet;
class vtkUnstructuredGrid;
 
class VTKFILTERSPARALLELFLOWPATHS_EXPORT vtkPLagrangianParticleTracker
  : public vtkLagrangianParticleTracker
{
public:
  vtkTypeMacro(vtkPLagrangianParticleTracker, vtkLagrangianParticleTracker);
  void PrintSelf(ostream& os, vtkIndent indent) override;
  static vtkPLagrangianParticleTracker* New();
 
protected:
  vtkPLagrangianParticleTracker();
  ~vtkPLagrangianParticleTracker() override;
 
  int RequestUpdateExtent(vtkInformation*, vtkInformationVector**, vtkInformationVector*) override;
 
  void GenerateParticles(const vtkBoundingBox* bounds, vtkDataSet* seeds,
    vtkDataArray* initialVelocities, vtkDataArray* initialIntegrationTimes, vtkPointData* seedData,
    int nVar, std::queue<vtkLagrangianParticle*>& particles) override;

  void GetParticleFeed(std::queue<vtkLagrangianParticle*>& particleQueue) override;
  int Integrate(vtkInitialValueProblemSolver* integrator, vtkLagrangianParticle*,
    std::queue<vtkLagrangianParticle*>& particleQueue, vtkPolyData* particlePathsOutput,
    vtkPolyLine* particlePath, vtkDataObject* interactionOutput) override;

  void ReceiveParticles(std::queue<vtkLagrangianParticle*>& particleQueue);

  void ReceiveTransferredParticleIds();
 
  bool FinalizeOutputs(vtkPolyData* particlePathsOutput, vtkDataObject* interactionOutput) override;
 
  bool UpdateSurfaceCacheIfNeeded(vtkDataObject*& surfaces) override;

  vtkIdType GetNewParticleId() override;

  void DeleteParticle(vtkLagrangianParticle* particle) override;

  vtkGetMacro(ParticleCounter, vtkIdType);
 
  void SetController(vtkMPIController*);
 
  vtkNew<vtkUnstructuredGrid> TmpSurfaceInput;
  vtkNew<vtkMultiBlockDataSet> TmpSurfaceInputMB;
  vtkMPIController* Controller;
  ParticleStreamManager* StreamManager;
  ParticleIdManager* TransferredParticleIdManager;
  ParticleFeedManager* FeedManager;
 
  std::mutex StreamManagerMutex;
  std::mutex OutOfDomainParticleMapMutex;
 
  std::map<vtkIdType, vtkLagrangianParticle*> OutOfDomainParticleMap;
 
private:
  vtkPLagrangianParticleTracker(const vtkPLagrangianParticleTracker&) = delete;
  void operator=(const vtkPLagrangianParticleTracker&) = delete;
};
VTK_ABI_NAMESPACE_END
#endif