vtkSplitField.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 vtkSplitField_h
#define vtkSplitField_h
 
#include "vtkDataSetAlgorithm.h"
#include "vtkFiltersGeneralModule.h" // For export macro
 
#include "vtkDataSetAttributes.h" // Needed for NUM_ATTRIBUTES
 
VTK_ABI_NAMESPACE_BEGIN
class vtkFieldData;
 
class VTKFILTERSGENERAL_EXPORT vtkSplitField : public vtkDataSetAlgorithm
{
public:
  vtkTypeMacro(vtkSplitField, vtkDataSetAlgorithm);
  void PrintSelf(ostream& os, vtkIndent indent) override;
 
  static vtkSplitField* New();
 
  void SetInputField(int attributeType, int fieldLoc);
 
  void SetInputField(const char* name, int fieldLoc);
 
  void SetInputField(const char* name, const char* fieldLoc);
 
  void Split(int component, const char* arrayName);
 
  enum FieldLocations
  {
    DATA_OBJECT = 0,
    POINT_DATA = 1,
    CELL_DATA = 2
  };
 
  struct Component
  {
    int Index;
    char* FieldName;
    Component* Next; // linked list
    void SetName(const char* name)
    {
      delete[] this->FieldName;
      this->FieldName = nullptr;
      if (name)
      {
        size_t len = strlen(name) + 1;
        this->FieldName = new char[len];
#ifdef _MSC_VER
        strncpy_s(this->FieldName, len, name, len - 1);
#else
        strncpy(this->FieldName, name, len);
#endif
      }
    }
    Component() { FieldName = nullptr; }
    ~Component() { delete[] FieldName; }
  };
 
protected:
  enum FieldTypes
  {
    NAME,
    ATTRIBUTE
  };
 
  vtkSplitField();
  ~vtkSplitField() override;
 
  int RequestData(vtkInformation*, vtkInformationVector**, vtkInformationVector*) override;
 
  char* FieldName;
  int FieldType;
  int AttributeType;
  int FieldLocation;
 
  static char FieldLocationNames[3][12];
  static char AttributeNames[vtkDataSetAttributes::NUM_ATTRIBUTES][10];
 
  vtkDataArray* SplitArray(vtkDataArray* da, int component);
 
  // Components are stored as a linked list.
  Component* Head;
  Component* Tail;
 
  // Methods to browse/modify the linked list.
  Component* GetNextComponent(Component* op) { return op->Next; }
  Component* GetFirst() { return this->Head; }
  void AddComponent(Component* op);
  Component* FindComponent(int index);
  void DeleteAllComponents();
 
  void PrintComponent(Component* op, ostream& os, vtkIndent indent);
  void PrintAllComponents(ostream& os, vtkIndent indent);
 
private:
  vtkSplitField(const vtkSplitField&) = delete;
  void operator=(const vtkSplitField&) = delete;
};
 
VTK_ABI_NAMESPACE_END
#endif