void CJtreeInfEngine::
MarginalizeCliqueToQuery( int clqNum, int querySz, const int *query,
int notExpandJPD )
{
// bad-args check
PNL_CHECK_RANGES( clqNum, 0, m_pJTree->GetNumberOfNodes() - 1 );
PNL_CHECK_RANGES( querySz, 1, m_pGraphicalModel->GetNumberOfNodes() );
PNL_CHECK_IS_NULL_POINTER(query);
// bad-args check end
// Note: cant call expand() for potentials, which contain continuous
// observed nodes in domain, cause those are to be expanded to
// mixture of gaussians, which we dont support right now.
delete m_pQueryJPD;
m_pQueryJPD = NULL;
delete m_pPotMPE;
m_pPotMPE = NULL;
delete m_pEvidenceMPE;
m_pEvidenceMPE = NULL;
bool bExpandAllowed = true;
CPotential* clqPotWithQuery = m_pJTree->GetNodePotential(clqNum);
EDistributionType dtClqWithQuery = clqPotWithQuery->GetDistributionType();
if( std::find_first_of( query, query + querySz,
m_actuallyObsNodes.begin(), m_actuallyObsNodes.end() )
!= ( query + querySz ) )
{
const int *queryIt = query, *query_end = query + querySz;
for( ; queryIt != query_end; ++queryIt )
{
if( std::find( m_actuallyObsNodes.begin(),
m_actuallyObsNodes.end(), *queryIt )
!= m_actuallyObsNodes.end() )
{
int shrNodebDiscrete =
m_pGraphicalModel->GetNodeType(*queryIt)->IsDiscrete();
if(((dtClqWithQuery == dtTabular)&&( !shrNodebDiscrete ))
||(( dtClqWithQuery == dtGaussian )&&( shrNodebDiscrete )))
{
bExpandAllowed = false;
break;
}
}
}
}
if( ( bExpandAllowed == false ) && ( notExpandJPD == false ) )
{
PNL_THROW( CAlgorithmicException,
" JPD expansion not possible technically " );
}
bExpandAllowed = notExpandJPD ? false : bExpandAllowed;
CPotential *pMargJPot = clqPotWithQuery->Marginalize( query, querySz,
m_bMaximize );
if( bExpandAllowed )
{
CPotential *pExpObsJPot = pMargJPot->ExpandObservedNodes(m_pEvidence);
if( m_bMaximize )
{
if( pMargJPot->GetDistributionType() == dtScalar )
{
m_pPotMPE = pExpObsJPot->GetNormalized();
m_pEvidenceMPE = m_pPotMPE->GetMPE();
}
else
{
m_pPotMPE = pMargJPot->GetNormalized();
m_pEvidenceMPE = m_pPotMPE->GetMPE();
}
}
else
{
m_pQueryJPD = pExpObsJPot->GetNormalized();
}
delete pExpObsJPot;
}
else
{
if( m_bMaximize )
{
m_pPotMPE = pMargJPot->GetNormalized();
m_pEvidenceMPE = m_pPotMPE->GetMPE();
}
else
{
m_pQueryJPD = pMargJPot->GetNormalized();
}
}
if((!m_bMaximize)&&(m_pQueryJPD->GetDistributionType() == dtGaussian))
{
static_cast<CGaussianDistribFun*>(
m_pQueryJPD->GetDistribFun())->UpdateMomentForm();
}
if((m_bMaximize)&&(m_pPotMPE->GetDistributionType() == dtGaussian))
{
static_cast<CGaussianDistribFun*>(
m_pPotMPE->GetDistribFun())->UpdateMomentForm();
}
delete pMargJPot;
}
