/******************************************************************************
 *
 * 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.
 *
******************************************************************************/

#include <sstream>
#include "api/pybind11_stl_include.h"
#include "api/python_export_utils.h"
#include "gen_component.h"
#include "api/component.h"
#include "api/component_type.h"

namespace MCell {
namespace API {

void GenComponent::check_semantics() const {
  if (!is_set(component_type)) {
    throw ValueError("Parameter 'component_type' must be set.");
  }
}

void GenComponent::set_initialized() {
  if (is_set(component_type)) {
    component_type->set_initialized();
  }
  initialized = true;
}

void GenComponent::set_all_attributes_as_default_or_unset() {
  class_name = "Component";
  component_type = nullptr;
  state = "STATE_UNSET";
  bond = BOND_UNBOUND;
}

std::shared_ptr<Component> GenComponent::copy_component() const {
  std::shared_ptr<Component> res = std::make_shared<Component>(DefaultCtorArgType());
  res->class_name = class_name;
  res->component_type = component_type;
  res->state = state;
  res->bond = bond;

  return res;
}

std::shared_ptr<Component> GenComponent::deepcopy_component(py::dict) const {
  std::shared_ptr<Component> res = std::make_shared<Component>(DefaultCtorArgType());
  res->class_name = class_name;
  res->component_type = is_set(component_type) ? component_type->deepcopy_component_type() : nullptr;
  res->state = state;
  res->bond = bond;

  return res;
}

bool GenComponent::__eq__(const Component& other) const {
  return
    (
      (is_set(component_type)) ?
        (is_set(other.component_type) ?
          (component_type->__eq__(*other.component_type)) : 
          false
        ) :
        (is_set(other.component_type) ?
          false :
          true
        )
     )  &&
    state == other.state &&
    bond == other.bond;
}

bool GenComponent::eq_nonarray_attributes(const Component& other, const bool ignore_name) const {
  return
    (
      (is_set(component_type)) ?
        (is_set(other.component_type) ?
          (component_type->__eq__(*other.component_type)) : 
          false
        ) :
        (is_set(other.component_type) ?
          false :
          true
        )
     )  &&
    state == other.state &&
    bond == other.bond;
}

std::string GenComponent::to_str(const bool all_details, const std::string ind) const {
  std::stringstream ss;
  ss << get_object_name() << ": " <<
      "\n" << ind + "  " << "component_type=" << "(" << ((component_type != nullptr) ? component_type->to_str(all_details, ind + "  ") : "null" ) << ")" << ", " << "\n" << ind + "  " <<
      "state=" << state << ", " <<
      "bond=" << bond;
  return ss.str();
}

py::class_<Component> define_pybinding_Component(py::module& m) {
  return py::class_<Component, std::shared_ptr<Component>>(m, "Component", "Instance of a component type belonging to a molecule instance.\nA component instance must have its state set if there is at least one allowed state.\nIt is also used to connect molecule instance in a complex instance through bonds.\n")
      .def(
          py::init<
            std::shared_ptr<ComponentType>,
            const std::string&,
            const int
          >(),
          py::arg("component_type"),
          py::arg("state") = "STATE_UNSET",
          py::arg("bond") = BOND_UNBOUND
      )
      .def("check_semantics", &Component::check_semantics)
      .def("__copy__", &Component::copy_component)
      .def("__deepcopy__", &Component::deepcopy_component, py::arg("memo"))
      .def("__str__", &Component::to_str, py::arg("all_details") = false, py::arg("ind") = std::string(""))
      .def("__eq__", &Component::__eq__, py::arg("other"))
      .def("to_bngl_str", &Component::to_bngl_str, "Creates a string that corresponds to this component's BNGL representation.")
      .def("dump", &Component::dump)
      .def_property("component_type", &Component::get_component_type, &Component::set_component_type, "Reference to a component type.")
      .def_property("state", &Component::get_state, &Component::set_state, "Specific state value of this component instance.")
      .def_property("bond", &Component::get_bond, &Component::set_bond, "Specific bond for this component instance.\nIt is either a numberical value such as in A(c!1),\nor one of special values BOND_UNBOUND in A(c), \nBOND_BOUND in A(c!+) or BOND_ANY in A(c!?).\n   \n")
    ;
}

std::string GenComponent::export_to_python(std::ostream& out, PythonExportContext& ctx) {
  if (!export_even_if_already_exported() && ctx.already_exported(this)) {
    return ctx.get_exported_name(this);
  }
  std::string exported_name = "component_" + std::to_string(ctx.postinc_counter("component"));
  if (!export_even_if_already_exported()) {
    ctx.add_exported(this, exported_name);
  }

  bool str_export = export_as_string_without_newlines();
  std::string nl = "";
  std::string ind = " ";
  std::stringstream ss;
  if (!str_export) {
    nl = "\n";
    ind = "    ";
    ss << exported_name << " = ";
  }
  ss << "m.Component(" << nl;
  ss << ind << "component_type = " << component_type->export_to_python(out, ctx) << "," << nl;
  if (state != "STATE_UNSET") {
    ss << ind << "state = " << "'" << state << "'" << "," << nl;
  }
  if (bond != BOND_UNBOUND) {
    ss << ind << "bond = " << bond << "," << nl;
  }
  ss << ")" << nl << nl;
  if (!str_export) {
    out << ss.str();
    return exported_name;
  }
  else {
    return ss.str();
  }
}

} // namespace API
} // namespace MCell