ElectronicsMap.h

Go to the documentation of this file.

 
#ifndef TOOLS_ELECTRONICSMAP_H_
#define TOOLS_ELECTRONICSMAP_H_
 
#include <map>
#include <sstream>
#include <vector>
 
namespace ldmx {
 
template <class ElectronicsID, class DetID>
class ElectronicsMap {
 public:
  ElectronicsMap(bool want_d2e = false)
      : eid2did_{ElectronicsID::MAX_INDEX, 0}, makeD2E_(want_d2e) {}
 
  void clear() {
    eid2did_ = std::vector<DetectorID::RawValue>(ElectronicsID::MAX_INDEX, 0);
    did2eid_.clear();
  }
 
  void addEntry(ElectronicsID eid, DetID did) {
    unsigned int index = eid.index();
    if (index >= eid2did_.size()) {
      EXCEPTION_RAISE("ElectronicsMapOverflow",
                      "Attempted to insert electronics with index " +
                          std::to_string(index) +
                          " which is larger than allowed in this map");
    }
    eid2did_[index] = did.raw();
    if (makeD2E_) did2eid_.insert(std::make_pair(did.raw(), eid));
  }
 
  bool exists(ElectronicsID eid) const {
    return (eid.index() < eid2did_.size() && eid2did_[eid.index()] != 0);
  }
 
  bool exists(DetID did) const {
    if (makeD2E_) {
      return did2eid_.find(did.raw()) == did2eid_.end();
    } else {
      for (auto i : eid2did_) {
        if (i == did.raw()) return true;
      }
      return false;
    }
  }
 
  DetID get(ElectronicsID eid) const {
    if (eid.index() >= eid2did_.size())
      return DetID();
    else
      return DetID(eid2did_[eid.index()]);
  }
 
  ElectronicsID get(DetID did) const {
    if (makeD2E_) {
      auto itr = did2eid_.find(did.raw());
      if (itr == did2eid_.end()) {
        std::stringstream ss;
        ss << "Unable to find mapping for det id " << did;
        EXCEPTION_RAISE("ElectronicsMapNotFound", ss.str());
      }
      return itr->second;
    } else {
      for (unsigned int i = 0; i < eid2did_.size(); i++) {
        if (eid2did_[i] == did.raw()) return ElectronicsID::idFromIndex(i);
      }
      std::stringstream ss;
      ss << "Unable to find mapping for det id " << did;
      EXCEPTION_RAISE("ElectronicsMapNotFound", ss.str());
    }
  }
 
 private:
  std::vector<DetectorID::RawValue> eid2did_;
  bool makeD2E_;
 
  std::map<DetectorID::RawValue, ElectronicsID> did2eid_;
};
 
}  // namespace ldmx
#endif  // TOOLS_ELECTRONICSMAP_H_