viewmaindialog

viewmaindialog

// MainDialog.cpp : Implementation of CMainDialog

// This will fill a tree control like this:

// libname

// memname

// variableName

// memname

// ...

#include "stdafx.h"

#include "MainDialog.h"

// atlcontrols.h is shipped with a sample atl program with vc

#include "atlcontrols.h"

#import "sas.tlb"

#import "SASWMan.dll"

#import "c:\program files\common files\system\ado\msado15.dll" rename("EOF", "adoEOF")

/////////////////////////////////////////////////////////////////////////////

// CMainDialog

LRESULT CMainDialog::OnInitDialog(UINT uMsg, WPARAM wParam, LPARAM lParam, BOOL& bHandled)

{

unsigned long i;

long j;

// This is defined in atlcontrols.h; it is not part of Win32.

ATLControls::CTreeViewCtrl ctrlTreeView;

ctrlTreeView.Attach(GetDlgItem(IDC_TREE1));

BSTR xmlInfo;

HRESULT hr;

_bstr_t strSource;

SASWorkspaceManager::IWorkspaceManager2Ptr pIWorkspaceManager;

SAS::IWorkspacePtr pIWorkspace;

SASWorkspaceManager::IServerDef2Ptr pIServerDef = NULL;

hr = pIWorkspaceManager.CreateInstance("SASWorkspaceManager.WorkspaceManager.1");

ADODB::_ConnectionPtr obConnection = NULL;

ADODB::_RecordsetPtr obRecordset = NULL;

hr = obRecordset.CreateInstance( __uuidof( ADODB::Recordset ));

hr = obConnection.CreateInstance( __uuidof( ADODB::Connection ));

// The smart pointers will throw an exception if any function returns a bad hr, so we

// don't need to check for hr's.

try{

pIWorkspaceManager->Workspaces->UseXMLInErrorInfo = false;

// Start SAS and get the Workspace

//***** Local SAS Code ****************

pIWorkspace = pIWorkspaceManager->Workspaces->CreateWorkspaceByServer(

_bstr_t("localsas"), SASWorkspaceManager::VisibilityProcess, NULL,

_bstr_t(""), _bstr_t(""), &xmlInfo);

//***** End Local SAS Code ************

//***** Remote SAS Code ***************

//pIServerDef.CreateInstance("SASWorkspaceManager.ServerDef");

//pIServerDef->Port = 5307;

//pIServerDef->Protocol = SASWorkspaceManager::ProtocolBridge;

//pIServerDef->PutMachineDNSName(_bstr_t("your.iomserver.com"));

//pIWorkspace = pIWorkspaceManager->Workspaces->CreateWorkspaceByServer(

// _bstr_t("remotesas"), SASWorkspaceManager::VisibilityProcess,

// pIServerDef, _bstr_t("username"), _bstr_t("password"), &xmlInfo);

//***** End Remote SAS Code ***********

//Valid Protocols

//SASWorkspaceManager::ProtocolCom

//SASWorkspaceManager::ProtocolBridge

// We can run some SAS code here to create a dataset or assign a libname

// or whatever. You can also use a StoredProcess, which is where the SAS

// code is already on the SAS server and you just tell SAS to run it; you

// can also pass in parameters from your program to the StoredProcess.

// That call might look something like this (assuming there is already a

// program on the SAS server called init.sas)

// SAS::IStoredProcessServicePtr pIStoredProcess;

// pIStoredProcess = pIWorkspace->LanguageService->StoredProcessService;

// pIStoredProcess->PutRepository(_bstr_t("c:\\stocks\\Repository"));

// pIStoredProcess->Execute(_bstr_t("init"), _bstr_t("x=5"));

pIWorkspace->LanguageService->Submit(

"libname stocks 'c:\\'; data a; x=5; y=10; run;");

// Output the SAS log to the Debug window

OutputDebugString(pIWorkspace->LanguageService->FlushLog(10000));

// Connect the ADO Connection to the Workspace we created

_bstr_t strConnect( "Provider=SAS.IOMProvider.1; SAS Workspace ID=" );

strConnect = strConnect + pIWorkspace->UniqueIdentifier;

hr = obConnection->Open(strConnect, _bstr_t(""), _bstr_t(""), NULL);

// Get a safearray containing all the assigned libraries

SAFEARRAY *psaLibrefNames;

pIWorkspace->DataService->ListLibrefs(&psaLibrefNames);

BSTR *pData;

HTREEITEM treeLibnameItem;

HTREEITEM treeMemnameItem;

SafeArrayAccessData(psaLibrefNames, (void **)&pData);

ADODB::_RecordsetPtr obMembers = NULL;

hr = obMembers.CreateInstance( __uuidof( ADODB::Recordset ));

// Add each assigned library to the treeview

for (i=0; i<psaLibrefNames->rgsabound->cElements; i++)

{

_bstr_t strLibname(pData[i]);

treeLibnameItem = ctrlTreeView.InsertItem(strLibname, NULL, NULL);

// Add each dataset from each library to the treeview.

// sashelp.vmember is a dataset the SAS system maintains that contains all the

// contents of all libraries on the system.

strSource = "select * from sashelp.vmember where libname='"

+ strLibname + "' and memtype='DATA'";

obMembers->Open( strSource, _variant_t((IDispatch *)obConnection),

ADODB::adOpenForwardOnly,

ADODB::adLockReadOnly, ADODB::adCmdText );

// Add each variable(observation) from each dataset to the treeview

while (!obMembers->adoEOF)

{

_bstr_t strMemname(obMembers->Fields->GetItem( _variant_t( "memname" ) )->Value);

treeMemnameItem = ctrlTreeView.InsertItem(strMemname, treeLibnameItem, NULL);

strSource = strLibname + "." + strMemname;

try{

obRecordset->Open( strSource, _variant_t((IDispatch *)obConnection),

ADODB::adOpenForwardOnly,

ADODB::adLockReadOnly, ADODB::adCmdTableDirect );

long fieldCount = obRecordset->Fields->Count;

for (j=0; j< fieldCount; j++)

{

ctrlTreeView.InsertItem(

(_bstr_t)(obRecordset->Fields->GetItem(

_variant_t( j ) )->Name), treeMemnameItem, NULL);

}

obRecordset->Close();

}

catch(_com_error e){

OutputDebugString(e.Description());

}

obMembers->MoveNext();

}

obMembers->Close();

}

SafeArrayUnaccessData(psaLibrefNames);

SafeArrayDestroy(psaLibrefNames);

pIWorkspaceManager->Workspaces->RemoveWorkspace(pIWorkspace);

pIWorkspace->Close(); // Ensure that the SAS process is terminated.

}

catch(_com_error e){

OutputDebugString(e.Description());

::MessageBox(NULL, e.Description(), "Error Caught!", MB_OK);

for (j=0; j<obConnection->Errors->Count; j++)

{

OutputDebugString(_bstr_t(obConnection->Errors->GetItem(j)->Description));

}

OutputDebugString(_bstr_t(obConnection->Errors->Count));

return 1;

}

::MessageBox( NULL, "Success!", "", MB_OK );

return 1; // Let the system set the focus

}