#include "utility.h"

dist::DissimilarityMeasure obj2m(py::str o) {
    auto s = py::cast<std::string>(py::cast<py::str>(o));
    for(auto m: dist::USABLE_MEASURES) if(!std::strcmp(dist::prob2str(m), s.data())) return m;
    throw std::invalid_argument("invalid key");
}
dist::DissimilarityMeasure obj2m(py::int_ o) {
    auto i = py::cast<py::ssize_t>(o);
    if(!dist::is_valid_measure((dist::DissimilarityMeasure)i)) {
        std::fprintf(stderr, "Warning: measure %td is not considered valid. Downstream work may break.\n", std::ptrdiff_t(i));
    }
    return static_cast<dist::DissimilarityMeasure>(i);
}

coresets::SensitivityMethod obj2sm(py::str o) {
    auto s = py::cast<std::string>(py::cast<py::str>(o));
    for(auto m: coresets::CORESET_CONSTRUCTIONS)
        if(s == coresets::sm2str(m)) return m;
    throw std::invalid_argument("invalid key");
}

coresets::SensitivityMethod obj2sm(py::int_ o) {
    auto i = py::cast<py::ssize_t>(o);
    return static_cast<coresets::SensitivityMethod>(i);
}

dist::DissimilarityMeasure obj2m(py::object o) {
    if(py::isinstance<py::str>(o))
        return obj2m(py::cast<py::str>(o));
    if(py::isinstance<py::int_>(o))
        return obj2m(py::cast<py::int_>(o));
    throw std::runtime_error("Unexpected object");
}
coresets::SensitivityMethod obj2sm(py::object o) {
    if(py::isinstance<py::str>(o))
        return obj2sm(py::cast<py::str>(o));
    if(py::isinstance<py::int_>(o))
        return obj2sm(py::cast<py::int_>(o));
    throw std::runtime_error("Unexpected object");
}