logplus/ModuleConsole/src/SubmitWorkflowDialog.cpp

/**
 * @file SubmitWorkflowDialog.cpp
 * @date 2014-12-10
 */
#include "SubmitWorkflowDialog.h"
#include "PaiLabel.h"
#include "PaiProgressBar.h"
#include "PaiPushButton.h"
#include "PaiRadioButton.h"
#include "PaiNameLineEdit.h"

#include "PaiCollapsibleWidget.h"
#include "error.h"

#include <QVBoxLayout>
#include <QHBoxLayout>
#include <QGridLayout>
#include <QSpacerItem>
#include <QLabel>
#include <QDebug>
#include <QTableWidget>
#include <QRadioButton>
#include <QHeaderView>
#include <QStackedWidget>
#include <QButtonGroup>

using namespace pai::gui;

SubmitWorkflowDialog::SubmitWorkflowDialog(QWidget *pParent) :
    PaiDialog(pParent),
    m_pTable(NULL),
    m_priority(pai::ios::job::Normal),
    m_currentSQueue("")
{
    //布局1区
    QLabel *pText = new QLabel("Select queue type:",this);
    PaiRadioButton *pNRadioBtn = new PaiRadioButton("Normal queue", this);
    pNRadioBtn->setChecked(true);
    PaiRadioButton *pSRadioBtn = new PaiRadioButton("Special queue", this);
    //设置互斥
    QButtonGroup *pGroup = new QButtonGroup(this);
    pGroup->addButton(pNRadioBtn);
    pGroup->addButton(pSRadioBtn);
    pGroup->setExclusive(true);

    connect(pNRadioBtn,SIGNAL(toggled(bool)),this,SLOT(ToggleNormalRadio(bool)));
    connect(pSRadioBtn,SIGNAL(toggled(bool)),this,SLOT(ToggleSpecialRadio(bool)));

    QHBoxLayout *pRadioHLayout = new QHBoxLayout();
    pRadioHLayout->addWidget(pNRadioBtn);
    pRadioHLayout->addWidget(pSRadioBtn);
    pRadioHLayout->addStretch();

    //一条线
    QFrame *pLineOne = new QFrame(this);
    pLineOne->setFrameShape(QFrame::HLine);
    pLineOne->setFrameShadow(QFrame::Sunken);

    QVBoxLayout *pRadioVLayout = new QVBoxLayout();
    pRadioVLayout->addWidget(pText);
    pRadioVLayout->addLayout(pRadioHLayout);
    pRadioVLayout->addWidget(pLineOne);
    //布局2区
    m_pStack = new QStackedWidget(this);
    m_pCpuWgt = new QWidget(this);
    m_pSpecialCpuWgt = new QWidget(this);
    m_pStack->addWidget(m_pCpuWgt);
    m_pStack->addWidget(m_pSpecialCpuWgt);
    m_pStack->setCurrentWidget(m_pCpuWgt);
    m_pStack->setMinimumHeight(155);
    m_pStack->setMinimumWidth(155);
    //布局3区
    m_pStorage = new QWidget(this);

    //一条线
    QFrame *pLineTwo = new QFrame(this);
    pLineTwo->setFrameShape(QFrame::HLine);
    pLineTwo->setFrameShadow(QFrame::Sunken);
    //布局4区
    QLabel *pJNText = new QLabel("Job Name:",this);
    m_pJobNameEdit = new PaiNameLineEdit(this);
    PaiLabel *pMaxLenMsg = new PaiLabel(this);
    m_priorityCheckBox = new QCheckBox("Submit with high priority",this);
    connect(m_priorityCheckBox,SIGNAL(stateChanged(int)),this,SLOT(SetPriority(int)));
    QVBoxLayout *pBoxLayout = new QVBoxLayout();
    pBoxLayout->addWidget(pJNText);
    pBoxLayout->addWidget(m_pJobNameEdit);
    pBoxLayout->addWidget(pMaxLenMsg);
    pBoxLayout->addWidget(m_priorityCheckBox);
    pBoxLayout->addWidget(pLineTwo);
    pBoxLayout->addItem(new QSpacerItem(40, 20, QSizePolicy::Expanding, QSizePolicy::Minimum));
    m_pJobNameEdit->SetPromptLabel(pMaxLenMsg);
    m_pJobNameEdit->setMinimumWidth(350);
    //布局5区
    m_pContinuePBtn = new PaiPushButton(tr("&Continue"), this);
    m_pCancelPBtn   = new PaiPushButton(tr("&Cancel"), this);

    QHBoxLayout *pBtnHLayout = new QHBoxLayout();
    pBtnHLayout->addStretch();
    pBtnHLayout->addWidget(m_pContinuePBtn);
    pBtnHLayout->addWidget(m_pCancelPBtn);
    //总布局
    QVBoxLayout *pMainLayout = new QVBoxLayout(GetContainer());
    pMainLayout->addLayout(pRadioVLayout);
    pMainLayout->addWidget(m_pStack);
    pMainLayout->addWidget(m_pStorage);
    pMainLayout->addLayout(pBoxLayout);
    pMainLayout->addLayout(pBtnHLayout);

    connect(m_pContinuePBtn, SIGNAL(clicked()), this, SLOT(accept()));
    connect(m_pCancelPBtn, SIGNAL(clicked()), this, SLOT(reject()));
    //绘制特殊队列的2区
    SetSpecialCpuWidget();
}

SubmitWorkflowDialog::~SubmitWorkflowDialog()
{
}

void SubmitWorkflowDialog::SetPriority(int checked)
{
    if(checked)
    {
        m_priority = pai::ios::job::High;
    }
    else
    {
        m_priority = pai::ios::job::Normal;
    }
}

pai::ios::job::JobPriority SubmitWorkflowDialog::GetPriority()
{
    return m_priority;
}

void SubmitWorkflowDialog::ClearCpuWidget()
{
    delete m_pCpuWgt->layout();

    QObjectList objects = m_pCpuWgt->children();
    foreach(QObject *pObj, objects)
    {
        pObj->setParent(NULL);
        delete pObj;
    }
    update();
}

QString SubmitWorkflowDialog::GetQueueName()
{
    return m_currentSQueue;
}

void SubmitWorkflowDialog::SetJobName(const QString& jobName)
{
    if(m_pJobNameEdit && !jobName.isEmpty())
    {
        m_pJobNameEdit->setFocus();
        m_pJobNameEdit->setText(jobName);
        m_pJobNameEdit->selectAll();
    }
}

QWidget* SubmitWorkflowDialog::AddItemWidget(const ResItem& item, PaiLabel *pIcon)
{
    if(item.queueName.isEmpty())
    {
        return new QWidget();
    }
    //item主标题
    PaiLabel *pMainTitleLabel = new PaiLabel(item.mainTitle, this);
    pMainTitleLabel->setObjectName("MainTitleLabel");
    QSizePolicy sizePolicy(QSizePolicy::Expanding, QSizePolicy::Expanding);
    sizePolicy.setHorizontalStretch(0);
    sizePolicy.setVerticalStretch(0);
    sizePolicy.setHeightForWidth(pMainTitleLabel->sizePolicy().hasHeightForWidth());
    pMainTitleLabel->setSizePolicy(sizePolicy);

    QHBoxLayout *pMainTitleHLayout = new QHBoxLayout();
    pMainTitleHLayout->addWidget(pIcon);
    pMainTitleHLayout->addWidget(pMainTitleLabel);

    
    //一条分割线
    QFrame *pLine = new QFrame(this);
    pLine->setFrameShape(QFrame::HLine);
    pLine->setFrameShadow(QFrame::Sunken);
    //当前展示的Widget
    QWidget *pItemWidget = new QWidget();
    pItemWidget->setObjectName("itemWidget");
    QVBoxLayout *pItemVLayout = new QVBoxLayout(pItemWidget);
    pItemVLayout->setContentsMargins(0, 0, 0, 0);
    pItemVLayout->addLayout(pMainTitleHLayout);
    pItemVLayout->addWidget(pLine);
    return pItemWidget;
}

void SubmitWorkflowDialog::SetCpuWidget()
{
    //清理上次的布局
    ClearCpuWidget();

    //在容器上布局当前item的Widget
    QVBoxLayout *pPanelLayout = new QVBoxLayout(m_pCpuWgt);
    pPanelLayout->setContentsMargins(0, 0 ,0, 0);
    pPanelLayout->addItem(new QSpacerItem(40, 25, QSizePolicy::Expanding, QSizePolicy::Minimum));

    ResItem item = m_cpuItem;
    PaiLabel *pIconLabel = new PaiLabel(this);
    pIconLabel->setPixmap(QPixmap(":/cpu.png"));
    m_currentSQueue = item.queueName;
    pPanelLayout->addWidget(AddItemWidget(item, pIconLabel));

}

QString SubmitWorkflowDialog::GetJobName()
{
    if(m_pJobNameEdit)
    {
        return m_pJobNameEdit->text();
    }
    return QString("");
}

void SubmitWorkflowDialog::UpdateContinueBtn(const QString& name)
{
    
}

void SubmitWorkflowDialog::ToggleNormalRadio(bool toggled)
{
    //如果普通队列radioButton被选中
    if(toggled)
    {
        //作业优先级checkBox使能，并且显示普通队列的item的Widget
        m_priorityCheckBox->setEnabled(true);
        m_pStack->setCurrentWidget(m_pCpuWgt);
        SetPriority(m_priorityCheckBox->isChecked());
        m_currentSQueue = m_cpuItem.queueName;
        //设置continue是否可用
        UpdateContinueBtn(m_currentSQueue);
    }
}

void SubmitWorkflowDialog::ToggleSpecialRadio(bool toggled)
{
    //如果特殊队列radioButton被选中
    if(toggled)
    {
        //作业优先级checkBox非使能，并且显示特殊队列的item的Widget
        m_priorityCheckBox->setEnabled(false);
        m_pStack->setCurrentWidget(m_pSpecialCpuWgt);
        m_priority = pai::ios::job::Normal;

        //如果特殊队列数量不为0
        if(m_pTable && m_pTable->rowCount() > 0)
        {
            QRadioButton *pRadioButton = dynamic_cast<QRadioButton*>(m_pTable->cellWidget(0,0));
            if(pRadioButton)
            {
                pRadioButton->setChecked(true);
                m_currentSQueue = m_pTable->item(0,1)->text();
            }
        }
        else
        {
            m_currentSQueue = "";
        }
        //设置continue按钮状态
        UpdateContinueBtn(m_currentSQueue);
    }
}

void SubmitWorkflowDialog::SelectCurrentRow(bool toggled)
{
    //如果radioButton被选中
    if(toggled)
    {
        //停止循环标志
        bool stop = false;
        for(int col = 0; col < m_pTable->columnCount(); ++col)
        {
            for(int row = 0; row < m_pTable->rowCount(); ++row)
            {
                QRadioButton *pRadioButton = dynamic_cast<QRadioButton*>(m_pTable->cellWidget(row,col));
                if(pRadioButton && pRadioButton->isChecked())
                {
                    //如果是当前选中radioButton
                    if(sender() == pRadioButton)
                    {
                        m_pTable->selectRow(row);
                        m_currentSQueue = m_pTable->item(row,col+1)->text();
                        m_pExtendCtrl->SetLabelText("<a><font color=#CC6600>"+m_currentSQueue+"</font></a>",m_currentSQueue);
                        //设置continue按钮状态
                        UpdateContinueBtn(m_currentSQueue);
                        stop = true;
                    }
                }
            }
            if(stop)
            {
                break;
            }
        }
    }
}

void SubmitWorkflowDialog::SelectCurrentRadioBtn()
{
    QList<QTableWidgetItem *>items = m_pTable->selectedItems();
    if(items.count() > 0)
    {
        int selectRow = items[0]->row();
        for(int col = 0; col < m_pTable->columnCount(); ++col)
        {
            QRadioButton *pRadioButton = dynamic_cast<QRadioButton*>(m_pTable->cellWidget(selectRow,col));
            if(pRadioButton && !pRadioButton->isChecked())
            {
                pRadioButton->setChecked(true);
                //更新continue按钮的状态
                UpdateContinueBtn(m_pTable->item(selectRow,col+1)->text());
                break;
            }
        }
    }
}

SubmitWorkflowDialog::ResItem SubmitWorkflowDialog::GetCurrentItem(const QString& name)
{
    SubmitWorkflowDialog::ResItem resItem;
    for(int i = 0; i < m_items.count(); ++i)
    {
        if(m_items[i].queueName == name)
        {
            resItem = m_items[i];
            break;
        }
    }
    return resItem;
}

void SubmitWorkflowDialog::SetSpecialCpuWidget()
{
  

}

SubmitWorkflowDialog::ResItem SubmitWorkflowDialog::GetSpecialQueueResItem(const QString& name)
{
    ResItem cpuItem;
    
    return cpuItem;
}

void SubmitWorkflowDialog::SetCpuItem(const ResItem& item)
{
    m_cpuItem = item;
}

void SubmitWorkflowDialog::SetStorageItem(const ResItem& item)
{
    m_storageItem = item;
}

void SubmitWorkflowDialog::SetStorageWidget()
{
    QVBoxLayout *pStorageLayout = new QVBoxLayout(m_pStorage);
    pStorageLayout->setContentsMargins(0, 0 ,0, 0);

    ResItem item = m_storageItem;
    PaiLabel *pIconLabel = new PaiLabel(this);
    QPixmap storage(":/storage.png");
    pIconLabel->setPixmap(storage.scaled(48,48,Qt::KeepAspectRatio));

    pStorageLayout->addWidget(AddItemWidget(item, pIconLabel));
}

void SubmitWorkflowDialog::ExpandAllSpecilaQueue()
{
    QWidget *pWgt = m_pSpecialCpuWgt->findChild<QWidget *>("itemWidget");
    if(pWgt)
    {
        pWgt->hide();
    }
    update();
}

void SubmitWorkflowDialog::CollapseAllSpecilaQueue()
{
    if(m_pExtendCtrl)
    {
        m_pExtendCtrl->SetLabelText("<a><font color=#000000>"+m_currentSQueue+"</font></a>",m_currentSQueue);
    }

    QWidget *pWgt = m_pSpecialCpuWgt->findChild<QWidget *>("itemWidget");
    if(pWgt)
    {
        ResItem item = GetCurrentItem(m_currentSQueue);

        PaiLabel *pMainTitleLabel = m_pSpecialCpuWgt->findChild<PaiLabel *>("MainTitleLabel");
        if(pMainTitleLabel)
        {
            pMainTitleLabel->setText(item.mainTitle);
        }

        PaiProgressBar *pResPrgsBar = m_pSpecialCpuWgt->findChild<PaiProgressBar *>("ResPrgsBar");
        if(pResPrgsBar)
        {
            int perc = 0;
            if(item.total != 0)
            {
                perc = (int)(100.0 * item.used / item.total);
            }
            pResPrgsBar->setValue(perc > 100 ? 100 : perc);
            pResPrgsBar->SetErrorState(item.required > (item.total - item.used));
        }

        PaiLabel *pTotalLabel = m_pSpecialCpuWgt->findChild<PaiLabel *>("TotalLabel");
        if(pTotalLabel)
        {
            pTotalLabel->setText(item.totalTitle);
        }

        PaiLabel *pFreeTitleLabel = m_pSpecialCpuWgt->findChild<PaiLabel *>("FreeTitleLabel");
        if(pFreeTitleLabel)
        {
            pFreeTitleLabel->setText(item.freeTitle);
        }

        PaiLabel *pUsedTitleLabel = m_pSpecialCpuWgt->findChild<PaiLabel *>("UsedTitleLabel");
        if(pUsedTitleLabel)
        {
            pUsedTitleLabel->setText(item.usedTitle);
        }
        pWgt->show();
    }
}

QList<SubmitWorkflowDialog::ResItem> SubmitWorkflowDialog::GetSpecialQueueResItems()
{
    QList<SubmitWorkflowDialog::ResItem> items;


    return items;
}

void SubmitWorkflowDialog::SetWidgets()
{
    SetCpuWidget();
    SetStorageWidget();
}