RecoTracker/LSTCore/src/TiltedGeometry.cc - master - githubmirror/cmssw

RecoTracker/LSTCore/src/TiltedGeometry.cc

63 строки2 KB

Andres Rios Tascon

Added LSTGeometry package and associated ESProducer

22 май 2026, 15:39

22 май 2026, 15:39f954f8e

#include "RecoTracker/LSTCore/interface/TiltedGeometry.h"

#include <fstream>
#include <iostream>
#include <limits>
#include <sstream>
#include <stdexcept>

lst::TiltedGeometry::TiltedGeometry(std::string const& filename) { load(filename); }

lst::TiltedGeometry::TiltedGeometry(lstgeometry::Slopes const& slopes) { load(slopes); }

void lst::TiltedGeometry::load(std::string const& filename) {
  drdzs_.clear();
  dxdys_.clear();

  std::ifstream ifile(filename, std::ios::binary);
  if (!ifile.is_open()) {
    throw std::runtime_error("Unable to open file: " + filename);
  }

  while (!ifile.eof()) {
    unsigned int detid;
    float drdz, dxdy;

    // Read the detid, drdz, and dxdy from binary file
    ifile.read(reinterpret_cast<char*>(&detid), sizeof(detid));
    ifile.read(reinterpret_cast<char*>(&drdz), sizeof(drdz));
    ifile.read(reinterpret_cast<char*>(&dxdy), sizeof(dxdy));

    if (ifile) {
      // TODO: This is needed for now in case old geometry files are used. Remove once deemed unnecessary.
      constexpr float kLegacyVerticalSlope = 123456789.0f;
      drdzs_[detid] = (drdz == kLegacyVerticalSlope) ? std::numeric_limits<float>::infinity() : drdz;
      dxdys_[detid] = (dxdy == kLegacyVerticalSlope) ? std::numeric_limits<float>::infinity() : dxdy;
    } else {
      // End of file or read failed
      if (!ifile.eof()) {
        throw std::runtime_error("Failed to read Tilted Geometry binary data.");
      }
    }
  }
}

void lst::TiltedGeometry::load(lstgeometry::Slopes const& slopes) {
  drdzs_.clear();
  dxdys_.clear();

  for (const auto& [detId, slope] : slopes) {
    drdzs_[detId] = slope.drdz;
    dxdys_[detId] = slope.dxdy;
  }
}

float lst::TiltedGeometry::getDrDz(unsigned int detid) const {
  auto res = drdzs_.find(detid);
  return res == drdzs_.end() ? 0.f : res->second;
}

float lst::TiltedGeometry::getDxDy(unsigned int detid) const {
  auto res = dxdys_.find(detid);
  return res == dxdys_.end() ? 0.f : res->second;
}