discretepdf.h

Go to the documentation of this file.

// $Id$
// Copyright (C) 2002 Klaas Gadeyne <first dot last at gmail dot com>
//               2008 Tinne De Laet <first dot last at mech dot kuleuven dot be>
//
// This program is free software; you can redistribute it and/or modify
// it under the terms of the GNU Lesser General Public License as published by
// the Free Software Foundation; either version 2.1 of the License, or
// (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU Lesser General Public License for more details.
//
// You should have received a copy of the GNU Lesser General Public License
// along with this program; if not, write to the Free Software
// Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
//
#ifndef DISCRETEPDF_H
#define DISCRETEPDF_H

#include "pdf.h"
#include "../wrappers/matrix/vector_wrapper.h"
#include "../wrappers/matrix/matrix_wrapper.h"
#include <vector>

namespace BFL
{

  class DiscretePdf : public Pdf<int> // inherit abstract_template_class
    {
    protected:
      unsigned int _num_states;

      vector<Probability> *_Values_p;

      bool NormalizeProbs();

      vector<double> _CumPDF;

      bool CumPDFUpdate();

    public:

      DiscretePdf(unsigned int num_states=0);

      DiscretePdf(const DiscretePdf &);

      virtual ~DiscretePdf();

      virtual DiscretePdf* Clone() const;

      unsigned int NumStatesGet()const;

      Probability ProbabilityGet(const int& state) const;


      bool ProbabilitySet(int state, Probability a);

      bool SampleFrom (vector<Sample<int> >& list_samples,
                       const unsigned int num_samples,
                       int method = DEFAULT,
                       void * args = NULL) const;
      bool SampleFrom (Sample<int>& one_sample, int method = DEFAULT, void * args = NULL) const;

      vector<Probability> ProbabilitiesGet() const;


      bool ProbabilitiesSet(vector<Probability> & values);

      int MostProbableStateGet();

    };

} // End namespace

#endif // DISCRETEPDF_H