73 lines
2.2 KiB
C++
73 lines
2.2 KiB
C++
/*
|
||
* SpecialModuleDecompose.h
|
||
*
|
||
* Created on: Mar 12, 2015
|
||
* Author: dev
|
||
*/
|
||
|
||
#ifndef PAI_FRAME_WORKFLOWENGINE_SPECIALMODULEDECOMPOSE_H
|
||
#define PAI_FRAME_WORKFLOWENGINE_SPECIALMODULEDECOMPOSE_H
|
||
|
||
#include "WorkFlowFileWrapper.h"
|
||
#include "WorkFlowFile.h"
|
||
#include "UnrelateWorkFlowFileHelper.h"
|
||
#include "ModuleConnection.h"
|
||
#include "ModuleInformation.h"
|
||
#include "WorkFlowDecompose.h"
|
||
|
||
#include <vector>
|
||
#include <string>
|
||
|
||
class SpecialModuleDecomposeTest;
|
||
namespace pai{
|
||
namespace workflow{
|
||
|
||
class CSpecialModuleDecompose:public CWorkFlowDecompose
|
||
{
|
||
|
||
public:
|
||
CSpecialModuleDecompose(const int &startModuleId);
|
||
virtual ~CSpecialModuleDecompose();
|
||
/**
|
||
* @brief 拆分工作流的入口函数，其他类需要实现此函数进行分类处理
|
||
* @param[in] workflowFileWrapper
|
||
*/
|
||
virtual void DecomposeWorkFlowFile(CWorkFlowFileWrapper* workflowFileWrapper);
|
||
private:
|
||
/**
|
||
* @brief 对有特殊模块的工作流进行拆分
|
||
*@param[in] workflowFile 原始的CWorkFlowFile的备份
|
||
*/
|
||
void InternalDecomposeWorkFlowFile(CWorkFlowFile* workflowFile);
|
||
/**
|
||
* @brief 处理特殊模块，在特殊模块的直接下游添加input和output，将工作流拆分成两个
|
||
*@param[in] workflowFile 原始的CWorkFlowFile的备份
|
||
*@param[in] moduleID
|
||
*@param[in] moduleSize 如果等于1，则判断下面是否直接连着output模块
|
||
*/
|
||
void ProcessDownStreamModule(CWorkFlowFile* workflowFile, const int moduleID, const int moduleSize);
|
||
/**
|
||
* @brief 处理特殊模块，处理特殊模块的所有上游模块的id存在多个connection（sourceid=模块id）情况
|
||
*@param[in] workflowFile 原始的CWorkFlowFile的备份
|
||
*@param[in] moduleID
|
||
*/
|
||
void ProcessUpStreamModule(CWorkFlowFile* workflowFile, const int moduleID);
|
||
/**
|
||
* @brief 从workflowFile中获取所有特殊模块，保存到specialModules中
|
||
*@param[in] workflowFile
|
||
*@return
|
||
*/
|
||
std::vector<CModuleInformation*> GetSpecialModules(CWorkFlowFile* workflowFile);
|
||
|
||
private:
|
||
std::vector<std::string> spcialModuleNameList;
|
||
std::map<int, string> inputId2InputPath;//保存临时输入模块id-》输入路径
|
||
|
||
friend class ::SpecialModuleDecomposeTest;
|
||
};
|
||
|
||
}
|
||
}
|
||
|
||
#endif /* SPECIALMODULEDECOMPOSE_H_ */
|