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;
|
||
}
|
||
}
|
||
|