VDBWriter.h

Go to the documentation of this file.

// ==============================================================================================
// Microflow 3D, http://www.microflow.pwr.edu.pl/
// Created by Roman Szafran on 26.03.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

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

namespace MF {
    namespace RW {

        template <class T_VDBGridType> // T_VDBGridType = <openvdb::FloatGrid::Ptr>, <openvdb::Int32Grid::Ptr>, <openvdb::DoubleGrid::Ptr> etc.
        class VDBWriter {
        public:
            VDBWriter(const std::shared_ptr<T_VDBGridType>& Grid_Ptr, const std::string &FileName) {
                std::string FileNameExt = FileName + ".vdb";
                // Writing the openvdb grid to a file.
                // Create a VDB file object.
                openvdb::io::File file(FileNameExt);
                // Add the grid pointer to a container.
                openvdb::GridPtrVec grids;
                grids.push_back(Grid_Ptr);
                // Write out the contents of the container.
                file.write(grids);
                file.close();
                std::cout << "Grid was saved successfully in VDB file: ----> " << FileNameExt << std::endl;
            };

            ~VDBWriter() = default;

            // T_VDBGridType = <openvdb::FloatGrid::Ptr>, <openvdb::Int32Grid::Ptr>, <openvdb::DoubleGrid::Ptr> etc.
            static std::shared_ptr<MF::RW::VDBWriter<T_VDBGridType>> New(const std::shared_ptr<T_VDBGridType>& Grid_Ptr, const std::string &FileName) {
                auto VDBWriter_Ptr = std::make_shared<MF::RW::VDBWriter<T_VDBGridType>>(Grid_Ptr, FileName);
                return VDBWriter_Ptr;
            };
        };
    } /* namespace RW */
} /* namespase MF */