normal.h

Go to the documentation of this file.

/* Copyright (C) 2000-2013 CEA
 *
 * This software and supporting documentation were developed by
 *     bioPICSEL
 *     CEA/DSV/I²BM/MIRCen/LMN, Batiment 61,
 *     18, route du Panorama
 *     92265 Fontenay-aux-Roses
 *     France
 */

#ifndef PRIMATOLOGIST_PROBA_NORMAL_H
#define PRIMATOLOGIST_PROBA_NORMAL_H

//--- primatologist ----------------------------------------------------------
#include <primatologist/utility/volume.h>                        // vol::empty
//--- cartodata --------------------------------------------------------------
#include <cartodata/volume/volume.h>                       // carto::VolumeRef
//--- std --------------------------------------------------------------------
#include <vector>
//----------------------------------------------------------------------------

namespace aims {
namespace proba {

  //==========================================================================
  //
  //                           PDF
  //
  //==========================================================================

  double gauss( double x, double mu, double sigma );
  double normal( double x, double mu, double sigma );

} // namespace proba

  //==========================================================================
  //
  //                      NORMAL DISTRIBUTION
  //
  //==========================================================================
  class NormalDistribution
  {
  public:
    NormalDistribution( double mu = 0., double sigma = 1. );
    NormalDistribution( const NormalDistribution & other );
    virtual ~NormalDistribution();
    NormalDistribution & operator= ( const NormalDistribution & other );

    double pdf( double x ) const;

    // accessors
    double getMu() const;
    double getSigma() const;

    // modifiers
    void setMu( double mu );
    void setSigma( double sigma );

  private:
    double _mu;
    double _sigma;
  };

  //==========================================================================
  //
  //                      NORMAL ESTIMATION
  //
  //==========================================================================
  template <typename DataType   = double,
            typename WeightType = double,
            typename MaskType   = char>
  class NormalEstimation
  {
  public:
    //--- constructor --------------------------------------------------------
    NormalEstimation();
    ~NormalEstimation();

    //--- set volumes --------------------------------------------------------
    void setData( const carto::VolumeRef<DataType> & data = vol::empty<DataType>() );
    void setWeight( const carto::VolumeRef<WeightType> & weight = vol::empty<WeightType>() );
    void setMask( const carto::VolumeRef<MaskType> & mask = vol::empty<MaskType>() );

    //--- options ------------------------------------------------------------
    void setBessel( bool bessel = true );
    void setRobust( bool robust = true );
    void setMultiWeights( bool multi = true );

    //--- fit ----------------------------------------------------------------
    void fit();

    //--- results ------------------------------------------------------------
    double getMu() const;
    double getSigma() const;
    NormalDistribution getDistribution() const;
    std::vector<NormalDistribution> getDistributionVector() const;

  protected:
    bool inMask( long x = 0, long y = 0, long z = 0, long t = 0 );
    void fitOne( const carto::VolumeRef<WeightType> & volweight,
                 NormalDistribution & distrib );

  private:
    std::vector<NormalDistribution> _distrib;
    carto::VolumeRef<DataType>      _data;
    carto::VolumeRef<WeightType>    _weight;
    carto::VolumeRef<MaskType>      _mask;
    bool _bessel;
    bool _robust;
    bool _multi;
    std::vector<float> _vs_data;
    std::vector<float> _vs_mask;

  };

} // namespace aims

#endif // PRIMATOLOGIST_PROBA_NORMAL_H