graphChangesGeneratorOnSubDiGraph_tpl.h

Go to the documentation of this file.

#ifndef DOXYGEN_SHOULD_SKIP_THIS

namespace gum {

  namespace learning {

    template < typename STRUCT_CONSTRAINT >
    GraphChangesGeneratorOnSubDiGraph< STRUCT_CONSTRAINT >::
       GraphChangesGeneratorOnSubDiGraph(STRUCT_CONSTRAINT& constraint) :
        _constraint(&constraint) {
      GUM_CONSTRUCTOR(GraphChangesGeneratorOnSubDiGraph);
    }

    template < typename STRUCT_CONSTRAINT >
    GraphChangesGeneratorOnSubDiGraph< STRUCT_CONSTRAINT >::
       GraphChangesGeneratorOnSubDiGraph(
          const GraphChangesGeneratorOnSubDiGraph& from) :
        _constraint(from._constraint),
        _target_nodes(from._target_nodes), _tail_nodes(from._tail_nodes),
        _legal_changes(from._legal_changes),
        __max_threads_number(from.__max_threads_number) {
      GUM_CONS_CPY(GraphChangesGeneratorOnSubDiGraph);
    }

    template < typename STRUCT_CONSTRAINT >
    GraphChangesGeneratorOnSubDiGraph< STRUCT_CONSTRAINT >::
       GraphChangesGeneratorOnSubDiGraph(
          GraphChangesGeneratorOnSubDiGraph&& from) :
        _constraint(from._constraint),
        _target_nodes(std::move(from._target_nodes)),
        _tail_nodes(std::move(from._tail_nodes)),
        _legal_changes(std::move(from._legal_changes)),
        __max_threads_number(from.__max_threads_number) {
      GUM_CONS_MOV(GraphChangesGeneratorOnSubDiGraph);
    }

    template < typename STRUCT_CONSTRAINT >
    GraphChangesGeneratorOnSubDiGraph<
       STRUCT_CONSTRAINT >::~GraphChangesGeneratorOnSubDiGraph() {
      GUM_DESTRUCTOR(GraphChangesGeneratorOnSubDiGraph);
    }

    template < typename STRUCT_CONSTRAINT >
    GraphChangesGeneratorOnSubDiGraph< STRUCT_CONSTRAINT >&
       GraphChangesGeneratorOnSubDiGraph< STRUCT_CONSTRAINT >::operator=(
          const GraphChangesGeneratorOnSubDiGraph< STRUCT_CONSTRAINT >& from) {
      if (this != &from) {
        _constraint = from._constraint;
        _target_nodes = from._target_nodes;
        _tail_nodes = from._tail_nodes;
        _legal_changes = from._legal_changes;
        __max_threads_number = from.__max_threads_number;
      }
      return *this;
    }

    template < typename STRUCT_CONSTRAINT >
    GraphChangesGeneratorOnSubDiGraph< STRUCT_CONSTRAINT >&
       GraphChangesGeneratorOnSubDiGraph< STRUCT_CONSTRAINT >::operator=(
          GraphChangesGeneratorOnSubDiGraph< STRUCT_CONSTRAINT >&& from) {
      if (this != &from) {
        _constraint = std::move(from._constraint);
        _target_nodes = std::move(from._target_nodes);
        _tail_nodes = std::move(from._tail_nodes);
        _legal_changes = std::move(from._legal_changes);
        __max_threads_number = from.__max_threads_number;
      }
      return *this;
    }

    template < typename STRUCT_CONSTRAINT >
    void GraphChangesGeneratorOnSubDiGraph< STRUCT_CONSTRAINT >::_createChanges() {
      _legal_changes.clear();

      // for all the pairs of nodes, consider adding, reverse and removing arcs
      std::vector< Set< GraphChange > > legal_changes;
#  pragma omp parallel num_threads(__max_threads_number)
      {
        int num_threads = getNumberOfRunningThreads();

#  pragma omp single
        {
          // resize the change vectors so that each thread can write to its
          // own vector
          legal_changes.resize(num_threads);
        }

        const Size this_thread = getThreadNumber();

        Idx i = 0;
        for (const auto node1 : _tail_nodes) {
          if (i == this_thread) {
            for (const auto node2 : _target_nodes) {
              if (node1 != node2) {
                // try arc additions
                ArcAddition arc_add(node1, node2);
                if (!_constraint->isAlwaysInvalid(arc_add)) {
                  legal_changes[this_thread].insert(std::move(arc_add));
                }

                // try arc deletion
                ArcDeletion arc_del(node1, node2);
                if (!_constraint->isAlwaysInvalid(arc_del)) {
                  legal_changes[this_thread].insert(std::move(arc_del));
                }

                // try arc reversal
                ArcReversal arc_rev(node1, node2);
                if (!_constraint->isAlwaysInvalid(arc_rev)) {
                  legal_changes[this_thread].insert(std::move(arc_rev));
                }
              }
            }
          }
          ++i;
          i %= num_threads;
        }
      }

      // now store the changes into the protected vectors of the
      // GraphChangesGeneratorOnSubDiGraph
      for (const auto& changes : legal_changes) {
        for (const auto& change : changes) {
          _legal_changes.insert(std::move(change));
        }
      }
    }

    template < typename STRUCT_CONSTRAINT >
    INLINE void GraphChangesGeneratorOnSubDiGraph< STRUCT_CONSTRAINT >::setGraph(
       const DiGraph& graph) {
      // generate the set of all changes
      _createChanges();
    }

    template < typename STRUCT_CONSTRAINT >
    INLINE void GraphChangesGeneratorOnSubDiGraph< STRUCT_CONSTRAINT >::setTargets(
       const NodeSet& nodes) {
      _target_nodes = nodes;
    }

    template < typename STRUCT_CONSTRAINT >
    INLINE void GraphChangesGeneratorOnSubDiGraph< STRUCT_CONSTRAINT >::addTarget(
       NodeId node) {
      _target_nodes.insert(node);
    }

    template < typename STRUCT_CONSTRAINT >
    INLINE void
       GraphChangesGeneratorOnSubDiGraph< STRUCT_CONSTRAINT >::eraseTarget(
          NodeId node) {
      _target_nodes.erase(node);
    }

    template < typename STRUCT_CONSTRAINT >
    INLINE void GraphChangesGeneratorOnSubDiGraph< STRUCT_CONSTRAINT >::setTails(
       const NodeSet& nodes) {
      _tail_nodes = nodes;
    }

    template < typename STRUCT_CONSTRAINT >
    INLINE void GraphChangesGeneratorOnSubDiGraph< STRUCT_CONSTRAINT >::setTails(
       Size nb_nodes) {
      _tail_nodes.clear();
      for (Idx i = 0; i < nb_nodes; ++i) {
        _tail_nodes.insert(i);
      }
    }

    template < typename STRUCT_CONSTRAINT >
    INLINE void GraphChangesGeneratorOnSubDiGraph< STRUCT_CONSTRAINT >::addTail(
       NodeId node) {
      _tail_nodes = node;
    }

    template < typename STRUCT_CONSTRAINT >
    INLINE void GraphChangesGeneratorOnSubDiGraph< STRUCT_CONSTRAINT >::eraseTail(
       NodeId node) {
      _tail_nodes.erase(node);
    }

    template < typename STRUCT_CONSTRAINT >
    INLINE void GraphChangesGeneratorOnSubDiGraph<
       STRUCT_CONSTRAINT >::clearChanges() noexcept {
      _legal_changes.clear();
    }

    template < typename STRUCT_CONSTRAINT >
    INLINE
       typename GraphChangesGeneratorOnSubDiGraph< STRUCT_CONSTRAINT >::iterator
       GraphChangesGeneratorOnSubDiGraph< STRUCT_CONSTRAINT >::begin() const {
      return _legal_changes.cbegin();
    }

    template < typename STRUCT_CONSTRAINT >
    INLINE const typename GraphChangesGeneratorOnSubDiGraph<
       STRUCT_CONSTRAINT >::iterator&
       GraphChangesGeneratorOnSubDiGraph< STRUCT_CONSTRAINT >::end() const {
      return _legal_changes.cend();
    }

    template < typename STRUCT_CONSTRAINT >
    INLINE void
       GraphChangesGeneratorOnSubDiGraph< STRUCT_CONSTRAINT >::modifyGraph(
          const ArcAddition& change) {}

    template < typename STRUCT_CONSTRAINT >
    INLINE void
       GraphChangesGeneratorOnSubDiGraph< STRUCT_CONSTRAINT >::modifyGraph(
          const ArcDeletion& change) {}

    template < typename STRUCT_CONSTRAINT >
    INLINE void
       GraphChangesGeneratorOnSubDiGraph< STRUCT_CONSTRAINT >::modifyGraph(
          const ArcReversal& change) {}

    template < typename STRUCT_CONSTRAINT >
    INLINE void
       GraphChangesGeneratorOnSubDiGraph< STRUCT_CONSTRAINT >::modifyGraph(
          const GraphChange& change) {}

    template < typename STRUCT_CONSTRAINT >
    INLINE void GraphChangesGeneratorOnSubDiGraph<
       STRUCT_CONSTRAINT >::notifyGetCompleted() {
      if (_legal_changes.size()) _legal_changes.clear();
    }

    template < typename STRUCT_CONSTRAINT >
    INLINE void
       GraphChangesGeneratorOnSubDiGraph< STRUCT_CONSTRAINT >::setMaxNbThreads(
          Size nb) noexcept {
#  if defined(_OPENMP) && !defined(GUM_DEBUG_MODE)
      if (nb == 0) nb = getMaxNumberOfThreads();
      __max_threads_number = nb;
#  else
      __max_threads_number = 1;
#  endif /* _OPENMP && GUM_DEBUG_MODE */
    }

    template < typename STRUCT_CONSTRAINT >
    INLINE STRUCT_CONSTRAINT&
           GraphChangesGeneratorOnSubDiGraph< STRUCT_CONSTRAINT >::constraint() const
       noexcept {
      return *_constraint;
    }

  } /* namespace learning */

} /* namespace gum */

#endif /* DOXYGEN_SHOULD_SKIP_THIS */