otbMassOfBelief.h

Go to the documentation of this file.

/*=========================================================================

  Program:   ORFEO Toolbox
  Language:  C++
  Date:      $Date$
  Version:   $Revision$


  Copyright (c) Centre National d'Etudes Spatiales. All rights reserved.
  See OTBCopyright.txt for details.


     This software is distributed WITHOUT ANY WARRANTY; without even
     the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
     PURPOSE.  See the above copyright notices for more information.

=========================================================================*/
#ifndef __otbMassOfBelief_h
#define __otbMassOfBelief_h

#include "itkDataObject.h"
#include "itkObjectFactory.h"
#include "itkNumericTraits.h"

#include <set>
#include <map>
#include <ostream>


namespace otb
{

template <class TLabel = unsigned short, class TMass = double>
class ITK_EXPORT MassOfBelief : public itk::DataObject
{
public:
  typedef MassOfBelief                  Self;
  typedef itk::DataObject               Superclass;
  typedef itk::SmartPointer<Self>       Pointer;
  typedef itk::SmartPointer<const Self> ConstPointer;
  
  itkNewMacro(Self);
  
  itkTypeMacro(MassOfBelief, DataObject);

  typedef TLabel                          LabelType;
  typedef TMass                           MassType;
  typedef std::set<LabelType>             LabelSetType;
  typedef std::map<LabelSetType, MassType> MassMapType;
  typedef std::set<LabelSetType>          LabelSetOfSetType;

  void SetMass(const LabelSetType & labelSet, const MassType & mass);
  
  MassType GetMass(const LabelSetType & labelSet) const;

  void RemoveMass(const LabelSetType & labelSet);
  
  LabelSetOfSetType GetSupport() const;

  LabelSetType GetUniverse() const;

  void Normalize();

  void EstimateUncertainty();

  void Clear();

  void InitializePowerSetMasses(const LabelSetType & universe);

  MassType GetBelief(const LabelSetType & labelSet) const;

  MassType GetPlausibility(const LabelSetType & labelSet) const;

  MassType GetBelief(const LabelSetOfSetType & containedLabelSet) const;

  MassType GetPlausibility(const LabelSetOfSetType & intersectedLabelSet) const;

  void Copy(const Self * massOfBelief);

  bool IsEmpty() const;

  static std::ostream & PrintLabelSet(std::ostream & out,
                                      const LabelSetType & labelSet);

  static std::ostream & PrintLabelSetOfSet(std::ostream & out,
                                           const LabelSetOfSetType & labelSet);

protected:
  MassOfBelief() {}
  
  virtual ~MassOfBelief() {}
  
  void PrintSelf(std::ostream& os, itk::Indent indent) const;

private:
  MassOfBelief(const Self&); //purposely not implemented
  void operator=(const Self&); //purposely not implemented

  MassMapType  m_MassesMap;
};

} // end namespace otb


#ifndef OTB_MANUAL_INSTANTIATION
#include "otbMassOfBelief.txx"
#endif

#endif