MixtureOfDiscreteDistributions.h

Go to the documentation of this file.

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

#ifndef BPP_NUMERIC_PROB_MIXTUREOFDISCRETEDISTRIBUTIONS_H
#define BPP_NUMERIC_PROB_MIXTUREOFDISCRETEDISTRIBUTIONS_H


#include "AbstractDiscreteDistribution.h"

namespace bpp
{
class MixtureOfDiscreteDistributions :
  public AbstractDiscreteDistribution
{
protected:
  std::vector<std::unique_ptr<DiscreteDistributionInterface>> vdd_;

  std::vector<double> probas_;

  std::vector<std::string> vNestedPrefix_;

public:
  MixtureOfDiscreteDistributions(const std::vector<std::unique_ptr<DiscreteDistributionInterface>>& distributions, const std::vector<double>& probas);

  virtual ~MixtureOfDiscreteDistributions();

  MixtureOfDiscreteDistributions(const MixtureOfDiscreteDistributions& mdd);

  MixtureOfDiscreteDistributions& operator=(const MixtureOfDiscreteDistributions& mdd);

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

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

  size_t getNumberOfDistributions() const { return vdd_.size(); }

  const DiscreteDistributionInterface& nDistribution(size_t n) const
  {
    return *vdd_[n];
  }

  double getNProbability(size_t n) const
  {
    return probas_[n];
  }

  void setNumberOfCategories(size_t nbClasses);

  void fireParameterChanged(const ParameterList& parameters);

  double qProb(double x) const;

  double pProb(double x) const;

  double Expectation(double a) const;

  void setMedian(bool median);

  void restrictToConstraint(const ConstraintInterface& c);

  void discretize();

  void setNamespace(const std::string& prefix);

protected:
  void updateDistribution();
};
} // end of namespace bpp.
#endif // BPP_NUMERIC_PROB_MIXTUREOFDISCRETEDISTRIBUTIONS_H