spanningForestPrim.h

Go to the documentation of this file.

#ifndef GUM_SPANNING_FOREST_PRIM_H
#define GUM_SPANNING_FOREST_PRIM_H

#include <agrum/core/priorityQueue.h>
#include <agrum/graphs/algorithms/spanningForest.h>

namespace gum {

  /* ===========================================================================
   */
  /* ===========================================================================
   */
  class SpanningForestPrim : public SpanningForest {
    public:
    // ############################################################################
    // ############################################################################


    SpanningForestPrim(const UndiGraph*             graph,
                       const EdgeProperty< float >* costTable);

    SpanningForestPrim(const SpanningForestPrim& toCopy);

    virtual ~SpanningForestPrim();


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


    const EdgeSet& edgesInSpanningForest();


    const UndiGraph& spanningForest();


    float costOfSpanningForest();


    private:
    const UndiGraph& __graph;

    const EdgeProperty< float >& __costTable;

    PriorityQueue< Edge, float > __edgesToExplore;

    UndiGraph __spanning_tree;

    float __spanning_tree_cost;

    bool __require_computation;

    void __compute();

    void __computeInAComponent(const NodeId id);

    void __exploreNode(const NodeId id);

    SpanningForestPrim& operator=(const SpanningForestPrim& toCopy);
  };

} /* namespace gum */

#endif /* GUM_SPANNING_FOREST_PRIM_H */