Topology.h

#pragma once
 
#include <bits/stdc++.h>
 
#include <map>
#include <memory>
#include <mutex>
#include <string>
#include <unordered_map>
#include <vector>
 
#define NUMBER_NODES 32
#define NUMBER_RINGS 12
 
class FrontEndBase;  // Forward class declaration
using FENRegMap = std::unordered_map<std::string, uint32_t>;
using TopologyMap = std::unordered_map<uint32_t, std::unordered_map<uint32_t, std::shared_ptr<FrontEndBase>>>;

struct FrontEndEntry {
  std::string name;                        
  std::shared_ptr<FrontEndBase> instance;  
  int ring;                                
  int node;                                
};
 
using TopologyList = std::vector<FrontEndEntry>;
using NodesList = std::map<int, FrontEndEntry>;

class Topology {
 public:
  Topology();
  const TopologyList& getAllEntries() const { return topology; };
 
  void addEntry(const FrontEndEntry& entry);
  const NodesList getNodesForRing(int ring);
  const uint8_t getNumberOfNodesInRing(int ring);
  const FENRegMap& getAddrMap(int ring, int node);
  const FrontEndBase* getFEN(int ring, int node);
  const std::vector<int> getActiveRings();
  void fillNodesArray(uint8_t* rings, size_t size = NUMBER_RINGS);
 
 private:
  TopologyList topology;
  TopologyMap topMap;
  size_t mapped_topology[NUMBER_RINGS][NUMBER_NODES]{};
  uint8_t active_nodes[NUMBER_RINGS]{};
  std::mutex topology_mutex;
};