vtkCONVERGECFDCGNSReader.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 vtkCONVERGECFDCGNSReader_h
#define vtkCONVERGECFDCGNSReader_h

#include "vtkIOCGNSReaderModule.h" // For export macro

#include "vtkNew.h" // for vtkNew
#include "vtkPartitionedDataSetCollectionAlgorithm.h"
#include "vtkSmartPointer.h" // for vtkSmartPointer

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

VTK_ABI_NAMESPACE_BEGIN
class vtkCGNSReader;
class vtkDataArraySelection;
class vtkPolyData;

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

  vtkGetMacro(PointDataArraySelection, vtkDataArraySelection*);

  vtkGetMacro(CellDataArraySelection, vtkDataArraySelection*);

  vtkGetMacro(ParcelDataArraySelection, vtkDataArraySelection*);

  virtual int CanReadFile(VTK_FILEPATH const std::string& filename);


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

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

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

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

  void FindVectorNodeIds(int cgioId, const std::vector<double>& arrayIds, const std::string& prefix,
    double& vectorXId, double& vectorYId, double& vectorZId) const;

  bool CreateParcelPoints(
    int cgioId, double parcelXId, double parcelYId, double parcelZId, vtkPolyData* parcel) const;

  vtkSmartPointer<vtkDataArray> ReadParcelDataArray(int cgioId, double dataNodeId,
    const std::string& name, const std::vector<double>& arrayIds, bool isVector) const;

  std::string FileName;
  bool DataArraysInitialized = false;
  vtkNew<vtkDataArraySelection> PointDataArraySelection;
  vtkNew<vtkDataArraySelection> CellDataArraySelection;
  vtkNew<vtkDataArraySelection> ParcelDataArraySelection;
  vtkNew<vtkCGNSReader> CGNSReader;
};

VTK_ABI_NAMESPACE_END
#endif // vtkCONVERGECFDCGNSReader_h