/******************************************************************************
 *
 * Copyright (C) 2020 by
 * The Salk Institute for Biological Studies
 *
 * Use of this source code is governed by an MIT-style
 * license that can be found in the LICENSE file or at
 * https://opensource.org/licenses/MIT.
 *
******************************************************************************/

#ifndef API_MODEL_H
#define API_MODEL_H

#include "generated/gen_model.h"
#include "api/api_common.h"
#include "api/globals.h"
#include "api/subsystem.h"
#include "api/instantiation.h"
#include "api/observables.h"
#include "api/api_config.h"
#include "api/warnings.h"
#include "api/notifications.h"
#include "api/shared_structs.h"
#include "api/callbacks.h"
#include "api/introspection.h"
#include "api/geometry_object.h"

namespace MCell {

class World;

namespace API {

class MolWallHitInfo;
class WallWallHitInfo;

class Model: public GenModel {
public:

  Model() : initialized(false), world(nullptr), callbacks(this) {
  }
  Model(DefaultCtorArgType) : initialized(false), world(nullptr), callbacks(this) {
  }
  virtual ~Model();

  // from generated template
  void initialize(const bool print_copyright = true) override;
  uint64_t run_iterations(const double iterations) override;
  void end_simulation(const bool print_final_report = true) override;

  void add_subsystem(std::shared_ptr<Subsystem> subsystem) override;
  void add_instantiation(std::shared_ptr<Instantiation> instantiation) override;
  void add_observables(std::shared_ptr<Observables> observables) override;

  void dump_internal_state(const bool with_geometry = false) override;

  void export_data_model(const std::string& file = STR_UNSET) override {
    export_data_model_viz_or_full(file, false, NAME_EXPORT_DATA_MODEL);
  }
  void export_viz_data_model(const std::string& file = STR_UNSET) override {
    export_data_model_viz_or_full(file, true, NAME_EXPORT_VIZ_DATA_MODEL);
  }
  void export_geometry(const std::string& output_files_prefix = STR_UNSET) override;

  void release_molecules(std::shared_ptr<ReleaseSite> release_site) override;

  std::vector<int> run_reaction(
      std::shared_ptr<ReactionRule> reaction_rule,
      const std::vector<int> reactant_ids,
      const double time) override;

  void add_vertex_move(
      std::shared_ptr<GeometryObject> object, const int vertex_index, const std::vector<double> displacement
  ) override;

  std::vector<std::shared_ptr<WallWallHitInfo>> apply_vertex_moves(
      const bool collect_wall_wall_hits = false,
      const bool randomize_order = true
  ) override;

  void pair_molecules(const int id1, const int id2) override;
  void unpair_molecules(const int id1, const int id2) override;
  int get_paired_molecule(const int id) override;
  std::map<uint, uint> get_paired_molecules() override;

  void register_mol_wall_hit_callback(
      const std::function<void(std::shared_ptr<MolWallHitInfo>, py::object)> function,
      py::object context,
      std::shared_ptr<GeometryObject> object = nullptr,
      std::shared_ptr<Species> species = nullptr
  ) override;

  void register_reaction_callback(
      const std::function<bool(std::shared_ptr<ReactionInfo>, py::object)> function,
      py::object context,
      std::shared_ptr<ReactionRule> reaction_rule
  ) override;

  void load_bngl(
      const std::string& file_name,
      const std::string& observables_path_or_file = "",
      std::shared_ptr<Region> default_release_region = nullptr,
      const std::map<std::string, double>& parameter_overrides = std::map<std::string, double>(),
      const CountOutputFormat observables_output_format = CountOutputFormat::AUTOMATIC_FROM_EXTENSION
  ) override;

  void export_to_bngl(
      const std::string& file_name,
      const BNGSimulationMethod simulation_method = BNGSimulationMethod::NONE) override;

  void save_checkpoint(const std::string& custom_dir = STR_UNSET) override;

  void schedule_checkpoint(
      const uint64_t iteration = 0,
      const bool continue_simulation = false,
      const std::string& custom_dir = STR_UNSET) override;

  // overrides from derived classes Subsystem, Instantiation, and Observables
  void add_species(std::shared_ptr<Species> s) override;
  void add_reaction_rule(std::shared_ptr<ReactionRule> r) override;

  void add_surface_class(std::shared_ptr<SurfaceClass> sc) override;
  void add_release_site(std::shared_ptr<ReleaseSite> s) override;
  void add_geometry_object(std::shared_ptr<GeometryObject> o) override;

  void add_viz_output(std::shared_ptr<VizOutput> viz_output) override;
  void add_count(std::shared_ptr<Count> count) override;

  // TODO: this belongs to Instantiation
  std::shared_ptr<GeometryObject> get_geometry_object_with_id(const geometry_object_id_t id);

  // TODO: this belongs to Subsystem
  std::shared_ptr<ReactionRule> get_reaction_rule_with_fwd_id(const BNG::rxn_rule_id_t id);

  bool is_initialized() const {
    return initialized;
  }

  void error_if_initialized(const char* what) {
    if (initialized) {
      throw RuntimeError(S("It is not possible to add ") + what + " once a model was initialized.");
    }
  }

  void dump() const;

  const World* get_world() const {
    return world;
  }
  World* get_world() {
    return world;
  }

private:
  void update_api_vertex_position_using_vertex_move(
      const VertexMoveInfo& move_info);

  void export_data_model_viz_or_full(
      const std::string& file,
      const bool only_for_visualization,
      const char* method_name
  );

  volatile bool initialized;
  World* world;

  Callbacks callbacks;

  std::vector<VertexMoveInfo> vertex_moves;

  // used in get_geometry_object_with_id
  std::map<geometry_object_id_t, std::shared_ptr<API::GeometryObject>> geometry_object_id_to_obj_cache;
};

} // namespace API
} // namespace MCell

#endif // API_MODEL_H