Skip to content
Snippets Groups Projects
Engine.h 6.72 KiB
Newer Older
/*
 * Capsule.h
 *
 *  Created on: Nov 7, 2016
 *      Author: wfg
 */

#ifndef ENGINE_H_
#define ENGINE_H_


/// \cond EXCLUDE_FROM_DOXYGEN
#include <vector>
#include <string>
#include <memory> //std::shared_ptr
#include <map>
#include <utility> //std::pair
/// \endcond

#ifdef HAVE_MPI
  #include <mpi.h>
#else
  #include "mpidummy.h"
#endif

wfg's avatar
wfg committed
#include "ADIOS.h"
#include "core/Method.h"
#include "core/Variable.h"
#include "core/Transform.h"
#include "core/Transport.h"
#include "core/Capsule.h"


namespace adios
{

/**
 * Base class for Engine operations managing shared-memory, and buffer and variables transform and transport operations
 */
class Engine
{

public:

    #ifdef HAVE_MPI
wfg's avatar
wfg committed
    MPI_Comm m_MPIComm = MPI_COMM_NULL; ///< only used as reference to MPI communicator passed from parallel constructor, MPI_Comm is a pointer itself. Public as called from C
    #else
    MPI_Comm m_MPIComm = 0; ///< only used as reference to MPI communicator passed from parallel constructor, MPI_Comm is a pointer itself. Public as called from C
    #endif

    const std::string m_EngineType; ///< from derived class
wfg's avatar
wfg committed
    const std::string m_Name; ///< name used for this engine
    const std::string m_AccessMode; ///< accessMode for buffers used by this engine
wfg's avatar
wfg committed
    const Method& m_Method; ///< associated method containing engine metadata
wfg's avatar
wfg committed

    int m_RankMPI = 0; ///< current MPI rank process
    int m_SizeMPI = 1; ///< current MPI processes size

wfg's avatar
wfg committed
    const std::string m_HostLanguage = "C++";

wfg's avatar
wfg committed
     * Unique constructor
     * @param engineType
     * @param name
     * @param accessMode
     * @param mpiComm
wfg's avatar
wfg committed
     * @param method
wfg's avatar
wfg committed
     * @param debugMode
     * @param cores
     * @param endMessage
wfg's avatar
wfg committed
    Engine( ADIOS& adios, const std::string engineType, const std::string name, const std::string accessMode,
            MPI_Comm mpiComm, const Method& method, const bool debugMode, const unsigned int cores,
            const std::string endMessage );

    virtual ~Engine( );

    /**
     * Write function that adds static checking on the variable to be passed by values
     * It then calls its corresponding derived class virtual function
     * This version uses m_Group to look for the variableName.
     * @param variableName name of variable to the written
     * @param values pointer passed from the application
     */
wfg's avatar
wfg committed
    void Write( Variable<T>& variable, const T* values )
wfg's avatar
wfg committed
        Write( variable, values );
wfg's avatar
wfg committed
    virtual void Write( Variable<char>& variable, const char* values ) = 0;
    virtual void Write( Variable<unsigned char>& variable, const unsigned char* values ) = 0;
    virtual void Write( Variable<short>& variable, const short* values ) = 0;
    virtual void Write( Variable<unsigned short>& variable, const unsigned short* values ) = 0;
    virtual void Write( Variable<int>& variable, const int* values ) = 0;
    virtual void Write( Variable<unsigned int>& variable, const unsigned int* values ) = 0;
    virtual void Write( Variable<long int>& variable, const long int* values ) = 0;
    virtual void Write( Variable<unsigned long int>& variable, const unsigned long int* values ) = 0;
    virtual void Write( Variable<long long int>& variable, const long long int* values ) = 0;
    virtual void Write( Variable<unsigned long long int>& variable, const unsigned long long int* values ) = 0;
    virtual void Write( Variable<float>& variable, const float* values ) = 0;
    virtual void Write( Variable<double>& variable, const double* values ) = 0;
    virtual void Write( Variable<long double>& variable, const long double* values ) = 0;
wfg's avatar
wfg committed
    /**
     * @brief Write functions can be overridden by derived classes. Base class behavior is to:
     * 1) Write to Variable values (m_Values) using the pointer to default group *m_Group set with SetDefaultGroup function
     * 2) Transform the data
     * 3) Write to all capsules -> data and metadata
     * @param variableName
     * @param values coming from user app
     */
    virtual void Write( const std::string variableName, const char* values ) = 0;
    virtual void Write( const std::string variableName, const unsigned char* values ) = 0;
    virtual void Write( const std::string variableName, const short* values ) = 0;
    virtual void Write( const std::string variableName, const unsigned short* values ) = 0;
    virtual void Write( const std::string variableName, const int* values ) = 0;
    virtual void Write( const std::string variableName, const unsigned int* values ) = 0;
    virtual void Write( const std::string variableName, const long int* values ) = 0;
    virtual void Write( const std::string variableName, const unsigned long int* values ) = 0;
    virtual void Write( const std::string variableName, const long long int* values ) = 0;
    virtual void Write( const std::string variableName, const unsigned long long int* values ) = 0;
    virtual void Write( const std::string variableName, const float* values ) = 0;
    virtual void Write( const std::string variableName, const double* values ) = 0;
    virtual void Write( const std::string variableName, const long double* values ) = 0;
wfg's avatar
wfg committed

    virtual void Close( const int transportIndex = -1  ); ///< Closes a particular transport, or all if -1
wfg's avatar
wfg committed


protected:
wfg's avatar
wfg committed
    ADIOS& m_ADIOS; ///< reference to ADIOS object that creates this Engine at Open
wfg's avatar
wfg committed
    std::vector< std::shared_ptr<Transport> > m_Transports; ///< transports managed
    const bool m_DebugMode = false; ///< true: additional checks, false: by-pass checks
    unsigned int m_Cores = 1;
    const std::string m_EndMessage; ///< added to exceptions to improve debugging
wfg's avatar
wfg committed
    std::set<std::string> m_WrittenVariables; ///< contains the names of the variables that are being written
    virtual void Init( ); ///< Initialize m_Capsules and m_Transports, called from constructor
    virtual void InitCapsules( ); ///< Initialize transports from Method, called from Init in constructor.
    virtual void InitTransports( ); ///< Initialize transports from Method, called from Init in constructor.
    /**
     * Used to verify parameters in m_Method containers
     * @param itParam iterator to a certain parameter
     * @param parameters map of parameters, from m_Method
     * @param parameterName used if exception is thrown to provide debugging information
     * @param hint used if exception is thrown to provide debugging information
     */
    void CheckParameter( const std::map<std::string, std::string>::const_iterator itParam,
                         const std::map<std::string, std::string>& parameters,
                         const std::string parameterName,
                         const std::string hint ) const;

    bool TransportNamesUniqueness( ) const; ///< checks if transport names are unique among the same types (file I/O)

};


} //end namespace

#endif /* ENGINE_H_ */