imstkVTKMeshIO.h

/*
** This file is part of the Interactive Medical Simulation Toolkit (iMSTK)
** iMSTK is distributed under the Apache License, Version 2.0.
** See accompanying NOTICE for details.
*/

#pragma once

#include "imstkMeshIO.h"

namespace imstk
{
class AbstractCellMesh;
class HexahedralMesh;
class ImageData;
class LineMesh;
class SurfaceMesh;
class TetrahedralMesh;

class VTKMeshIO
{
public:
    VTKMeshIO() = default;
    virtual ~VTKMeshIO() = default;

    static std::shared_ptr<PointSet> read(const std::string& filePath, MeshFileType meshType);

    static bool write(const std::shared_ptr<PointSet> imstkMesh, const std::string& filePath, const MeshFileType meshType);

protected:

    template<typename ReaderType>
    static std::shared_ptr<PointSet> readVtkGenericFormatData(const std::string& filePath);

    template<typename ReaderType>
    static std::shared_ptr<SurfaceMesh> readVtkPolyData(const std::string& filePath);

    template<typename WriterType>
    static bool writeVtkImageData(const std::shared_ptr<ImageData> imstkMesh, const std::string& filePath);

    template<typename WriterType>
    static bool writeVtkPolyData(const std::shared_ptr<SurfaceMesh> imstkMesh, const std::string& filePath);

    template<typename WriterType>
    static bool writeVtkPolyData(const std::shared_ptr<LineMesh> imstkMesh, const std::string& filePath);

    template<typename WriterType>
    static bool writeVtkPointSet(const std::shared_ptr<PointSet> imstkMesh, const std::string& filePath);

    template<typename WriterType>
    static bool writeVtkUnstructuredGrid(const std::shared_ptr<TetrahedralMesh> tetMesh, const std::string& filePath);
    template<typename WriterType>
    static bool writeVtkUnstructuredGrid(const std::shared_ptr<HexahedralMesh> hMesh, const std::string& filePath);

    template<typename ReaderType>
    static std::shared_ptr<AbstractCellMesh> readVtkUnstructuredGrid(const std::string& filePath);

    template<typename ReaderType>
    static std::shared_ptr<ImageData> readVtkImageData(const std::string& filePath);

    static std::shared_ptr<ImageData> readVtkImageDataDICOM(const std::string& filePath);

    static std::shared_ptr<ImageData> readVtkImageDataNIFTI(const std::string& filePath);

    static bool writeVtkImageDataNIFTI(std::shared_ptr<ImageData> imageData, const std::string& filePath);

    static bool writeMetaImageData(std::shared_ptr<ImageData> imageData, const std::string& filePath);
};
} // namespace imstk