root/components/AutomaticGainControl/trunk/AutomaticGainControl/AutomaticGainControl.cpp @ 2433

/****************************************************************************

Copyright 2006 Virginia Polytechnic Institute and State University

This file is part of the OSSIE AutomaticGainControl.

OSSIE AutomaticGainControl 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 2 of the License, or
(at your option) any later version.

OSSIE AutomaticGainControl 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 OSSIE AutomaticGainControl; if not, write to the Free Software
Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA

****************************************************************************/


#include <string>
#include <iostream>
#include "AutomaticGainControl.h"

#undef VERBOSE

AutomaticGainControl_i::AutomaticGainControl_i(const char *uuid, omni_condition *condition) : Resource_impl(uuid), component_running(condition) 
{
    dataOut_0 = new standardInterfaces_i::complexShort_u("data_out");
    dataIn_0 = new standardInterfaces_i::complexShort_p("data_in");

    // set initial values
    // these particular values are good for 8kHz audio
    energy_lo = 4000.0f;
    energy_hi = 8000.0f;
    k_attack = 0.002f;
    k_release = 0.0005f;
    g_min = 1.0f;
    g_max = 100.0f;

    //Create the thread for the writer's processing function 
    processing_thread = new omni_thread(run, (void *) this);

    //Start the thread containing the writer's processing function 
    processing_thread->start();
}

AutomaticGainControl_i::~AutomaticGainControl_i(void)
{   
    delete dataOut_0;
    delete dataIn_0;
}

void AutomaticGainControl_i::run(void * data)
{
    ((AutomaticGainControl_i*) data)->run_loop();
}

CORBA::Object_ptr AutomaticGainControl_i::getPort( const char* portName ) throw (CORBA::SystemException, CF::PortSupplier::UnknownPort)
{
    std::cout << "AutomaticGainControl_i getPort called with : " << portName << std::endl;
    
    CORBA::Object_var p;

    p = dataOut_0->getPort(portName);

    if (!CORBA::is_nil(p))
        return p._retn();

    p = dataIn_0->getPort(portName);

    if (!CORBA::is_nil(p))
        return p._retn();

    /*exception*/
    throw CF::PortSupplier::UnknownPort();
}

void AutomaticGainControl_i::start() throw (CORBA::SystemException, CF::Resource::StartError)
{
    std::cout << "start called on AutomaticGainControl" << std::endl;
}

void AutomaticGainControl_i::stop() throw (CORBA::SystemException, CF::Resource::StopError) 
{  
    std::cout << "stop called on AutomaticGainControl" << std::endl;
}

void AutomaticGainControl_i::releaseObject() throw (CORBA::SystemException, CF::LifeCycle::ReleaseError)
{
    std::cout << "releaseObject called on AutomaticGainControl" << std::endl;
    
    component_running->signal();
}

void AutomaticGainControl_i::initialize() throw (CF::LifeCycle::InitializeError, CORBA::SystemException)
{
    std::cout << "initialize called on AutomaticGainControl" << std::endl;
}

void AutomaticGainControl_i::configure(const CF::Properties& props) throw (CORBA::SystemException, CF::PropertySet::InvalidConfiguration, CF::PropertySet::PartialConfiguration)
{
    CORBA::Float simple_temp;

    std::cout << "configure called on AutomaticGainControl" << std::endl;
    
    std::cout << "props length : " << props.length() << std::endl;

    for (unsigned int i = 0; i <props.length(); i++)
    {
        std::cout << "Property id : " << props[i].id << std::endl;

        if (strcmp(props[i].id, "DCE:aaf97fa0-d184-4d88-9954-3a1334c73d6d") == 0)
        {
            // energy_lo
            props[i].value >>= simple_temp;
            omni_mutex_lock oml(accessPrivateData);
            energy_lo = simple_temp;
#ifdef VERBOSE
            std::cout << "AGC prop (energy_lo): " << simple_temp << std::endl;
#endif
        }

        if (strcmp(props[i].id, "DCE:346e17c9-6678-483a-bffb-1909c64bddc0") == 0)
        {
            // energy_hi
            props[i].value >>= simple_temp;
            omni_mutex_lock oml(accessPrivateData);
            energy_hi = simple_temp;
#ifdef VERBOSE
            std::cout << "AGC prop (energy_hi): " << simple_temp << std::endl;
#endif

        }

        if (strcmp(props[i].id, "DCE:4608b943-4fe2-49df-91fb-afa287b609d4") == 0)
        {
            // k_attack
            props[i].value >>= simple_temp;
            omni_mutex_lock oml(accessPrivateData);
            k_attack = simple_temp;
#ifdef VERBOSE
            std::cout << "AGC prop (k_attack): " << simple_temp << std::endl;
#endif
        }

        if (strcmp(props[i].id, "DCE:491ec3de-ed45-48af-a6fc-ca2d6465e136") == 0)
        {
            // k_release
            props[i].value >>= simple_temp;
            omni_mutex_lock oml(accessPrivateData);
            k_release = simple_temp;
#ifdef VERBOSE
            std::cout << "AGC prop (k_release): " << simple_temp << std::endl;
#endif
        }

        if (strcmp(props[i].id, "DCE:312f63fe-709a-4217-933b-c584c8d6a9bb") == 0)
        {
            // g_min
            props[i].value >>= simple_temp;
            omni_mutex_lock oml(accessPrivateData);
            g_min = simple_temp;
#ifdef VERBOSE
            std::cout << "AGC prop (g_min): " << simple_temp << std::endl;
#endif
        }

        if (strcmp(props[i].id, "DCE:8357ee0d-2417-46d9-8475-2e5778d797e4") == 0)
        {
            // g_max
            props[i].value >>= simple_temp;
            omni_mutex_lock oml(accessPrivateData);
            g_max = simple_temp;
#ifdef VERBOSE
            std::cout << "AGC prop (g_max): " << simple_temp << std::endl;
#endif
        }

    }
}


void AutomaticGainControl_i::run_loop()
{
    std::cout << "AutomaticGainControl's process_data thread started" << std::endl;

    PortTypes::ShortSequence I_out, Q_out;
    
    PortTypes::ShortSequence *I_in(NULL), *Q_in(NULL);
    //CORBA::UShort I_in_length, Q_in_length;

    unsigned int i, N;
    short I, Q;

    SigProc::agc * agc;

    agc = new SigProc::agc();
    agc->set_values(
           energy_lo,
           energy_hi,
           k_attack,
           k_release,
           g_min,
           g_max
         );

    while( true )
    {
        dataIn_0->getData(I_in, Q_in);

        // overwrite agc values with properties
        // lock mutex for asynchronous configure() invocation
        {
            omni_mutex_lock oml(accessPrivateData);
            agc->set_values(
                   energy_lo,
                   energy_hi,
                   k_attack,
                   k_release,
                   g_min,
                   g_max
                 );
        }

        N = I_in->length();

        I_out.length(N); // define output length
        Q_out.length(N); // define output length

        for (i=0; i<N; i++)
        {
            I = (*I_in)[i];
            Q = (*Q_in)[i];
            agc->do_work(I, Q);
            I_out[i] = I;
            Q_out[i] = Q;
        }
#ifdef VERBOSE
        float gain(0.0f), energy(0.0f);
        agc->get_status(gain, energy);
        std::cout << "AutomaticGainControl: gain: " << gain
                  << ", energy: " << energy << std::endl;
#endif

        dataIn_0->bufferEmptied();
        dataOut_0->pushPacket(I_out, Q_out);
    }
}