/*------------------------------------------------------------------------------
--  This file is a part of the SocExplorer Software
--  Copyright (C) 2011, Laboratory of Plasmas Physic - 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 : Alexis Jeandet
--                     Mail : alexis.jeandet@lpp.polytechnique.fr
----------------------------------------------------------------------------*/
#include "dsu3plugin.h"
#include <QFileDialog>
#include <QDir>
#include "dsu3pluginpywrapper.h"
#include <socexplorerengine.h>

dsu3plugin::dsu3plugin(QWidget *parent):socexplorerplugin(parent,false)
{
    this->UI = new dsu3pluginui();
    this->setWidget((QWidget*)this->UI);
    this->elfparserInst = new elfparser();
    this->pyObject = new dsu3pluginPywrapper(this);
    QObject::connect(((dsu3pluginPywrapper*)this->pyObject),SIGNAL(openFile(QString)),this,SLOT(openFile(QString)));
    QObject::connect(((dsu3pluginPywrapper*)this->pyObject),SIGNAL(loadFile()),this,SLOT(flashTarget()));
    QObject::connect(((dsu3pluginPywrapper*)this->pyObject),SIGNAL(run()),this,SLOT(run()));
    connect(this->UI,SIGNAL(openFile()),this,SLOT(openFile()));
    connect(this->UI,SIGNAL(flashTarget()),this,SLOT(flashTarget()));
    connect(this->UI,SIGNAL(run()),this,SLOT(run()));
    connect(this,SIGNAL(updateInfo(elfparser*)),this->UI,SIGNAL(updateInfo(elfparser*)));
}


dsu3plugin::~dsu3plugin()
{

}


void dsu3plugin::openFile()
{
    QString filename = QFileDialog::getOpenFileName(this,tr("Open elf File"), QDir::homePath(), tr("Elf Files (*)"));
    if(filename!="")
    {
        this->openFile(filename);
    }
}

void dsu3plugin::openFile(QString fileName)
{
    this->elfparserInst->setFilename(fileName);
    emit this->updateInfo(this->elfparserInst);
}

bool dsu3plugin::configureTarget()
{
    if(parent==NULL)
        return false;
    unsigned int DSUBASEADDRESS = SocExplorerEngine::self()->getEnumDeviceBaseAddress(this,0x01 , 0x004,0);
    if(DSUBASEADDRESS == (unsigned int)-1)
        DSUBASEADDRESS = 0x90000000;
    unsigned int MCTRLBASEADDRESS = SocExplorerEngine::self()->getEnumDeviceBaseAddress(this,0x04 , 0x00F,0);
    if(MCTRLBASEADDRESS == (unsigned int)-1)
        return false;

    //Force a debug break
    WriteRegs(uIntlist()<<0x0000002f,(unsigned int)DSUBASEADDRESS);
    WriteRegs(uIntlist()<<0x0000ffff,(unsigned int)DSUBASEADDRESS+0x20);
    //Clear time tag counter
    WriteRegs(uIntlist()<<0,(unsigned int)DSUBASEADDRESS+0x8);

    //Clear ASR registers
    WriteRegs(uIntlist()<<0<<0<<0,(unsigned int)DSUBASEADDRESS+0x400040);
    WriteRegs(uIntlist()<<0x2,(unsigned int)DSUBASEADDRESS+0x400024);
    WriteRegs(uIntlist()<<0<<0<<0<<0<<0<<0<<0<<0,(unsigned int)DSUBASEADDRESS+0x400060);
    WriteRegs(uIntlist()<<0,(unsigned int)DSUBASEADDRESS+0x48);
    WriteRegs(uIntlist()<<0,(unsigned int)DSUBASEADDRESS+0x000004C);
    WriteRegs(uIntlist()<<0,(unsigned int)DSUBASEADDRESS+0x400040);



    WriteRegs(uIntlist()<<0x2FF<<0xE60<<0,(unsigned int)MCTRLBASEADDRESS);


    WriteRegs(uIntlist()<<0<<0<<0<<0,(unsigned int)DSUBASEADDRESS+0x400060);
    WriteRegs(uIntlist()<<0x0000FFFF,(unsigned int)DSUBASEADDRESS+0x24);

    unsigned int buff=0;
//    for(int i=0;i<1567;i++)
//    {
//        parent->Write(&buff,(unsigned int)1,DSUBASEADDRESS+0x300000+(4*i));
//    }
    memSet(DSUBASEADDRESS+0x300000,0,1567);
    WriteRegs(uIntlist()<<0<<0xF30000E0<<0x00000002<<0x40000000<<0x40000000<<0x40000004<<0x1000000,(unsigned int)DSUBASEADDRESS+0x400000);
    WriteRegs(uIntlist()<<0<<0<<0<<0<<0<<0<<0x403ffff0<<0<<0<<0<<0<<0<<0<<0<<0<<0<<0<<0<<0<<0<<0<<0<<0<<0,(unsigned int)DSUBASEADDRESS+0x300020);
    WriteRegs(uIntlist()<<0x000002EF,(unsigned int)DSUBASEADDRESS);

    //Disable interrupts
    unsigned int APBIRQCTRLRBASEADD = (unsigned int)SocExplorerEngine::self()->getEnumDeviceBaseAddress(this,1,0x0d,0);
    if(APBIRQCTRLRBASEADD == (unsigned int)-1)
        return false;
    WriteRegs(uIntlist()<<0x00000000,APBIRQCTRLRBASEADD+0x040);
    WriteRegs(uIntlist()<<0xFFFE0000,APBIRQCTRLRBASEADD+0x080);
    WriteRegs(uIntlist()<<0<<0,APBIRQCTRLRBASEADD);

    //Set up timer
    unsigned int APBTIMERBASEADD = (unsigned int)SocExplorerEngine::self()->getEnumDeviceBaseAddress(this,1,0x11,0);
    if(APBTIMERBASEADD == (unsigned int)-1)
        return false;
    WriteRegs(uIntlist()<<0xffffffff,APBTIMERBASEADD+0x014);
    WriteRegs(uIntlist()<<0x00000018,APBTIMERBASEADD+0x04);
    WriteRegs(uIntlist()<<0x00000007,APBTIMERBASEADD+0x018);
    return true;
}

bool dsu3plugin::flashTarget()
{
    configureTarget();

    /*Write .text*/
    this->writeSection(".text");
    /*Write .data*/
    this->writeSection(".data");
    return true;
}

void dsu3plugin::run()
{
    unsigned int DSUBASEADDRESS = SocExplorerEngine::self()->getEnumDeviceBaseAddress(this,0x01 , 0x004,0);
    if(DSUBASEADDRESS == (unsigned int)-1)
        DSUBASEADDRESS = 0x90000000;
    WriteRegs(uIntlist()<<0,DSUBASEADDRESS+0x020);
}

void dsu3plugin::WriteRegs(uIntlist Values, unsigned int address)
{
    unsigned int* buff;
    buff = (unsigned int*)malloc(Values.count()*sizeof(unsigned int));
    for(int i=0;i<Values.count();i++)
    {
        buff[i]=Values.at(i);
    }
    parent->Write(buff,(unsigned int)Values.count(),address);
    free(buff);
}

void dsu3plugin::writeSection(int index)
{
    char* buffch=NULL;
    unsigned int* buff;
    int size = this->elfparserInst->getSectionDatasz(index);
    int sizeInt = size/4;
    if(parent==NULL)
        return;
    this->elfparserInst->getSectionData(index,&buffch);
    buff = (unsigned int*)malloc(((size/4)+1)*sizeof(unsigned int));
    for(int i=0;i<sizeInt;i++)
    {
        buff[i] = 0x0FF & ((unsigned int)buffch[4*i]);
        buff[i] =  (buff[i]<<8) + (0x0FF & ((unsigned int)buffch[(4*i)+1]));
        buff[i] =  (buff[i]<<8) + (0x0FF & ((unsigned int)buffch[(4*i)+2]));
        buff[i] =  (buff[i]<<8) + (0x0FF & ((unsigned int)buffch[(4*i)+3]));
    }
    if(size%4)
    {
        buff[sizeInt]=0;
        for(int i=(size%4);i>0;i--)
        {
            buff[sizeInt] =  (buff[sizeInt]<<8) + (0x0FF & ((unsigned int)buffch[size-i]));
        }
        sizeInt++;
    }
    parent->Write(buff,(unsigned int)sizeInt,(unsigned int)this->elfparserInst->getSectionPaddr(index));
    free(buff);
}

void dsu3plugin::writeSection(const QString &name)
{
    for(int i=0;i<this->elfparserInst->getSectioncount();i++)
    {
        if(!this->elfparserInst->getSectionName(i).compare(name))
        {
            printf("about to write section %s @ 0x%x size = %d\n",elfparserInst->getSectionName(i).toStdString().c_str(),elfparserInst->getSectionPaddr(i),elfparserInst->getSectionMemsz(i));
            writeSection(i);
        }
    }
}


unsigned int dsu3plugin::Write(unsigned int *Value,unsigned int count,unsigned  int address)
{
    if(parent!=NULL)
        return parent->Write(Value,count,address);
    return 0;
}

bool dsu3plugin::memSet(unsigned int address,int value, unsigned int count)
{
    unsigned int* buffer = (unsigned int*)malloc(count*sizeof(unsigned int));
    if(buffer!=NULL)
    {
        memset((void*)buffer,value,count*sizeof(unsigned int));
        parent->Write(buffer,count,address);
        free(buffer );
        return true;
    }
    return false;
}

unsigned int dsu3plugin::Read(unsigned int *Value,unsigned int count, unsigned int address)
{
    if(parent!=NULL)
        return parent->Read(Value,count,address);
    return 0;
}