|
|
#include <pybind11/pybind11.h> |
|
|
namespace py = pybind11; |
|
|
|
|
|
void init_encoders(py::module &handle) { |
|
|
|
|
|
py::enum_<enums::ENCODER_TYPE>(handle, "ENCODER_TYPE", py::arithmetic()) |
|
|
.value("EXPRESSIVE_ENCODER", enums::ENCODER_TYPE::EXPRESSIVE_ENCODER) |
|
|
.value("NO_ENCODER", enums::ENCODER_TYPE::NO_ENCODER) |
|
|
.export_values(); |
|
|
|
|
|
py::class_<encoder::ExpressiveEncoder>(handle, "ExpressiveEncoder") |
|
|
.def(py::init<>()) |
|
|
.def("encode", &encoder::ExpressiveEncoder::encode) |
|
|
.def("decode", &encoder::ExpressiveEncoder::decode) |
|
|
.def("midi_to_json", &encoder::ExpressiveEncoder::midi_to_json) |
|
|
.def("midi_to_tokens", &encoder::ExpressiveEncoder::midi_to_tokens) |
|
|
.def("json_to_midi", &encoder::ExpressiveEncoder::json_to_midi) |
|
|
.def("json_track_to_midi", &encoder::ExpressiveEncoder::json_track_to_midi) |
|
|
.def("json_to_tokens", &encoder::ExpressiveEncoder::json_to_tokens) |
|
|
.def("tokens_to_json", &encoder::ExpressiveEncoder::tokens_to_json) |
|
|
.def("resample_delta_json", &encoder::ExpressiveEncoder::resample_delta_json) |
|
|
.def("tokens_to_midi", &encoder::ExpressiveEncoder::tokens_to_midi) |
|
|
.def("pretty", &encoder::ExpressiveEncoder::pretty) |
|
|
.def("vocab_size", &encoder::ExpressiveEncoder::vocab_size) |
|
|
.def("get_attribute_control_types", &encoder::ExpressiveEncoder::get_attribute_control_types) |
|
|
.def("set_scheme", &encoder::ExpressiveEncoder::set_scheme) |
|
|
.def_readonly("config", &encoder::ExpressiveEncoder::config) |
|
|
.def_readonly("rep", &encoder::ExpressiveEncoder::rep); |
|
|
|
|
|
} |
|
|
|
