GridVDB.h

Go to the documentation of this file.

// ==============================================================================================
// Microflow 3D, http://www.microflow.pwr.edu.pl/
// Created by Roman Szafran on 04.05.19.
// Copyright (c) 2019 Wroclaw University of Science and Technology.
// Distributed under the Apache License, Version 2.0. You may obtain a copy of the License at
// http://www.apache.org/licenses/LICENSE-2.0 or see accompanying file license.txt.
// Redistributions of source code must retain the above copyright and license notice.
// ==============================================================================================

# pragma once

// OpenVDB
#include <openvdb/openvdb.h>

#include <memory>

#include "ReadersAndWriters/VDBWriter.h"
#include "ReadersAndWriters/VTIWriter.h"
#include "MFDatabase/ConfigData.h"

namespace MF {
    namespace Database {


        template<class T_VDBGridType> // T_VDBGridType = <openvdb::FloatGrid>, <openvdb::Int32Grid>, <openvdb::DobleGrid> etc.
        class GridVDB {
        public:
            GridVDB() = default;
            ~GridVDB() = default;

            static std::shared_ptr<MF::Database::GridVDB<T_VDBGridType>> New() {
                auto newGrid_Ptr = std::make_shared<MF::Database::GridVDB<T_VDBGridType>>();
                newGrid_Ptr->m_VDBGrid_Ptr = T_VDBGridType::create();
                return newGrid_Ptr;
            }

            std::shared_ptr<T_VDBGridType> getVDBGridPtr() {return m_VDBGrid_Ptr;}; 

            inline void setGridName(const std::string &GridName) { m_VDBGrid_Ptr->setName(GridName);}; 

            inline void insertMetaDataFloat(const std::string &MetaName, float value) { m_VDBGrid_Ptr->insertMeta(MetaName, openvdb::FloatMetadata(value));}; 

            inline void insertMetaDataInt(const std::string &MetaName, int32_t value) { m_VDBGrid_Ptr->insertMeta(MetaName, openvdb::Int32Metadata(value));}; 

            inline void insertMetaDataDouble(const std::string &MetaName, double value) {m_VDBGrid_Ptr->insertMeta(MetaName, openvdb::DoubleMetadata(value));}; 

            inline void insertMetaDataString(const std::string &MetaName, const std::string &string) {m_VDBGrid_Ptr->insertMeta(MetaName, openvdb::StringMetadata(string)); }; 

            template <typename T_ValueType> // T_ValueType = <int>, <double>, <string> etc.
            const T_ValueType &getMetaData(const std::string &MetaName) { return m_VDBGrid_Ptr->template metaValue<T_ValueType>(MetaName); }; 

            inline void setGridToFogVolume(bool voxelizeActiveTiles = false) {openvdb::tools::sdfToFogVolume(m_VDBGrid_Ptr); 
                if (voxelizeActiveTiles) m_VDBGrid_Ptr->treePtr()->voxelizeActiveTiles();};

            inline void setGridBackgroundValue(float value) { openvdb::tools::changeBackground(m_VDBGrid_Ptr->tree(), value); }; 

            inline T_VDBGridType getGridNewCopy() { return m_VDBGrid_Ptr->deepCopy(); }; 

            inline void saveToVDBFile(const std::string &FileName) {auto w = MF::RW::VDBWriter<T_VDBGridType>::New(m_VDBGrid_Ptr, FileName);}; 

            template<typename T1_VTKArrayType> // T1_VTKArrayType = <vtkFloatArray>, <vtkDoubleArray>, <vtkIntArray> etc.
            void saveToVTIFile(const std::string &FileName, const std::string &DataName, int numComponents, float voxelSize) {
                auto w = MF::RW::VTIWriter<T_VDBGridType>::New(m_VDBGrid_Ptr, FileName, voxelSize);
                w->template addDataFromGrid<T1_VTKArrayType>(DataName,numComponents);
                w->addNodeCoordinateFromGrid();
                w->writeFile();
            };

            void saveAllDataToVTIFile(const std::shared_ptr<MF::Database::ThreadArray> & ThreadArray_Ptr, const std::shared_ptr<MF::Database::ConfigData>& ConfigData_Ptr,
                                      const std::string &FileName) {
                double PhysVoxelSize = ConfigData_Ptr->getCaseFloatParam("GeometryParams.PhysicalVoxelSize");
                auto w = MF::RW::VTIWriter<T_VDBGridType>::New(m_VDBGrid_Ptr, FileName, PhysVoxelSize);
                if (ConfigData_Ptr->getCaseBooleanParam("VTKFileParams.NodeID_Save"))
                    w->addNodeIDFromGrid();
                if (ConfigData_Ptr->getCaseBooleanParam("VTKFileParams.NodeUidThread_Save"))
                    w->addUidThreadFromThread(ThreadArray_Ptr);
                if (ConfigData_Ptr->getCaseBooleanParam("VTKFileParams.NodeType_Save"))
                    w->addNodeTypeFromThread(ThreadArray_Ptr);
                if (ConfigData_Ptr->getCaseBooleanParam("VTKFileParams.NodeCoordinate_Save"))
                    w->addNodeCoordinateFromThread(ThreadArray_Ptr);
                if (ConfigData_Ptr->getCaseBooleanParam("VTKFileParams.VelocityLB_Save"))
                    w->addVelocityLBFromThread(ThreadArray_Ptr);
                if (ConfigData_Ptr->getCaseBooleanParam("VTKFileParams.RhoLB_Save"))
                    w->addRhoLBFromThread(ThreadArray_Ptr);
                if (ConfigData_Ptr->getCaseBooleanParam("VTKFileParams.RhoPhys_Save"))
                    w->addRhoPhysFromThread(ThreadArray_Ptr);
                if (ConfigData_Ptr->getCaseBooleanParam("VTKFileParams.PressurePhys_Save"))
                    w->addPressPhysFromThread(ThreadArray_Ptr);
                if (ConfigData_Ptr->getCaseBooleanParam("VTKFileParams.VelocityPhys_Save"))
                    w->addVelocityPhysFromThread(ThreadArray_Ptr);
                if (ConfigData_Ptr->getCaseBooleanParam("VTKFileParams.FQ19_Save"))
                    w->addFQ19FromThread(ThreadArray_Ptr);
                if (ConfigData_Ptr->getCaseBooleanParam("VTKFileParams.PropagationDirections_Save"))
                    w->addPropagationDirFromThread(ThreadArray_Ptr);
                if (ConfigData_Ptr->getCaseBooleanParam("VTKFileParams.InitialVelocityLB_Save"))
                    w->addInitialVelocityLBFromThread(ThreadArray_Ptr);
                if (ConfigData_Ptr->getCaseBooleanParam("VTKFileParams.InitialRhoLB_Save"))
                    w->addInitialRhoLBFromThread(ThreadArray_Ptr);
                if (ConfigData_Ptr->getCaseBooleanParam("VTKFileParams.ThreadsNamed_Save"))
                    w->addNodeThreadsByNameThread(ThreadArray_Ptr);
                w->writeFile();
            };

        private:
            std::shared_ptr<T_VDBGridType> m_VDBGrid_Ptr;

        };

    } /* namespace Database */
} /* namespace MF */