CTPPSDetId.h

Go to the documentation of this file.

/****************************************************************************
 *
 * This is a part of TOTEM offline software.
 * Authors:
 *  Jan Kašpar (jan.kaspar@gmail.com)
 *
 ****************************************************************************/

#ifndef DataFormats_CTPPSDetId_CTPPSDetId
#define DataFormats_CTPPSDetId_CTPPSDetId

#include "DataFormats/DetId/interface/DetId.h"

#include "FWCore/Utilities/interface/Exception.h"

#include <iosfwd>
#include <iostream>
#include <string>

class CTPPSDetId : public DetId {
public:
  enum SubDetector { sdTrackingStrip = 3, sdTrackingPixel = 4, sdTimingDiamond = 5, sdTimingFastSilicon = 6 };

  explicit CTPPSDetId(uint32_t id);

  CTPPSDetId(uint32_t SubDet, uint32_t Arm, uint32_t Station, uint32_t RomanPot = 0);

  //-------------------- bit assignment --------------------

  static const uint32_t startArmBit, maskArm, maxArm, lowMaskArm;
  static const uint32_t startStationBit, maskStation, maxStation, lowMaskStation;
  static const uint32_t startRPBit, maskRP, maxRP, lowMaskRP;

  //-------------------- component getters and setters --------------------

  uint32_t arm() const { return ((id_ >> startArmBit) & maskArm); }

  void setArm(uint32_t arm) {
    id_ &= ~(maskArm << startArmBit);
    id_ |= ((arm & maskArm) << startArmBit);
  }

  uint32_t station() const { return ((id_ >> startStationBit) & maskStation); }

  void setStation(uint32_t station) {
    id_ &= ~(maskStation << startStationBit);
    id_ |= ((station & maskStation) << startStationBit);
  }

  uint32_t rp() const { return ((id_ >> startRPBit) & maskRP); }

  void setRP(uint32_t rp) {
    id_ &= ~(maskRP << startRPBit);
    id_ |= ((rp & maskRP) << startRPBit);
  }

  //-------------------- id getters for higher-level objects --------------------

  CTPPSDetId armId() const { return CTPPSDetId(rawId() & (~lowMaskArm)); }

  CTPPSDetId stationId() const { return CTPPSDetId(rawId() & (~lowMaskStation)); }

  CTPPSDetId rpId() const { return CTPPSDetId(rawId() & (~lowMaskRP)); }

  //-------------------- name methods --------------------

  enum NameFlag { nShort, nFull, nPath };

  inline void subDetectorName(std::string &name, NameFlag flag = nFull) const {
    if (flag == nPath)
      name = subDetectorPaths[subdetId()];
    else
      name = subDetectorNames[subdetId()];
  }

  inline void armName(std::string &name, NameFlag flag = nFull) const {
    switch (flag) {
      case nShort:
        name = "";
        break;
      case nFull:
        subDetectorName(name, flag);
        name += "_";
        break;
      case nPath:
        subDetectorName(name, flag);
        name += "/sector ";
        break;
    }

    name += armNames[arm()];
  }

  inline void stationName(std::string &name, NameFlag flag = nFull) const {
    switch (flag) {
      case nShort:
        name = "";
        break;
      case nFull:
        armName(name, flag);
        name += "_";
        break;
      case nPath:
        armName(name, flag);
        name += "/station ";
        break;
    }

    name += stationNames[station()];
  }

  inline void rpName(std::string &name, NameFlag flag = nFull) const {
    switch (flag) {
      case nShort:
        name = "";
        break;
      case nFull:
        stationName(name, flag);
        name += "_";
        break;
      case nPath:
        stationName(name, flag);
        name += "/";
        break;
    }

    name += rpNames[rp()];
  }

private:
  static const std::string subDetectorNames[];
  static const std::string subDetectorPaths[];
  static const std::string armNames[];
  static const std::string stationNames[];
  static const std::string rpNames[];
};

std::ostream &operator<<(std::ostream &os, const CTPPSDetId &id);

namespace std {
  template <>
  struct hash<CTPPSDetId> {
    typedef CTPPSDetId argument_type;
    typedef std::size_t result_type;
    result_type operator()(const argument_type &id) const noexcept { return std::hash<uint64_t>()(id.rawId()); }
  };
}  // namespace std

#endif