DirichletDiscreteDistribution.h

Go to the documentation of this file.

// SPDX-FileCopyrightText: The Bio++ Development Group
//
// SPDX-License-Identifier: CECILL-2.1

#ifndef BPP_NUMERIC_PROB_DIRICHLETDISCRETEDISTRIBUTION_H
#define BPP_NUMERIC_PROB_DIRICHLETDISCRETEDISTRIBUTION_H


#include "../../Exceptions.h"
#include "../../Io/OutputStream.h"
#include "../ParameterAliasable.h"
#include "../VectorTools.h"
#include "BetaDiscreteDistribution.h"
#include "MultipleDiscreteDistribution.h"

namespace bpp
{
class DirichletDiscreteDistribution :
  public MultipleDiscreteDistribution,
  public AbstractParameterAliasable
{
private:
  std::vector<BetaDiscreteDistribution* > vpBDD_;

public:
  DirichletDiscreteDistribution(std::vector<size_t> vn, Vdouble valpha);

  ~DirichletDiscreteDistribution();

  DirichletDiscreteDistribution* clone() const
  {
    return new DirichletDiscreteDistribution(*this);
  }

protected:
  void applyParameters();

public:
  std::string getName() const {return "Dirichlet";}

  void fireParameterChanged(const ParameterList& parameters);

  size_t getNumberOfCategories() const;

  Vdouble getValueCategory(Vdouble& Vvalue) const;

  VVdouble getCategories() const;

  virtual double getProbability(Vdouble& category) const;

  Vdouble rand() const;

  Vdouble randC() const;

protected:
  void discretize(Vdouble& valpha);
};
} // end of namespace bpp.
#endif // BPP_NUMERIC_PROB_DIRICHLETDISCRETEDISTRIBUTION_H