SmartDiscretePriorFactor.h

Go to the documentation of this file.

 
#pragma once
 
#include <vector>
 
#include "dcsam/DiscretePriorFactor.h"
 
namespace dcsam {
 
class SmartDiscretePriorFactor : public DiscretePriorFactor {
 public:
  using Base = DiscretePriorFactor;
 
  SmartDiscretePriorFactor() = default;
 
  SmartDiscretePriorFactor(const gtsam::DiscreteKey& dk,
                           const std::vector<double> probs)
      : Base(dk, probs) {}
 
  void updateProbs(const std::vector<double>& probs) {
    assert(probs.size() == probs_.size());
    probs_ = probs;
  }
 
  std::string markdown(const gtsam::KeyFormatter& keyFormatter,
                       const Names& names) const override {
    return toDecisionTreeFactor().markdown(keyFormatter, names);
  }
 
  std::string html(const gtsam::KeyFormatter& keyFormatter,
                   const Names& names) const override {
    return toDecisionTreeFactor().markdown(keyFormatter, names);
  }
};
 
}  // namespace dcsam