vtkSortedTableStreamer.h

Go to the documentation of this file.

// SPDX-FileCopyrightText: Copyright (c) Kitware Inc.
// SPDX-FileCopyrightText: Copyright (c) Ken Martin, Will Schroeder, Bill Lorensen
// SPDX-License-Identifier: BSD-3-Clause
#ifndef vtkSortedTableStreamer_h
#define vtkSortedTableStreamer_h

#include "vtkPVVTKExtensionsFiltersRenderingModule.h" // needed for export macro
#include "vtkSmartPointer.h"                          // for vtkSmartPointer
#include "vtkTableAlgorithm.h"
#include <utility> // for std::pair

class vtkDataArray;
class vtkIdTypeArray;
class vtkMultiProcessController;
class vtkPartitionedDataSet;
class vtkStringArray;
class vtkTable;
class vtkUnsignedIntArray;

class VTKPVVTKEXTENSIONSFILTERSRENDERING_EXPORT vtkSortedTableStreamer : public vtkTableAlgorithm
{
private:
  class InternalsBase;
  template <class T>
  class Internals;
  InternalsBase* Internal;

public:
  static void PrintInfo(vtkTable* input);

  static bool TestInternalClasses();
  static vtkSortedTableStreamer* New();
  vtkTypeMacro(vtkSortedTableStreamer, vtkTableAlgorithm);
  void PrintSelf(ostream& os, vtkIndent indent) override;

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


  vtkGetMacro(Block, vtkIdType);
  vtkSetMacro(Block, vtkIdType);


  vtkGetMacro(BlockSize, vtkIdType);
  vtkSetMacro(BlockSize, vtkIdType);


  vtkGetMacro(SelectedComponent, int);
  vtkSetMacro(SelectedComponent, int);


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


  void SetController(vtkMultiProcessController*);
  vtkGetObjectMacro(Controller, vtkMultiProcessController);

  const char* GetColumnNameToSort();

  // Update column to sort has well as invalidating the pre-processing
  void SetColumnNameToSort(const char* columnName);

  // Choose if the sorting order should be inverted or not
  void SetInvertOrder(int newValue);
  vtkGetMacro(InvertOrder, int);

protected:
  vtkSortedTableStreamer();
  ~vtkSortedTableStreamer() override;

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

  void CreateInternalIfNeeded(vtkTable* input, vtkDataArray* data);
  vtkDataArray* GetDataArrayToProcess(vtkTable* input);


  vtkGetStringMacro(ColumnToSort);
  vtkSetStringMacro(ColumnToSort);

  vtkIdType Block;
  vtkIdType BlockSize;
  vtkMultiProcessController* Controller;

  char* ColumnToSort;
  bool ShowFieldData = false;
  int SelectedComponent;
  int InvertOrder;

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

  vtkSmartPointer<vtkTable> MergeBlocks(vtkPartitionedDataSet* cd);

  void PopulateFieldDataArrays(vtkPartitionedDataSet* ptd, vtkSmartPointer<vtkTable> outTable);

  vtkSmartPointer<vtkUnsignedIntArray> GenerateCompositeIndexArray(
    vtkPartitionedDataSet* cd, vtkIdType maxSize);
  vtkSmartPointer<vtkStringArray> GenerateBlockNameArray(vtkPartitionedDataSet* cd);
  vtkSmartPointer<vtkIdTypeArray> GenerateBlockIndicesArray(
    vtkPartitionedDataSet* cd, vtkStringArray* blockNames, vtkIdType maxSize);
};

#endif