BigWheelCoincidenceLUT.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
#include "TrigT1TGC/BigWheelCoincidenceLUT.h"
 
#include <iostream>
#include <fstream>
#include <sstream>
#include <string>
 
#include "TrigT1TGC/TGCNumbering.h"
#include "TrigT1TGC/TGCDatabaseManager.h"
 
#include "AthenaKernel/getMessageSvc.h"
#include "AthenaPoolUtilities/CondAttrListCollection.h"
#include "PathResolver/PathResolver.h"
#include "StoreGate/ReadCondHandle.h"
 
namespace LVL1TGC {
 
int8_t BigWheelCoincidenceLUT::test(int sideId, int octantId, int moduleId, int subsector, 
                                    int type, int dr, int dphi) const {
  if (type < TGCTriggerLUTs::COIN_HH || type > TGCTriggerLUTs::COIN_LL) return 0;  // no candidate
 
  int phimod2 = 0;
  if (moduleId == 2 || moduleId == 5 || moduleId == 8) {   // forward sectors
    int sector = (moduleId-2) / 3 + octantId * 3;   // sector number assuming the forward sector
    phimod2 = (sector%2 == 1) ? 1 : 0;
  }
 
  uint32_t addr = ((type & TGCTriggerLUTs::TYPE_MASK)<<TGCTriggerLUTs::TYPE_SHIFT) +
                  ((phimod2 & TGCTriggerLUTs::PHIMOD2_MASK)<<TGCTriggerLUTs::PHIMOD2_SHIFT) +
                  ((moduleId & TGCTriggerLUTs::MODULE_MASK)<<TGCTriggerLUTs::MODULE_SHIFT) +
                  ((subsector & TGCTriggerLUTs::ROI_MASK)<<TGCTriggerLUTs::ROI_SHIFT) +
                  (((dr+TGCTriggerLUTs::DR_HIGH_RANGE) & TGCTriggerLUTs::DR_MASK)<<TGCTriggerLUTs::DR_SHIFT) +
                  (((dphi+TGCTriggerLUTs::DPHI_HIGH_RANGE) & TGCTriggerLUTs::DPHI_MASK)<<TGCTriggerLUTs::DPHI_SHIFT);
 
  int8_t content = 0x0;   // outside from defined window, i.e. pT=0
 
  if(tgcArgs()->USE_CONDDB()) {
    SG::ReadCondHandle<TGCTriggerLUTs> readHandle{m_readCondKey};
    const TGCTriggerLUTs* readCdo{*readHandle};
    bool fullCW = (readCdo->getType(TGCTriggerLUTs::CW_BW) == "full");
    if(fullCW) addr += (sideId<<TGCTriggerLUTs::SIDE_SHIFT) +
                       ((octantId & TGCTriggerLUTs::OCTANT_MASK)<<TGCTriggerLUTs::OCTANT_SHIFT);
 
    content = readCdo->getBigWheelPt(addr);
  } else {
    if(m_fullCW) addr += (sideId<<TGCTriggerLUTs::SIDE_SHIFT) +
                         ((octantId & TGCTriggerLUTs::OCTANT_MASK)<<TGCTriggerLUTs::OCTANT_SHIFT);
 
    std::unordered_map<uint32_t, char>::const_iterator it = m_lut.find(addr);
    if(it != m_lut.end()) {
      char pt_char = it->second;
      content = m_pTdef.find(pt_char)->second;
    }
  }
 
  return content;
}
 
 
BigWheelCoincidenceLUT::BigWheelCoincidenceLUT(LVL1TGCTrigger::TGCArguments* tgcargs,
                           const SG::ReadCondHandleKey<TGCTriggerLUTs>& readKey,
                                               const std::string& version)
: m_verName(version),
  m_tgcArgs(tgcargs),
  m_readCondKey(readKey) {
 
  MsgStream log(Athena::getMessageSvc(), "LVL1TGC::BigWheelCoincidenceLUT");
  log << MSG::INFO
      << " BigWheel LUT version of " << m_verName << " is selected." << endmsg;
 
  if (!tgcArgs()->USE_CONDDB()) {
    // read LUT contents from local files
    this->readMap();
  } 
}
 
BigWheelCoincidenceLUT::~BigWheelCoincidenceLUT() {
}
 
bool BigWheelCoincidenceLUT::readMap() 
{
  const uint8_t kNMODULETYPE = 12;
  const uint8_t modulenumber[kNMODULETYPE] = {0, 1, 2, 2, 3, 4, 5, 5, 6, 7, 8, 8};
  const std::string modulename[kNMODULETYPE] = {"0a","1a","2a","2b","3a","4a","5a","5b","6a","7a","8a","8b"};
  const std::string sidename[TGCSide::kNSide] = {"a","c"};
  const std::string capitalsidename[TGCSide::kNSide] = {"A", "C"};
 
  const std::string octantName[kNOctant] =
    {  "0", "1", "2", "3", "4", "5", "6", "7"};
  const std::string coincidenceTypeName[N_COIN_TYPE] = {"HH", "HL", "LH", "LL"};
 
  MsgStream log(Athena::getMessageSvc(), "LVL1TGC::BigWheelCoincidenceLUT");
 
  // Automatic identification of octant-symmetry or full-CW from version
  std::string tryname = PathResolver::FindCalibDirectory("dev")+"/TrigT1TGC/BW/cm_a0aHH_Octant_" + m_verName + ".db";
  std::ifstream tryfile(tryname.c_str(), std::ios::in);
  if (tryfile) {
    m_fullCW = false;
    tryfile.close();
  } else {
    tryname = PathResolver::FindCalibDirectory("dev")+"/TrigT1TGC/BW/cm_mod0aHH_A0_" + m_verName + ".db";
    tryfile = std::ifstream(tryname.c_str(), std::ios::in);
    if (tryfile) {
      m_fullCW = true;
      tryfile.close();
    } else {
      log << MSG::ERROR << "Could not found the expected file!" << endmsg;
    }
  }
 
  const uint8_t num_sides = (m_fullCW) ? TGCSide::kNSide : 1;
  const uint8_t num_octants = (m_fullCW) ? kNOctant : 1;
 
  for (uint8_t iside=0; iside < num_sides; iside++) {
    for (uint8_t ioctant=0; ioctant < num_octants; ioctant++) {
      uint32_t octaddr = (iside<<TGCTriggerLUTs::SIDE_SHIFT) +
                         ((ioctant & TGCTriggerLUTs::OCTANT_MASK)<<TGCTriggerLUTs::OCTANT_SHIFT);
 
      // loop over all files...
      for (int iModule=0; iModule < kNMODULETYPE; iModule+=1) {
        uint32_t phimod2 = (modulename[iModule].find("b") != std::string::npos) ? 1 : 0;
 
        if (m_fullCW && iModule%4 > 1) {   // only forward sectors
          if ((ioctant%2 == 0 && uint32_t(iModule/4)%2 != phimod2) ||   // A0, A2, A4, ...
              (ioctant%2 == 1 && uint32_t(iModule/4)%2 == phimod2)) {   // A1, A3, A5, ...
            continue;   // only one of phimod2 sectors should be used
          }
        }
        uint32_t modaddr = ((modulenumber[iModule] & TGCTriggerLUTs::MODULE_MASK)<<TGCTriggerLUTs::MODULE_SHIFT) +
                           ((phimod2 & TGCTriggerLUTs::PHIMOD2_MASK)<<TGCTriggerLUTs::PHIMOD2_SHIFT);
 
        for (int iCoinType=0; iCoinType != N_COIN_TYPE; iCoinType++) {
          std::string fn = "/BW/cm_";
          if (m_fullCW) {
            fn += "mod" + modulename[iModule] + coincidenceTypeName[iCoinType] + "_" + capitalsidename[iside] + octantName[ioctant] + "_";
          } else {
            fn += sidename[iside] + modulename[iModule] + coincidenceTypeName[iCoinType] + "_Octant_";
          }
          fn += m_verName + ".db";
 
          int type = -1;
          int lDR, hDR, lDPhi, hDPhi;
          std::string fullName = PathResolver::FindCalibDirectory("dev")+"/TrigT1TGC"+fn;
          if( fullName.length() == 0 ) {
            log << MSG::ERROR << " Could not found " << fn.c_str() << endmsg;
            continue;
          }
 
          std::ifstream file(fullName.c_str(),std::ios::in);
          if(!file){
            log << MSG::ERROR << " Could not found " << fullName.c_str() << endmsg;
            continue;
          }
 
          std::string buf, tag;
          char delimiter = '\n';
          while (getline(file,buf,delimiter)){
            std::istringstream header(buf);
            header>>tag;
 
            if (tag == "#") {    // read header part
              int roi;
              header >> roi >> lDR >> hDR >> lDPhi >> hDPhi;
              type = getTYPE(lDR, hDR, lDPhi, hDPhi);
              // check moduleNumber and ptLevel
              if(type < 0) {
                log << MSG::WARNING
                    << " illegal parameter in database header : " << header.str() << " in file " << fn << endmsg;
                break;
              }
 
              uint32_t cwaddr = ((uint8_t(type) & TGCTriggerLUTs::TYPE_MASK)<<TGCTriggerLUTs::TYPE_SHIFT) +
                                ((roi & TGCTriggerLUTs::ROI_MASK)<<TGCTriggerLUTs::ROI_SHIFT);
 
              for(uint8_t ir=lDR+TGCTriggerLUTs::DR_HIGH_RANGE; ir <= hDR+TGCTriggerLUTs::DR_HIGH_RANGE; ir++) {
                uint32_t draddr = (ir & TGCTriggerLUTs::DR_MASK)<<TGCTriggerLUTs::DR_SHIFT;
 
                // get window data
                getline(file, buf, delimiter);
 
                for(uint8_t iphi=lDPhi+TGCTriggerLUTs::DPHI_HIGH_RANGE; iphi <= hDPhi+TGCTriggerLUTs::DPHI_HIGH_RANGE; iphi++) {
                  uint32_t theaddr = octaddr + modaddr + cwaddr + draddr + iphi;
                  char pt = buf[iphi-lDPhi-TGCTriggerLUTs::DPHI_HIGH_RANGE];
                  if (pt == 'X') continue;   // not opened
          if(m_lut.count(theaddr)==0){
            m_lut[theaddr] = pt;
          }else{
            log << MSG::ERROR
            << " Problem with loading TGC BW Trigger LUT: duplicated entry at address=" << theaddr << " with pt=" << pt << endmsg;
          }
                }
              }
            }   // if (tag == "#")
          }   // while (getline(...))
        }   // for (int iCoinType)
      }   // for (int iModule)
    }   // for (uint8_t ioctant)
  }   // for (uint8_t iside)
 
  return true;
}
 
 
} //end of namespace bracket