//
// Copyright 2017-2018 Ettus Research, a National Instruments Company
// Copyright 2019 Ettus Research, a National Instruments Brand
//
// SPDX-License-Identifier: GPL-3.0-or-later
//

#ifndef INCLUDED_UHD_USRP_FE_CONNECTION_PYTHON_HPP
#define INCLUDED_UHD_USRP_FE_CONNECTION_PYTHON_HPP

#include <uhd/usrp/fe_connection.hpp>

void export_fe_connection(py::module& m)
{
    using fe_connection_t = uhd::usrp::fe_connection_t;
    using sampling_t      = fe_connection_t::sampling_t;

    py::enum_<sampling_t>(m, "sampling")
        .value("QUADRATURE", sampling_t::QUADRATURE)
        .value("HETERODYNE", sampling_t::HETERODYNE)
        .value("REAL", sampling_t::REAL);

    py::class_<fe_connection_t>(m, "fe_connection")

        // Constructors
        .def(py::init<sampling_t, bool, bool, bool, double>())
        .def(py::init<const std::string&, double>())

        // Methods
        .def("get_sampling_mode", &fe_connection_t::get_sampling_mode)
        .def("is_iq_swapped", &fe_connection_t::is_iq_swapped)
        .def("is_i_inverted", &fe_connection_t::is_i_inverted)
        .def("is_q_inverted", &fe_connection_t::is_q_inverted)
        .def("get_if_freq", &fe_connection_t::get_if_freq)
        .def("set_if_freq", &fe_connection_t::set_if_freq);
}

#endif /* INCLUDED_UHD_USRP_FE_CONNECTION_PYTHON_HPP */