about summary refs log blame commit diff stats
path: root/src/network_set.h
blob: cec3f3929114535fa8c39c11a2e1ebd8abc95ee6 (plain) (tree)
1
2
3
4
5
6
7
8
9
10
11
12
13











                              
                                 
 
                                                          
                                     


                                                                     
                                                                

                                                         
#ifndef NETWORK_SET_H_3036B8E3
#define NETWORK_SET_H_3036B8E3

#include <map>
#include <optional>
#include <set>
#include <utility>
#include <vector>

class NetworkSet {
 public:
  void Clear();

  void AddLink(int id1, int id2);

  void AddLinkToNetwork(int network_id, int id1, int id2);

  bool IsItemInNetwork(int id) const;

  const std::set<std::pair<int, int>>& GetNetworkGraph(int id) const;

 private:

  std::map<int, std::set<std::pair<int, int>>> network_by_item_;
};

#endif /* end of include guard: NETWORK_SET_H_3036B8E3 */