MultiCouplingScheme.hpp

Go to the documentation of this file.

#pragma once
 
#include <map>
#include <string>
#include <vector>
#include "BaseCouplingScheme.hpp"
#include "cplscheme/Constants.hpp"
#include "logging/Logger.hpp"
#include "m2n/SharedPointer.hpp"
#include "mesh/SharedPointer.hpp"
#include "utils/Helpers.hpp"
 
namespace precice::cplscheme {
class CouplingData;
struct ExchangeData;

class MultiCouplingScheme : public BaseCouplingScheme {
public:
  MultiCouplingScheme(
      double                             maxTime,
      int                                maxTimeWindows,
      double                             timeWindowSize,
      const std::string                 &localParticipant,
      std::map<std::string, m2n::PtrM2N> m2ns,
      const std::string                 &controller,
      int                                minIterations,
      int                                maxIterations);

  void addDataToSend(
      const mesh::PtrData &data,
      mesh::PtrMesh        mesh,
      bool                 requiresInitialization,
      bool                 exchangeSubsteps,
      const std::string   &to);

  void addDataToReceive(
      const mesh::PtrData &data,
      mesh::PtrMesh        mesh,
      bool                 requiresInitialization,
      bool                 exchangeSubsteps,
      const std::string   &from);
 
  void determineInitialDataExchange() override;
 
  std::vector<std::string> getCouplingPartners() const final override;
 
  bool hasAnySendData() final override;
 
private:
  std::map<std::string, m2n::PtrM2N> _m2ns;

  std::map<std::string, DataMap> _receiveDataVector;

  std::map<std::string, DataMap> _sendDataVector;

  std::set<std::string> _receiveInitialFrom;

  std::set<std::string> _sendInitialTo;
 
  logging::Logger _log{"cplscheme::MultiCouplingScheme"};
 
  void exchangeFirstData() final override;
 
  void exchangeSecondData() final override;
 
  bool sendsInitializedDataTo(const std::string &to) const;
 
  bool receivesInitializedDataFrom(const std::string &from) const;
 
  DataMap &getAccelerationData() final override;

  void initializeReceiveDataStorage() final override;

  void exchangeInitialData() final override;

  std::string _controller;

  bool _isController;
};
 
} // namespace precice::cplscheme