From cb8c309d5d3063a2d6ef435fce285b8ea5932c16 Mon Sep 17 00:00:00 2001
From: Sebastian Junges <sebastian.junges@rwth-aachen.de>
Date: Sun, 20 Aug 2017 21:52:27 +0200
Subject: [PATCH] Constraint collector: general constructor

---
 src/core/analysis.cpp | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/src/core/analysis.cpp b/src/core/analysis.cpp
index e7ab72e..ba7dfc9 100644
--- a/src/core/analysis.cpp
+++ b/src/core/analysis.cpp
@@ -7,7 +7,9 @@ void define_graph_constraints(py::module& m) {
 
     // ConstraintCollector
     py::class_<storm::analysis::ConstraintCollector<storm::RationalFunction>, std::shared_ptr<storm::analysis::ConstraintCollector<storm::RationalFunction>>>(m, "ConstraintCollector", "Collector for constraints on parametric Markov chains")
-        .def(py::init<storm::models::sparse::Dtmc<storm::RationalFunction>>(), "model")
+            .def("__init__", [](storm::analysis::ConstraintCollector<storm::RationalFunction> &instance, storm::models::sparse::Model<storm::RationalFunction> const& model) -> void {
+                new (&instance) storm::analysis::ConstraintCollector<storm::RationalFunction>(model);
+            }, py::arg("model"))
         .def_property_readonly("wellformed_constraints", &storm::analysis::ConstraintCollector<storm::RationalFunction>::getWellformedConstraints, "Get the constraints ensuring a wellformed model")
         .def_property_readonly("graph_preserving_constraints", &storm::analysis::ConstraintCollector<storm::RationalFunction>::getGraphPreservingConstraints, "Get the constraints ensuring the graph is preserved")
     ;