#include "prism.h"

// Define python bindings
void define_prism(py::module& m) {

    // Parse prism program
    m.def("parse_program", &storm::parseProgram, "Parse program", py::arg("path"));

    // PrismType
    py::enum_<storm::prism::Program::ModelType>(m, "PrismModelType", "Type of the prism model")
        .value("DTMC", storm::prism::Program::ModelType::DTMC)
        .value("CTMC", storm::prism::Program::ModelType::CTMC)
        .value("MDP", storm::prism::Program::ModelType::MDP)
        .value("CTMDP", storm::prism::Program::ModelType::CTMDP)
        .value("MA", storm::prism::Program::ModelType::MA)
        .value("UNDEFINED", storm::prism::Program::ModelType::UNDEFINED)
    ;
    
    // PrismProgram
    py::class_<storm::prism::Program>(m, "Program", "Prism program")
        .def("nr_modules", &storm::prism::Program::getNumberOfModules, "Get number of modules")
        .def("model_type", &storm::prism::Program::getModelType, "Get model type")
        .def("has_undefined_constants", &storm::prism::Program::hasUndefinedConstants, "Check if program has undefined constants")
    ;
    
}