discreteconditionalpdf.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 __DISCRETE_CONDITIONAL_PDF__
#define __DISCRETE_CONDITIONAL_PDF__

#include "conditionalpdf.h"

namespace BFL
{


  class DiscreteConditionalPdf : public ConditionalPdf<int, int>
    {
    protected:
      unsigned int _num_states;


      double * _probability_p;
      int * _cond_arg_dims_p;

      int _total_dimension;

      int IndexGet(const int& input, const std::vector<int>& condargs) const;

      // variables to prevent memory allocation on the heap during call of
      //SampleFrom
      mutable std::vector<double> _probs;
      mutable std::vector<double> _valuelist;

    public:

      DiscreteConditionalPdf(int num_states=1,
                             int num_conditional_arguments=1,
                             int cond_arg_dimensions[] = NULL);
      DiscreteConditionalPdf(const DiscreteConditionalPdf & pdf);
      virtual ~DiscreteConditionalPdf();

      virtual DiscreteConditionalPdf* Clone() const;

      unsigned int NumStatesGet()const;

      // Redefine all pure virtuals!
      Probability ProbabilityGet(const int& input) const;
      virtual bool SampleFrom (Sample<int>& one_sample, int method, void * args) const;
      virtual bool SampleFrom (vector<Sample<int> >& list_samples, int num_samples, int method, void * args) const;

      void ProbabilitySet(const double& prob, const int& input, const std::vector<int>& condargs) const;


    };

} // End namespace BFL

#endif // __DISCRETE_CONDITIONAL_PDF__