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

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

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"

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");


    //Create the thread for the writer's processing function 
    processing_thread = new omni_thread(process_data, (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;
}

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)
{
    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)
        {
            CORBA::Float simple_temp;
            props[i].value >>= simple_temp;
            simple_0_value = simple_temp;
        }

        if (strcmp(props[i].id, "DCE:346e17c9-6678-483a-bffb-1909c64bddc0") == 0)
        {
            CORBA::Float simple_temp;
            props[i].value >>= simple_temp;
            simple_1_value = simple_temp;
        }

        if (strcmp(props[i].id, "DCE:4608b943-4fe2-49df-91fb-afa287b609d4") == 0)
        {
            CORBA::Float simple_temp;
            props[i].value >>= simple_temp;
            simple_2_value = simple_temp;
        }

        if (strcmp(props[i].id, "DCE:491ec3de-ed45-48af-a6fc-ca2d6465e136") == 0)
        {
            CORBA::Float simple_temp;
            props[i].value >>= simple_temp;
            simple_3_value = simple_temp;
        }

        if (strcmp(props[i].id, "DCE:312f63fe-709a-4217-933b-c584c8d6a9bb") == 0)
        {
            CORBA::Float simple_temp;
            props[i].value >>= simple_temp;
            simple_4_value = simple_temp;
        }

        if (strcmp(props[i].id, "DCE:8357ee0d-2417-46d9-8475-2e5778d797e4") == 0)
        {
            CORBA::Float simple_temp;
            props[i].value >>= simple_temp;
            simple_5_value = simple_temp;
        }

    }
}

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

    AutomaticGainControl_i *mem = (AutomaticGainControl_i *) data;
    
    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;
    float gain(0.0f), energy(0.0f);

    SigProc::agc * agc;

    // override properties
    mem->energy_lo = 4000.0f;
    mem->energy_hi = 16000.0f;
    mem->k_attack = 0.5f;
    mem->k_release = 0.0f;
    mem->g_max = 10.0f;
    mem->g_min = 0.1f;

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


    while( true )
    {
        std::cout << "AGC waiting for data..." << std::endl;
        mem->dataIn_0->getData(I_in, Q_in);
        std::cout << "AGC got " << I_in->length() << " samples" << std::endl;
        
        N = I_in->length();

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

        for (i=0; i<N; i++)
        {
            I_out[i] = (*I_in)[i];
            Q_out[i] = (*Q_in)[i];
            agc->do_work(I_out[i], Q_out[i], gain, energy);
        }
        std::cout << "gain: " << gain << ", energy: " << energy << std::endl;

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