Line data    Source code

       1              : // Copyright (C) 2015 Technische Universitaet Muenchen
       2              : // This file is part of the Mamico project. For conditions of distribution
       3              : // and use, please see the copyright notice in Mamico's main folder, or at
       4              : // www5.in.tum.de/mamico
       5              : #ifndef _MAMICO_COUPLING_SCENARIO_SCENARIO_H_
       6              : #define _MAMICO_COUPLING_SCENARIO_SCENARIO_H_
       7              : 
       8              : class Scenario;
       9              : 
      10              : #include "coupling/CouplingMDDefinitions.h"
      11              : #include "coupling/solvers/CouetteSolver.h"
      12              : #include "mamico_git_version_info.h"
      13              : #include <iostream>
      14              : #include <string>
      15              : 
      16              : #define __MAMICO_STRINGIFY_EXPAND(x) #x
      17              : #define MAMICO_STRINGIFY(x) __MAMICO_STRINGIFY_EXPAND(x)
      18              : 
      19              : class Scenario {
      20              : public:
      21              :   Scenario(std::string scenarioname) : _scenarioname(scenarioname) {
      22              :     int rank = 0;
      23              : #if (COUPLING_MD_PARALLEL == COUPLING_MD_YES)
      24              :     MPI_Comm_rank(MPI_COMM_WORLD, &rank);
      25              : #endif
      26              :     _isRootRank = (rank == 0);
      27              :     if (_isRootRank) {
      28              :       std::cout << "Run " << scenarioname << "..." << std::endl;
      29              :       std::cout << "MaMiCo git commit hash = " << MAMICO_STRINGIFY(MAMICO_COMMIT_HASH) << std::endl;
      30              :     }
      31              :   }
      32              :   virtual ~Scenario() {
      33              :     if (_isRootRank) {
      34              :       std::cout << "Shut down " << _scenarioname << std::endl;
      35              :     }
      36              :   }
      37              : 
      38              :   virtual void run() = 0;
      39              :   virtual void init() = 0;
      40              :   virtual void runOneCouplingCycle(int cycle) = 0;
      41              :   virtual void equilibrateMicro() = 0;
      42              : 
      43              :   virtual coupling::solvers::AbstractCouetteSolver<3>* getSolver() = 0;
      44            0 :   const coupling::services::ParallelTimeIntegrationService<3>* getTimeIntegrationService() const { return _timeIntegrationService.get(); }
      45              : 
      46              : protected:
      47              :   std::unique_ptr<coupling::services::ParallelTimeIntegrationService<3>> _timeIntegrationService;
      48              : 
      49              :   /** @brief if this is the world global root process */
      50              :   bool _isRootRank;
      51              : 
      52              : private:
      53              :   const std::string _scenarioname;
      54              : };
      55              : 
      56              : #endif // _MAMICO_COUPLING_SCENARIO_SCENARIO_H_