vtkAxisAlignedCutter.h

Go to the documentation of this file.

// SPDX-FileCopyrightText: Copyright (c) Kitware Inc.
// SPDX-License-Identifier: BSD-3-Clause
#ifndef vtkAxisAlignedPlaneCutter_h
#define vtkAxisAlignedPlaneCutter_h

#include "vtkAMRSliceFilter.h" // for vtkAMRSliceFilter
#include "vtkContourValues.h"  // for vtkContourValues
#include "vtkDataObjectAlgorithm.h"
#include "vtkHyperTreeGridAxisCut.h"                // for vtkHyperTreeGridAxisCut
#include "vtkPVVTKExtensionsFiltersGeneralModule.h" // for export macro
#include "vtkSmartPointer.h"                        // for vtkSmartPointer

class vtkDataAssembly;
class vtkImplicitFunction;
class vtkHyperTreeGridAxisCut;
class vtkMultiProcessController;
class vtkPartitionedDataSet;
class vtkPartitionedDataSetCollection;
class vtkPlane;

class VTKPVVTKEXTENSIONSFILTERSGENERAL_EXPORT vtkAxisAlignedCutter : public vtkDataObjectAlgorithm
{
public:
  vtkTypeMacro(vtkAxisAlignedCutter, vtkDataObjectAlgorithm);
  void PrintSelf(ostream& os, vtkIndent indent) override;
  static vtkAxisAlignedCutter* New();


  void SetCutFunction(vtkImplicitFunction* function);
  vtkImplicitFunction* GetCutFunction();

  vtkMTimeType GetMTime() override;


  void SetOffsetValue(int i, double value);
  double GetOffsetValue(int i);


  void SetNumberOfOffsetValues(int number);
  int GetNumberOfOffsetValues();


  vtkSetMacro(LevelOfResolution, int);
  vtkGetMacro(LevelOfResolution, int);


  void SetController(vtkMultiProcessController* controller);
  vtkMultiProcessController* GetController();

protected:
  vtkAxisAlignedCutter();
  ~vtkAxisAlignedCutter() override = default;

  int RequestDataObject(vtkInformation*, vtkInformationVector**, vtkInformationVector*) override;
  int RequestData(vtkInformation*, vtkInformationVector**, vtkInformationVector*) override;
  int FillInputPortInformation(int, vtkInformation* info) override;

private:
  vtkAxisAlignedCutter(const vtkAxisAlignedCutter&) = delete;
  void operator=(const vtkAxisAlignedCutter&) = delete;

  bool ProcessHTG(
    vtkHyperTreeGrid* inputHTG, vtkPlane* plane, vtkPartitionedDataSetCollection* outputSlices);

  bool ProcessPDS(vtkPartitionedDataSet* inputPDS, vtkPlane* plane,
    vtkPartitionedDataSetCollection* outputPDC, vtkDataAssembly* outputHierarchy, int nodeId);

  void CutHTGWithAAPlane(
    vtkHyperTreeGrid* input, vtkHyperTreeGrid* output, vtkPlane* plane, double offset);

  void CutAMRWithAAPlane(vtkOverlappingAMR* input, vtkOverlappingAMR* output, vtkPlane* plane);

  vtkSmartPointer<vtkImplicitFunction> CutFunction;
  vtkNew<vtkHyperTreeGridAxisCut> HTGCutter;
  vtkNew<vtkAMRSliceFilter> AMRCutter;
  int LevelOfResolution = 0;
  vtkNew<vtkContourValues> OffsetValues;

  vtkSmartPointer<vtkMultiProcessController> Controller;
};

#endif