artis-star/src/artis-star/kernel/dsde/GraphManager.hpp

/**
 * @file kernel/dsde/GraphManager.hpp
 * @author The ARTIS Development Team
 * See the AUTHORS or Authors.txt file
 */

/*
 * ARTIS - the multimodeling and simulation environment
 * This file is a part of the ARTIS environment
 *
 * Copyright (C) 2013-2019 ULCO http://www.univ-littoral.fr
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

#ifndef DSDE_GRAPH_MANAGER
#define DSDE_GRAPH_MANAGER

#include <artis-star/kernel/pdevs/GraphManager.hpp>

#include <vector>

namespace artis {
namespace dsde {

template<class Time,
    class Parameters = common::NoParameters,
    class GraphParameters = common::NoParameters>
class GraphManager : public pdevs::GraphManager<Time, Parameters, GraphParameters> {
public:
  typedef pdevs::GraphManager<Time, Parameters, GraphParameters> super;
  typedef GraphManager<Time, Parameters, GraphParameters> type;

  GraphManager(common::Coordinator<Time> *coordinator,
               const Parameters &parameters,
               const GraphParameters &graph_parameters)
      :
      pdevs::GraphManager<Time, Parameters, GraphParameters>(coordinator, parameters,
                                                             graph_parameters) {}

  ~GraphManager() override {
    for (auto m: _models) {
      delete m;
    }
  }

  virtual void add_dynamic_child(unsigned int index, common::Model<Time> *child) {
    super::add_child(index, child);
    _models.push_back(child);
    _new_models.push_back(child);
  }

  virtual void add_dynamic_children(unsigned int index, common::Model<Time> *child) {
    super::add_children(index, child);
    _models.push_back(child);
    _new_models.push_back(child);
  }

  void add_link(unsigned int source_model_index, unsigned int source_port_index,
                unsigned int destination_model_index, unsigned int destination_port_index) {
    common::Model<Time> *source_model = super::child_map(source_model_index);
    common::Model<Time> *destination_model = super::child_map(destination_model_index);

    assert(source_model and destination_model);

    this->out({source_model, source_port_index})
        >> this->in({destination_model, destination_port_index});
  }

  void clear_new_models() { _new_models.clear(); }

  std::vector<common::Model<Time> *> &new_models() { return _new_models; }

  void remove_dynamic_child(const typename Time::type &t, unsigned int index) {
    size_t new_index = index
        - (pdevs::GraphManager<Time, Parameters, GraphParameters>::_children.size()
            - _models.size());

    new_index -= std::count_if(_models.begin(), _models.end(),
                               [](common::Model<Time> *model) { return model == nullptr; });
    type::coordinator()->remove_model(t, _models[new_index]);
    super::remove_child(index);
    this->remove_links(_models[new_index]);
    delete _models[new_index];
    _models[new_index] = nullptr;
  }

  void remove_link(unsigned int source_model_index, unsigned int source_port_index,
                   unsigned int destination_model_index, unsigned int destination_port_index) {
    common::Model<Time> *source_model = super::child_map(source_model_index);
    common::Model<Time> *destination_model = super::child_map(destination_model_index);

    assert(source_model and destination_model);

    super::remove_link(source_model, source_port_index, destination_model,
                       destination_port_index);
  }

private:
  std::vector<common::Model<Time> *> _models;
  std::vector<common::Model<Time> *> _new_models;
};

}
}

#endif