#include "Capture.h"
#include "Json.h"
#include <QFile>
#include "Config.h"

Capture::Capture(QObject *parent) : QObject(parent)
{
    srcV = NULL;
    srcA = NULL;
    enc = NULL;
    res = NULL;
    uvc = NULL;
    uac = NULL;
}

void Capture::init()
{
    confPath = CAPTURE_CONF;
    if(QFile::exists(CAPTURE_USB_CONF))
        confPath = CAPTURE_USB_CONF;

    if(!QFile::exists(confPath))
        return;

    QVariantMap conf = Json::loadFile(confPath).toMap();

    enc=Link::create("EncodeV");
    res=Link::create("Resample");
    uvc=Link::create("OutputUVC");
    uac=Link::create("OutputAlsa");

    enc->linkV(uvc);
    res->linkA(uac);

    connect(uvc,SIGNAL(newEvent(QString,QVariant)),this,SLOT(onNewEvent(QString,QVariant)));
    update(conf);
}

bool Capture::update(QVariantMap cfg)
{
    if(cfg.isEmpty())
        return false;

    config=cfg;
    if(config["enable"].toBool())
    {
        int srcId=config["src"].toInt();
        Channel *chn=Config::findChannelById(srcId);
        if(chn==NULL)
            return false;

        LinkObject *curSrcV=chn->overlay;
        LinkObject *curSrcA=chn->gain;

        if(srcV != NULL && srcV != curSrcV)
            srcV->unLinkV(enc);

        if(srcA != NULL && srcA != curSrcA)
            srcA->unLinkA(res);

        QVariantMap data;
        data["lnk"]=config["lnk"];
        uac->start(data);

        res->start();
        uvc->start();

        data.clear();
        data["codec"]="jpeg";
        data["snap"]=false;
        data["framerate"]=config["framerate"];
        data["memsize"]=2048000;
        data["bitrate"]=config["bitrate"];
        data["share"]=0;
        enc->start(data);

        if(curSrcV!=NULL && curSrcV != srcV)
            curSrcV->linkV(enc);
        if(curSrcA!=NULL && curSrcA != srcA)
            curSrcA->linkA(res);

        srcV = curSrcV;
        srcA = curSrcA;
    }
    else
    {
        uvc->stop();
        enc->stop();
        uac->stop();
        res->stop();

        if(srcV != NULL)
        {
            srcV->unLinkV(enc);
            srcV = NULL;
        }
        if(srcA != NULL)
        {
            srcA->unLinkA(res);
            srcA = NULL;
        }
    }

    if(confPath == CAPTURE_CONF)
        Json::saveFile(cfg,confPath);
    return true;
}

void Capture::onNewEvent(QString type, QVariant info)
{
    if(type=="resize" && enc!=NULL)
    {
        enc->setData(info.toMap());
    }
}