influenceDiagramGenerator.h

Go to the documentation of this file.

#ifndef GUM_INF_DIAG_GENERATOR_H
#define GUM_INF_DIAG_GENERATOR_H

#include <cstdio>
#include <cstdlib>
#include <iostream>
#include <vector>

#include <agrum/BN/generator/simpleCPTGenerator.h>
#include <agrum/ID/generator/UTGenerator.h>
#include <agrum/ID/generator/simpleUTGenerator.h>
#include <agrum/ID/influenceDiagram.h>
#include <agrum/variables/labelizedVariable.h>

namespace gum {

  template < typename GUM_SCALAR >
  class InfluenceDiagramGenerator {
    public:
    // ############################################################################
    // ############################################################################

    InfluenceDiagramGenerator();

    explicit InfluenceDiagramGenerator(ICPTGenerator< GUM_SCALAR >* cptGenerator);

    explicit InfluenceDiagramGenerator(UTGenerator* utGenerator);

    InfluenceDiagramGenerator(ICPTGenerator< GUM_SCALAR >* cptGenerator,
                              UTGenerator*                 utGenerator);

    ~InfluenceDiagramGenerator();

    // ############################################################################
    // ############################################################################

    InfluenceDiagram< GUM_SCALAR >* generateID(Size       nbrNodes,
                                               GUM_SCALAR arcDensity,
                                               GUM_SCALAR chanceNodeDensity,
                                               GUM_SCALAR utilityNodeDensity,
                                               Size       max_modality = 2);

    private:
    // Check if a temporal order exists and creates ones if not
    void __checkTemporalOrder(InfluenceDiagram< GUM_SCALAR >* infdiag);
    // The Conditional Probability Table generator
    ICPTGenerator< GUM_SCALAR >* __cptGenerator;
    // The Utility Table generator
    UTGenerator* __utGenerator;
  };

} /* namespace gum */

#include <agrum/ID/generator/influenceDiagramGenerator_tpl.h>
#endif /* GUM_INF_DIAG_GENERATOR_H */