//
// Copyright 2024 Ettus Research, a National Instruments Brand
//
// SPDX-License-Identifier: GPL-3.0-or-later
//

#pragma once

#include <uhd/rfnoc/block_controller_factory_python.hpp>
#include <rfnoc/gain/gain_block_control.hpp>

using namespace rfnoc::gain;

void export_gain_block_control(py::module& m)
{
    py::class_<gain_block_control, uhd::rfnoc::noc_block_base, gain_block_control::sptr>(
        m, "gain_block_control")
        .def(py::init(
            &uhd::rfnoc::block_controller_factory<gain_block_control>::make_from))
        .def("set_gain_value", &gain_block_control::set_gain_value, py::arg("gain"))
        .def("get_gain_value", &gain_block_control::get_gain_value)

        ;
}
