simpleCPTGenerator_tpl.h

Go to the documentation of this file.

#include <agrum/BN/generator/simpleCPTGenerator.h>

#include <ctime>

namespace gum {

  // Default constructor.
  template < typename GUM_SCALAR >
  INLINE SimpleCPTGenerator< GUM_SCALAR >::SimpleCPTGenerator() :
      ICPTGenerator< GUM_SCALAR >() {
    GUM_CONSTRUCTOR(SimpleCPTGenerator);
  }

  // Destructor.
  template < typename GUM_SCALAR >
  INLINE SimpleCPTGenerator< GUM_SCALAR >::~SimpleCPTGenerator() {
    GUM_DESTRUCTOR(SimpleCPTGenerator);
  }

  // Generates a CPT using floats.
  // @param varID The variable id of the CPT owner.
  // @param cpt A reference on the CPT to fill.

  template < typename GUM_SCALAR >
  void SimpleCPTGenerator< GUM_SCALAR >::generateCPT(
     const Idx& varId, const Potential< GUM_SCALAR >& cpt) {
    cpt.random();

    const auto&   var = cpt.variable(varId);
    Instantiation cptInst(cpt);

    for (cptInst.setFirstNotVar(var); !cptInst.end(); cptInst.incNotVar(var)) {
      GUM_SCALAR sum = (GUM_SCALAR)0;

      for (cptInst.setFirstVar(var); !cptInst.end(); cptInst.incVar(var)) {
        sum += cpt[cptInst];
      }

      for (cptInst.setFirstVar(var); !cptInst.end(); cptInst.incVar(var)) {
        cpt.set(cptInst, cpt[cptInst] / sum);
      }

      cptInst.unsetEnd();
    }
  }

} /* namespace gum */