/*------------------------------------------------------------------------------
--  This file is a part of the LPPMON Software
--  Copyright (C) 2012, Laboratory of Plasma Physics - CNRS
--
--  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, write to the Free Software
--  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
-------------------------------------------------------------------------------*/
/*--                  Author : Paul LEROY
--                     Mail : paul.leroy@lpp.polytechnique.fr
----------------------------------------------------------------------------*/
#include "rmapplugin.h"
#include <QHostAddress>
#include <QIODevice>
#include <QApplication>
#include <rmappluginpythonwrapper.h>
#include <PythonQt.h>
#include <QTimer>

rmapplugin::rmapplugin(QWidget *parent)
  :lppmonplugin(parent,false)
{
    this->UI = new rmapPluginUI();
    this->setWindowTitle(tr("RMAP and SPW Communication"));
    this->setWidget((QWidget*)this->UI);

    timeCode = 0;
    time_COARSE = 0;
    time_FINE = 0;
    currentBridge = selectedBridgeIsUnknown;

    //**************
    //Python wrapper
    this->pyObject = new rmappluginPythonWrapper();
    connect(this->pyObject,SIGNAL(ReadSig(uint*,uint,uint)),this,SLOT(Read(uint*,uint,uint)));
    connect(this->pyObject,SIGNAL(WriteSig(uint*,uint,uint)),this,SLOT(Write(uint*,uint,uint)));
    //**
    connect( (rmappluginPythonWrapper*)this->pyObject, SIGNAL( WriteSPWSig(char*,uint,char,char) ),
            this, SLOT( WriteSPW(char*,uint,char,char)), Qt::DirectConnection );
    //**
    connect( (rmappluginPythonWrapper*)this->pyObject, SIGNAL( updateTargetAddress(unsigned char) ),
            this, SLOT( setValueTargetAddress(unsigned char)) );
    //**
    connect( (rmappluginPythonWrapper*)this->pyObject, SIGNAL( updateSourceAddress(unsigned char) ),
            this, SLOT( setValueSourceAddress(unsigned char)) );
    //**
    connect( (rmappluginPythonWrapper*)this->pyObject, SIGNAL(sendMessage(QString)),
             this, SLOT(displayOnConsole(QString)) );
    //**
    connect( (rmappluginPythonWrapper*)this->pyObject, SIGNAL(fetchPacketSig()),
             this, SLOT(fetchPacket()), Qt::DirectConnection );
    //***
    connect( (rmappluginPythonWrapper*)this->pyObject, SIGNAL(nbPacketHasChanged(int)),
             this, SLOT(nbPacketHasChanged(int)));
    //**************

    //**************
    // get a smart pointer to the __main__ module of the Python interpreter
    PythonQtObjectPtr context = PythonQt::self()->getMainModule();
    // add a QObject as variable of name "BUTTON_rmapOpenCommunication" to the namespace of the __main__ module
    context.addObject("BUTTON_rmapOpenCommunication", UI->rmapOpenCommunicationButton);
    context.addObject("BUTTON_rmapCloseCommunication", UI->rmapCloseCommunicationButton);
    context.addObject("BUTTON_selectStarDundee", UI->selectStarDundee_BUTTON);
    context.addObject("BUTTON_selectGRESB", UI->selectGRESB_BUTTON);
    context.addObject("GRESB_Bridge", UI->gresbBridge);
    //**************

    connect(UI->rmapOpenCommunicationButton, SIGNAL(clicked()), this, SLOT(openBridge()));
    connect(UI->rmapCloseCommunicationButton, SIGNAL(clicked()), this, SLOT(closeBridge()));
    connect(this, SIGNAL(
                updateStatistics(unsigned char,unsigned char,unsigned char,unsigned char,uint,uint,uint,uint)
                ),
            this->UI->tmStatistics, SLOT(
                updateStatistics(unsigned char,unsigned char,unsigned char,unsigned char,uint,uint,uint,uint)
                ));

    //******
    // GRESB
    connect(this->UI->gresbBridge, SIGNAL(sendMessage(QString)), this, SLOT(displayOnConsole(QString)));
    connect(this->UI->gresbBridge, SIGNAL(isOpen(bool)), this, SLOT(activatePlugin(bool)));
    connect(this->UI->gresbBridge, SIGNAL(RMAP_write_reply_setText(QString)), this, SLOT(RMAP_write_reply_setText(QString)));
    connect(this->UI->gresbBridge, SIGNAL(appendToLog(QString)), this, SLOT(appendToLog(QString)));
    connect(this, SIGNAL(ccsdsPacketIsProcessed()), this->UI->gresbBridge, SLOT(ccsdsPacketIsProcessed()));
    connect(this->UI->rmapTargetLogicalAddressSpinBox, SIGNAL(valueChanged(int)),
            this->UI->gresbBridge, SLOT(targetHasChanged(int)));
    connect(this->UI->rmapSourceLogicalAddressSpinBox, SIGNAL(valueChanged(int)),
            this->UI->gresbBridge, SLOT(sourceHasChanged(int)));
    connect(this->UI->gresbBridge, SIGNAL(sendPacket(TMPacketToRead*)),
            this, SLOT(receivePacketFromBridge(TMPacketToRead*)), Qt::DirectConnection);

    //************
    // Star Dundee
    connect(this->UI->starDundee, SIGNAL(sendMessage(QString)), this, SLOT(displayOnConsole(QString)));
    connect(this->UI->starDundee, SIGNAL(isOpen(bool)), this, SLOT(activatePlugin(bool)));
    connect(this->UI->starDundee, SIGNAL(RMAP_write_reply_setText(QString)), this, SLOT(RMAP_write_reply_setText(QString)));
    connect(this->UI->starDundee, SIGNAL(appendToLog(QString)), this, SLOT(appendToLog(QString)));
    connect(this, SIGNAL(ccsdsPacketIsProcessed()), this->UI->starDundee, SLOT(ccsdsPacketIsProcessed()));
    connect(this->UI->rmapTargetLogicalAddressSpinBox, SIGNAL(valueChanged(int)),
            this->UI->starDundee, SLOT(targetHasChanged(int)));
    connect(this->UI->rmapSourceLogicalAddressSpinBox, SIGNAL(valueChanged(int)),
            this->UI->starDundee, SLOT(sourceHasChanged(int)));
    connect(this->UI->starDundee, SIGNAL(sendPacket(TMPacketToRead*)),
            this, SLOT(receivePacketFromBridge(TMPacketToRead*)), Qt::DirectConnection);

    connect(this->UI, SIGNAL(bridgeHasChanged(selectedBridge)), this, SLOT(bridgeHasChanged(selectedBridge)));

    ((rmappluginPythonWrapper*)this->pyObject)->ccsdsPacketStore = &(this->generalCCSDSPacketStore);
}

rmapplugin::~rmapplugin()
{
    switch(currentBridge)
    {
    case selectedBridgeIsGRESB :
        if (RMAPSend_SOCKET->isOpen()) RMAPSend_SOCKET->disconnectFromHost();
        if (RMAPReceive_SOCKET->isOpen()) RMAPReceive_SOCKET->disconnectFromHost();
        if (GRESBStatusQuery_SOCKET->isOpen()) GRESBStatusQuery_SOCKET->disconnectFromHost();
        break;
    case selectedBridgeIsStarDundee :
        break;
    default:
        break;
    }
}

unsigned int rmapplugin::Write(unsigned int *Value, unsigned int count, unsigned int address)
{
    unsigned int result;
    switch(currentBridge)
    {
    case selectedBridgeIsGRESB :
        result = UI->gresbBridge->Write(Value, count, address);
        break;
    case selectedBridgeIsStarDundee :
        result = UI->starDundee->Write(Value, count, address);
        break;
    default:
        result = 1;
        break;
    }
    return result;
}

unsigned int rmapplugin::Read(unsigned int *Value, unsigned int count, unsigned int address)
{
    unsigned int result;
    switch(currentBridge)
    {
    case selectedBridgeIsGRESB :
        result = UI->gresbBridge->Read(Value, count, address);
        break;
    case selectedBridgeIsStarDundee :
        result = UI->starDundee->Read(Value, count, address);
        break;
    default:
        result = 1;
        break;
    }
    return result;
}

////////
// SLOTS

unsigned int rmapplugin::WriteSPW(char *Value, unsigned int count, char targetLogicalAddress, char userApplication) // SLOT
{
    unsigned int result;
    switch(currentBridge)
    {
    case selectedBridgeIsGRESB :
        result = UI->gresbBridge->WriteSPW(Value, count, targetLogicalAddress, userApplication);
        break;
    case selectedBridgeIsStarDundee :
        result = UI->starDundee->WriteSPW(Value, count, targetLogicalAddress, userApplication);
        break;
    default:
        result = 1;
        break;
    }
    return result;
}

void rmapplugin::openBridge()
{
    switch(currentBridge)
    {
    case selectedBridgeIsGRESB :
        this->UI->gresbBridge->Open();
        break;
    case selectedBridgeIsStarDundee :
        this->UI->starDundee->Open();
        break;
    default:
        break;
    }
}

void rmapplugin::closeBridge()
{
    switch(currentBridge)
    {
    case selectedBridgeIsGRESB :
        this->UI->gresbBridge->Close();
        break;
    case selectedBridgeIsStarDundee :
        this->UI->starDundee->Close();
        break;
    default:
        break;
    }
}

void rmapplugin::RMAP_write_reply_setText(QString text)
{
    this->UI->RMAP_write_reply->setText(text);
}

void rmapplugin::appendToLog(QString text)
{
    APPENDTOLOG(text);
}

void rmapplugin::setValueTargetAddress(unsigned char newAddress)
{
    this->UI->rmapTargetLogicalAddressSpinBox->setValue(newAddress);
}

void rmapplugin::setValueSourceAddress(unsigned char newAddress)
{
    this->UI->rmapSourceLogicalAddressSpinBox->setValue(newAddress);
}

void rmapplugin::receivePacketFromBridge(TMPacketToRead *packet)
{
    preProcessPacket(packet);

    // Send the packet to the TM echo bridge for processing
    this->UI->tmEchoBridge->sendTMPacket(packet);
    //this->UI->tmEchoBridge->sendTMPacket_alternative(packet);

    this->generalCCSDSPacketStore.append(packet);
    this->UI->nbPacketInStore->setText("nb packets in store: " + QString::number(generalCCSDSPacketStore.size()));
    processPacketStore();
}

void rmapplugin::preProcessPacket(TMPacketToRead *packet)
{
    unsigned char pid = 0;
    unsigned char cat = 0;
    unsigned char typ = 0;
    unsigned char sub = 0;
    unsigned int sid = 0;
    unsigned int length = 0;
    unsigned int coarse_t = 0;
    unsigned int fine_t = 0;

    //*********************************
    // get the parameters of the packet
    pid = ((packet->Value[4] & 0x07) << 4) + ((packet->Value[5] & 0xf0) >> 4);
    cat = packet->Value[5] & 0x0f;
    typ = packet->Value[11];    // TYPE
    sub = packet->Value[12];    // SUBTYPE
    sid = 0;
    length = packet->Value[8] * 256 + packet->Value[9];
    coarse_t = packet->Value[14] * pow(2, 24) + packet->Value[15] * pow(2, 16)
            + packet->Value[16] * pow(2, 8) + packet->Value[17];
    fine_t = packet->Value[18] * pow(2, 8) + packet->Value[19];

    if ((pid == 76) & (cat == 1) & (typ == 1) & (sub == 8))
            sid = packet->Value[20] * 256 + packet->Value[21];
    else if ((pid == 76) & (cat == 4) & (typ == 3) & (sub == 25))
            sid = 1;
    else if ((pid == 76) & (cat == 12) & (typ == 21) & (sub == 3))
            sid = packet->Value[20];
    else if ((pid == 79) & (cat == 12) & (typ == 21) & (sub == 3))
            sid = packet->Value[20];

    emit updateStatistics(pid, cat, typ, sub, sid, length, coarse_t, fine_t);


    //****************************************
    // if the packet is a waveform, display it
    /*if ( (typ == 21) & (sub == 3) )
    {
        sid = packet->Value[20];    // SID
        switch (sid){
            case SID_NORMAL_SWF_F0:
                buildWFAndDisplay(packet, &wfPacketNormal[0], 0);
                break;
            case SID_NORMAL_SWF_F1:
                buildWFAndDisplay(packet, &wfPacketNormal[1], 1);
                break;
            case SID_NORMAL_SWF_F2:
                buildWFAndDisplay(packet, &wfPacketNormal[2], 2);
                break;
            case SID_NORMAL_CWF_F3:
                buildWFAndDisplay(packet, &wfPacketNormal[3], 3);
                break;
        }
    }*/
}

void rmapplugin::nbPacketHasChanged(int nb)
{
    this->UI->nbPacketInStore->setText("nb packets in store: " + QString::number(nb));
}

void rmapplugin::buildWFAndDisplay(TMPacketToRead *packet, WFPacket *wfPacket, unsigned char num_page)
{
    unsigned int i = 0;
    unsigned int j = 0;
    unsigned char *data;
    unsigned char pkt_nr = 0;
    unsigned int blk_nr = 0;

    pkt_nr = packet->Value[23]; // PKT_NR
    blk_nr = packet->Value[24] * 256 + packet->Value[25];
    data = &packet->Value[26]; // start of the first data block;
    j = (pkt_nr-1) * 340;
    for ( i=0; i<blk_nr; i++ ){
        wfPacket->wf_e1[j + i] = (short) ( (data[ (i * BLK_SIZE)     ] << 8) + (data[ (i*BLK_SIZE) + 1])  );
        wfPacket->wf_v[j + i]  = (short) ( (data[ (i * BLK_SIZE)  + 2] << 8) + (data[ (i*BLK_SIZE) + 3])  );
        wfPacket->wf_b1[j + i] = (short) ( (data[ (i * BLK_SIZE)  + 4] << 8) + (data[ (i*BLK_SIZE) + 5])  );
        wfPacket->wf_e2[j + i] = (short) ( (data[ (i * BLK_SIZE)  + 6] << 8) + (data[ (i*BLK_SIZE) + 7])  );
        wfPacket->wf_b3[j + i] = (short) ( (data[ (i * BLK_SIZE)  + 8] << 8) + (data[ (i*BLK_SIZE) + 9])  );
        wfPacket->wf_b2[j + i] = (short) ( (data[ (i * BLK_SIZE) + 10] << 8) + (data[ (i*BLK_SIZE) + 11]) );
    }
    if (pkt_nr == 7)
    {
        this->UI->wfDisplay->displayOnPlot(wfPacket->wf_v,  num_page, 0);
        this->UI->wfDisplay->displayOnPlot(wfPacket->wf_e1, num_page, 1);
        this->UI->wfDisplay->displayOnPlot(wfPacket->wf_e2, num_page, 2);
        this->UI->wfDisplay->displayOnPlot(wfPacket->wf_b1, num_page, 3);
        this->UI->wfDisplay->displayOnPlot(wfPacket->wf_b2, num_page, 4);
        this->UI->wfDisplay->displayOnPlot(wfPacket->wf_b3, num_page, 5);
    }
}

/////////////////////
// INTERNAL FUNCTIONS

void rmapplugin::processCCSDSPacket(unsigned char *ccsdsPacket, unsigned int size) // SLOT
{
    QString message;
    unsigned int fine_time_value = 0;
    fine_time_value = ((unsigned int) ccsdsPacket[7]<<24)
            + ((unsigned int) ccsdsPacket[6]<<16)
            + ((unsigned int) ccsdsPacket[5]<<8)
            + ((unsigned int) ccsdsPacket[4]);
    message.append(QTime::currentTime().toString() +":" + QString::number(QTime::currentTime().msec()) + ": ");
    message.append("size "
                   + QString::number(size)
                   +" *** header "
                   + QString::number(ccsdsPacket[0], 16)
                   + " "
                   + QString::number(ccsdsPacket[1], 16)
                   + " "
                   + QString::number(ccsdsPacket[2], 16)
                   + " "
                   + QString::number(ccsdsPacket[3], 16)
                   + " *** coarse time "
                   + QString::number(fine_time_value));
                   //+ QString::number(ccsdsPacket[4], 16)
                   //+" "
                   //+ QString::number(ccsdsPacket[5], 16)
                   //+" "
                   //+ QString::number(ccsdsPacket[6], 16)
                   //+" "
                   //+ QString::number(ccsdsPacket[7], 16));
    displayOnConsole(message);
    //((rmappluginPythonWrapper*)this->pyObject)->storeCCSDSPacket(ccsdsPacket, size);
    emit ccsdsPacketIsProcessed();
}

void rmapplugin::processPacketStore()
{
    ((rmappluginPythonWrapper*)this->pyObject)->processPacketStore();
}

int rmapplugin::fetchPacket()
{
    int ret = 0;

    switch(currentBridge)
    {
    case selectedBridgeIsGRESB :
        break;
    case selectedBridgeIsStarDundee :
        ret = this->UI->starDundee->receiveSPWPacketLoop();
        break;
    default:
        break;
    }

    return ret;
}