spanning_tree.h

Go to the documentation of this file.

#ifndef SPANNING_TREE_H
#define SPANNING_TREE_H

#include <queue>
#include <unordered_set>
#include <valarray>
#include <ros/ros.h>
#include <tf2/utils.h>
#include <geometry_msgs/PoseArray.h>
#include <geometry_msgs/Vector3.h>
#include <nav_msgs/OccupancyGrid.h>
#include "lib/edge.h"

using namespace std;
using namespace ros;

class spanning_tree
{
public:
    spanning_tree ();

    vector<edge> get_mst_edges ();

    geometry_msgs::PoseArray get_tree ();

    void initialize_graph (nav_msgs::OccupancyGrid gridmap, bool vertical = false, bool connect4 = true);

    void construct ();

private:
    void add_edge (int from, int to, int cost);

     bool different_sets (int a, int b);

    vector<unordered_set<int>> nodes;

    priority_queue<edge, vector<edge>, compare_edge> edges;

    vector<edge> mst_edges;

    nav_msgs::OccupancyGrid map;

    double rotation;

    geometry_msgs::Vector3 translation;

    bool vertical;
};

#endif // SPANNING_TREE_H