ttkTrackingFromOverlap.h

Go to the documentation of this file.

 
 
#pragma once
 
// VTK includes
#include <vtkInformation.h>
#include <vtkMultiBlockDataSet.h>
#include <vtkSmartPointer.h>
#include <vtkUnstructuredGridAlgorithm.h>
 
// VTK Module
#include <ttkTrackingFromOverlapModule.h>
 
// TTK includes
#include <TrackingFromOverlap.h>
#include <ttkAlgorithm.h>
 
class TTKTRACKINGFROMOVERLAP_EXPORT ttkTrackingFromOverlap
  : public ttkAlgorithm,
    protected ttk::TrackingFromOverlap {
 
public:
  static ttkTrackingFromOverlap *New();
  vtkTypeMacro(ttkTrackingFromOverlap, ttkAlgorithm);
 
  vtkSetMacro(LabelFieldName, const std::string &);
  vtkGetMacro(LabelFieldName, std::string);
 
protected:
  ttkTrackingFromOverlap() {
    SetNumberOfInputPorts(1);
    SetNumberOfOutputPorts(1);
  }
 
  int reset();
 
  int packInputData(vtkDataObject *inputDataObject,
                    vtkMultiBlockDataSet *packedData) const;
  int packStreamedData(vtkMultiBlockDataSet *streamedData,
                       vtkMultiBlockDataSet *packedData) const;
  int checkData(vtkMultiBlockDataSet *data);
 
  int storeStreamedData(vtkMultiBlockDataSet *data);
  int computeNodes(vtkMultiBlockDataSet *data);
  int computeTrackingGraphs(vtkMultiBlockDataSet *data);
  int computeNestingTrees(vtkMultiBlockDataSet *data);
  int computeBranches();
 
  int meshNestedTrackingGraph(vtkDataObject *trackingGraph);
 
  int RequestData(vtkInformation *request,
                  vtkInformationVector **inputVector,
                  vtkInformationVector *outputVector) override;
 
  int FillInputPortInformation(int port, vtkInformation *info) override {
    switch(port) {
      case 0:
        info->Remove(vtkAlgorithm::INPUT_REQUIRED_DATA_TYPE());
        info->Append(
          vtkAlgorithm::INPUT_REQUIRED_DATA_TYPE(), "vtkMultiBlockDataSet");
        info->Append(vtkAlgorithm::INPUT_REQUIRED_DATA_TYPE(), "vtkDataSet");
        break;
      default:
        return 0;
    }
    return 1;
  }
 
  int FillOutputPortInformation(int port, vtkInformation *info) override {
    switch(port) {
      case 0:
        info->Set(vtkDataObject::DATA_TYPE_NAME(), "vtkUnstructuredGrid");
        break;
      default:
        return 0;
    }
    return 1;
  }
 
private:
  int LabelDataType;
  std::string LabelFieldName{"RegionId"};
 
  vtkSmartPointer<vtkMultiBlockDataSet> previousIterationData;
 
  // Containers for nodes and edges
  std::vector<std::vector<Nodes>> levelTimeNodesMap; // N
  std::vector<std::vector<Edges>> levelTimeEdgesTMap; // E_T
  std::vector<std::vector<Edges>> timeLevelEdgesNMap; // E_N
};