gsParaviewCollection.h

Go to the documentation of this file.

#pragma once

#include <gsCore/gsForwardDeclarations.h>
#include <gsIO/gsFileManager.h>
#include <gsIO/gsParaviewDataSet.h>



#include<fstream>

namespace gismo {
class GISMO_EXPORT gsParaviewCollection
{
public:
    typedef std::string String;
public:

    gsParaviewCollection(String const  &fn,
                         gsExprEvaluator<> * evaluator=nullptr)
                        : m_filename(fn),
                        m_isSaved(false),
                        m_time(-1),
                        m_evaluator(evaluator),
                        m_options(gsParaviewDataSet::defaultOptions()),
                        counter(0)
    {
        std::string path = gsFileManager::getPath(m_filename);

        // If the path does not start with ./ or / , it is assumed to be a relative path
        if ( !gsFileManager::isFullyQualified(path) )
            path = gsFileManager::getCurrentPath() + path;

        m_filename = path + gsFileManager::getBasename(m_filename) + ".pvd";
        gsFileManager::mkdir( path );

        // if ( "" != m_filename.parent_path())
        // GISMO_ENSURE( fsystem::exists( m_filename.parent_path() ), 
        //     "The specified folder " << m_filename.parent_path() << " does not exist, please create it first.");  
        mfile <<"<?xml version=\"1.0\"?>\n";
        mfile <<"<VTKFile type=\"Collection\" version=\"0.1\">\n";
        mfile <<"<Collection>\n";
    }

    void addPart(String const & fn, real_t tStep=-1, std::string name="", index_t part=-1)
    {   
        std::string ext = "";
        if (gsFileManager::getExtension(fn) == "")
        {
            if (name=="Mesh" || name=="mesh")
                ext = ".vtp";
            else if (name=="Geometry" || name=="geometry" || name=="Solution" || name=="solution")
                ext = ".vts";
            else
                GISMO_ERROR("No extension could be found for file "<<fn<<". Try to add an extension or add name 'Mesh','Solution','Geometry'");
        }

        GISMO_ASSERT( !m_isSaved , "Error: collection has been already saved." );
        mfile << "<DataSet ";
        if (part != -1)   mfile << "part=\""<< part <<"\" ";
        if (tStep != -1)  mfile << "timestep=\""<< tStep <<"\" ";
        if (name != "") mfile << "name=\"" << name << "\" ";
        mfile << "file=\"" << fn+ext <<"\"/>\n";
    }
    // CAUTION! 
    // The previous 3 versions of gsParaviewCollection::addPart() have been combined into the one above
    // since they were all doing basiacally the same thing. Below you can see a 'conversion table' that
    // can help you adapt your code to the new syntax. For questions contact C. Karampatzakis (Github @ckarampa )
    // OLD SYNTAX ( Deprecated )                              |   NEW SYNTAX 
    //-----------------------------------------------------------------------------------------------------------
    // addPart(String const & fn)                             |   addPart( fn, -1, "", counter++);               |
    //                                                                                                           |
    // addPart(String const & fn, String const & ext)         |   addPart( fn+ext, -1, "", counter++);           |
    //                                                                                                           |
    // addPart(String const & fn, int i, String const & ext)  |   addPart( fn+std::to_string(i)+ext, -1, "", i); |
    // ----------------------------------------------------------------------------------------------------------

    // The following functions are all deprecated, only here for backwards compatibility!

    GISMO_DEPRECATED void addTimestep(String const & fn, double tstep, String const & ext)
    {
        // mfile << "<DataSet timestep=\""<<tstep<<"\" file=\""<<fn<<ext<<"\"/>\n";
        addPart( fn+ext, tstep);
    }

    GISMO_DEPRECATED void addTimestep(String const & fn, int part, double tstep, String const & ext)
    {
        // mfile << "<DataSet part=\""<<part<<"\" timestep=\""<<tstep<<"\" file=\""<<fn<<"_"<<part<<ext<<"\"/>\n";
        addPart( fn+"_"+std::to_string(part)+ext, tstep, "", part);
    }

    // End of deprecated functions

    GISMO_DEPRECATED void addPart(String const & fn, String extension)
    {
        addPart( fn+extension);
    }

    void addDataSet(gsParaviewDataSet & dataSet, real_t time=-1);

    void newTimeStep(gsMultiPatch<real_t> * geometry, real_t time=-1);

 
    template <typename... Rest>
    void addField(Rest... rest)
    {
        GISMO_ENSURE( !m_dataset.isEmpty(), "The gsParaviewDataSet, stored internally by gsParaviewCollection, is empty! Try running newTimestep() before addField().");
        m_dataset.addField(rest...);
    }

    template <typename... Rest>
    void addFields(Rest... rest)
    {
        GISMO_ENSURE( !m_dataset.isEmpty(), "The gsParaviewDataSet, stored internally by gsParaviewCollection, is empty! Try running newTimestep() before addFields().");
        m_dataset.addFields(rest...);
    }

    void saveTimeStep(){
        GISMO_ENSURE( !m_dataset.isEmpty(), "The gsParaviewDataSet, stored internally by gsParaviewCollection, is empty! Try running newTimestep() before saveTimeStep().");
        addDataSet(m_dataset,m_time);
    };

    void save()
    {
        GISMO_ASSERT(!m_isSaved, "Error: gsParaviewCollection::save() already called." );
        if (!m_isSaved)
        {
            mfile <<"</Collection>\n";
            mfile <<"</VTKFile>\n";

            gsDebug << "Exporting to " << m_filename << "\n";
            std::ofstream f( m_filename.c_str() );
            GISMO_ASSERT(f.is_open(), "Error creating "<< m_filename );
            f << mfile.rdbuf();
            f.close();
            mfile.str("");
            m_isSaved=true;
            counter = -1;
        }
    }

    gsOptionList & options() {return m_options;}

private:
    std::stringstream mfile;

    std::string m_filename;

    bool m_isSaved;

    int m_time;

    gsExprEvaluator<> * m_evaluator;

    gsParaviewDataSet m_dataset;

    gsOptionList m_options;

    index_t counter;

private:
    // Construction without a filename is not allowed
    gsParaviewCollection();
};

//=================================================================================================



inline void makeCollection(std::string const & fn, std::string const & ext, int n = 0)
{
    gsParaviewCollection pc(fn);
    if ( n > 0)
    {
        for (int i=0; i<n ; i++)
            pc.addPart(fn + std::to_string(i) + ext);
    }
    else
        pc.addPart(fn + ext);

    pc.save();
}


} // end namespace gismo