RasterToMesh.h

Go to the documentation of this file.

 
#pragma once
 
#include "GeoLib/Raster.h"
#include "MeshLib/Location.h"
#include "MeshLib/MeshEnums.h"
#include "MeshLib/Properties.h"
 
class vtkImageData; // For conversion from Image to QuadMesh
 
namespace MeshLib {
 
class Mesh;
class Node;
class Element;
template <typename T>
class PropertyVector;
 
class RasterToMesh
{
public:
    static std::unique_ptr<MeshLib::Mesh> convert(
        GeoLib::Raster const& raster,
        MeshElemType elem_type,
        UseIntensityAs intensity_type,
        std::string const& array_name = "Colour");
 
    static std::unique_ptr<MeshLib::Mesh> convert(
        vtkImageData* img,
        const double origin[3],
        const double scalingFactor,
        MeshElemType elem_type,
        UseIntensityAs intensity_type,
        std::string const& array_name = "Colour");
 
    static std::unique_ptr<MeshLib::Mesh> convert(
        const double* const img,
        GeoLib::RasterHeader const& header,
        MeshElemType elem_type,
        UseIntensityAs intensity_type,
        std::string const& array_name = "Colour");
 
private:
    template<typename T>
    static void fillPropertyVector(
        MeshLib::PropertyVector<T> &prop_vec,
        double const*const img,
        GeoLib::RasterHeader const& header,
        MeshElemType elem_type)
    {
        for (std::size_t k = 0; k < header.n_depth; k++)
        {
            std::size_t const layer_idx = (k*header.n_rows*header.n_cols);
            for (std::size_t i = 0; i < header.n_cols; i++)
            {
                std::size_t idx(i * header.n_rows + layer_idx);
                for (std::size_t j = 0; j < header.n_rows; j++)
                {
                    auto val(static_cast<T>(img[idx + j]));
                    prop_vec.push_back(val);
                    if (elem_type == MeshElemType::TRIANGLE ||
                        elem_type == MeshElemType::PRISM)
                    {
                        prop_vec.push_back(val); // because each pixel is represented by two cells
                    }
                }
            }
        }
    }
};
 
} // end namespace MeshLib