//
// Copyright 2010 Ettus Research LLC
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with this program.  If not, see <http://www.gnu.org/licenses/>.
//

#include "usrp1_impl.hpp"
#include <uhd/utils/assert.hpp>
#include <uhd/usrp/codec_props.hpp>
#include <boost/bind.hpp>
#include <boost/foreach.hpp>
#include <boost/format.hpp>

using namespace uhd;
using namespace uhd::usrp;

/***********************************************************************
 * Helper Methods
 **********************************************************************/
void usrp1_impl::codec_init(void)
{
    //make proxies
    BOOST_FOREACH(dboard_slot_t dboard_slot, _dboard_slots){
        _rx_codec_proxies[dboard_slot] = wax_obj_proxy::make(
              boost::bind(&usrp1_impl::rx_codec_get, this, _1, _2, dboard_slot),
              boost::bind(&usrp1_impl::rx_codec_set, this, _1, _2, dboard_slot));

        _tx_codec_proxies[dboard_slot] = wax_obj_proxy::make(
              boost::bind(&usrp1_impl::tx_codec_get, this, _1, _2, dboard_slot),
              boost::bind(&usrp1_impl::tx_codec_set, this, _1, _2, dboard_slot));
    }
}    

/***********************************************************************
 * RX Codec Properties
 **********************************************************************/
static const std::string adc_pga_gain_name = "PGA";

void usrp1_impl::rx_codec_get(const wax::obj &key_, wax::obj &val, dboard_slot_t dboard_slot)
{
    named_prop_t key = named_prop_t::extract(key_);

    //handle the get request conditioned on the key
    switch(key.as<codec_prop_t>()) {
    case CODEC_PROP_NAME:
        val = str(boost::format("usrp1 adc - ad9862 - slot %c") % char(dboard_slot));
        return;

    case CODEC_PROP_OTHERS:
        val = prop_names_t();
        return;

    case CODEC_PROP_GAIN_NAMES:
        val = prop_names_t(1, adc_pga_gain_name);
        return;

    case CODEC_PROP_GAIN_RANGE:
        UHD_ASSERT_THROW(key.name == adc_pga_gain_name);
        val = usrp1_codec_ctrl::rx_pga_gain_range;
        return;

    case CODEC_PROP_GAIN_I:
        UHD_ASSERT_THROW(key.name == adc_pga_gain_name);
        val = _codec_ctrls[dboard_slot]->get_rx_pga_gain('A');
        return;

    case CODEC_PROP_GAIN_Q:
        UHD_ASSERT_THROW(key.name == adc_pga_gain_name);
        val = _codec_ctrls[dboard_slot]->get_rx_pga_gain('B');
        return;

    default: UHD_THROW_PROP_GET_ERROR();
    }
}

void usrp1_impl::rx_codec_set(const wax::obj &key_, const wax::obj &val, dboard_slot_t dboard_slot)
{
    named_prop_t key = named_prop_t::extract(key_);

    //handle the set request conditioned on the key
    switch(key.as<codec_prop_t>()) {
    case CODEC_PROP_GAIN_I:
        UHD_ASSERT_THROW(key.name == adc_pga_gain_name);
        _codec_ctrls[dboard_slot]->set_rx_pga_gain(val.as<float>(), 'A');
        return;

    case CODEC_PROP_GAIN_Q:
        UHD_ASSERT_THROW(key.name == adc_pga_gain_name);
        _codec_ctrls[dboard_slot]->set_rx_pga_gain(val.as<float>(), 'B');
        return;

    default: UHD_THROW_PROP_SET_ERROR();
    }
}

/***********************************************************************
 * TX Codec Properties
 **********************************************************************/
static const std::string dac_pga_gain_name = "PGA";

void usrp1_impl::tx_codec_get(const wax::obj &key_, wax::obj &val, dboard_slot_t dboard_slot)
{
    named_prop_t key = named_prop_t::extract(key_);

    //handle the get request conditioned on the key
    switch(key.as<codec_prop_t>()) {
    case CODEC_PROP_NAME:
        val = str(boost::format("usrp1 dac - ad9862 - slot %c") % char(dboard_slot));
        return;

    case CODEC_PROP_OTHERS:
        val = prop_names_t();
        return;

    case CODEC_PROP_GAIN_NAMES:
        val = prop_names_t(1, dac_pga_gain_name);
        return;

    case CODEC_PROP_GAIN_RANGE:
        UHD_ASSERT_THROW(key.name == dac_pga_gain_name);
        val = usrp1_codec_ctrl::tx_pga_gain_range;
        return;

    case CODEC_PROP_GAIN_I: //only one gain for I and Q
    case CODEC_PROP_GAIN_Q:
        UHD_ASSERT_THROW(key.name == dac_pga_gain_name);
        val = _codec_ctrls[dboard_slot]->get_tx_pga_gain();
        return;

    default: UHD_THROW_PROP_GET_ERROR();
    }
}

void usrp1_impl::tx_codec_set(const wax::obj &key_, const wax::obj &val, dboard_slot_t dboard_slot)
{
    named_prop_t key = named_prop_t::extract(key_);

    //handle the set request conditioned on the key
    switch(key.as<codec_prop_t>()){
    case CODEC_PROP_GAIN_I: //only one gain for I and Q
    case CODEC_PROP_GAIN_Q:
        UHD_ASSERT_THROW(key.name == dac_pga_gain_name);
        _codec_ctrls[dboard_slot]->set_tx_pga_gain(val.as<float>());
        return;

    default: UHD_THROW_PROP_SET_ERROR();
    }
}