OpalSimulation.h

Go to the documentation of this file.

#ifndef __OPALSIMULATION_H__
#define __OPALSIMULATION_H__

#include <string>
#include <map>

#include "mpi.h"

#include "Util/Types.h"
#include "Util/CmdArguments.h"
#include "Simulation/Simulation.h"
#include "Simulation/GenerateOpalSimulation.h"

#include "boost/smart_ptr.hpp"

class OpalSimulation : public Simulation {

public:

    OpalSimulation(Expressions::Named_t objectives,
                   Expressions::Named_t constraints,
                   Param_t params, std::string name, MPI_Comm comm,
                   CmdArguments_t args,
                   std::map<std::string, std::string> uvars);

    virtual ~OpalSimulation();

    void run();

    std::map<std::string, std::vector<double> > getData(const std::vector<std::string> &statVariables);

    void collectResults();

    void cleanUp();

    void cleanUp(const std::vector<std::string>& keep);

    reqVarContainer_t getResults() { return requestedVars_; }

    void setFilename(int id) { id_m = id; }

private:

    std::string simulationName_;
    std::string simulationDirName_;
    std::string simTmpDir_;

    std::string pwd_;

    std::streambuf* strm_buffer_;
    std::streambuf* strm_err_;

    std::map<std::string, std::string> userVariables_;

    reqVarContainer_t requestedVars_;

    Expressions::Named_t objectives_;
    Expressions::Named_t constraints_;
    std::set<std::string> dvarNames_;

    MPI_Comm comm_;

    std::unique_ptr<GenerateOpalSimulation> gs_;

    int id_m;

    void invalidBunch();

    bool hasResultsAvailable();

    void createSymlink_m(const std::string& path);

    void copyH5_m();

    void setupSimulation();

    void setupFSStructure();

    void getVariableDictionary(variableDictionary_t& dictionary,
                               const std::string& filename,
                               const Expressions::Expr_t* const expression);

    void redirectOutToFile();
    void restoreOut();
};

#endif