vtkExplodeDataSet.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 vtkExplodeDataSet_h
#define vtkExplodeDataSet_h

#include "vtkFiltersGeneralModule.h" // For export macro
#include "vtkPartitionedDataSetCollectionAlgorithm.h"

#include <map> // for std::map

VTK_ABI_NAMESPACE_BEGIN
class vtkDataArray;
class vtkDataSet;
class vtkIdList;
class vtkPointSet;
class vtkStringArray;

class VTKFILTERSGENERAL_EXPORT vtkExplodeDataSet : public vtkPartitionedDataSetCollectionAlgorithm
{
public:
  static vtkExplodeDataSet* New();
  vtkTypeMacro(vtkExplodeDataSet, vtkPartitionedDataSetCollectionAlgorithm);
  void PrintSelf(ostream& os, vtkIndent indent) override;


  vtkSetMacro(UsePartitionNamesFromFieldData, bool);
  vtkGetMacro(UsePartitionNamesFromFieldData, bool);
  vtkBooleanMacro(UsePartitionNamesFromFieldData, bool);


  vtkSetMacro(PartitionNamesArray, std::string);
  vtkGetMacro(PartitionNamesArray, std::string);


  vtkSetMacro(PartitionValuesArray, std::string);
  vtkGetMacro(PartitionValuesArray, std::string);

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

  int RequestData(vtkInformation*, vtkInformationVector**, vtkInformationVector*) override;

  int FillInputPortInformation(int port, vtkInformation* info) override;

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

  std::map<double, vtkSmartPointer<vtkIdList>> GetIdListsByValue(vtkDataArray* scalars);

  vtkPointSet* CreatePartition(vtkDataSet* input, vtkIdList* partCellIds);

  std::string GetPartitionName(vtkIdType id, const std::map<double, std::string>& providedNames,
    double value, std::string basename);

  std::map<double, std::string> GetProvidedNames(vtkDataSet* input);

  vtkStringArray* GetNamesArray(vtkDataSet* input);

  vtkSmartPointer<vtkDataArray> GetValuesArray(vtkDataSet* input, vtkIdType expectedSize);

  bool UsePartitionNamesFromFieldData = false;
  std::string PartitionNamesArray;
  std::string PartitionValuesArray;
};

VTK_ABI_NAMESPACE_END
#endif