dm_apriori.cpp

用VC做的一个挖掘系统的关联规则算法。COM组件的源代码

// DM_Apriori.cpp : Implementation of CDM_Apriori
#include "stdafx.h"
#include "DM_AprioriCOM.h"
#include "DM_Apriori.h"
#include <comdef.h>

/////////////////////////////////////////////////////////////////////////////
// CDM_Apriori
extern "C" __declspec(dllimport)  int  DM_GenLargeItemSets(const TCHAR* strConnection,
														const TCHAR* strDataSourceTableName,
														const TCHAR* strTID,
														const TCHAR* strItemID,
														const TCHAR* strLargeSetsTableName,
														float fSupport);

extern "C" __declspec(dllimport)  int  DM_GenAssociationRules(const TCHAR* strConnection,
															const TCHAR* strLargeSetsTableName,
															const TCHAR* strAssociationRulesTableName,
															float fConfidence);




STDMETHODIMP CDM_Apriori::GenLargeItemSets(BSTR strDBDSN, BSTR strDataSourceTableName, BSTR strTID, BSTR strItemID, BSTR strLargeSetsTableName, float fSupport, int *piErrorCode)
{
	// TODO: Add your implementation code here
	::CoInitialize(NULL);

	bstr_t bstrDBDSN=strDBDSN;
	bstr_t bstrDataSourceTableName=strDataSourceTableName;
	bstr_t bstrTID=strTID;
	bstr_t bstrItemID=strItemID;
	bstr_t bstrLargeSetsTableName=strLargeSetsTableName;

	*piErrorCode = DM_GenLargeItemSets((const TCHAR*)bstrDBDSN,
									(const TCHAR*)bstrDataSourceTableName,
									(const TCHAR*)bstrTID,
									(const TCHAR*)bstrItemID,
									(const TCHAR*)bstrLargeSetsTableName,
									fSupport
									);

	
	return S_OK;
}

STDMETHODIMP CDM_Apriori::GenAssociationRules(BSTR strDBDSN, BSTR strLargeSetsTableName, BSTR strAssociationRulesTableName, float fConfidence, int *piErrorCode)
{
	// TODO: Add your implementation code here
	bstr_t bstrDBDSN=strDBDSN;
	bstr_t bstrLargeSetsTableName=strLargeSetsTableName;
	bstr_t bstrARules = strAssociationRulesTableName;
    
	*piErrorCode = DM_GenAssociationRules((const char*)bstrDBDSN,
										(const char*) bstrLargeSetsTableName,
										(const char*) bstrARules,
										fConfidence);

	return S_OK;
}