MultipliedForegroundInferenceAlgorithm.cpp

Go to the documentation of this file.

#include "inference/model/foreground/MultipliedForegroundInferenceAlgorithm.h"

namespace ProbabilisticSceneRecognition {
 
  MultipliedForegroundInferenceAlgorithm::MultipliedForegroundInferenceAlgorithm(boost::shared_ptr<std::vector<boost::shared_ptr<SceneObjectDescription> > >& pSceneObjects)
  : ForegroundInferenceAlgorithm(pSceneObjects)
  , mProbability(1.0)
  {
  }
  
  MultipliedForegroundInferenceAlgorithm::~MultipliedForegroundInferenceAlgorithm()
  {
  }

  void MultipliedForegroundInferenceAlgorithm::doInference(std::vector<ISM::Object> pEvidenceList, std::ofstream& pRuntimeLogger)

  {
    // Reset the scene probability. We MARGINALZE here, so we set it to ZERO.
    mProbability = 1.0;

    ROS_INFO_STREAM("Calculating scene probability.");
    
    // Iterate over all scene objects, evaluate them and summarize the results.
    for(boost::shared_ptr<SceneObjectDescription> sceneObject : *mSceneObjects)
    {
      // Update the scene objects with the new evidence.
      sceneObject->update(pEvidenceList, pRuntimeLogger);
      
      // Add the score of the scene object to the scene probabilty.
      mProbability *= sceneObject->getSceneObjectProbability() * sceneObject->getSceneObjectPriori();
      
      ROS_INFO_STREAM(" > Multiplying to scene probability: '" << sceneObject->getSceneObjectProbability() << "' with priori '" << sceneObject->getSceneObjectPriori() << "'.");
      
      // TODO Think about a normalization term.
    }
  }
  
  double MultipliedForegroundInferenceAlgorithm::getProbability()
  {
    return mProbability;
  }
  
}