GenericOutput.hpp

/*---------------------------------------------------------------------------*\
 *
 *  mimmo
 *
 *  Copyright (C) 2015-2021 OPTIMAD engineering Srl
 *
 *  -------------------------------------------------------------------------
 *  License
 *  This file is part of mimmo.
 *
 *  mimmo is free software: you can redistribute it and/or modify it
 *  under the terms of the GNU Lesser General Public License v3 (LGPL)
 *  as published by the Free Software Foundation.
 *
 *  mimmo is distributed in the hope that it will be useful, but WITHOUT
 *  ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
 *  FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public
 *  License for more details.
 *
 *  You should have received a copy of the GNU Lesser General Public License
 *  along with mimmo. If not, see <http://www.gnu.org/licenses/>.
 *
\*---------------------------------------------------------------------------*/
#ifndef __OUTPUTDOF_HPP__
#define __OUTPUTDOF_HPP__
 
#include "BaseManipulation.hpp"
#include "IOData.hpp"
 
namespace mimmo{
 
namespace outputCSVStream{
    template<typename T>
    std::fstream&  ofstreamcsv(std::fstream &in, const T &x);
    template<typename T>
    std::fstream&  ofstreamcsvend(std::fstream &in, const T &x);
    template<typename T>
    std::fstream&  ofstreamcsv(std::fstream &in, const std::vector< T > &x);
    template<typename T, size_t d>
    std::fstream&  ofstreamcsv(std::fstream &in, const std::array< T,d > &x);
    template<typename T>
    std::fstream&  ofstreamcsvend(std::fstream &in, const std::vector< T > &x);
    template<typename T, size_t d>
    std::fstream&  ofstreamcsvend(std::fstream &in, const std::array< T,d > &x);
    template<typename T>
    std::fstream&  ofstreamcsv(std::fstream &in, const MimmoPiercedVector< T > &x);
    template<typename T, size_t d>
    std::fstream&  ofstreamcsvend(std::fstream &in, const MimmoPiercedVector< T > &x);
}
class GenericOutput: public BaseManipulation{
private:
    std::string             m_dir;          
    std::string             m_filename;     
    bool                    m_csv;          
    std::unique_ptr<IOData> m_input;        
public:
    GenericOutput(std::string dir = "./", std::string filename = "output.txt", bool csv = false);
    GenericOutput(const bitpit::Config::Section & rootXML);
    ~GenericOutput();
 
    GenericOutput(const GenericOutput & other);
    GenericOutput& operator=(GenericOutput other);
 
    void buildPorts();
 
    BITPIT_DEPRECATED( template<typename T>
    T*      getInput());
 
    template<typename T>
    void    setInput(T* data);
 
    template<typename T>
    void    setInput(T data);
 
    void    clearInput();
 
    void setWriteDir(std::string dir);
    void setFilename(std::string filename);
    void setCSV(bool csv);
 
    void    execute();
 
    virtual void absorbSectionXML(const bitpit::Config::Section & slotXML, std::string name = "");
    virtual void flushSectionXML(bitpit::Config::Section & slotXML, std::string name= "");
 
protected:
    void swap(GenericOutput & x) noexcept;
 
private:
    template<typename T>
    void _setInput(T & data);
};
 
class GenericOutputMPVData: public BaseManipulation{
private:
    std::string             m_dir;          
    std::string             m_filename;     
    bool                    m_csv;          
    std::unique_ptr<IOData> m_input;        
    bool            m_binary;       
public:
    GenericOutputMPVData(std::string dir = "./", std::string filename = "output.txt", bool csv = false);
    GenericOutputMPVData(const bitpit::Config::Section & rootXML);
    ~GenericOutputMPVData();
 
    GenericOutputMPVData(const GenericOutputMPVData & other);
    GenericOutputMPVData& operator=(GenericOutputMPVData other);
 
    void buildPorts();
 
    BITPIT_DEPRECATED(template<typename T>
    MimmoPiercedVector< T >*    getInput());
 
    template<typename T>
    void    setInput(MimmoPiercedVector< T > *data);
 
    template<typename T>
    void    setInput(MimmoPiercedVector< T > data);
 
    void    clearInput();
 
    void setWriteDir(std::string dir);
    void setFilename(std::string filename);
    void setCSV(bool csv);
    void setBinary(bool binary);
 
    void    execute();
 
    virtual void absorbSectionXML(const bitpit::Config::Section & slotXML, std::string name = "");
    virtual void flushSectionXML(bitpit::Config::Section & slotXML, std::string name= "");
 
protected:
    void swap(GenericOutputMPVData & x) noexcept;
#if MIMMO_ENABLE_MPI
    template<typename T>
    void    collectDataFromAllProcs(MimmoPiercedVector<T> & locdata, MimmoPiercedVector<T> * dataglobal);
#endif
 
private:
    template<typename T>
    void _setInput(MimmoPiercedVector< T > & data);
 
};
 
REGISTER_PORT(M_COORDS, MC_VECARR3, MD_FLOAT,__OUTPUTDOF_HPP__)
REGISTER_PORT(M_DISPLS, MC_VECARR3, MD_FLOAT,__OUTPUTDOF_HPP__)
REGISTER_PORT(M_DATAFIELD, MC_VECTOR, MD_FLOAT,__OUTPUTDOF_HPP__)
REGISTER_PORT(M_VECTORFIELD, MC_SCALAR, MD_MPVECARR3FLOAT_,__OUTPUTDOF_HPP__)
REGISTER_PORT(M_SCALARFIELD, MC_SCALAR, MD_MPVECFLOAT_,__OUTPUTDOF_HPP__)
REGISTER_PORT(M_POINT, MC_ARRAY3, MD_FLOAT,__OUTPUTDOF_HPP__)
REGISTER_PORT(M_SPAN, MC_ARRAY3, MD_FLOAT,__OUTPUTDOF_HPP__)
REGISTER_PORT(M_DIMENSION, MC_ARRAY3, MD_INT,__OUTPUTDOF_HPP__)
REGISTER_PORT(M_VALUED, MC_SCALAR, MD_FLOAT,__OUTPUTDOF_HPP__)
REGISTER_PORT(M_VALUEI, MC_SCALAR, MD_INT,__OUTPUTDOF_HPP__)
REGISTER_PORT(M_VALUEB, MC_SCALAR, MD_BOOL,__OUTPUTDOF_HPP__)
REGISTER_PORT(M_DEG, MC_ARRAY3, MD_INT,__OUTPUTDOF_HPP__)
 
REGISTER(BaseManipulation, GenericOutput, "mimmo.GenericOutput")
REGISTER(BaseManipulation, GenericOutputMPVData, "mimmo.GenericOutputMPVData")
 
}
 
#include "GenericOutput.tpp"
 
#endif /* __OUTPUTDOF_HPP__ */