logplus/Workflow/WFEngine/WorkflowEngine/src/parser/ModuleInformation.cpp

/**
 * @file ModuleInfomation.cpp
 *
 * @author 黄军
 * @date 2011-7-27
 */
#include "ModuleManager.h"
#include "ModuleInformation.h"
// #include "Utils.h"
// #include "Log.h"
#include <string>
#include <iostream>
#include <vector>
#include <map>
#include <cstdlib>
#include <fstream>
// #include "Utils.h"
namespace pai {
namespace workflow {

CModuleInformation::CModuleInformation():m_oModule(NULL), m_iStepId(0), m_iId(0), m_strNamespace(), m_strName(),\
		m_strVersion(), m_bQC(false), m_bBreakpoint(false), m_strComments(),\
		m_bEnabled(true), m_ModParam(NULL), inModules(), outModules(), m_bBlankModule(false)
{
}


CModuleInformation::CModuleInformation(const std::string& moduleClass):m_oModule(NULL), m_iStepId(0), m_iId(0), m_strNamespace(), m_strName(),\
		m_strVersion(), m_bQC(false), m_bBreakpoint(false), m_strComments(),\
		m_bEnabled(true), m_ModParam(NULL), inModules(), outModules(), m_bBlankModule(false)
{
    if (moduleClass.empty())
    {
    	m_bBlankModule = true;
    	m_ModParam = NULL;
    	m_oModule = NULL;
    }
    else
    {
    	m_bBlankModule = false;
		m_oModule = GetModuleManager()->CreateModuleByClassName(moduleClass);
		if (!m_oModule)
		{
			// exception
			//pai::log::Error(pai::utils::CUtils::GetFormatStr("Failed to create moduel for %s", moduleClass.c_str()));
            // throw pai::error::runtime_error("Failed to create module for " + moduleClass + ",please check this module library!");
		}

		m_strVersion = GetModuleMetaData()->GetVersion();
    }
}
CModuleInformation::CModuleInformation(const CModuleInformation & modInfo):m_oModule(NULL), m_iStepId(modInfo.GetStepID()), m_iId(modInfo.GetId()), m_strNamespace(modInfo.GetNamespace()), m_strName(modInfo.GetName()),\
		m_strVersion(modInfo.m_strVersion), m_bQC(modInfo.GetQC()), m_bBreakpoint(modInfo.GetBreakpoint()), m_strComments(modInfo.GetComments()),\
		m_bEnabled(modInfo.GetEnabled()), m_ModParam(NULL), inModules(), outModules(), m_bBlankModule(modInfo.IsBlankModule())
{
	Clone(modInfo);
}
void CModuleInformation::Clone(const workflow::CModuleInformation& modInfo)
{
	if(this == &modInfo){
		return;
	}
    m_iStepId = modInfo.GetStepID();
    m_iId = modInfo.GetId();
    m_strNamespace = modInfo.GetNamespace();
    m_strName == modInfo.GetName();
    m_bBlankModule = modInfo.IsBlankModule();
    m_strVersion = modInfo.GetVersion();
    if (!m_bBlankModule)
    {
    	if(m_ModParam != NULL){
    		delete m_ModParam;
    	}
    	m_ModParam = NULL; //是否以后要删除
    	if(m_oModule != NULL){
    		delete m_oModule;
    	}
		m_oModule = GetModuleManager()->CreateModuleByClassName(modInfo.GetClassName());
		if(m_oModule)
		{
			m_oModule->GetModuleParameter()->Clone(*(modInfo.GetModule()->GetModuleParameter()));
			*(m_oModule->GetMetaData()) = *(modInfo.GetModule()->GetMetaData());
		}
    }
    else
    {
    	if(m_ModParam != NULL){
    	    delete m_ModParam;
    	}
    	if(m_oModule != NULL){
    	    delete m_oModule;
    	}
    	m_ModParam = NULL;
    	m_oModule = NULL;
    }
    m_bQC = modInfo.GetQC();
    m_bBreakpoint = modInfo.GetBreakpoint();
    m_strComments = modInfo.GetComments();
    m_bEnabled = modInfo.GetEnabled();
}

std::string CModuleInformation::GetComments2() const
{
    return m_strComments;
}

CModuleInformation::~CModuleInformation()
{
    if (m_ModParam != NULL)
    {
        delete m_ModParam;
        m_ModParam = NULL;
    }

    if (m_oModule)
    {
        delete m_oModule;
        m_oModule = NULL;
    }

    // // START 为检查工作流崩溃问题，暂时记录相关类析构时的堆栈信息，以便崩溃后跟踪。
    // std::string logPath;
    // std::ostringstream log;
    // log << "workflow_coredump_debug_log_pid" << getpid() << ".log";
    // std::string logName = log.str();
    // char *pEnv = (char*)pai::utils::CUtils::GetPaiHome().c_str();//getenv("PAI_HOME");

    // if(pEnv == NULL)
    // {
    //     logPath = std::string("/home/") + logName;
    // }
    // else
    // {
    //     logPath = std::string(pEnv) + "/log/" + logName;
    // }
    /*
    std::ofstream fcout(logPath.c_str(), std::ofstream::app);
    if(fcout)
    {
        fcout << "#####start_CModuleInformation#####" << std::endl;
        std::vector<std::string> stack_trace = GetStackTrace();
        for (size_t i = 0; i < stack_trace.size(); ++i)
        {
            fcout << stack_trace[i] << std::endl;
        }
        fcout << "#####end_CModuleInformation#####" << std::endl;
        fcout.close();
    }
*/
    // END
}



const CModuleMetaData* CModuleInformation::GetModuleMetaData() const
{
	if (m_oModule)
	{
		return m_oModule->GetMetaData();
	}
	else
	{
		return NULL;
	}
}

std::string CModuleInformation::GetName() const
{
    if (m_oModule)
    {
        return m_oModule->GetMetaData()->GetID();
    }
    // pai::log::Error(pai::utils::CUtils::GetFormatStr("module is NULL"));
	return "";
}

bool CModuleInformation::IsBlankModule() const
{
	return m_bBlankModule;
}

void CModuleInformation::SetBlankModule(bool bBlankModule)
{
	m_bBlankModule = bBlankModule;
}

CModuleParameter* CModuleInformation::GetModuleParameter() const
{
    if (m_oModule)
    {
        return m_oModule->GetModuleParameter();
    }
    // pai::log::Error(pai::utils::CUtils::GetFormatStr("%s : module is NULL", __func__));
    return NULL;
}

void CModuleInformation::Print()
{
    CModuleInformation* modInfo = this;
    CModule* module = modInfo->GetModule();
    std::cout << "\t\t\tid=" << modInfo->GetStepID() << std::endl;
    // std::cout << "\t\t\tsortId=" << module->GetModuleContext().module_sort_id << std::endl;
    // std::cout << "\t\t\tindegree=" << module->GetModuleContext().module_indegree << std::endl;
    // std::cout << "\t\t\toutdegree=" << module->GetModuleContext().module_outdegree << std::endl;
    std::cout << "\t\t\tnamespace=" << modInfo->GetNamespace() << std::endl;
    std::cout << "\t\t\tname=" << module->GetClassName() << std::endl;
    std::cout << "\t\t\tqc=" << modInfo->GetQC() << std::endl;
    std::cout << "\t\t\tbreakpoint=" << modInfo->GetBreakpoint() << std::endl;
    std::cout << "\t\t\tcomments=" << modInfo->GetComments() << std::endl;
    std::cout << "\t\t\tenabled=" << modInfo->GetEnabled() << std::endl;
    std::cout << "\t\t\tparametervalues:" << std::endl;
    std::cout << "\t\t\t{" << std::endl;
    int i = 0;
    std::vector < CParameterItem > paramItems = module->GetModuleParameter()->GetParameterItems();
    for (std::vector<CParameterItem>::iterator it = paramItems.begin(); it != paramItems.end(); ++it)
    {
        std::cout << "\t\t\t\tParameterValue[" << i << "]" << std::endl;
        std::cout << "\t\t\t\t{" << std::endl;
        CParameterItem item = *it;
        std::cout << "\t\t\t\t\tname=" << item.GetName().toStdString() << std::endl;
        std::cout << "\t\t\t\t\ttype=" << ParamItemTypeToString(item.GetType()) << std::endl;
        std::cout << "\t\t\t\t\tvalue=" << item.ValueToString() << std::endl;
        std::cout << "\t\t\t\t\tdefault=" << item.GetDefault() << std::endl;
        std::cout << "\t\t\t\t\tmax=" << item.GetMax() << std::endl;
        std::cout << "\t\t\t\t\tmin=" << item.GetMin() << std::endl;
        std::cout << "\t\t\t\t}" << std::endl;
        i++;
    }
    std::cout << "\t\t\t}" << std::endl;
}

std::string CModuleInformation::ParamItemTypeToString(ParameterType type)
{
    std::string strIntType = "int";
    std::string strFloatType = "float";
    std::string strDoubleType = "double";
    std::string strStringType = "string";

    std::string strType = "";
    switch (type)
    {
    case pai::module::ParmType_INT:
        strType = strIntType;
        break;
    case pai::module::ParmType_FLOAT:
        strType = strFloatType;
        break;
    case pai::module::ParmType_DOUBLE:
        strType = strDoubleType;
        break;
    case pai::module::ParmType_STRING:
        strType = strStringType;
        break;
    default:
        strType = strStringType;
    }
    return strType;
}

unsigned long CModuleInformation::GetMemorySize()
{
    unsigned long total = 0;
    total = sizeof(*this);
    if(!inModules.empty())
    {
        std::vector<CModuleInformation*>::iterator it = inModules.begin();
        while(it !=inModules.end())
        {
            total += (*it)->GetMemorySize();
            ++it;
        }
    }

    if(!outModules.empty())
    {
        std::vector<CModuleInformation*>::iterator it = outModules.begin();
        while(it !=outModules.end())
        {
            total += (*it)->GetMemorySize();
            ++it;
        }
    }
    if(m_oModule)
    {
        total += m_oModule->GetMemorySize();
    }
    return total;
}

bool operator ==(CModuleInformation &info, CModuleInformation &info2)
{
    std::string moduleId = info.GetModuleID();
    std::string moduleId2 = info2.GetModuleID();
    if (moduleId != moduleId2)
    {
        // pai::log::Error( "两模块的ModuleId不等，分别为：" + moduleId + ";" + moduleId2 );
        return false;
    }
    CModuleParameter*parm = info.GetModuleParameter();
    CModuleParameter* parm2 = info2.GetModuleParameter();
    if (!((*parm) == (*parm2)))
    {
        return false;
    }
    return true;
}

}
}