// 'demographicPara' is a 1x6 matrix.
double MaxPosterior::computeJointDensity_MPI_overSubSample(Eigen::MatrixXd demographicPara, IM im, popTree* poptree, Chain coldCh, unsigned int nProcs, unsigned int crr_procID)
{
// REMOVE
//if(crr_procID == 0)
// std::cout << "\n In MaxPosterior::computeJointDensity_MPI_overSubSample\n";
poptree->replacePara(demographicPara);
coldCh.compute_partialJointPosteriorDensity_overSubSample(poptree, im, crr_procID, nProcs);
std::vector<long double> expectationOfEachCoalProb = coldCh.get_expectationOfCoalProb();
unsigned int nloci = coldCh.GetNumLoci();
double posterior = 1;
if(nloci != expectationOfEachCoalProb.size())
{
std::cout << "** Error In MaxPosterior::computeJointDensity_MPI_overSubSample() **\n";
std::cout << "nloci = " << nloci
<< " expectationOfEachCoalProb.size() = " << expectationOfEachCoalProb.size() <<"\n\n";
}
for(unsigned int lc=0; lc< nloci; lc++)
{
double each_local = expectationOfEachCoalProb.at(lc);
double each_global =0;
MPI::COMM_WORLD.Barrier();
MPI::COMM_WORLD.Allreduce(&each_local, &each_global, 1, MPI_DOUBLE, MPI_SUM);
MPI::COMM_WORLD.Barrier();
if(crr_procID ==0)
{
posterior *=each_global;
}
MPI::COMM_WORLD.Barrier();
}
if(crr_procID ==0)
{
Eigen::Vector3d paraMax = im.get_paraMax();
double priorPopTree = poptree->computeJointPrior(paraMax);
posterior *=priorPopTree;
}
if(crr_procID ==0)
{
totalComputingTime_eigen += coldCh.get_eachComputingTime_eigen();
totalComputingTime_condiProb += coldCh.get_eachComputingTime_condiProb();
totalNum_eigenFunctionCalls++;
totalNum_condiProbFunctionCalls += coldCh.get_countCondiProbFunctionCalls();
// std::cout << "totalComputingTime_eigen = " << totalComputingTime_eigen.count()/1000 << "milliseconds\n";
}
return posterior;
}
