261 lines
8.4 KiB
C++
261 lines
8.4 KiB
C++
|
/*
|
|||
|
|
* AddLineCmd.cpp
|
|||
|
|
*
|
|||
|
|
* Created on: 2012-10-10
|
|||
|
|
* Author: limengzhuo
|
|||
|
|
*/
|
|||
|
|
#include <QDebug>
|
|||
|
|
#include "AddLineCmd.h"
|
|||
|
|
#include "ModuleGraphicsItem.h"
|
|||
|
|
#include "ModuleConnectGraphicsItem.h"
|
|||
|
|
#include "PaiWorkflowDataModel.h"
|
|||
|
|
#include "WorkflowSceneManager.h"
|
|||
|
|
#include "ModuleConnection.h"
|
|||
|
|
#include "WorkFlowFile.h"
|
|||
|
|
#include "PaiMessageBox.h"
|
|||
|
|
#include "ConsoleGUIService.h"
|
|||
|
|
|
|||
|
|
using namespace pai::graphics2d;
|
|||
|
|
using namespace pai::workflow;
|
|||
|
|
using namespace pai::objectmodel;
|
|||
|
|
using namespace pai::gui;
|
|||
|
|
|
|||
|
|
namespace pai
|
|||
|
|
{
|
|||
|
|
AddLineCmd::AddLineCmd(WorkflowSceneManager *pSceneManager,
|
|||
|
|
PaiWorkflowDataModel *pWorkflow,
|
|||
|
|
ModuleGraphicsItem *pStartItem,
|
|||
|
|
ModuleGraphicsItem *pEndItem,
|
|||
|
|
QUndoCommand *pParent)
|
|||
|
|
:QUndoCommand(pParent)
|
|||
|
|
,m_pSceneManager(pSceneManager)
|
|||
|
|
,m_pWorkflow(pWorkflow)
|
|||
|
|
,m_pConnection(new CModuleConnection())
|
|||
|
|
,m_nStepIdFirst(0)
|
|||
|
|
,m_nStepIdSecond(0)
|
|||
|
|
,m_nStartPortIndex(0)
|
|||
|
|
,m_nEndPortIndex(0)
|
|||
|
|
,m_type(AddLineType_General)
|
|||
|
|
,m_bPortBeenLined(false)
|
|||
|
|
{
|
|||
|
|
if (pStartItem && (pStartItem->GetOutputPortCount() > 0)
|
|||
|
|
&& pEndItem && (pEndItem->GetInputPortCount() > 0))
|
|||
|
|
{
|
|||
|
|
if(pStartItem->GetModule())
|
|||
|
|
{
|
|||
|
|
m_nStepIdFirst = pStartItem->GetModule()->GetStepID();
|
|||
|
|
}
|
|||
|
|
if(pEndItem->GetModule())
|
|||
|
|
{
|
|||
|
|
m_nStepIdSecond = pEndItem->GetModule()->GetStepID();
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
else if (pEndItem->GetOutputPortCount() > 0)
|
|||
|
|
{
|
|||
|
|
if(pEndItem->GetModule())
|
|||
|
|
{
|
|||
|
|
m_nStepIdFirst = pEndItem->GetModule()->GetStepID();
|
|||
|
|
}
|
|||
|
|
if(pStartItem->GetModule())
|
|||
|
|
{
|
|||
|
|
m_nStepIdSecond = pStartItem->GetModule()->GetStepID();
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
AddLineCmd::AddLineCmd(WorkflowSceneManager *pSceneManager,
|
|||
|
|
PaiWorkflowDataModel *pWorkflow,
|
|||
|
|
ModuleGraphicsItem *pStartItem,
|
|||
|
|
ModuleGraphicsItem *pEndItem,
|
|||
|
|
int nStartPortIndex,
|
|||
|
|
int nEndPortIndex,
|
|||
|
|
QUndoCommand *pParent)
|
|||
|
|
:QUndoCommand(pParent)
|
|||
|
|
,m_pSceneManager(pSceneManager)
|
|||
|
|
,m_pWorkflow(pWorkflow)
|
|||
|
|
,m_pConnection(new CModuleConnection())
|
|||
|
|
,m_nStepIdFirst(0)
|
|||
|
|
,m_nStepIdSecond(0)
|
|||
|
|
,m_nStartPortIndex(nStartPortIndex)
|
|||
|
|
,m_nEndPortIndex(nEndPortIndex)
|
|||
|
|
,m_type(AddLineType_Destinated)
|
|||
|
|
,m_bPortBeenLined(false)
|
|||
|
|
{
|
|||
|
|
if(pStartItem->GetModule())
|
|||
|
|
{
|
|||
|
|
m_nStepIdFirst = pStartItem->GetModule()->GetStepID();
|
|||
|
|
}
|
|||
|
|
if(pEndItem->GetModule())
|
|||
|
|
{
|
|||
|
|
m_nStepIdSecond = pEndItem->GetModule()->GetStepID();
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
AddLineCmd::~AddLineCmd()
|
|||
|
|
{
|
|||
|
|
if(m_pConnection)
|
|||
|
|
{
|
|||
|
|
delete m_pConnection;
|
|||
|
|
m_pConnection = NULL;
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
void AddLineCmd::undo()
|
|||
|
|
{
|
|||
|
|
RemoveLine();
|
|||
|
|
}
|
|||
|
|
void AddLineCmd::RemoveLine()
|
|||
|
|
{
|
|||
|
|
if (m_pWorkflow && m_pConnection)
|
|||
|
|
{
|
|||
|
|
CModuleConnection* pConnectNeedRemove = m_pWorkflow->GetConnection(m_pConnection);
|
|||
|
|
if (pConnectNeedRemove)
|
|||
|
|
{
|
|||
|
|
//删除连线前，确保线程池中线程结束
|
|||
|
|
m_pSceneManager->WaitThreadForDone();
|
|||
|
|
m_pSceneManager->DeleteConnection(pConnectNeedRemove);
|
|||
|
|
if (m_pWorkflow->RemoveConnection(pConnectNeedRemove) == false)
|
|||
|
|
{
|
|||
|
|
pai::log::Debug(_FLF("Add connection undo failure."));
|
|||
|
|
}
|
|||
|
|
else
|
|||
|
|
{
|
|||
|
|
//删除连线成功，开启提交校验
|
|||
|
|
m_pSceneManager->StartSubmitValidateThread();
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
void AddLineCmd::redo()
|
|||
|
|
{
|
|||
|
|
AddLine();
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
bool AddLineCmd::GetPortHasbeenLined()
|
|||
|
|
{
|
|||
|
|
return m_bPortBeenLined;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
bool AddLineCmd::DetectCircle(ModuleGraphicsItem *pStartItem, ModuleGraphicsItem *pEndItem)
|
|||
|
|
{
|
|||
|
|
//获取pEndItem对应的所有的输出模块
|
|||
|
|
QList<ModuleGraphicsItem*> lstModule = m_pSceneManager->GetOutputModules(pEndItem);
|
|||
|
|
// check whether the module item is pStartItem
|
|||
|
|
foreach(ModuleGraphicsItem* pModuleItem, lstModule)
|
|||
|
|
{
|
|||
|
|
// found a circle
|
|||
|
|
if (pModuleItem == pStartItem)
|
|||
|
|
{
|
|||
|
|
return true;
|
|||
|
|
}
|
|||
|
|
// Depth first search
|
|||
|
|
if (DetectCircle(pStartItem, pModuleItem))
|
|||
|
|
{
|
|||
|
|
return true;
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
return false;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
void AddLineCmd::AddLine()
|
|||
|
|
{
|
|||
|
|
if (!m_pSceneManager || m_nStartPortIndex < 0 ||
|
|||
|
|
m_nEndPortIndex < 0)
|
|||
|
|
{
|
|||
|
|
return;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
ModuleGraphicsItem *pStartItem = m_pSceneManager->FindModule(m_nStepIdFirst);
|
|||
|
|
ModuleGraphicsItem *pEndItem = m_pSceneManager->FindModule(m_nStepIdSecond);
|
|||
|
|
|
|||
|
|
if(!pStartItem || !pEndItem)
|
|||
|
|
{
|
|||
|
|
return;
|
|||
|
|
}
|
|||
|
|
/////////////////改
|
|||
|
|
// 获取输出端口数量
|
|||
|
|
int nOutputPort = pStartItem->GetOutputPortCount();
|
|||
|
|
if (nOutputPort < m_nStartPortIndex + 1)
|
|||
|
|
{
|
|||
|
|
return;
|
|||
|
|
}
|
|||
|
|
//获取输入端口数量
|
|||
|
|
int nInputPort = pEndItem->GetInputPortCount();
|
|||
|
|
if (nInputPort < m_nEndPortIndex + 1)
|
|||
|
|
{
|
|||
|
|
return;
|
|||
|
|
}
|
|||
|
|
//如果结束item存在多个输入端口，查找合理的输入端口
|
|||
|
|
if(m_type == AddLineType_General)
|
|||
|
|
{
|
|||
|
|
//设置即将被连线的输入端口索引
|
|||
|
|
m_nEndPortIndex = pEndItem->GetFreeInPortIndex();
|
|||
|
|
}
|
|||
|
|
//检查结束item的输入端口是否连线
|
|||
|
|
if (m_nEndPortIndex == -1 || pEndItem->InPortHasBeenLined(m_nEndPortIndex))
|
|||
|
|
{
|
|||
|
|
m_bPortBeenLined = true;
|
|||
|
|
|
|||
|
|
QString strMessage = QString("This port has been lined, you should connect other port or delete the line on the port first.");
|
|||
|
|
PaiMessageBox::Critical(NULL, QObject::tr("Error!"), strMessage);
|
|||
|
|
pai::log::Error(_FLF(QObject::tr("QUndoCommand::ShowMessage ").toStdString()+strMessage.toStdString()));
|
|||
|
|
pai::log::Debug(_FLF("port already lined. Add connection failure."));
|
|||
|
|
return;
|
|||
|
|
}
|
|||
|
|
//检查将要连线的模块之间是否已经存在连线
|
|||
|
|
if(m_pSceneManager->CheckConnection(pStartItem,pEndItem))
|
|||
|
|
{
|
|||
|
|
QString strMessage = QObject::tr("You can only connect two modules with one line!");
|
|||
|
|
PaiMessageBox::Critical(NULL, QObject::tr("Error!"), strMessage);
|
|||
|
|
pai::log::Error(_FLF(QObject::tr("QUndoCommand::ShowMessage ").toStdString()+strMessage.toStdString()));
|
|||
|
|
return;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
//删除循环的连线
|
|||
|
|
if (DetectCircle(pStartItem, pEndItem))
|
|||
|
|
{
|
|||
|
|
QString strMessage = QObject::tr("You will create a circle, it's not allowed!");
|
|||
|
|
PaiMessageBox::Critical(NULL, QObject::tr("Error!"), strMessage);
|
|||
|
|
pai::log::Error(_FLF(QObject::tr("QUndoCommand::ShowMessage ").toStdString()+strMessage.toStdString()));
|
|||
|
|
pai::log::Debug(_FLF("Circle will be created. Add connection failure."));
|
|||
|
|
return;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
//新建连线
|
|||
|
|
workflow::CModuleConnection *pConnection = new workflow::CModuleConnection();
|
|||
|
|
pConnection->SetSourceId(pStartItem->GetModule()->GetStepID());
|
|||
|
|
pConnection->SetDestId(pEndItem->GetModule()->GetStepID());
|
|||
|
|
|
|||
|
|
pConnection->SetOutPort(QString::number(m_nStartPortIndex).toStdString()); // first output port
|
|||
|
|
pConnection->SetInPort(QString::number(m_nEndPortIndex).toStdString()); // first input port
|
|||
|
|
|
|||
|
|
//记录新建的连线
|
|||
|
|
pConnection->CopyTo(*m_pConnection);
|
|||
|
|
//工作流中添加连线
|
|||
|
|
if (!m_pWorkflow->AddConnection(pConnection))
|
|||
|
|
{
|
|||
|
|
delete pConnection;
|
|||
|
|
pConnection = NULL;
|
|||
|
|
pai::log::Debug(_FLF("Add connection failure."));
|
|||
|
|
}
|
|||
|
|
else
|
|||
|
|
{
|
|||
|
|
//场景中添加连线item
|
|||
|
|
m_pSceneManager->AddConnection(pConnection);
|
|||
|
|
//添加连线成功以后开启提交校验线程
|
|||
|
|
m_pSceneManager->StartSubmitValidateThread();
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
void AddLineCmd::SetFirstStepID(int first)
|
|||
|
|
{
|
|||
|
|
m_nStepIdFirst = first;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
void AddLineCmd::SetSecondStepID(int second)
|
|||
|
|
{
|
|||
|
|
m_nStepIdSecond = second;
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|