trackerTFP::LayerEncoding Class Reference

Class to encode layer ids for Kalman Filter Layers consitent with rough r-z track parameters are counted from 0 onwards. More...

#include <LayerEncoding.h>

Public Member Functions
	LayerEncoding ()
	LayerEncoding (const DataFormats *dataFormats)
const std::vector< int > &	layerEncoding (int binEta, int binZT, int binCot) const
const std::map< int, const tt::SensorModule * > &	layerEncodingMap (int binEta, int binZT, int binCot) const
const int	layerIdKF (int binEta, int binZT, int binCot, int layerId) const
const std::vector< int > &	maybeLayer (int binEta, int binZT, int binCot) const
TTBV	maybePattern (int binEta, int binZT, int binCot) const
	~LayerEncoding ()

Private Attributes
const DataFormat *	cot_
const DataFormats *	dataFormats_
std::vector< std::vector< std::vector< std::vector< int > > > >	layerEncoding_
std::vector< std::vector< std::vector< std::map< int, const tt::SensorModule * > > > >	layerEncodingMap_
std::vector< std::vector< std::vector< std::vector< int > > > >	maybeLayer_
const tt::Setup *	setup_
const DataFormat *	zT_

Detailed Description

Class to encode layer ids for Kalman Filter Layers consitent with rough r-z track parameters are counted from 0 onwards.

Author

Thomas Schuh

Date

2020, July

Definition at line 19 of file LayerEncoding.h.

Constructor & Destructor Documentation

LayerEncoding() [1/2]

trackerTFP::LayerEncoding::LayerEncoding ( )

inline

Definition at line 21 of file LayerEncoding.h.

{}

LayerEncoding() [2/2]

trackerTFP::LayerEncoding::LayerEncoding

(

const DataFormats *

dataFormats

)

Definition at line 15 of file LayerEncoding.cc.

References funct::abs(), trackerTFP::DataFormat::base(), trackerTFP::begin, tt::Setup::boundarieEta(), tt::Setup::chosenRofZ(), ALPAKA_ACCELERATOR_NAMESPACE::brokenline::constexpr(), trackerTFP::cot, cot_, ztail::d, dumpMFGeometry_cfg::delta, HLT_2024v10_cff::distance, alcaDQMUpload::encode(), trackerTFP::end, geometryDiff::file, spr::find(), trackerTFP::DataFormat::floating(), mps_fire::i, ALPAKA_ACCELERATOR_NAMESPACE::vertexFinder::it, dqmiolumiharvest::j, MainPageGenerator::l, trackerTFP::layer, layerEncoding(), layerEncoding_, layerEncodingMap_, PostProcessor_cff::layerIds, hgcalTBTopologyTester_cfi::layers, visualization-live-secondInstance_cfg::m, maybeLayer(), maybeLayer_, edm_modernize_messagelogger::ml, tt::Setup::numLayers(), tt::Setup::numSectorsEta(), MillePedeFileConverter_cfg::out, funct::pow(), print(), tt::SensorModule::r(), tt::Setup::sensorModules(), setup_, ALPAKA_ACCELERATOR_NAMESPACE::ecal::reconstruction::internal::barrel::sm(), contentValuesCheck::ss, trackerTFP::DataFormat::toSigned(), tier0::unique(), trackerTFP::DataFormat::width(), tt::SensorModule::z(), trackerTFP::zht, trackerTFP::zT, and zT_.

      : setup_(dataFormats->setup()),
        dataFormats_(dataFormats),
        zT_(&dataFormats->format(Variable::zT, Process::zht)),
        cot_(&dataFormats->format(Variable::cot, Process::zht)),
        layerEncoding_(setup_->numSectorsEta(),
                       vector<vector<vector<int>>>(pow(2, zT_->width()), vector<vector<int>>(pow(2, cot_->width())))),
        layerEncodingMap_(setup_->numSectorsEta(),
                          vector<vector<map<int, const SensorModule*>>>(
                              pow(2, zT_->width()), vector<map<int, const SensorModule*>>(pow(2, cot_->width())))),
        maybeLayer_(setup_->numSectorsEta(),
                    vector<vector<vector<int>>>(pow(2, zT_->width()), vector<vector<int>>(pow(2, cot_->width())))) {
    // number of boundaries of fiducial area in r-z plane for a given set of rough r-z track parameter
    static constexpr int boundaries = 2;
    // find unique sensor mouldes in r-z
    // allowed distance in r and z in cm between modules to consider them not unique
    static constexpr double delta = 1.e-3;
    vector<const SensorModule*> sensorModules;
    sensorModules.reserve(setup_->sensorModules().size());
    for (const SensorModule& sm : setup_->sensorModules())
      sensorModules.push_back(&sm);
    auto smallerR = [](const SensorModule* lhs, const SensorModule* rhs) { return lhs->r() < rhs->r(); };
    auto smallerZ = [](const SensorModule* lhs, const SensorModule* rhs) { return lhs->z() < rhs->z(); };
    auto equalRZ = [](const SensorModule* lhs, const SensorModule* rhs) {
      return abs(lhs->r() - rhs->r()) < delta && abs(lhs->z() - rhs->z()) < delta;
    };
    stable_sort(sensorModules.begin(), sensorModules.end(), smallerR);
    stable_sort(sensorModules.begin(), sensorModules.end(), smallerZ);
    sensorModules.erase(unique(sensorModules.begin(), sensorModules.end(), equalRZ), sensorModules.end());
    // find set of moudles for each set of rough r-z track parameter
    // loop over eta sectors
    for (int binEta = 0; binEta < setup_->numSectorsEta(); binEta++) {
      // cotTheta of eta sector centre
      const double sectorCot = (sinh(setup_->boundarieEta(binEta + 1)) + sinh(setup_->boundarieEta(binEta))) / 2.;
      // z at radius choenRofZ of eta sector centre
      const double sectorZT = setup_->chosenRofZ() * sectorCot;
      // loop over bins in zT
      for (int binZT = 0; binZT < pow(2, zT_->width()); binZT++) {
        // z at radius chosenRofZ wrt zT of sectorZT of this bin centre
        const double zT = zT_->floating(zT_->toSigned(binZT));
        // z at radius chosenRofZ wrt zT of sectorZT of this bin boundaries
        const vector<double> zTs = {sectorZT + zT - zT_->base() / 2., sectorZT + zT + zT_->base() / 2.};
        // loop over bins in cotTheta
        for (int binCot = 0; binCot < pow(2, cot_->width()); binCot++) {
          // cotTheta wrt sectorCot of this bin centre
          const double cot = cot_->floating(cot_->toSigned(binCot));
          // layer ids crossed by left and right rough r-z parameter shape boundaries
          vector<set<int>> layers(boundaries);
          map<int, const SensorModule*>& layermaps = layerEncodingMap_[binEta][binZT][binCot];
          // cotTheta wrt sectorCot of this bin boundaries
          const vector<double> cots = {sectorCot + cot - cot_->base() / 2., sectorCot + cot + cot_->base() / 2.};
          // loop over all unique modules
          for (const SensorModule* sm : sensorModules) {
            // check if module is crossed by left and right rough r-z parameter shape boundaries
            for (int i = 0; i < boundaries; i++) {
              const int j = boundaries - i - 1;
              const double zTi = zTs[sm->r() > setup_->chosenRofZ() ? i : j];
              const double coti = cots[sm->r() > setup_->chosenRofZ() ? j : i];
              // distance between module and boundary in moudle tilt angle direction
              const double d =
                  (zTi - sm->z() + (sm->r() - setup_->chosenRofZ()) * coti) / (sm->cosTilt() - sm->sinTilt() * coti);
              // compare distance with module size and add module layer id to layers if module is crossed
              if (abs(d) < sm->numColumns() * sm->pitchCol() / 2.) {
                layers[i].insert(sm->layerId());
                layermaps[sm->layerId()] = sm;
              }
            }
          }
          // mayber layers are given by layer ids crossed by only one booundary
          set<int> maybeLayer;
          set_symmetric_difference(layers[0].begin(),
                                   layers[0].end(),
                                   layers[1].begin(),
                                   layers[1].end(),
                                   inserter(maybeLayer, maybeLayer.end()));
          // layerEncoding is given by sorted layer ids crossed by any booundary
          set<int> layerEncoding;
          set_union(layers[0].begin(),
                    layers[0].end(),
                    layers[1].begin(),
                    layers[1].end(),
                    inserter(layerEncoding, layerEncoding.end()));
          vector<int>& le = layerEncoding_[binEta][binZT][binCot];
          le = vector<int>(layerEncoding.begin(), layerEncoding.end());
          vector<int>& ml = maybeLayer_[binEta][binZT][binCot];
          ml.reserve(maybeLayer.size());
          for (int m : maybeLayer) {
            int layer = distance(le.begin(), find(le.begin(), le.end(), m));
            if (layer >= setup_->numLayers())
              layer = setup_->numLayers() - 1;
            ml.push_back(layer);
          }
        }
      }
    }
    const bool print = false;
    if (!print)
      return;
    static constexpr int widthLayer = 3;
    static constexpr auto layerIds = {1, 2, 3, 4, 5, 6, 11, 12, 13, 14, 15};
    stringstream ss;
    for (int layer : layerIds) {
      auto encode = [layer, this](const vector<int>& layers, int& l) {
        const auto it = find(layers.begin(), layers.end(), layer);
        if (it == layers.end())
          return false;
        l = distance(layers.begin(), it);
        if (l >= setup_->numLayers())
          l = setup_->numLayers() - 1;
        return true;
      };
      for (int binEta = 0; binEta < setup_->numSectorsEta(); binEta++) {
        for (int binZT = 0; binZT < pow(2, zT_->width()); binZT++) {
          for (int binCot = 0; binCot < pow(2, cot_->width()); binCot++) {
            const int zT =
                binZT < pow(2, zT_->width() - 1) ? binZT + pow(2, zT_->width() - 1) : binZT - pow(2, zT_->width() - 1);
            const int cot = binCot < pow(2, cot_->width() - 1) ? binCot + pow(2, cot_->width() - 1)
                                                               : binCot - pow(2, cot_->width() - 1);
            const vector<int>& layers = layerEncoding_[binEta][zT][cot];
            const vector<int>& maybes = maybeLayer_[binEta][zT][cot];
            int layerKF(-1);
            if (encode(layers, layerKF))
              ss << "1" << TTBV(layerKF, widthLayer) << (encode(maybes, layerKF) ? "1" : "0");
            else
              ss << "00000";
            ss << endl;
          }
        }
      }
    }
    fstream file;
    file.open("layerEncoding.txt", ios::out);
    file << ss.rdbuf();
    file.close();
  }

~LayerEncoding()

trackerTFP::LayerEncoding::~LayerEncoding ( )

inline

Definition at line 23 of file LayerEncoding.h.

{}

Member Function Documentation

layerEncoding()

const std::vector<int>& trackerTFP::LayerEncoding::layerEncoding ( int  binEta, int  binZT, int  binCot  ) const

inline

Definition at line 25 of file LayerEncoding.h.

References layerEncoding_.

Referenced by LayerEncoding(), and hph::Setup::layerEncoding().

                                                                                 {
      return layerEncoding_.at(binEta).at(binZT).at(binCot);
    }

layerEncodingMap()

const std::map<int, const tt::SensorModule*>& trackerTFP::LayerEncoding::layerEncodingMap ( int  binEta, int  binZT, int  binCot  ) const

inline

Definition at line 28 of file LayerEncoding.h.

References layerEncodingMap_.

Referenced by hph::Setup::layerEncodingMap().

                                                                                                        {
      return layerEncodingMap_.at(binEta).at(binZT).at(binCot);
    }

layerIdKF()

const int trackerTFP::LayerEncoding::layerIdKF	(	int	binEta,
		int	binZT,
		int	binCot,
		int	layerId
	)		const

Definition at line 152 of file LayerEncoding.cc.

References HLT_2024v10_cff::distance, spr::find(), ALPAKA_ACCELERATOR_NAMESPACE::vertexFinder::it, trackerTFP::layer, layerEncoding_, hgcalTBTopologyTester_cfi::layers, tt::Setup::numLayers(), and setup_.

Referenced by trklet::DRin::produce(), and trackerTFP::ProducerKFin::produce().

                                                                                         {
    const vector<int>& layers = layerEncoding_[binEta][binZT][binCot];
    const auto it = find(layers.begin(), layers.end(), layerId);
    if (it == layers.end())
      return -1;
    int layer = distance(layers.begin(), it);
    if (layer >= setup_->numLayers())
      layer = setup_->numLayers() - 1;
    return layer;
  }

maybeLayer()

const std::vector<int>& trackerTFP::LayerEncoding::maybeLayer ( int  binEta, int  binZT, int  binCot  ) const

inline

Definition at line 32 of file LayerEncoding.h.

References maybeLayer_.

Referenced by LayerEncoding().

                                                                              {
      return maybeLayer_.at(binEta).at(binZT).at(binCot);
    }

maybePattern()

TTBV trackerTFP::LayerEncoding::maybePattern	(	int	binEta,
		int	binZT,
		int	binCot
	)		const

Definition at line 164 of file LayerEncoding.cc.

References HLT_2024v10_cff::distance, spr::find(), layerEncoding_, hgcalTBTopologyTester_cfi::layers, visualization-live-secondInstance_cfg::m, maybeLayer_, tt::Setup::numLayers(), and setup_.

Referenced by trklet::KFin::consume(), trklet::DRin::produce(), and trackerTFP::ProducerKFin::produce().

                                                                          {
    TTBV ttBV(0, setup_->numLayers());
    const vector<int>& layers = layerEncoding_[binEta][binZT][binCot];
    const vector<int>& maybes = maybeLayer_[binEta][binZT][binCot];
    for (int m : maybes)
      ttBV.set(distance(layers.begin(), find(layers.begin(), layers.end(), m)));
    return ttBV;
  }

Member Data Documentation

cot_

const DataFormat* trackerTFP::LayerEncoding::cot_

private

Definition at line 48 of file LayerEncoding.h.

Referenced by LayerEncoding().

dataFormats_

const DataFormats* trackerTFP::LayerEncoding::dataFormats_

private

Definition at line 44 of file LayerEncoding.h.

layerEncoding_

std::vector<std::vector<std::vector<std::vector<int> > > > trackerTFP::LayerEncoding::layerEncoding_

private

Definition at line 50 of file LayerEncoding.h.

Referenced by LayerEncoding(), layerEncoding(), layerIdKF(), and maybePattern().

layerEncodingMap_

std::vector<std::vector<std::vector<std::map<int, const tt::SensorModule*> > > > trackerTFP::LayerEncoding::layerEncodingMap_

private

Definition at line 51 of file LayerEncoding.h.

Referenced by LayerEncoding(), and layerEncodingMap().

maybeLayer_

std::vector<std::vector<std::vector<std::vector<int> > > > trackerTFP::LayerEncoding::maybeLayer_

private

Definition at line 53 of file LayerEncoding.h.

Referenced by LayerEncoding(), maybeLayer(), and maybePattern().

setup_

const tt::Setup* trackerTFP::LayerEncoding::setup_

private

Definition at line 42 of file LayerEncoding.h.

Referenced by LayerEncoding(), layerIdKF(), maybePattern(), and upgradeWorkflowComponents.UpgradeWorkflow::setup().

zT_

const DataFormat* trackerTFP::LayerEncoding::zT_

private

Definition at line 46 of file LayerEncoding.h.

Referenced by LayerEncoding().