ConditionalProbability.cpp

Go to the documentation of this file.

#include "inference/model/foreground/ocm/shape/ConditionalProbability.h"

namespace ProbabilisticSceneRecognition {

ConditionalProbability::ConditionalProbability(): mWasRead(false)
{ }

ConditionalProbability::~ConditionalProbability()
{ }

void ConditionalProbability::setProbability(std::string pParentId, double pProbability)
{
    if (mWasRead) throw std::runtime_error("In ConditionalProbability::setProbability(" + pParentId + ", "
                                           + std::to_string(pProbability) + "): trying to add to a probability that has already been read.");
    mParentProbabilities[pParentId] = pProbability; // overwrites older probabilities.
}

double ConditionalProbability::getProbability()
{
    mWasRead = true;

    // to increase speed: if there is only one parent anyways, return its probability immediately.
    if (mParentProbabilities.size() == 1)
        return mParentProbabilities.begin()->second;

    if (mParentProbabilities.empty())
        throw std::runtime_error("In MinimumConditionalProbability::getProbability(): trying to access probability that has not been set.");

    return calculateProbability();
}

std::string ConditionalProbability::printParentProbabilities()
{
    std::string result;
    for (std::pair<std::string, double> parentProbability: mParentProbabilities)
        result += "(" + parentProbability.first + ", " + std::to_string(parentProbability.second) + ")";
    return result;
}

}