vtkNetCDFUGRIDReader.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 vtkNetCDFUGRIDReader_h
#define vtkNetCDFUGRIDReader_h

#include "vtkDataArraySelection.h" // For vtkSmartPointer downcast
#include "vtkIONetCDFModule.h"     // For export macro
#include "vtkSmartPointer.h"       // For vtkSmartPointer
#include "vtkUnstructuredGridAlgorithm.h"

#include <cstdlib> // For std::size_t
#include <vector>  // For std::vector

VTK_ABI_NAMESPACE_BEGIN

class VTKIONETCDF_EXPORT vtkNetCDFUGRIDReader : public vtkUnstructuredGridAlgorithm
{
public:
  static vtkNetCDFUGRIDReader* New();
  vtkTypeMacro(vtkNetCDFUGRIDReader, vtkUnstructuredGridAlgorithm);
  void PrintSelf(ostream& os, vtkIndent indent) override;


  vtkSetFilePathMacro(FileName);
  vtkGetFilePathMacro(FileName);


  vtkGetObjectMacro(PointDataArraySelection, vtkDataArraySelection);
  vtkGetObjectMacro(CellDataArraySelection, vtkDataArraySelection);


  int GetNumberOfPointArrays();
  int GetNumberOfCellArrays();


  const char* GetPointArrayName(int index);
  const char* GetCellArrayName(int index);


  int GetPointArrayStatus(const char* name);
  int GetCellArrayStatus(const char* name);
  void SetPointArrayStatus(const char* name, int status);
  void SetCellArrayStatus(const char* name, int status);


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

protected:
  vtkNetCDFUGRIDReader();
  ~vtkNetCDFUGRIDReader() override;

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

  bool Open();
  bool ParseHeader();
  bool FillArraySelection(const std::vector<int>& ids, vtkDataArraySelection* selection);
  bool FillPoints(vtkUnstructuredGrid* output);
  bool FillCells(vtkUnstructuredGrid* output);
  bool FillArrays(vtkUnstructuredGrid* output, std::size_t timeStep);
  void Close();

  bool CheckError(int error);
  std::string GetAttributeString(int var, std::string name);
  std::string GetVariableName(int var);
  std::string GetAttributeName(int var, int att);
  std::string GetDimensionName(int dim);
  vtkSmartPointer<vtkDataArray> GetArrayData(int var, std::size_t time, std::size_t size);

private:
  char* FileName = nullptr;

  int NcId = -1;
  int MeshVarId = -1;
  int FaceVarId = -1;
  int FaceFillValue = -1;
  int FaceStartIndex = 0;
  int NodeXVarId = -1;
  int NodeYVarId = -1;
  int NodeType = -1;
  std::size_t NodeCount = 0;
  std::size_t FaceCount = 0;
  std::size_t NodesPerFace = 0;
  std::size_t FaceStride = 0;
  std::size_t NodesPerFaceStride = 0;
  bool ReplaceFillValueWithNan = false;
  std::vector<int> NodeArrayVarIds; // data variables linked to nodes (points)
  std::vector<int> FaceArrayVarIds; // data variables linked to face (cells)
  std::vector<double> TimeSteps;

  vtkSmartPointer<vtkDataArraySelection> PointDataArraySelection;
  vtkSmartPointer<vtkDataArraySelection> CellDataArraySelection;

  vtkNetCDFUGRIDReader(const vtkNetCDFUGRIDReader&) = delete;
  void operator=(const vtkNetCDFUGRIDReader&) = delete;
};

VTK_ABI_NAMESPACE_END

#endif