ProbabilisticRewardMapping.h

Go to the documentation of this file.

// SPDX-FileCopyrightText: The Bio++ Development Group
//
// SPDX-License-Identifier: CECILL-2.1
 
#ifndef _LEGACY_PROBABILISTIC_REWARD_MAPPING_H_
#define _LEGACY_PROBABILISTIC_REWARD_MAPPING_H_
 
#include "RewardMapping.h"
#include "../../Mapping/Reward.h" // We use the new implementation here
#include "../../Tree/TreeExceptions.h"
 
#include <Bpp/Text/TextTools.h>
 
// From the STL:
#include <vector>
 
namespace bpp
{
class LegacyProbabilisticRewardMapping :
  public LegacyAbstractRewardMapping
{
private:
  std::shared_ptr<const Reward> reward_;
 
  std::vector<std::vector<double>> mapping_;
 
public:
  LegacyProbabilisticRewardMapping(const Tree& tree, std::shared_ptr<const Reward> reward, size_t numberOfSites) :
    LegacyAbstractRewardMapping(tree),
    reward_(reward),
    mapping_(0)
  {
    setNumberOfSites(numberOfSites);
  }
 
  LegacyProbabilisticRewardMapping(const Tree& tree) :
    LegacyAbstractRewardMapping(tree),
    reward_(nullptr),
    mapping_(0)
  {}
 
 
  LegacyProbabilisticRewardMapping* clone() const override
  {
    return new LegacyProbabilisticRewardMapping(*this);
  }
 
  LegacyProbabilisticRewardMapping(const LegacyProbabilisticRewardMapping& prm) = default;
 
  LegacyProbabilisticRewardMapping& operator=(const LegacyProbabilisticRewardMapping& prm) = default;
 
  virtual ~LegacyProbabilisticRewardMapping() {}
 
public:
  virtual double getReward(int nodeId, size_t siteIndex) const
  {
    return mapping_[siteIndex][getNodeIndex(nodeId)];
  }
 
  virtual void setTree(const Tree& tree);
 
  virtual void setNumberOfSites(size_t numberOfSites) override;
 
  virtual double& operator()(size_t nodeIndex, size_t siteIndex) override
  {
    return mapping_[siteIndex][nodeIndex];
  }
 
  virtual const double& operator()(size_t nodeIndex, size_t siteIndex) const override
  {
    return mapping_[siteIndex][nodeIndex];
  }
 
  std::vector<double>& operator[](size_t siteIndex)
  {
    return mapping_[siteIndex];
  }
 
  const std::vector<double>& operator[](size_t siteIndex) const
  {
    return mapping_[siteIndex];
  }
};
} // end of namespace bpp.
 
#endif // _LEGACY_PROBABILISTIC_REWARD_MAPPING_H_