#include "elementreduction.h"
#include <iostream>
#include <QFile>
#include <QTextStream>
ElementReduction::ElementReduction(const QList<BranchStruc *> &branchList):branchList(branchList)
{

}

ElementReduction::~ElementReduction()
{

}

void ElementReduction::calibration(const QString &id,QHash<QString,BranchStruc *> &idToBranch,QHash<QString,QVector<BranchStruc *> > &linkage)
{
    QString to=idToBranch[id]->toID;
    if(linkage[to].length()==1)
    {
        idToBranch[id]->fromID.swap(idToBranch[id]->toID);
    }
}

void ElementReduction::doIt(const QString& rootID)
{
    QHash<QString,QVector<BranchStruc *> > linkage;
    //先记录元件之间的连接关系
    QHash<QString,BranchStruc *> idToBranch;
    foreach(BranchStruc* v,this->branchList)
    {
        idToBranch[v->id]=v;
        linkage[v->fromID].push_back(v);
        linkage[v->toID].push_back(v);
        std::cout<<v->fromID.toStdString()<<" "<<v->id.toStdString()<<" "<<v->toID.toStdString()<<std::endl;
    }
    QDomDocument root;

//    QDomElement element=root.createElement(this->branchList.at(2)->id);
    QDomElement element=root.createElement(rootID);
    this->calibration(rootID,idToBranch,linkage);
    std::cout<<"start "<<rootID.toStdString()<<std::endl;
    root.appendChild(element);
    this->buildTreeTo(element,root,linkage,idToBranch);
//    this->visited.remove(this->branchList.at(2)->id);
    this->buildTreeFrom(element,root,linkage,idToBranch);
    this->merge(element,idToBranch);
    QFile file("1.xml");
    if(file.open(QFile::WriteOnly))
    {
//        root.setContent(&file);
        QTextStream writer(&file);

        root.save(writer,4);
        file.close();
    }
    //看看是不是每个都访问到了
    foreach(BranchStruc* v,this->branchList)
    {
        if(!this->visited.contains(v->id))
        {
            std::cout<<"unvisited "<<v->id.toStdString()<<std::endl;
        }
    }
    //合并

    //从新给连接关系编号

}



void ElementReduction::merge(QDomElement &element,QHash<QString,BranchStruc *>& idToBranch)
{
    return;
    QString parentID=idToBranch[element.tagName()]->id;
    QStringList sep=parentID.split('-');
    QString toID=sep.at(1)+sep.at(0);
    idToBranch[element.tagName()]->toID=toID;
    element.setAttribute("toID",toID);
    //顺便编号
    //采用广度优先
    bool merged=true;
    while(merged)
    {
        merged=false;
        QDomNodeList list=element.childNodes();
        int length=list.length();
        for(int i=0;i<length;i++)
        {
            QDomNode nod=list.at(i);
            QDomElement &node=(QDomElement &)nod;
            QString id;
            id=node.nodeName();
            BranchStruc *branch=idToBranch[id];
            branch->fromID=toID;
            node.setAttribute("fromID",toID);
            if(branch->isZeroBranch)
            {
                branch->dispose=true;
                while(node.childNodes().length()>0)
                {
                    std::cout<<"append "<<node.firstChild().nodeName().toStdString()<<" to "<<element.tagName().toStdString()<<std::endl;
                    element.appendChild(node.removeChild(node.firstChild()));
                }
                element.removeChild(node);
                merged=true;
                break;
            }
        }
    }
    QDomNodeList list=element.childNodes();
    for(int i=0;i<list.length();i++)
    {
        QDomNode nod=list.at(i);
        QDomElement &node=(QDomElement &)nod;
        this->merge((QDomElement &)node,idToBranch);
    }


}


void ElementReduction::buildTreeTo(QDomElement &element,QDomDocument &root,QHash<QString,QVector<BranchStruc *> >& linkage,QHash<QString,BranchStruc *>& idToBranch)
{

    QString id=element.tagName();
        if(this->visited.contains(id))
        {
//            std::cout<<"visited "<<branch->id.toStdString()<<std::endl;
            return;
        }
        this->visited[id]=0;
    BranchStruc * branch=idToBranch[id];
    QString nextTo;
    nextTo=branch->toID;
    QVector<BranchStruc *> nextVec;
    nextVec=linkage[nextTo];
//    std::cout<<nextVec.length()<<" of "<<element.tagName().toStdString()<<std::endl;
    foreach(BranchStruc *n,nextVec)
    {

        if(this->visited.contains(n->id))
        {
            continue;
        }
//        std::cout<<element.tagName().toStdString()<<"to can connect to "<<n->id.toStdString()<<std::endl;
//        std::cout<<"add "<<n->id.toStdString()<<std::endl;
        QDomElement newEle=root.createElement(n->id);
//        std::cout<<"create "<<n->id.toStdString()<<std::endl;
        element.appendChild(newEle);
        this->buildTreeTo(newEle,root,linkage,idToBranch);
    }
}

void ElementReduction::buildTreeFrom(QDomElement &element,QDomDocument &root,QHash<QString,QVector<BranchStruc *> >& linkage,QHash<QString,BranchStruc *>& idToBranch)
{

    QString id=element.tagName();
        if(this->visited.contains(id))
        {
            std::cout<<"visited "<<id.toStdString()<<std::endl;
            return;
        }
        this->visited[id]=0;
    BranchStruc * branch=idToBranch[id];
    QString nextFrom=branch->fromID;
    QVector<BranchStruc *> nextFromVec;
    nextFromVec=linkage[nextFrom];

    foreach(BranchStruc *n,nextFromVec)
    {

        if(this->visited.contains(n->id))
        {
            continue;
        }
        std::cout<<element.tagName().toStdString()<<"from can connect to "<<n->id.toStdString()<<std::endl;
//        std::cout<<"add "<<n->id.toStdString()<<std::endl;
        QDomElement newEle=root.createElement(n->id);
//        std::cout<<"create "<<n->id.toStdString()<<std::endl;
        element.appendChild(newEle);
        this->buildTreeFrom(newEle,root,linkage,idToBranch);
    }
}