/*
 * 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_ */