QGDReceiver.cc

Go to the documentation of this file.

#include "QGDReceiver.hh"
#include "QGDMessageHandler.hh"
#include <TMessage.h> 
#include <TServerSocket.h>
#include <TThread.h>
#include <iostream>
 
ClassImp(QGDReceiver);
 
//TMessage* QGDReceiver::fMessage = NULL;
//TSocket* QGDReceiver::fSocket = NULL;
//QGDMessageHandler* QGDReceiver::fMainFrame = NULL;
//QGDReceiver::Status QGDReceiver::fStatus;
//Int_t QGDReceiver::fPort = 1130;
 
QGDReceiver::QGDReceiver(QGDMessageHandler* mainFrame)
{
    fMessage = NULL;
    fSocket = NULL;
    fReceiveThread = NULL;
    fGetNewDataThread = NULL;
    fMainFrame = mainFrame;
    fStatus = ST_STOP;
    SetStatus(ST_STOP);
}
 
 
QGDReceiver::~QGDReceiver()
{
    Stop();
}
 
void QGDReceiver::Start(Int_t port)
{
    Stop();
    fMessage = NULL;
    fPort = port;
    SetStatus(ST_WAITCLIENT); 
    fReceiveThread = new TThread("WaitNewData",(void(*) (void *))&WaitNewData,(void*) this);
    fReceiveThread->Run();
    fGetNewDataThread = new TThread("GetNewData",(void(*) (void *))&GetNewData,(void*) this);
    fGetNewDataThread->Run();
}
 
void QGDReceiver::Stop()
{
    SetStatus(ST_STOP);
 
 
    if(fGetNewDataThread) {
        while (fGetNewDataThread->GetState() == TThread::kRunningState) usleep (100000);
        TThread::Delete(fGetNewDataThread);
        delete fGetNewDataThread;
    }
    fGetNewDataThread = NULL;
 
    if(fReceiveThread) {
        while (fReceiveThread->GetState() == TThread::kRunningState) usleep (100000);
        TThread::Delete(fReceiveThread);
        delete fReceiveThread;
    }
    fReceiveThread = NULL;
 
    fMessage = NULL;
    if(fSocket != NULL && fSocket != (TSocket*)-1) {
        fSocket->Close();
        delete fSocket;
    }
    fSocket = NULL;
 
 
}
 
void QGDReceiver::SetStatus(QGDReceiver::Status status)
{
    if(fStatus != status) {
        fStatus = status;
        fMainFrame->NotifyReceiverStatus(status);
    }
}
 
void* QGDReceiver::WaitNewData(void* mainframe)
{
    QGDReceiver* me = (QGDReceiver*) mainframe;
    me->SetStatus(ST_WAITCLIENT);
    while(me->GetStatus() != ST_STOP) {
        // create server
        TServerSocket ss(me->fPort,kTRUE);
        ss.SetOption(kNoBlock,1);
        // wait the client for establishing connection
        me->SetStatus(ST_WAITCLIENT);
        me->fSocket = NULL;
        do {
            me->fSocket = ss.Accept();
            usleep(200000);
            if (me->GetStatus()==ST_STOP) {ss.Close(); return NULL;}
        } while(me->fSocket == NULL || me->fSocket == (TSocket*)-1);
        // connection established
        ss.Close();
        //fSocket->Send("Go");
        me->SetStatus(ST_CONNECTED);
 
        // wait for a message and receive it if we are not proccessing one already.
        while(me->fSocket != NULL && me->fSocket->IsValid()) {
            if(me->fMessage == NULL) {
                me->fSocket->Recv(me->fMessage);
                //fSocket->Send("GotIt");
            }
            usleep(10000);
            if (me->GetStatus()==ST_STOP) break;
        }
 
        // connection closed 
        if(me->fSocket  != NULL) {
            me->fSocket->Close();
            delete me->fSocket;
        }
        me->fSocket = NULL;
    }
    return NULL;
}
 
void* QGDReceiver::GetNewData(void* mainframe) 
{ 
    QGDReceiver* me = (QGDReceiver*) mainframe;
    while(me->GetStatus() != ST_STOP) {
//    while(1) {
        usleep(10000);
        if(me->fMessage)  {
            TObject* obj = me->fMessage->ReadObject(me->fMessage->GetClass());
            if(QGDCommandList* cmdList = dynamic_cast<QGDCommandList*>(obj)) {
                QGDCommand cmd = me->fMainFrame->GetCommand(*cmdList);
                if(me->fSocket != NULL && me->fSocket->IsValid()) {
                    TMessage mess(kMESS_OBJECT);
                    mess.WriteObject(&cmd);
                    me->fSocket->Send(mess);
                }
            } else if (QGDModuleList* modList = dynamic_cast<QGDModuleList*>(obj)) {
                me->fMainFrame->DisplayModuleList(*modList);
            } else if (QGDModuleParameterList* modList = dynamic_cast<QGDModuleParameterList*>(obj)) {
                me->fMainFrame->DisplayModuleParameters(*modList);
            } else {
                me->fMainFrame->Display(obj);
            }
            delete me->fMessage;
            me->fMessage = NULL;
        }
    }
    return NULL;
}