graphChangesGeneratorOnSubDiGraph.h

Go to the documentation of this file.

#ifndef GUM_LEARNING_GRAPH_CHANGES_GENERATOR_ON_SUBDIGRAPH_H
#define GUM_LEARNING_GRAPH_CHANGES_GENERATOR_ON_SUBDIGRAPH_H

#include <agrum/agrum.h>
#include <agrum/core/OMPThreads.h>
#include <agrum/core/set.h>
#include <agrum/graphs/diGraph.h>
#include <agrum/learning/structureUtils/IGraphChangesGenerator4DiGraph.h>
#include <agrum/learning/structureUtils/graphChange.h>

namespace gum {

  namespace learning {

    template < typename STRUCT_CONSTRAINT >
    class GraphChangesGeneratorOnSubDiGraph
        : public IGraphChangesGenerator4DiGraph {
      public:
      using iterator = typename Set< GraphChange >::const_iterator;

      using const_iterator = iterator;

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

      GraphChangesGeneratorOnSubDiGraph(STRUCT_CONSTRAINT& constraint);

      GraphChangesGeneratorOnSubDiGraph(
         const GraphChangesGeneratorOnSubDiGraph< STRUCT_CONSTRAINT >& from);

      GraphChangesGeneratorOnSubDiGraph(
         GraphChangesGeneratorOnSubDiGraph< STRUCT_CONSTRAINT >&& from);

      virtual ~GraphChangesGeneratorOnSubDiGraph();


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

      GraphChangesGeneratorOnSubDiGraph< STRUCT_CONSTRAINT >& operator=(
         const GraphChangesGeneratorOnSubDiGraph< STRUCT_CONSTRAINT >& from);

      GraphChangesGeneratorOnSubDiGraph< STRUCT_CONSTRAINT >&
         operator=(GraphChangesGeneratorOnSubDiGraph< STRUCT_CONSTRAINT >&& from);


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

      iterator begin() const;

      const iterator& end() const;


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

      STRUCT_CONSTRAINT& constraint() const noexcept;

      void setGraph(const DiGraph& graph);

      void setTargets(const NodeSet& nodes);

      void addTarget(NodeId node);

      void eraseTarget(NodeId node);

      void setTails(const NodeSet& nodes);

      void setTails(Size nb_nodes);

      void addTail(NodeId node);

      void eraseTail(NodeId node);

      void modifyGraph(const ArcAddition& change);

      void modifyGraph(const ArcDeletion& change);

      void modifyGraph(const ArcReversal& change);

      void modifyGraph(const GraphChange& change);

      void clearChanges() noexcept;

      void notifyGetCompleted();

      void setMaxNbThreads(Size nb) noexcept;


      protected:
      STRUCT_CONSTRAINT* _constraint;

      NodeSet _target_nodes;

      NodeSet _tail_nodes;

      Set< GraphChange > _legal_changes;

      void _createChanges();

      private:
#if defined(_OPENMP) && !defined(GUM_DEBUG_MODE)
      Size __max_threads_number{getMaxNumberOfThreads()};
#else
      Size __max_threads_number{1};
#endif /* GUM_DEBUG_MODE */
    };

  } /* namespace learning */

} /* namespace gum */

#include <agrum/learning/structureUtils/graphChangesGeneratorOnSubDiGraph_tpl.h>

#endif /* GUM_LEARNING_GRAPH_CHANGES_GENERATOR_ON_SUBDIGRAPH_H */