130 lines
4.0 KiB
C++
130 lines
4.0 KiB
C++
/*
|
||
* UnrelateWorkFlowFileHelper.h
|
||
*
|
||
* Created on: Mar 5, 2015
|
||
* Author: dev
|
||
*/
|
||
|
||
#ifndef PAI_FRAME_WORKFLOWENGINE_UNRELATEWORKFLOWFILEHELPER_H
|
||
#define PAI_FRAME_WORKFLOWENGINE_UNRELATEWORKFLOWFILEHELPER_H
|
||
|
||
#include <set>
|
||
#include <vector>
|
||
|
||
#include "WorkFlowFile.h"
|
||
|
||
class CUnrelateWorkFlowFileHelperTest;
|
||
namespace pai{
|
||
namespace workflow{
|
||
|
||
struct UnrelateRoot2LeafPair
|
||
{
|
||
std::set<int> roots;
|
||
std::set<int> leafs;
|
||
bool HasRoots(const int root)
|
||
{
|
||
if(roots.count(root) > 0)
|
||
{
|
||
return true;
|
||
}else
|
||
{
|
||
return false;
|
||
}
|
||
}
|
||
void InsertPair(int root, std::set<int>& leaf)
|
||
{
|
||
roots.insert(root);
|
||
for(std::set<int>::iterator iter = leaf.begin(); iter != leaf.end(); iter++)
|
||
{
|
||
leafs.insert(*iter);
|
||
}
|
||
}
|
||
};
|
||
|
||
class CUnrelateWorkFlowFileHelper
|
||
{
|
||
private:
|
||
std::vector<CWorkFlowFile*>* unrelatedWorkflowFileList;
|
||
|
||
friend class ::CUnrelateWorkFlowFileHelperTest; //for test
|
||
|
||
private:
|
||
/**
|
||
* @brief 获取workflowfile中的所有输出
|
||
* @param[in] workflowFile
|
||
* @param[in/out] leafList 存储输出模块的moduleid
|
||
*/
|
||
void GetLeafList(CWorkFlowFile* workflowFile, std::set<int>& leafList);
|
||
/**
|
||
* @brief 获取workflowfile中的所有输入
|
||
* @param[in] workflowFile
|
||
* @param[in/out] rootList 存储输入模块的moduleid
|
||
*/
|
||
void GetRootList(CWorkFlowFile* workflowFile, std::set<int>& rootList);
|
||
/**
|
||
* @brief 获取从输入可以到达的输出的映射
|
||
* @param[in] connections
|
||
* @param[in] rootList 所有的根节点
|
||
* @param[in] leafList 所有的叶子节点
|
||
* @param[in/out] root2LeafList 存储输入模块的moduleid 与输出模块id的映射
|
||
*/
|
||
void GetRoot2LeafList(std::vector<CModuleConnection*>* connections, std::set<int>& rootList,
|
||
std::set<int>& leafList, std::map<int, std::set<int> >& root2LeafList);
|
||
/**
|
||
* @brief 递归获取从输入可以到达的输出的映射
|
||
* @param[in] connections
|
||
* @param[in] leafList 所有的叶子节点
|
||
* @param[in] moduleID 模块id
|
||
* @param[in/out] findLeafs
|
||
*/
|
||
void InternalGetRoot2LeafList(std::vector<CModuleConnection*>* connections,
|
||
std::set<int>& leafList, int moduleID, std::set<int>& findLeafs);
|
||
/**
|
||
* @brief 获取不相关的的输入和输出对
|
||
* @param[in] root2LeafList 储输入模块的moduleid 与输出模块id的映射
|
||
* @param[in/out] root2LeafPairList 存储不相关的的输入和输出对
|
||
*/
|
||
void GetUnrelateRoot2LeafPair(std::map<int, std::set<int> >& root2LeafList,
|
||
std::vector<UnrelateRoot2LeafPair>& root2LeafPairList);
|
||
/**
|
||
* @brief 通过rootlist中跟节点，获取所有下游节点
|
||
* @param[in] rootList 存储跟节点
|
||
* @param[in] connections
|
||
* @param[in/out] unrelateWorkflowModuleIDs 存储moduleid
|
||
*/
|
||
void GetUnrelateModuleIdByRootId(std::set<int>& rootList, std::vector<CModuleConnection*>* connections,
|
||
std::set<int>& unrelateWorkflowModuleIDs);
|
||
/**
|
||
* @brief 递归获取所有下游节点
|
||
* @param[in] sourceId
|
||
* @param[in] connections
|
||
* @param[in/out] unrelateWorkflowModuleIDs 存储moduleid
|
||
*/
|
||
void InneralGetUnrelateModuleIdByRootId(const int sourceId, std::vector<CModuleConnection*>* connections,
|
||
std::set<int>& unrelateWorkflowModuleIDs);
|
||
/**
|
||
* @brief 在sourceWorkflowFile的基础上创建新的workflowfile对象
|
||
* @param[in] sourceWorkflowFile
|
||
* @param[in] unrelateWorkflowModuleIDs 新workflowfile中包含的moduleid
|
||
* @return 新的workflowfile对象
|
||
*/
|
||
CWorkFlowFile* GetNewWorkFlowFile(CWorkFlowFile* sourceWorkflowFile,
|
||
std::set<int>& unrelateWorkflowModuleIDs);
|
||
public:
|
||
CUnrelateWorkFlowFileHelper(CWorkFlowFile* workflowFile);
|
||
~CUnrelateWorkFlowFileHelper();
|
||
/**
|
||
* @brief 对workflowFile拆分后，在workflowFile有可能出现多个互补相干的工作流
|
||
* @param[in] workflowFile
|
||
* @param[in/out] unrelatedWorkflowFileList 保存结果
|
||
*/
|
||
std::vector<CWorkFlowFile*>* GetUnrelatedWorkFlowFile();
|
||
|
||
void GetTopLevelWkf(std::vector<CWorkFlowFile*>& result, std::map<int, int>& outputID2inputID);
|
||
};
|
||
|
||
}
|
||
}
|
||
|
||
#endif /* WORKFLOWFILEWRAPPER_H_ */
|