DFSCode.h

Go to the documentation of this file.

#include <ostream>
#include <vector>

#include <agrum/PRM/gspan/edgeCode.h>

#ifndef GUM_DFS_CODE_H
#  define GUM_DFS_CODE_H

namespace gum {
  namespace prm {
    namespace gspan {

      class DFSCode {
        public:
        static bool validNeighbors(EdgeCode* e1, EdgeCode* e2);

        DFSCode();

        DFSCode(const DFSCode& source);

        ~DFSCode();

        std::vector< EdgeCode* > codes;

        DFSCode& operator=(const DFSCode& source);

        bool operator==(const DFSCode& code) const;

        bool operator!=(const DFSCode& code) const;

        bool operator<(const DFSCode& code) const;

        bool operator<=(const DFSCode& code) const;

        typedef std::vector< EdgeCode* >::iterator iterator;

        typedef std::vector< EdgeCode* >::const_iterator const_iterator;
      };

      std::ostream& operator<<(std::ostream& out, const DFSCode& code);

      inline bool DFSCode::validNeighbors(EdgeCode* e1, EdgeCode* e2) {
        if (e1->isBackward()) {
          if (e2->isForward()) {
            return (e2->i <= e1->i) && (e2->j = (e1->i + 1));
          } else {
            return (e2->i == e1->i) && (e1->j < e2->j);
          }
        } else {
          // e1 is a forward edge
          if (e2->isForward()) {
            return (e2->i <= e1->j) && (e2->j == (e1->j + 1));
          } else {
            return (e2->i == e1->j) && (e2->j < e1->i);
          }
        }
      }

    } /* namespace gspan */
  }   /* namespace prm */
} /* namespace gum */

#  ifndef GUM_NO_INLINE
#    include <agrum/PRM/gspan/DFSCode_inl.h>
#  endif   // GUM_NO_INLINE

#endif /* GUM_DFS_CODE_H */