frmdetails.cpp

#include "frmdetails.h"
#include "frmtrends.h"
#include "frmsource.h"
#include "frmdbtag.h"
#include "frmsavetags.h"

#include <QtSql/QSqlQuery>
#include <QtSql/QSqlError>
#include <QDate>
#include <QTime>
#include <QStandardItem>
#include <QModelIndex>
#include <QList>
#include <QString>
#include <QStringList>
#include <QTextStream>
#include <QDateTime>
#include <QMap>
#include <QPair>
#include <QFileDialog>

#include <TBranch.h>
#include <TGraph.h>
#include <fstream>

#include "Debug.h"
#include "cmssw/SiStripFedKey.h"
#include "cmssw/SiStripFecKey.h"

SelectionDetails::SelectionDetails(QWidget* p, const TreeViewerRunInfo& tinfo): 
    QConnectedTabWidget(p)
{
    setupUi(this);

    selModel = new QStandardItemModel(0, 17, this);
    listSelection->setModel(selModel);
    selModel = selModel;

    run    = tinfo.getCurrentRunNumber();
    refrun = tinfo.getReferenceRunNumber();
}

SelectionDetails::~SelectionDetails() {
}

void SelectionDetails::populate(TTree* tree, const QVector<int>& sel, const QString& varname) {
    var = varname;
    QString myQuery;        
    QTextStream queryss(&myQuery);
    queryss << "SELECT DISTINCT ticketId, deviceId, runNumber, tagDescription, tkComment, tkAuthor, timeStampInsert, tkanalysislog.tagNumber"
            << " FROM TkAnalysisLog, TkAnalysisTag"
            << " WHERE TkAnalysisLog.tagNumber = TkAnalysisTag.tagNumber"
            << " AND TkAnalysisLog.timeStampCloseTicket IS NULL" 
            << " order by ticketid, runnumber";
    QSqlQuery query(myQuery);

    QString myTagQuery;        
    QTextStream tagqueryss(&myTagQuery);
    tagqueryss << "select distinct TAGNUMBER,TAGDESCRIPTION from TkAnalysisTag where tagnumber <> 0 order by tagnumber";
    QSqlQuery tagquery(myTagQuery);

    while(tagquery.next()) {
        cmbTagList->addItem(tagquery.value(1).toString(), tagquery.value(0).toString()); 
        dbTags[tagquery.value(0).toInt()] = tagquery.value(1).toString().toStdString();
    }

    QStringList headerLabels;
    headerLabels << "DeviceId";
    headerLabels << varname.toStdString().c_str();
    headerLabels << "Power";
    headerLabels << "Detid";
    headerLabels << "FecCrate";
    headerLabels << "Fec";
    headerLabels << "Ring";
    headerLabels << "Ccu";
    headerLabels << "CcuArrangement";
    headerLabels << "I2CChannel";
    headerLabels << "I2CAddress";
    headerLabels << "lasChan";
    headerLabels << "FedId";
    headerLabels << "FeUnit";
    headerLabels << "FeChan";
    headerLabels << "FeApv";
    headerLabels << "Selected";
    selModel->setHorizontalHeaderLabels(headerLabels);

    TObjString* detector  = new TObjString("");
    double devId          = 0.0;
    double var            = 0.0;
    double power          = 0.0;
    double side           = 0.0;
    double layer          = 0.0;
    double cl             = 0.0;
    double cr             = 0.0;
    double detId          = 0.0;
    double FecCrate       = 0.0;
    double Fec            = 0.0;
    double Ring           = 0.0;
    double Ccu            = 0.0;
    double CcuArrangement = 0.0;
    double I2CChannel     = 0.0;
    double I2CAddress     = 0.0;
    double lasChan        = 0.0;
    double FedId          = 0.0;
    double FeUnit         = 0.0;
    double FeChan         = 0.0;
    double FeApv          = 0.0;

    tree->SetBranchAddress("Detector"                   , &detector);   
    tree->SetBranchAddress("DeviceId"                   , &devId);   
    tree->SetBranchAddress(varname.toStdString().c_str(), &var);
    tree->SetBranchAddress("Power"                      , &power);
    tree->SetBranchAddress("Side"                       , &side);
    tree->SetBranchAddress("Layer"                      , &layer);
    tree->SetBranchAddress("Cl"                         , &cl);
    tree->SetBranchAddress("Cr"                         , &cr);
    tree->SetBranchAddress("Detid"                      , &detId);
    tree->SetBranchAddress("FecCrate"                   , &FecCrate);
    tree->SetBranchAddress("Fec"                        , &Fec);
    tree->SetBranchAddress("Ring"                       , &Ring);
    tree->SetBranchAddress("Ccu"                        , &Ccu);
    tree->SetBranchAddress("CcuArrangement"             , &CcuArrangement);
    tree->SetBranchAddress("I2CChannel"                 , &I2CChannel);
    tree->SetBranchAddress("I2CAddress"                 , &I2CAddress);
    tree->SetBranchAddress("lasChan"                    , &lasChan);
    tree->SetBranchAddress("FedId"                      , &FedId);
    tree->SetBranchAddress("FeUnit"                     , &FeUnit);
    tree->SetBranchAddress("FeChan"                     , &FeChan);
    tree->SetBranchAddress("FeApv"                      , &FeApv);

    for(int i = 0; i < tree->GetEntries(); i++) {
        QList<QStandardItem*> list;
        tree->GetEntry(i);

        QString powerstr = QString(detector->GetString().Data());
        powerstr += ".";
        powerstr += QString::number(side);
        powerstr += ".";
        powerstr += QString::number(layer);
        powerstr += ".";
        powerstr += QString::number(cl);
        powerstr += ".";
        powerstr += QString::number(cr);
        powerstr += ".";
        powerstr += QString::number(power);

        QStandardItem* idevId          = new QStandardItem(QString::number(devId, 'g', 20));
        QStandardItem* ivar            = new QStandardItem(QString::number(var));
        QStandardItem* ipower          = new QStandardItem(powerstr);
        QStandardItem* idetId          = new QStandardItem(QString::number(detId, 'g', 20));
        QStandardItem* iFecCrate       = new QStandardItem(QString::number(FecCrate));
        QStandardItem* iFec            = new QStandardItem(QString::number(Fec));
        QStandardItem* iRing           = new QStandardItem(QString::number(Ring));
        QStandardItem* iCcu            = new QStandardItem(QString::number(Ccu));
        QStandardItem* iCcuArrangement = new QStandardItem(QString::number(CcuArrangement));
        QStandardItem* iI2CChannel     = new QStandardItem(QString::number(I2CChannel));
        QStandardItem* iI2CAddress     = new QStandardItem(QString::number(I2CAddress));
        QStandardItem* ilasChan        = new QStandardItem(QString::number(lasChan+1.0));
        QStandardItem* iFedId          = new QStandardItem(QString::number(FedId));
        QStandardItem* iFeUnit         = new QStandardItem(QString::number(FeUnit));
        QStandardItem* iFeChan         = new QStandardItem(QString::number(FeChan));
        QStandardItem* iFeApv          = new QStandardItem(QString::number(FeApv));
        QStandardItem* iSel            = new QStandardItem(QString::number(sel[i]));

        ivar           ->setTextAlignment(Qt::AlignHCenter);
        ipower         ->setTextAlignment(Qt::AlignHCenter);
        idetId         ->setTextAlignment(Qt::AlignHCenter);
        iFecCrate      ->setTextAlignment(Qt::AlignHCenter);
        iFec           ->setTextAlignment(Qt::AlignHCenter);
        iRing          ->setTextAlignment(Qt::AlignHCenter);
        iCcu           ->setTextAlignment(Qt::AlignHCenter);
        iCcuArrangement->setTextAlignment(Qt::AlignHCenter);
        iI2CChannel    ->setTextAlignment(Qt::AlignHCenter);
        iI2CAddress    ->setTextAlignment(Qt::AlignHCenter);
        ilasChan       ->setTextAlignment(Qt::AlignHCenter);
        iFedId         ->setTextAlignment(Qt::AlignHCenter);
        iFeUnit        ->setTextAlignment(Qt::AlignHCenter);
        iFeChan        ->setTextAlignment(Qt::AlignHCenter);
        iFeApv         ->setTextAlignment(Qt::AlignHCenter);
        iSel           ->setTextAlignment(Qt::AlignHCenter);

        idevId->setCheckable(true);

        query.first();
        while(query.next()) {
            if (query.value(1).toDouble() == devId) {
                QString strtemplate("Ticket Nr: %1(%3)\n from Run Nr: %2\n Author: %5\n Date: %6 Time: %7\n %4");
                QString str = strtemplate.arg(query.value(0).toInt())
                                         .arg(query.value(2).toInt())
                                         .arg(query.value(3).toString())
                                         .arg(query.value(4).toString())
                                         .arg(query.value(5).toString())
                                         .arg(query.value(6).toDate().toString())
                                         .arg(query.value(6).toTime().toString());
                QStandardItem* otkt = new QStandardItem(str);
                idevId->appendRow(otkt);
	            openTktMap.insert(std::pair<int,std::pair<int,QStandardItem*> >(query.value(1).toInt(),std::make_pair(query.value(7).toInt(),otkt)));
            }
        }

        if (sel[i] == 0 && idevId->rowCount() == 0) continue;

        list.append(idevId         );
        list.append(ivar           );
        list.append(ipower         );
        list.append(idetId         );
        list.append(iFecCrate      );
        list.append(iFec           );
        list.append(iRing          );
        list.append(iCcu           );
        list.append(iCcuArrangement);
        list.append(iI2CChannel    );
        list.append(iI2CAddress    );
        list.append(ilasChan       );
        list.append(iFedId         );
        list.append(iFeUnit        );
        list.append(iFeChan        );
        list.append(iFeApv         );
        list.append(iSel           );
        selModel->appendRow(list);
    }

    listSelection->hideColumn(16);
    for (int i = 0; i < selModel->rowCount(); i++) {
        if ((selModel->item(i, 16)->text()).toInt() == 0) listSelection->setRowHidden(i, QModelIndex(), true); 
    }

    listSelection->expandAll();
    for (int i = 0; i < 16; i++) listSelection->resizeColumnToContents(i);
    listSelection->collapseAll();

}

QVector<QStandardItem*> SelectionDetails::getSelectedList() {
    QVector<QStandardItem*> selectedList;
    for (int i = 0; i < selModel->rowCount(); i++) {
        if (selModel->item(i, 0)->checkState() == Qt::Checked) {
            selectedList.push_back(new QStandardItem(selModel->item(i, 0)->text()));
        }
    }
    return selectedList;
}

void SelectionDetails::on_btnSelectAll_clicked() {
    for (int i = 0; i < selModel->rowCount(); i++) {
        if (!listSelection->isRowHidden(i, QModelIndex())) selModel->item(i, 0)->setCheckState(Qt::Checked);
    }
}

void SelectionDetails::on_btnUnselectAll_clicked() {
    for (int i = 0; i < selModel->rowCount(); i++) {
        if (!listSelection->isRowHidden(i, QModelIndex())) selModel->item(i, 0)->setCheckState(Qt::Unchecked);
    }
}

void SelectionDetails::on_btnSelectTagged_clicked() {
    for (int i = 0; i < selModel->rowCount(); i++) {
        if (!listSelection->isRowHidden(i, QModelIndex()) && selModel->item(i, 0)->rowCount() > 0) selModel->item(i, 0)->setCheckState(Qt::Checked);
    }
}

void SelectionDetails::on_btnUnselectTagged_clicked() {
    for (int i = 0; i < selModel->rowCount(); i++) {
        if (!listSelection->isRowHidden(i, QModelIndex()) && selModel->item(i, 0)->rowCount() > 0) selModel->item(i, 0)->setCheckState(Qt::Unchecked);
    }
}

void SelectionDetails::on_btnShowTaggedSelected_clicked() {
    for (int i = 0; i < selModel->rowCount(); i++) {
        if (selModel->item(i, 0)->rowCount() == 0) listSelection->setRowHidden(i, QModelIndex(), true);
        if ((selModel->item(i, 16)->text()).toInt() == 0) listSelection->setRowHidden(i, QModelIndex(), true );
    }
}

void SelectionDetails::on_btnShowAllSelected_clicked() {
    for (int i = 0; i < selModel->rowCount(); i++) {
        if ((selModel->item(i, 16)->text()).toInt() != 0) listSelection->setRowHidden(i, QModelIndex(), false);
        if ((selModel->item(i, 16)->text()).toInt() == 0) listSelection->setRowHidden(i, QModelIndex(), true );
    }        
}

void SelectionDetails::on_btnShowTaggedAll_clicked() {
    for (int i = 0; i < selModel->rowCount(); i++) {
        if (selModel->item(i, 0)->rowCount() == 0) listSelection->setRowHidden(i, QModelIndex(), true);
        if ((selModel->item(i, 16)->text()).toInt() == 0) listSelection->setRowHidden(i, QModelIndex(), false);
    }        
}

void SelectionDetails::on_btnAddTag_clicked() {
    TagCreator* tagger = new TagCreator();
    tagger->show();
}

void SelectionDetails::on_btnTagSelected_clicked() {
    TagUpload* tagger = new TagUpload();

    tagger->setTagDeviceList(getSelectedList());
    tagger->setRunNumber(run.toInt());
    tagger->setUploadTag(cmbTagList->currentIndex()+1, cmbTagList->currentText());
    tagger->setCloseTicketFlag(chkCloseTkt->isChecked());
    tagger->setOpenTicketAndTagList(openTktMap, dbTags);
    tagger->exec();
}

void SelectionDetails::on_btnShowSource_clicked() {
    QVector<QPair<QString, QString> > devices;
    QVector<QPair<unsigned, unsigned> > keys;
    for (int i = 0; i < selModel->rowCount(); i++) {
        if (selModel->item(i, 0)->checkState() == Qt::Checked) {
            devices.push_back(QPair<QString, QString>(selModel->item(i, 3)->text(), selModel->item(i, 10)->text()));
            SiStripFecKey feckey(
                selModel->item(i, 4 )->text().toInt(),
                selModel->item(i, 5 )->text().toInt(),
                selModel->item(i, 6 )->text().toInt(),
                selModel->item(i, 7 )->text().toInt(),
                selModel->item(i, 9 )->text().toInt(),
                selModel->item(i, 11)->text().toInt(),
                selModel->item(i, 10)->text().toInt()
            );
            SiStripFedKey fedkey(
                selModel->item(i, 12)->text().toInt(),
                selModel->item(i, 13)->text().toInt(),
                selModel->item(i, 14)->text().toInt(),
                selModel->item(i, 15)->text().toInt()
            );
            keys.push_back(QPair<unsigned, unsigned>(feckey.key(), fedkey.key()));
        }
    }

    SourceDisplay* source = new SourceDisplay(NULL, run, refrun, devices, keys);
    emit showTabSignal(source, "Source");
}

void SelectionDetails::on_btnShowTrend_clicked() {
    Trends* trends = new Trends;

    QSqlQuery runinfoquery(QString("select runnumber, modedescription from viewallrun left outer join analysis using(runnumber) where runnumber=")+run);
    QString runtype;
    QString runtypetable;
    QString plotvar;
    while(runinfoquery.next()) runtype = runinfoquery.value(1).toString();
    if      (runtype == "TIMING")                                              runtypetable = "ANALYSISTIMING";
    else if (runtype == "OPTOSCAN" || runtype == "GAINSCAN")                   runtypetable = "ANALYSISOPTOSCAN";
    else if (runtype == "VPSPSCAN")                                            runtypetable = "ANALYSISVPSPSCAN";
    else if (runtype == "VERY_FAST_CONNECTION" || runtype == "FASTFEDCABLING") runtypetable = "ANALYSISFASTFEDCABLING";
    else if (runtype == "PEDESTALS" || runtype == "PEDESTAL")                  runtypetable = "ANALYSISPEDESTALS";
    else return;
    plotvar = runtypetable + "." + var;

    QMap<QPair<QString, QString>, TGraph*> graphMap;

    for (int i = 0; i < selModel->rowCount(); i++) {
        if (selModel->item(i, 0)->checkState() == Qt::Checked) {
            QString myQuery;
            QTextStream queryss(&myQuery);

            if (runtype == "PEDESTALS" || runtype == "PEDESTAL") {
            
              queryss << "with input as ( select ? fecslot, ? ringslot, ? ccuaddress, ? i2cchannel, ? i2caddress from dual) "
                  << "select viewdevice.fecslot, viewdevice.ringslot, viewdevice.ccuaddress,viewdevice.i2cchannel, "
                  << qPrintable(plotvar)      << ", viewallrun.runnumber, starttime "
                  << "from "                  << qPrintable(runtypetable)                 << " join analysis on " 
                  << qPrintable(runtypetable) << ".analysisid=analysis.analysisid "
                  << "join viewdevice on "    << qPrintable(runtypetable)                 << ".deviceid=viewdevice.deviceid join apvfec on apvfec.deviceid=viewdevice.deviceid join viewallrun "
                  << " on analysis.runnumber=viewallrun.runnumber and " 
                  << " apvfec.versionmajorid=viewallrun.FECVERSIONMAJORID and " 
                  << "apvfec.versionminorid=viewallrun.FECVERSIONMINORID " 
                  << " where apvmode = ( " 
                  << " select distinct apvmode from apvfec join viewdevice using(deviceid) " 
                  << "join viewallrun on " 
                  << " apvfec.versionmajorid=viewallrun.FECVERSIONMAJORID and " 
                  << "apvfec.versionminorid=viewallrun.FECVERSIONMINORID and fecslot = ( select fecslot from input) and ringslot = ( select ringslot from input) and ccuaddress = ( select ccuaddress from input) and i2cchannel = ( select i2cchannel from input) and i2caddress = ( select i2caddress from input) and viewallrun.runnumber = " << run.toInt() << ")"
                  << "and fecslot= ( select fecslot from input) and "
                  << "ringslot =  (select ringslot from input ) and " 
                  << "ccuaddress= (select ccuaddress from input ) and "
                  << "i2cchannel= (select i2cchannel from input ) and "
                  << "i2caddress= (select i2caddress from input ) order by runnumber";
            
            
            } else {
              
              if (plotvar == "ANALYSISTIMING.TickHeight")  {
                  plotvar = "( case when ANALYSISTIMING.HEIGHT = -131070 then 65535 else ANALYSISTIMING.HEIGHT end)";
              }  
              
              queryss << "select viewdevice.fecslot, viewdevice.ringslot, viewdevice.ccuaddress,viewdevice.i2cchannel, "
                  << qPrintable(plotvar)      << ", runnumber, starttime "
                  << "from "                  << qPrintable(runtypetable)                 << " join analysis on " 
                  << qPrintable(runtypetable) << ".analysisid=analysis.analysisid "
                  << "join viewdevice on "    << qPrintable(runtypetable)                 << ".deviceid=viewdevice.deviceid join run using(runnumber) "
                  << "where fecslot="         << "? and "
                  << "ringslot="              << "? and "
                  << "ccuaddress="            << "? and "
                  << "i2cchannel="            << "? and "
                  << "i2caddress="            << "? order by runnumber";
              
            }

            QVector<QPair<double, double> > graphEntries;
            QSqlQuery query;
            query.prepare(myQuery);
            query.addBindValue((selModel->item(i, 5)->text()).toInt());
            query.addBindValue((selModel->item(i, 6)->text()).toInt());
            query.addBindValue((selModel->item(i, 7)->text()).toInt());
            query.addBindValue((selModel->item(i, 9)->text()).toInt());
            query.addBindValue((selModel->item(i,10)->text()).toInt());
            query.exec();
            if( query.lastError().isValid() ) {
                if (Debug::Inst()->getEnabled()) qDebug() << query.lastError().text();
            } 

            while (query.next()) graphEntries.push_back(QPair<double, double>(query.value(4).toDouble(), double(query.value(6).toDateTime().toMSecsSinceEpoch()/1000)-9466.848e5));
            int n = graphEntries.size();
            double* xvals = new double[n];
            double* yvals = new double[n];
            for (int j = 0; j < n; j++) {
                xvals[j] = graphEntries[j].second;
                yvals[j] = graphEntries[j].first;
            }

            // need a check for zero entries here
            TGraph* gr = new TGraph(n, xvals, yvals);
            graphMap[QPair<QString, QString>(selModel->item(i, 3)->text(), selModel->item(i, 10)->text())] = gr;
        }
    }

    trends->getSelectionInfo(graphMap);
    emit showTabSignal(trends, "Trends");
}

void SelectionDetails::on_btnSave_clicked() {
 
    char* curDir;
    curDir = getenv("PWD");
    QString saveFileName = (QFileDialog::getSaveFileName(this, tr("Save Image"), curDir, tr("Text files (*.txt)")));
    if (Debug::Inst()->getEnabled()) qDebug() << saveFileName;
    if (saveFileName == NULL) return;
    else {
        std::ofstream file_out(qPrintable(saveFileName), std::ios::out);
        file_out << "Detid" << " and " << "I2CAddress" << std::endl; 
        for (int i = 0; i < selModel->rowCount(); i++) {
            if (!listSelection->isRowHidden(i, QModelIndex()) && selModel->item(i, 0)->checkState() == Qt::Checked) {
                file_out << selModel->item(i, 3)->text().toStdString() << '\t' << selModel->item(i, 10)->text().toStdString() << std::endl;
            }
        }
        file_out.close();
    }
}