gridwriter.hpp

#ifndef GRIDWRITER_H_
#define GRIDWRITER_H_

#include <fstream>

#include <fast_methods/ndgridmap/ndgridmap.hpp>

// TODO: include checks which ensure that the grids are adecuate for the functions used.
// TODO: there should be a check when writing grid: it is written already? erase and write. Something like that.
class GridWriter {
    public:
        template <class T, size_t ndims>
        static void saveGridValues
        (const char * filename, nDGridMap<T, ndims> & grid) {
            std::ofstream ofs;
            ofs.open (filename,  std::ofstream::out | std::ofstream::trunc);

            ofs << grid.getCell(0).type() << '\n';
            ofs << grid.getLeafSize() << '\n' << ndims;

            std::array<unsigned int, ndims> dimsize = grid.getDimSizes();
            for (unsigned int i = 0; i < ndims; ++i)
                ofs << '\n' << dimsize[i] << "\t";

            for (unsigned int i = 0; i < grid.size(); ++i)
                ofs << '\n' << grid.getCell(i).getValue();

            ofs.close();
        }

        template <class T, size_t ndims>
        static void saveVelocities
        (const char * filename, nDGridMap<T, ndims> & grid) {
            std::ofstream ofs;
            ofs.open (filename,  std::ofstream::out | std::ofstream::trunc);

            ofs << grid.getCell(0).type() << '\n';
            ofs << grid.getLeafSize() << '\n' << ndims;

            std::array<unsigned int, ndims> dimsize = grid.getDimSizes();
            for (unsigned int i = 0; i < ndims; ++i)
                ofs << '\n' << dimsize[i] << "\t";

            for (unsigned int i = 0; i < grid.size(); ++i)
                ofs << '\n' << grid.getCell(i).getVelocity();

            ofs.close();
        }

        template <class T, size_t ndims>
        static void savePath
        (const char * filename, nDGridMap<T, ndims> & grid, std::vector< std::array<double,ndims> > & path) {
            std::ofstream ofs;
            ofs.open (filename,  std::ofstream::out | std::ofstream::trunc);

            ofs << grid.getLeafSize() << '\n' << ndims;

            std::array<unsigned int, ndims> dimsize = grid.getDimSizes();
            for (unsigned int i = 0; i < ndims; ++i)
                ofs << '\n'<< dimsize[i] << "\t";

            for (unsigned int i = 0; i < path.size(); ++i) {
                ofs << '\n';
                for (unsigned int j = 0; j < ndims; ++j)
                    ofs << path[i][j] << "\t" ;
            }

            ofs.close();
        }

        template <class T, size_t ndims>
        static void savePathVelocity
        (const char * filename, nDGridMap<T, ndims> & grid, std::vector< std::array<double,ndims> > & path, std::vector <double> path_velocity) {
            std::ofstream ofs;
            ofs.open (filename,  std::ofstream::out | std::ofstream::trunc);

            ofs << grid.getLeafSize() << '\n' << ndims;

            std::array<unsigned int, ndims> dimsize = grid.getDimSizes();
            for (unsigned int i = 0; i < ndims; ++i)
                ofs << '\n' << dimsize[i];

            for (unsigned int i = 0; i < path.size(); ++i) {
                ofs <<'\n';
                for (unsigned int j = 0; j < ndims; ++j)
                    ofs << path[i][j] << "\t" ;
                ofs << path_velocity[i];
            }

            ofs.close();
        }
};

#endif /* GRIDWRITER_H_ */