//+-------------------------------------------------------------------------
//
//  Microsoft Windows
//
//  Copyright (C) Microsoft Corporation, 1997 - 1997
//
//  File:       cliqset.h
//
//--------------------------------------------------------------------------

//
// cliqset.h:  Definitions for the clique set object.
//
#ifndef _CLIQSET_H_
#define _CLIQSET_H_

#include "gmobj.h"
#include "infer.h"

class GEDGEMBN_CLIQ;

//  Counters maintained in the inference engine
struct CLIQSETSTAT
{
	long _cReload;		// Number of times clique tree was reloaded
	long _cCollect;		// Number of collect operations
	long _cEnterEv;		// Number of calls to EnterEvidence
	long _cGetBel;		// Number of calls to GetBelief
	long _cProbNorm;	// Number of calls to ProbNorm

	CLIQSETSTAT () { Clear(); }
	void Clear ()
	{
		_cReload = 0;
		_cCollect = 0;
		_cEnterEv = 0;
		_cGetBel = 0;
		_cProbNorm = 0;
	}
};

////////////////////////////////////////////////////////////////////
////////////////////////////////////////////////////////////////////
//
//	GOBJMBN_CLIQSET:
//
//		Since any model may decompose into a set of clique trees
//		(assemblages with no interconnections whatever), a CLIQSET
//		is defined as the join point or grouping for the clique
//		tree "forest".
//
////////////////////////////////////////////////////////////////////
////////////////////////////////////////////////////////////////////

class GOBJMBN_CLIQSET: public GOBJMBN_INFER_ENGINE
{
	friend class CLIQSETWORK;

  public:
	GOBJMBN_CLIQSET ( MBNET & model, 
					  REAL rMaxEstimatedSize = -1.0, 
					  int iInferEngID = 0 );

	virtual ~ GOBJMBN_CLIQSET ();
	
	virtual INT EType () const
		{ return EBNO_CLIQUE_SET; }

	virtual void Create ();
	virtual void Destroy ();
	virtual void Reload ();
	virtual void EnterEvidence ( GNODEMBN * pgnd, const CLAMP & clamp );
	virtual void GetEvidence ( GNODEMBN * pgnd, CLAMP & clamp );
	virtual void Infer ();
	virtual void GetBelief ( GNODEMBN * pgnd, MDVCPD & mdvBel );
	virtual PROB ProbNorm ();
	virtual void Dump ();

	//  Return true if the state of information is impossible
	bool BImpossible ();
		
	enum ESTATE							//  State of the junction tree
	{
		CTOR,							//  Just constructed
		UNDIR,							//  Undirected graph created
		MORAL,							//  Moralized
		CLIQUED,						//  Cliques constructed
		BUILT,							//  Fully constructed
		CONSISTENT,						//  Fully propagated
		EVIDENCE						//  Unpropagated evidence present
	};

	ESTATE EState () const				{ return _eState;		}
	MBNET & Model ()					{ return _model;		}
	INT IInferEngID () const			{ return _iInferEngID;	}

	//  Force reloading of the clique tree and full inference
	void SetReset ( bool bReset = true )
		{ _bReset = bReset ; }
	//  Force full collect/distribute cycle
	void SetCollect ( bool bCollect = true )
		{ _bCollect = bCollect; }

	//  Provide access to inference statistics
	CLIQSETSTAT & CqsetStat ()			{ return _cqsetStat;	}

  protected:	
	ESTATE _eState;						//  State of junction tree

	// Tallies
	int _cCliques;						//  Number of cliques
	int _cCliqueMemberArcs;				//  Number of clique member arcs
	int _cSepsetArcs;					//  Number of sepsest (arcs)
	int _cUndirArcs;					//  Undirected arcs in moral graph

	//  Inference control
	bool _bCollect;						//  Is "collect/distribute" pass necessary?
	bool _bReset;						//  Does tree need resetting?
	REAL _probNorm;						//  Residual prob of tree

	CLIQSETSTAT _cqsetStat;				//  Statistics

  protected:
	bool BCollect() const
		{ return _bCollect; }

	//  Cliquing helper functions
	int CNeighborUnlinked ( GNODEMBN * pgndmbn, bool bLinkNeighbors = false );
	void Eliminate ( GNODEMBN * pgndmbn, CLIQSETWORK & clqsetWork ) ;
	void GenerateCliques ( CLIQSETWORK & clqsetWork );
	void CreateUndirectedGraph( bool bMarryParents = true );
	void DestroyDirectedGraph ();

	//  Inference and tree maintenance
	void Reset ();
	void CollectEvidence ();
	void DistributeEvidence ();	

	//  Create (but don't init/load) all the clique and sepset marginals
	void CreateMarginals();
	//  Load probabilities into cliques; initialize all the sepsets
	void LoadMarginals ();

	//  Return the "family" or "self" clique for a node
	GOBJMBN_CLIQUE * PCliqueFromNode ( GNODEMBN * pgnd,
									   bool bFamily,
									   GEDGEMBN_CLIQ * * ppgedgeClique = NULL );

	// Typedefs for pointer-to-member-functions; used by Walk(). If bDownwards,
	//   then object is being enumerated on the way down the tree.
	typedef bool (GOBJMBN_CLIQSET::*PFNC_JTREE) ( GOBJMBN_CLIQUE & clique,
												  bool bDownwards /*= true*/ );
	typedef bool (GOBJMBN_CLIQSET::*PFNC_SEPSET) ( GEDGEMBN_SEPSET & sepset,
												  bool bDownwards /*= true*/ );

	//  Apply the given member function(s) to all cliques and/or sepsets,
	//	  depth first.
	int WalkTree ( bool bDepthFirst,
				   PFNC_JTREE pfJtree = NULL,
				   PFNC_SEPSET pfSepset = NULL );

	//  Apply the given member function to cliques and sepsets, depth first
	int WalkDepthFirst ( GOBJMBN_CLIQUE * pClique,
						 PFNC_JTREE pfJtree = NULL,
						 PFNC_SEPSET pfSepset = NULL );
	int WalkBreadthFirst ( GOBJMBN_CLIQUE * pClique,
						 PFNC_JTREE pfJtree = NULL,
						 PFNC_SEPSET pfSepset = NULL );

	//  Add an undirected arc iff there isn't one already.
	bool BAddUndirArc ( GNODEMBN * pgndbnSource, GNODEMBN * pgndbnSink );

	//  Clique and sepset helper functions used during WalkTree().
	bool BCreateClique	( GOBJMBN_CLIQUE & clique,  bool bDownwards );
	bool BLoadClique	( GOBJMBN_CLIQUE & clique,  bool bDownwards );
	bool BCreateSepset	( GEDGEMBN_SEPSET & sepset, bool bDownwards );
	bool BLoadSepset	( GEDGEMBN_SEPSET & sepset, bool bDownwards );

	bool BCollectEvidenceAtSepset ( GEDGEMBN_SEPSET & sepset, bool bDownwards );
	bool BDistributeEvidenceAtSepset ( GEDGEMBN_SEPSET & sepset, bool bDownwards );
	bool BCollectEvidenceAtRoot ( GOBJMBN_CLIQUE & clique, bool bDownwards );
	bool BDistributeEvidenceAtRoot ( GOBJMBN_CLIQUE & clique, bool bDownwards );

	bool BCollectInitEvidenceAtSepset ( GEDGEMBN_SEPSET & sepset, bool bDownwards );
	bool BDistributeInitEvidenceAtSepset ( GEDGEMBN_SEPSET & sepset, bool bDownwards );
	bool BCollectInitEvidenceAtRoot ( GOBJMBN_CLIQUE & clique, bool bDownwards );
	bool BDistributeInitEvidenceAtRoot ( GOBJMBN_CLIQUE & clique, bool bDownwards );

	//   Perform initial inference collect/distribute cycle
	void InferInit ();	
	void CollectEvidenceInit ();
	void DistributeEvidenceInit ();	

	bool BDumpSepset ( GEDGEMBN_SEPSET & sepset, bool bDownwards );
	bool BDumpClique ( GOBJMBN_CLIQUE & clique, bool bDownwards );
	void CheckConsistency ();
	bool BConsistentSepset ( GEDGEMBN_SEPSET & sepset, bool bDownwards );

private:
	void Clear ();
};

#endif // _CLIQSET_H_