/******************************************************************************
 *
 * Copyright (C) 2021 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_GEN_GEOMETRY_OBJECT_H
#define API_GEN_GEOMETRY_OBJECT_H

#include "api/api_common.h"
#include "api/region.h"


namespace MCell {
namespace API {

class GeometryObject;
class Color;
class InitialSurfaceRelease;
class Region;
class SurfaceClass;
class SurfaceRegion;
class PythonExportContext;

#define GEOMETRY_OBJECT_CTOR() \
    GeometryObject( \
        const std::string& name_, \
        const std::vector<std::vector<double>> vertex_list_, \
        const std::vector<std::vector<int>> wall_list_, \
        const bool is_bngl_compartment_ = false, \
        const std::string& surface_compartment_name_ = STR_UNSET, \
        const std::vector<std::shared_ptr<SurfaceRegion>> surface_regions_ = std::vector<std::shared_ptr<SurfaceRegion>>(), \
        std::shared_ptr<SurfaceClass> surface_class_ = nullptr, \
        const std::vector<std::shared_ptr<InitialSurfaceRelease>> initial_surface_releases_ = std::vector<std::shared_ptr<InitialSurfaceRelease>>(), \
        std::shared_ptr<Color> initial_color_ = nullptr, \
        const RegionNodeType node_type_ = RegionNodeType::UNSET, \
        std::shared_ptr<Region> left_node_ = nullptr, \
        std::shared_ptr<Region> right_node_ = nullptr \
    )  : GenGeometryObject(node_type_,left_node_,right_node_) { \
      class_name = "GeometryObject"; \
      name = name_; \
      vertex_list = vertex_list_; \
      wall_list = wall_list_; \
      is_bngl_compartment = is_bngl_compartment_; \
      surface_compartment_name = surface_compartment_name_; \
      surface_regions = surface_regions_; \
      surface_class = surface_class_; \
      initial_surface_releases = initial_surface_releases_; \
      initial_color = initial_color_; \
      node_type = node_type_; \
      left_node = left_node_; \
      right_node = right_node_; \
      postprocess_in_ctor(); \
      check_semantics(); \
    } \
    GeometryObject(DefaultCtorArgType) : \
      GenGeometryObject(DefaultCtorArgType()) { \
      set_all_attributes_as_default_or_unset(); \
      set_all_custom_attributes_to_default(); \
    }

class GenGeometryObject: public Region {
public:
  GenGeometryObject( 
      const RegionNodeType node_type_ = RegionNodeType::UNSET, 
      std::shared_ptr<Region> left_node_ = nullptr, 
      std::shared_ptr<Region> right_node_ = nullptr 
  )  : Region(node_type_,left_node_,right_node_)  {
  }
  GenGeometryObject(DefaultCtorArgType) {
  }
  void postprocess_in_ctor() override {}
  void check_semantics() const override;
  void set_initialized() override;
  void set_all_attributes_as_default_or_unset() override;

  std::shared_ptr<GeometryObject> copy_geometry_object() const;
  std::shared_ptr<GeometryObject> deepcopy_geometry_object(py::dict = py::dict()) const;
  virtual bool __eq__(const GeometryObject& other) const;
  virtual bool eq_nonarray_attributes(const GeometryObject& other, const bool ignore_name = false) const;
  bool operator == (const GeometryObject& other) const { return __eq__(other);}
  bool operator != (const GeometryObject& other) const { return !__eq__(other);}
  std::string to_str(const bool all_details=false, const std::string ind="") const override;

  virtual std::string export_to_python(std::ostream& out, PythonExportContext& ctx);
  virtual std::string export_vec_vertex_list(std::ostream& out, PythonExportContext& ctx, const std::string& parent_name);
  virtual std::string export_vec_wall_list(std::ostream& out, PythonExportContext& ctx, const std::string& parent_name);
  virtual std::string export_vec_surface_regions(std::ostream& out, PythonExportContext& ctx, const std::string& parent_name);
  virtual std::string export_vec_initial_surface_releases(std::ostream& out, PythonExportContext& ctx, const std::string& parent_name);


  // --- attributes ---
  std::vector<std::vector<double>> vertex_list;
  virtual void set_vertex_list(const std::vector<std::vector<double>> new_vertex_list_) {
    if (initialized) {
      throw RuntimeError("Value 'vertex_list' of object with name " + name + " (class " + class_name + ") "
                         "cannot be set after model was initialized.");
    }
    cached_data_are_uptodate = false;
    vertex_list = new_vertex_list_;
  }
  virtual std::vector<std::vector<double>>& get_vertex_list() {
    cached_data_are_uptodate = false; // arrays and other data can be modified through getters
    return vertex_list;
  }

  std::vector<std::vector<int>> wall_list;
  virtual void set_wall_list(const std::vector<std::vector<int>> new_wall_list_) {
    if (initialized) {
      throw RuntimeError("Value 'wall_list' of object with name " + name + " (class " + class_name + ") "
                         "cannot be set after model was initialized.");
    }
    cached_data_are_uptodate = false;
    wall_list = new_wall_list_;
  }
  virtual std::vector<std::vector<int>>& get_wall_list() {
    cached_data_are_uptodate = false; // arrays and other data can be modified through getters
    return wall_list;
  }

  bool is_bngl_compartment;
  virtual void set_is_bngl_compartment(const bool new_is_bngl_compartment_) {
    if (initialized) {
      throw RuntimeError("Value 'is_bngl_compartment' of object with name " + name + " (class " + class_name + ") "
                         "cannot be set after model was initialized.");
    }
    cached_data_are_uptodate = false;
    is_bngl_compartment = new_is_bngl_compartment_;
  }
  virtual bool get_is_bngl_compartment() const {
    cached_data_are_uptodate = false; // arrays and other data can be modified through getters
    return is_bngl_compartment;
  }

  std::string surface_compartment_name;
  virtual void set_surface_compartment_name(const std::string& new_surface_compartment_name_) {
    if (initialized) {
      throw RuntimeError("Value 'surface_compartment_name' of object with name " + name + " (class " + class_name + ") "
                         "cannot be set after model was initialized.");
    }
    cached_data_are_uptodate = false;
    surface_compartment_name = new_surface_compartment_name_;
  }
  virtual const std::string& get_surface_compartment_name() const {
    cached_data_are_uptodate = false; // arrays and other data can be modified through getters
    return surface_compartment_name;
  }

  std::vector<std::shared_ptr<SurfaceRegion>> surface_regions;
  virtual void set_surface_regions(const std::vector<std::shared_ptr<SurfaceRegion>> new_surface_regions_) {
    if (initialized) {
      throw RuntimeError("Value 'surface_regions' of object with name " + name + " (class " + class_name + ") "
                         "cannot be set after model was initialized.");
    }
    cached_data_are_uptodate = false;
    surface_regions = new_surface_regions_;
  }
  virtual std::vector<std::shared_ptr<SurfaceRegion>>& get_surface_regions() {
    cached_data_are_uptodate = false; // arrays and other data can be modified through getters
    return surface_regions;
  }

  std::shared_ptr<SurfaceClass> surface_class;
  virtual void set_surface_class(std::shared_ptr<SurfaceClass> new_surface_class_) {
    if (initialized) {
      throw RuntimeError("Value 'surface_class' of object with name " + name + " (class " + class_name + ") "
                         "cannot be set after model was initialized.");
    }
    cached_data_are_uptodate = false;
    surface_class = new_surface_class_;
  }
  virtual std::shared_ptr<SurfaceClass> get_surface_class() const {
    cached_data_are_uptodate = false; // arrays and other data can be modified through getters
    return surface_class;
  }

  std::vector<std::shared_ptr<InitialSurfaceRelease>> initial_surface_releases;
  virtual void set_initial_surface_releases(const std::vector<std::shared_ptr<InitialSurfaceRelease>> new_initial_surface_releases_) {
    if (initialized) {
      throw RuntimeError("Value 'initial_surface_releases' of object with name " + name + " (class " + class_name + ") "
                         "cannot be set after model was initialized.");
    }
    cached_data_are_uptodate = false;
    initial_surface_releases = new_initial_surface_releases_;
  }
  virtual std::vector<std::shared_ptr<InitialSurfaceRelease>>& get_initial_surface_releases() {
    cached_data_are_uptodate = false; // arrays and other data can be modified through getters
    return initial_surface_releases;
  }

  std::shared_ptr<Color> initial_color;
  virtual void set_initial_color(std::shared_ptr<Color> new_initial_color_) {
    if (initialized) {
      throw RuntimeError("Value 'initial_color' of object with name " + name + " (class " + class_name + ") "
                         "cannot be set after model was initialized.");
    }
    cached_data_are_uptodate = false;
    initial_color = new_initial_color_;
  }
  virtual std::shared_ptr<Color> get_initial_color() const {
    cached_data_are_uptodate = false; // arrays and other data can be modified through getters
    return initial_color;
  }

  // --- methods ---
  virtual void translate(const std::vector<double> move) = 0;
}; // GenGeometryObject

class GeometryObject;
py::class_<GeometryObject> define_pybinding_GeometryObject(py::module& m);
} // namespace API
} // namespace MCell

#endif // API_GEN_GEOMETRY_OBJECT_H