MultiMDMediator.h

// Copyright (C) 2016 Technische Universitaet Muenchen
// This file is part of the Mamico project. For conditions of distribution
// and use, please see the copyright notice in Mamico's main folder, or at
// www5.in.tum.de/mamico
#ifndef _COUPLING_MULTIMDMEDIATOR_H_
#define _COUPLING_MULTIMDMEDIATOR_H_
 
#include "coupling/InstanceHandling.h"
#include "coupling/services/MultiMDCellService.h"
 
namespace coupling {
template <class LinkedCell, unsigned int> class MultiMDMediator;
}

template <class LinkedCell, unsigned int dim> class coupling::MultiMDMediator {
public:
  MultiMDMediator(coupling::services::MultiMDCellService<LinkedCell, dim>& multiMDCellService, coupling::InstanceHandling<LinkedCell, dim>& instanceHandling,
                  tarch::utils::MultiMDService<dim>& multiMDService, coupling::interface::MacroscopicSolverInterface<dim>* macroscopicSolverInterface)
      : _multiMDCellService(multiMDCellService), _instanceHandling(instanceHandling), _multiMDService(multiMDService),
        _listActiveMDSimulations(_multiMDService.getNumberLocalComms(), std::vector<bool>()), _macroscopicSolverInterface(macroscopicSolverInterface) {
    for (auto& group : _listActiveMDSimulations) {
      group = std::vector<bool>(_multiMDService.getLocalNumberOfMDSimulations(), true);
    }
  }

  void addMDSimulation();

  void addMDSimulation(const unsigned int& communicator);

  void addNMDSimulations(const unsigned int& N);

  void addNMDSimulations(const unsigned int& communicator, const unsigned int& N);

  void rmMDSimulation();

  void rmMDSimulation(const unsigned int& communicator, const unsigned int& index);

  void rmMDSimulation(const unsigned int& communicator);

  void rmNMDSimulations(const unsigned int& N);

  void shutdownCommunicator(const unsigned int& communicator);

  unsigned int getNumberOfActiveMDSimulations(const unsigned int communicator);

  unsigned int getNumberOfActiveMDSimulations();
 
private:
  void addMDSimulationBlock();

  void rmMDSimulationBlock();

  unsigned int getAvgNumberOfActiveMDSimulations();

  unsigned int findFirstCommWithLowLoad();

  unsigned int findFirstCommWithHighLoad();

  unsigned int findInactiveLocalIndex(const unsigned int& communicator);

  unsigned int findActiveLocalIndex(const unsigned int& communicator);
 
  coupling::services::MultiMDCellService<LinkedCell, dim>& _multiMDCellService;
  coupling::InstanceHandling<LinkedCell, dim>& _instanceHandling;
  tarch::utils::MultiMDService<dim>& _multiMDService;
  std::vector<std::vector<bool>> _listActiveMDSimulations; // global list of active (true) and inactive
                                                           // (false) simulations
  coupling::interface::MacroscopicSolverInterface<dim>* _macroscopicSolverInterface;
};
 
#include "coupling/MultiMDMediator.cpph"
 
#endif //_COUPLING_MULTIMDMEDIATOR_H_