147 lines
3.4 KiB
C++
147 lines
3.4 KiB
C++
//+-------------------------------------------------------------------------
|
|
//
|
|
// Microsoft Windows
|
|
//
|
|
// Copyright (C) Microsoft Corporation, 1997 - 1997
|
|
//
|
|
// File: testinfo.h
|
|
//
|
|
//--------------------------------------------------------------------------
|
|
|
|
//
|
|
// testinfo.h: test file generation
|
|
//
|
|
#ifndef _TESTINFO_H_
|
|
#define _TESTINFO_H_
|
|
|
|
|
|
#include "cliqset.h" // Exact clique-based inference
|
|
#include "clique.h" // Clique structure details
|
|
#include "parmio.h" // Text Parameter streaming I/O
|
|
#include "utility.h" // Entropic utility
|
|
#include "recomend.h" // Troubleshooting recommendations
|
|
|
|
typedef unsigned long ULONG;
|
|
|
|
//
|
|
// Options flags; first 16 bits are the pass count; i.e., number of times to run
|
|
// the inference testing code for timing purposes.
|
|
//
|
|
const ULONG fPassCountMask = 0x0FFF;
|
|
const ULONG fDistributions = 1<<15;
|
|
const ULONG fImpossible = 1<<16;
|
|
const ULONG fVerbose = 1<<17;
|
|
const ULONG fCliquing = 1<<18;
|
|
const ULONG fInference = 1<<19;
|
|
const ULONG fMulti = 1<<20;
|
|
const ULONG fOutputFile = 1<<21;
|
|
const ULONG fShowTime = 1<<22;
|
|
const ULONG fSaveDsc = 1<<23;
|
|
const ULONG fPause = 1<<24;
|
|
const ULONG fSymName = 1<<25;
|
|
const ULONG fExpand = 1<<26;
|
|
const ULONG fClone = 1<<27;
|
|
const ULONG fUtil = 1<<28;
|
|
const ULONG fReg = 1<<29;
|
|
const ULONG fTSUtil = 1<<30;
|
|
const ULONG fInferStats = 1<<31;
|
|
|
|
// Declare a map from strings to pointers to nodes
|
|
typedef map<ZSTR, GNODEMBND *, less<ZSTR> > MPSTRPND;
|
|
|
|
class TESTINFO
|
|
{
|
|
public:
|
|
TESTINFO ( ULONG fCtl, MBNETDSC & mbnet, ostream * pos = NULL )
|
|
:_fCtl(fCtl),
|
|
_mbnet(mbnet),
|
|
_pos(pos),
|
|
_pInferEng(NULL),
|
|
_pmbUtil(NULL),
|
|
_pmbRecom(NULL),
|
|
_rImposs(-1.0),
|
|
_clOut(0)
|
|
{
|
|
_pInferEng = mbnet.PInferEngine();
|
|
assert( _pInferEng );
|
|
if ( fCtl & fUtil )
|
|
{
|
|
_pmbUtil = new MBNET_ENTROPIC_UTILITY( *_pInferEng );
|
|
}
|
|
if ( fCtl & fTSUtil )
|
|
{
|
|
GOBJMBN_CLIQSET * pCliqueSet;
|
|
DynCastThrow(_pInferEng, pCliqueSet);
|
|
_pmbRecom = new MBNET_RECOMMENDER( *pCliqueSet );
|
|
}
|
|
}
|
|
|
|
~ TESTINFO ()
|
|
{
|
|
delete _pmbUtil;
|
|
delete _pmbRecom;
|
|
}
|
|
|
|
void InferTest ();
|
|
|
|
MBNET_ENTROPIC_UTILITY & MbUtil ()
|
|
{
|
|
assert( _pmbUtil );
|
|
return *_pmbUtil;
|
|
}
|
|
MBNET_RECOMMENDER & MbRecom ()
|
|
{
|
|
assert( _pmbRecom );
|
|
return *_pmbRecom;
|
|
}
|
|
MBNETDSC & Mbnet ()
|
|
{ return _mbnet; }
|
|
GOBJMBN_INFER_ENGINE & InferEng ()
|
|
{
|
|
assert( _pInferEng );
|
|
return *_pInferEng;
|
|
}
|
|
ostream * Postream ()
|
|
{ return _pos; }
|
|
ostream & Ostream ()
|
|
{
|
|
assert( _pos );
|
|
return *_pos;
|
|
}
|
|
MPSTRPND & Mpstrpnd ()
|
|
{ return _mpstrpnd; }
|
|
ULONG FCtl ()
|
|
{ return _fCtl; }
|
|
|
|
void GetUtilities ();
|
|
void GetTSUtilities ();
|
|
void GetBeliefs ();
|
|
|
|
SZC SzcNdName ( GNODEMBN * pgnd );
|
|
SZC SzcNdName ( ZSREF zsSymName );
|
|
|
|
bool BFlag ( ULONG fFlag )
|
|
{
|
|
return (FCtl() & fFlag) > 0;
|
|
}
|
|
|
|
// Return a displayable string of the current options settings
|
|
static ZSTR ZsOptions ( ULONG fFlag );
|
|
|
|
public:
|
|
|
|
ULONG _fCtl; // Control flags
|
|
MBNETDSC & _mbnet; // The model to test
|
|
MPSTRPND _mpstrpnd; // The set of nodes to use
|
|
ostream * _pos; // The output stream or NULL
|
|
REAL _rImposs; // The value to report for impossible probs
|
|
int _clOut; // Output line counter
|
|
protected:
|
|
GOBJMBN_INFER_ENGINE * _pInferEng;
|
|
MBNET_ENTROPIC_UTILITY * _pmbUtil;
|
|
MBNET_RECOMMENDER * _pmbRecom;
|
|
};
|
|
|
|
|
|
#endif // _TESTINFO_H_
|