ttkPointDataSelector.h

Go to the documentation of this file.

 
#pragma once
 
#include <array>
#include <limits>
#include <string>
#include <vtkDataArraySelection.h>
 
// VTK Module
#include <ttkPointDataSelectorModule.h>
 
// ttk code includes
#include <ttkAlgorithm.h>
 
class vtkDataSet;
 
class TTKPOINTDATASELECTOR_EXPORT ttkPointDataSelector : public ttkAlgorithm {
 
public:
  static ttkPointDataSelector *New();
  vtkTypeMacro(ttkPointDataSelector, ttkAlgorithm);
 
  vtkSetMacro(RegexpString, const std::string &);
 
  void SetRangeId(int data0, int data1) {
    RangeId[0] = data0;
    RangeId[1] = data1;
    Modified();
  }
  int *GetRangeId() {
    return RangeId.data();
  }
 
  vtkSetMacro(RenameSelected, bool);
  vtkGetMacro(RenameSelected, bool);
 
  vtkSetMacro(SelectedFieldName, const std::string &);
  vtkGetMacro(SelectedFieldName, std::string);
 
  void AddScalarField(const std::string &s) {
    SelectedFields.emplace_back(s);
    Modified();
  }
 
  void ClearScalarFields() {
    SelectedFields.clear();
    Modified();
  }
 
  vtkDataArraySelection *GetRangeIds() {
    vtkDataArraySelection *arr = vtkDataArraySelection::New();
    arr->SetArraySetting("0", true);
    const auto nFields = AvailableFields.size();
    arr->SetArraySetting(
      std::to_string(nFields == 0 ? 0 : nFields - 1).c_str(), true);
    return arr;
  }
 
protected:
  ttkPointDataSelector();
 
  int RequestInformation(vtkInformation *request,
                         vtkInformationVector **inputVector,
                         vtkInformationVector *outputVector) override;
 
  int FillInputPortInformation(int port, vtkInformation *info) override;
  int FillOutputPortInformation(int port, vtkInformation *info) override;
  int RequestData(vtkInformation *request,
                  vtkInformationVector **inputVector,
                  vtkInformationVector *outputVector) override;
 
  void FillAvailableFields(vtkDataSet *input);
 
private:
  bool RenameSelected{false};
  std::string SelectedFieldName{"SelectedField"};
  std::vector<std::string> SelectedFields{};
  std::vector<std::string> AvailableFields{};
  std::string RegexpString{".*"};
  std::array<int, 2> RangeId{0, std::numeric_limits<int>::max()};
};