root/ossiedev/branches/hvolos/components/MIMODec/MIMODec.cpp @ 9519

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

Copyright 2009 by your_name_or_organization, all rights reserved.

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


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

using namespace arma;

MIMODec_i::MIMODec_i(const char *uuid, omni_condition *condition) :
        Resource_impl(uuid), component_running(condition) {
    dataOut = new standardInterfaces_i::complexShort_u("Symbols_OUT");
    dataIn = new standardInterfaces_i::complexShort_p("MUX_MIMO_Symbols_IN");
    Mr=1;
    Nt=1;
    MIMOmethod=0;
    Modmethod="BPSK";

    //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();

}

MIMODec_i::~MIMODec_i(void) {
    delete dataOut;
    delete dataIn;
}

// Static function for omni thread
void MIMODec_i::Run( void * data ) {
    ((MIMODec_i*)data)->ProcessData();
}

CORBA::Object_ptr MIMODec_i::getPort( const char* portName ) throw (
    CORBA::SystemException, CF::PortSupplier::UnknownPort) {
    DEBUG(3, MIMODec, "getPort() invoked with " << portName)

    CORBA::Object_var p;

    p = dataOut->getPort(portName);

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

    p = dataIn->getPort(portName);

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

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

void MIMODec_i::start() throw (CORBA::SystemException,
                               CF::Resource::StartError) {
    DEBUG(3, MIMODec, "start() invoked")
}

void MIMODec_i::stop() throw (CORBA::SystemException, CF::Resource::StopError) {
    DEBUG(3, MIMODec, "stop() invoked")
}

void MIMODec_i::releaseObject() throw (CORBA::SystemException,
                                       CF::LifeCycle::ReleaseError) {
    DEBUG(3, MIMODec, "releaseObject() invoked")

    component_running->signal();
}

void MIMODec_i::initialize() throw (CF::LifeCycle::InitializeError,
                                    CORBA::SystemException) {
    DEBUG(3, MIMODec, "initialize() invoked")
}

void MIMODec_i::configure(const CF::Properties& props)
throw (CORBA::SystemException,
       CF::PropertySet::InvalidConfiguration,
       CF::PropertySet::PartialConfiguration) {
    DEBUG(3, MIMODec, "configure() invoked")

    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:4c5f385e-3b3e-11de-b7e0-0000e80014f9") == 0) {
            CORBA::Short simple_temp;
            props[i].value >>= simple_temp;
            Nt = simple_temp;
        }

        if (strcmp(props[i].id, "DCE:74a38a7c-3b3e-11de-b7e0-0000e80014f9") == 0) {
            const char * prop_str;
            //string prop_str;
            props[i].value >>= prop_str;
            //prop_str=prop_char;

            if (strcmp(prop_str,"MRC")) {
                MIMOmethod=0;

            } else if (strcmp(prop_str,"STBC")) {
                MIMOmethod=1;

            } else if (strcmp(prop_str,"VBLAST")) {
                MIMOmethod=2;
            };
        }

        if (strcmp(props[i].id, "DCE:a1026f5c-3b3e-11de-b7e0-0000e80014f9") == 0) {
            CORBA::Short simple_temp;
            props[i].value >>= simple_temp;
            Mr = simple_temp;
        }

        if (strcmp(props[i].id, "DCE:17cabf40-3b3f-11de-b7e0-0000e80014f9") == 0) {
            const char * prop_str;
            props[i].value >>= prop_str;
            Modmethod=prop_str;
        }
    }
}

void MIMODec_i::ProcessData() {
    DEBUG(3, MIMODec, "ProcessData() invoked")

    PortTypes::ShortSequence I_out, Q_out;


    PortTypes::ShortSequence *I_in(NULL), *Q_in(NULL);
    CORBA::UShort Data_length;
    short int NH;

    while (1) {
        //Mr=4;
        //Nt=2;

        NH=Mr*Nt;
        dataIn->getData(I_in, Q_in);

        Data_length = I_in->length();


        I_out.length((Data_length-NH)/Mr); //must define length of output
        Q_out.length((Data_length-NH)/Mr); //must define length of output

        /*insert code here to do work*/


        cx_mat H(Mr,Nt);
        MIMOmethod=0;
        std::cout<<Mr<<"X"<<Nt<<" M:"<< MIMOmethod<<" DL:"<<Data_length<<std::endl;
        for (int i=0;i<Mr;i++) {
            for (int j=0;j<Nt;j++) {
                H(i,j)=cx_double((*I_in)[i*Mr+j],(*Q_in)[i*Mr+j])/10000.0;
                //std::cout<<"Iin:"<<(*I_in)[i*2+j]<<"  ";
                std::cout<<"H("<<i<<","<<j<<")="<<H(i,j)<<"  ";
            }
        }
        std::cout<<std::endl;

        cx_double r1;
        cx_double r2;
        cx_mat rx_symbols(((Data_length-NH)/Mr),1);
         short int j=0;

        switch (MIMOmethod) {
        case 0:
            j=0;

            for (short int i=NH; i<Data_length;i+=Mr) {
                rx_symbols(j,0)=0;

                for (short int k=0;k<Mr;k++) {
                    r1=cx_double((*I_in)[i+k],(*Q_in)[i+k]);
                    rx_symbols(j,0)+=r1*conj(H(k,0));
                    std::cout<<i<<","<<j<<","<<k<<std::endl;
                }

                j+=1;
            }

            break;
        case 1:
            //std::cout<<std::endl;

            /*Alamouti
            for i=1:2:length(RX_symbols1),
              r1=RX_symbols1(i);
              r2=RX_symbols1(i+1);
              RX_symbols(i)=r1*conj(H(1,1))+conj(r2)*H(1,2);
              RX_symbols(i+1)=r1*conj(H(1,2))-conj(r2)*H(1,1);
            end;
            */



            j=0;
            //std::cout<<"Decode"<<std::endl;
            //std::cout<<"Hdec="<<H;

            for (short int i=NH; i<Data_length;i+=2*Mr) {
                rx_symbols(j,0)=0;
                rx_symbols(j+1,0)=0;

                for (short int k=0;k<Mr;k++) {
                    r1=cx_double((*I_in)[i+k],(*Q_in)[i+k]);
                    r2=cx_double((*I_in)[i+Mr+k],(*Q_in)[i+Mr+k]);
                    rx_symbols(j,0)+=r1*conj(H(k,0))+conj(r2)*H(k,1);
                    rx_symbols(j+1,0)+=r1*conj(H(k,1))-conj(r2)*H(k,0);

                }

                j+=2;
            }

            break;
        case 2:
            break;
        }

        double scale=1;
        for (short int i=0; i<((Data_length-NH)/Mr);i++) {

            if ( (real(rx_symbols(i,0))>30000)||(imag(rx_symbols(i,0))>30000))
                scale=20;
            else
                scale=1;


            I_out[i]=(CORBA::Short)(real(rx_symbols(i,0))/scale);
            Q_out[i]=(CORBA::Short)(imag(rx_symbols(i,0))/scale);
            //std::cout<<"Symbol"<<rx_symbols(i,0)<<std::endl;
            //std::cout<<"("<<i<<")"<<"I:"<<I_out[i]<<"Q:"<<Q_out[i]<<std::endl;
        }

        dataIn->bufferEmptied();
        dataOut->pushPacket(I_out, Q_out);
    }
}