[py] added __hash__ for Interval, IntervalVar, IntervalVectorVar

codac-team · Apr 20, 2022 · 5fcaebf · 5fcaebf
1 parent 329f6bf
commit 5fcaebf
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 2 deletions.
diff --git a/python/pybind11 b/python/pybind11
diff --git a/python/src/core/cn/codac_py_Variable.cpp b/python/src/core/cn/codac_py_Variable.cpp
@@ -33,6 +33,11 @@ void export_IntervalVar(py::module& m)
 
     .def(py::init<>(),
       "todo") // todo: doc
+
+    //.attr("__hash__") = py::none()
+    .def("__eq__", [](const IntervalVar& s1, const IntervalVar& s2) { return reinterpret_cast<std::uintptr_t>(&s1) == reinterpret_cast<std::uintptr_t>(&s2); })
+
+    .def("__hash__", [](const IntervalVar& s1) { return reinterpret_cast<std::uintptr_t>(&s1); })
   ;
 }
 
@@ -57,5 +62,9 @@ void export_IntervalVectorVar(py::module& m)
       },
       "todo",
       py::return_value_policy::reference_internal)
-  ;
+
+    .def("__eq__", [](const IntervalVectorVar& s1, const IntervalVectorVar& s2) { return reinterpret_cast<std::uintptr_t>(&s1) == reinterpret_cast<std::uintptr_t>(&s2); })
+
+    .def("__hash__", [](const IntervalVectorVar& s1) { return reinterpret_cast<std::uintptr_t>(&s1); })
+    ;
 }
diff --git a/python/src/core/domains/interval/codac_py_Interval.cpp b/python/src/core/domains/interval/codac_py_Interval.cpp
@@ -199,6 +199,7 @@ void export_Interval(py::module& m)
     .def("bisect", &Interval::bisect, DOCS_INTERVAL_BISECT, py::arg("ratio")=0.5)
     .def("__get_item__", get_item, "x[0] returns the lb and x[1] returns ub")
     .def("copy", &interval_copy, "return a new object which is the copy of x")
+    .def("__hash__", [](const Interval& s1) { return reinterpret_cast<std::uintptr_t>(&s1); })
     //.def( "__pow__", pow__)
 
   // Constants