trklet::TrackletEngineDisplaced Class Reference

#include <TrackletEngineDisplaced.h>

Inheritance diagram for trklet::TrackletEngineDisplaced:

Public Member Functions
void	addInput (MemoryBase *memory, std::string input) override
void	addOutput (MemoryBase *memory, std::string output) override
void	execute ()
short	memNameToIndex (const std::string &name)
void	readTables ()
	TrackletEngineDisplaced (std::string name, Settings const &settings, Globals *global)
	~TrackletEngineDisplaced () override
Public Member Functions inherited from trklet::ProcessBase
unsigned int	getISeed (const std::string &name)
std::string const &	getName () const
void	initLayerDisk (unsigned int pos, int &layer, int &disk)
void	initLayerDisk (unsigned int pos, int &layer, int &disk, int &layerdisk)
unsigned int	initLayerDisk (unsigned int pos)
void	initLayerDisksandISeed (unsigned int &layerdisk1, unsigned int &layerdisk2, unsigned int &iSeed)
unsigned int	nbits (unsigned int power)
	ProcessBase (std::string name, Settings const &settings, Globals *global)
virtual	~ProcessBase ()=default

Private Attributes
int	disk1_
int	disk2_
int	firstphibits_
std::vector< VMStubsTEMemory * >	firstvmstubs_
int	iSeed_
int	layer1_
int	layer2_
int	secondphibits_
VMStubsTEMemory *	secondvmstubs_
std::vector< StubPairsMemory * >	stubpairs_
std::vector< std::set< short > >	table_

Additional Inherited Members
Protected Attributes inherited from trklet::ProcessBase
Globals *	globals_
std::string	name_
Settings const &	settings_

Detailed Description

Definition at line 20 of file TrackletEngineDisplaced.h.

Constructor & Destructor Documentation

TrackletEngineDisplaced()

TrackletEngineDisplaced::TrackletEngineDisplaced	(	std::string	name,
		Settings const &	settings,
		Globals *	global
	)

Definition at line 15 of file TrackletEngineDisplaced.cc.

References disk1_, disk2_, firstphibits_, firstvmstubs_, iSeed_, layer1_, layer2_, Skims_PA_cff::name, trklet::Settings::nfinephi(), secondphibits_, secondvmstubs_, trklet::ProcessBase::settings_, and stubpairs_.

    : ProcessBase(name, settings, global) {
  stubpairs_.clear();
  firstvmstubs_.clear();
  secondvmstubs_ = nullptr;
  layer1_ = 0;
  layer2_ = 0;
  disk1_ = 0;
  disk2_ = 0;
  string name1 = name.substr(1);  //this is to correct for "TED" having one more letter then "TE"
  if (name1[3] == 'L') {
    layer1_ = name1[4] - '0';
  }
  if (name1[3] == 'D') {
    disk1_ = name1[4] - '0';
  }
  if (name1[11] == 'L') {
    layer2_ = name1[12] - '0';
  }
  if (name1[11] == 'D') {
    disk2_ = name1[12] - '0';
  }
  if (name1[12] == 'L') {
    layer2_ = name1[13] - '0';
  }
  if (name1[12] == 'D') {
    disk2_ = name1[13] - '0';
  }

  iSeed_ = -1;
  if (layer1_ == 3 && layer2_ == 4)
    iSeed_ = 8;
  if (layer1_ == 5 && layer2_ == 6)
    iSeed_ = 9;
  if (layer1_ == 2 && layer2_ == 3)
    iSeed_ = 10;
  if (disk1_ == 1 && disk2_ == 2)
    iSeed_ = 11;

  firstphibits_ = settings_.nfinephi(0, iSeed_);
  secondphibits_ = settings_.nfinephi(1, iSeed_);
}

~TrackletEngineDisplaced()

TrackletEngineDisplaced::~TrackletEngineDisplaced ( )

override

Definition at line 58 of file TrackletEngineDisplaced.cc.

References table_.

{ table_.clear(); }

Member Function Documentation

addInput()

void TrackletEngineDisplaced::addInput ( MemoryBase *  memory, std::string  input  )

overridevirtual

Implements trklet::ProcessBase.

Definition at line 74 of file TrackletEngineDisplaced.cc.

References cms::cuda::assert(), Exception, firstvmstubs_, input, mps_setup::memory, trklet::ProcessBase::name_, secondvmstubs_, trklet::ProcessBase::settings_, createJobs::tmp, and trklet::Settings::writetrace().

                                                                       {
  if (settings_.writetrace()) {
    edm::LogVerbatim("Tracklet") << "In " << name_ << " adding input from " << memory->getName() << " to input "
                                 << input;
  }
  if (input == "firstvmstubin") {
    VMStubsTEMemory* tmp = dynamic_cast<VMStubsTEMemory*>(memory);
    assert(tmp != nullptr);
    firstvmstubs_.push_back(tmp);
    return;
  }
  if (input == "secondvmstubin") {
    VMStubsTEMemory* tmp = dynamic_cast<VMStubsTEMemory*>(memory);
    assert(tmp != nullptr);
    secondvmstubs_ = tmp;
    return;
  }
  throw cms::Exception("BadConfig") << __FILE__ << " " << __LINE__ << " Could not find input : " << input;
}

addOutput()

void TrackletEngineDisplaced::addOutput ( MemoryBase *  memory, std::string  output  )

overridevirtual

Implements trklet::ProcessBase.

Definition at line 60 of file TrackletEngineDisplaced.cc.

References cms::cuda::assert(), Exception, mps_setup::memory, trklet::ProcessBase::name_, convertSQLitetoXML_cfg::output, trklet::ProcessBase::settings_, stubpairs_, createJobs::tmp, and trklet::Settings::writetrace().

                                                                         {
  if (settings_.writetrace()) {
    edm::LogVerbatim("Tracklet") << "In " << name_ << " adding output to " << memory->getName() << " to output "
                                 << output;
  }
  if (output == "stubpairout") {
    StubPairsMemory* tmp = dynamic_cast<StubPairsMemory*>(memory);
    assert(tmp != nullptr);
    stubpairs_.push_back(tmp);
    return;
  }
  throw cms::Exception("BadConfig") << __FILE__ << " " << __LINE__ << " Could not find output : " << output;
}

execute()

void TrackletEngineDisplaced::execute

(

void

)

Definition at line 94 of file TrackletEngineDisplaced.cc.

References cms::cuda::assert(), trklet::VMStubTE::bend(), trklet::Settings::benddecode(), newFWLiteAna::bin, trklet::Settings::debugTracklet(), trklet::Stub::disk(), disk1_, disk2_, trklet::Settings::enableTripletTables(), trklet::VMStubTE::finephi(), firstphibits_, firstvmstubs_, groupFilesInBlocks::fout, trklet::MemoryBase::getName(), trklet::ProcessBase::getName(), trklet::VMStubsTEMemory::getVMStubTE(), trklet::VMStubsTEMemory::getVMStubTEBinned(), trklet::ProcessBase::globals_, mps_fire::i, iSeed_, trklet::VMStubTE::isPSmodule(), dqmiolumiharvest::j, trklet::Stub::l1tstub(), dqmdumpme::last, layer1_, layer2_, trklet::Settings::maxStep(), trklet::ProcessBase::name_, trklet::FPGAWord::nbits(), trklet::VMStubsTEMemory::nVMStubs(), trklet::VMStubsTEMemory::nVMStubsBinned(), trklet::Globals::ofstream(), trklet::L1TStub::phi(), trklet::L1TStub::r(), readTables(), secondphibits_, secondvmstubs_, trklet::ProcessBase::settings_, trklet::VMStubTE::stub(), stubpairs_, table_, trklet::Settings::useSeed(), trklet::FPGAWord::value(), trklet::VMStubTE::vmbits(), trklet::Settings::writeMonitorData(), trklet::Settings::writeTripletTables(), and trklet::L1TStub::z().

                                      {
  if (!settings_.useSeed(iSeed_))
    return;

  if (table_.empty() && (settings_.enableTripletTables() && !settings_.writeTripletTables()))
    readTables();

  unsigned int countall = 0;
  unsigned int countpass = 0;
  unsigned int nInnerStubs = 0;

  for (unsigned int iInnerMem = 0; iInnerMem < firstvmstubs_.size();
       nInnerStubs += firstvmstubs_.at(iInnerMem)->nVMStubs(), iInnerMem++)
    ;

  assert(!firstvmstubs_.empty());
  assert(secondvmstubs_ != nullptr);

  for (auto& iInnerMem : firstvmstubs_) {
    assert(iInnerMem->nVMStubs() == iInnerMem->nVMStubs());
    for (unsigned int i = 0; i < iInnerMem->nVMStubs(); i++) {
      const VMStubTE& firstvmstub = iInnerMem->getVMStubTE(i);
      if (settings_.debugTracklet()) {
        edm::LogVerbatim("Tracklet") << "In " << getName() << " have first stub";
      }

      if ((layer1_ == 3 && layer2_ == 4) || (layer1_ == 5 && layer2_ == 6)) {
        int lookupbits = firstvmstub.vmbits().value() & 1023;
        int zdiffmax = (lookupbits >> 7);
        int newbin = (lookupbits & 127);
        int bin = newbin / 8;

        int zbinfirst = newbin & 7;

        int start = (bin >> 1);
        int last = start + (bin & 1);

        assert(last < 8);

        if (settings_.debugTracklet()) {
          edm::LogVerbatim("Tracklet") << "Will look in zbins " << start << " to " << last;
        }
        for (int ibin = start; ibin <= last; ibin++) {
          for (unsigned int j = 0; j < secondvmstubs_->nVMStubsBinned(ibin); j++) {
            if (settings_.debugTracklet()) {
              edm::LogVerbatim("Tracklet") << "In " << getName() << " have second stub(1) " << ibin << " " << j;
            }

            if (countall >= settings_.maxStep("TE"))
              break;
            countall++;
            const VMStubTE& secondvmstub = secondvmstubs_->getVMStubTEBinned(ibin, j);

            int zbin = (secondvmstub.vmbits().value() & 7);
            if (start != ibin)
              zbin += 8;
            if (zbin < zbinfirst || zbin - zbinfirst > zdiffmax) {
              if (settings_.debugTracklet()) {
                edm::LogVerbatim("Tracklet") << "Stubpair rejected because of wrong zbin";
              }
              continue;
            }

            assert(firstphibits_ != -1);
            assert(secondphibits_ != -1);

            FPGAWord iphifirstbin = firstvmstub.finephi();
            FPGAWord iphisecondbin = secondvmstub.finephi();

            unsigned int index = (iphifirstbin.value() << secondphibits_) + iphisecondbin.value();

            FPGAWord firstbend = firstvmstub.bend();
            FPGAWord secondbend = secondvmstub.bend();

            index = (index << firstbend.nbits()) + firstbend.value();
            index = (index << secondbend.nbits()) + secondbend.value();

            if ((settings_.enableTripletTables() && !settings_.writeTripletTables()) &&
                (index >= table_.size() || table_.at(index).empty())) {
              if (settings_.debugTracklet()) {
                edm::LogVerbatim("Tracklet")
                    << "Stub pair rejected because of stub pt cut bends : "
                    << settings_.benddecode(firstvmstub.bend().value(), layer1_ - 1, firstvmstub.isPSmodule()) << " "
                    << settings_.benddecode(secondvmstub.bend().value(), layer2_ - 1, secondvmstub.isPSmodule());
              }

              //FIXME temporarily commented out until stub bend table fixed
              //if (!settings_.writeTripletTables())
              //  continue;
            }

            if (settings_.debugTracklet())
              edm::LogVerbatim("Tracklet") << "Adding layer-layer pair in " << getName();
            for (unsigned int isp = 0; isp < stubpairs_.size(); ++isp) {
              if (!settings_.enableTripletTables() || settings_.writeTripletTables() || table_.at(index).count(isp)) {
                if (settings_.writeMonitorData("Seeds")) {
                  ofstream fout("seeds.txt", ofstream::app);
                  fout << __FILE__ << ":" << __LINE__ << " " << name_ << " " << iSeed_ << endl;
                  fout.close();
                }
                stubpairs_.at(isp)->addStubPair(firstvmstub, secondvmstub, index, getName());
              }
            }

            countpass++;
          }
        }

      } else if (layer1_ == 2 && layer2_ == 3) {
        int lookupbits = firstvmstub.vmbits().value() & 1023;
        int zdiffmax = (lookupbits >> 7);
        int newbin = (lookupbits & 127);
        int bin = newbin / 8;

        int zbinfirst = newbin & 7;

        int start = (bin >> 1);
        int last = start + (bin & 1);

        assert(last < 8);

        if (settings_.debugTracklet()) {
          edm::LogVerbatim("Tracklet") << "Will look in zbins " << start << " to " << last;
        }
        for (int ibin = start; ibin <= last; ibin++) {
          for (unsigned int j = 0; j < secondvmstubs_->nVMStubsBinned(ibin); j++) {
            if (settings_.debugTracklet()) {
              edm::LogVerbatim("Tracklet") << "In " << getName() << " have second stub(2) ";
            }

            if (countall >= settings_.maxStep("TE"))
              break;
            countall++;

            const VMStubTE& secondvmstub = secondvmstubs_->getVMStubTEBinned(ibin, j);

            int zbin = (secondvmstub.vmbits().value() & 7);
            if (start != ibin)
              zbin += 8;
            if (zbin < zbinfirst || zbin - zbinfirst > zdiffmax) {
              if (settings_.debugTracklet()) {
                edm::LogVerbatim("Tracklet") << "Stubpair rejected because of wrong zbin";
              }
              continue;
            }

            assert(firstphibits_ != -1);
            assert(secondphibits_ != -1);

            FPGAWord iphifirstbin = firstvmstub.finephi();
            FPGAWord iphisecondbin = secondvmstub.finephi();

            unsigned int index = (iphifirstbin.value() << secondphibits_) + iphisecondbin.value();

            FPGAWord firstbend = firstvmstub.bend();
            FPGAWord secondbend = secondvmstub.bend();

            index = (index << firstbend.nbits()) + firstbend.value();
            index = (index << secondbend.nbits()) + secondbend.value();

            if ((settings_.enableTripletTables() && !settings_.writeTripletTables()) &&
                (index >= table_.size() || table_.at(index).empty())) {
              if (settings_.debugTracklet()) {
                edm::LogVerbatim("Tracklet")
                    << "Stub pair rejected because of stub pt cut bends : "
                    << settings_.benddecode(firstvmstub.bend().value(), layer1_ - 1, firstvmstub.isPSmodule()) << " "
                    << settings_.benddecode(secondvmstub.bend().value(), layer2_ - 1, secondvmstub.isPSmodule());
              }
              continue;
            }

            if (settings_.debugTracklet())
              edm::LogVerbatim("Tracklet") << "Adding layer-layer pair in " << getName();
            for (unsigned int isp = 0; isp < stubpairs_.size(); ++isp) {
              if ((!settings_.enableTripletTables() || settings_.writeTripletTables()) ||
                  (index < table_.size() && table_.at(index).count(isp))) {
                if (settings_.writeMonitorData("Seeds")) {
                  ofstream fout("seeds.txt", ofstream::app);
                  fout << __FILE__ << ":" << __LINE__ << " " << name_ << " " << iSeed_ << endl;
                  fout.close();
                }
                stubpairs_.at(isp)->addStubPair(firstvmstub, secondvmstub, index, getName());
              }
            }

            countpass++;
          }
        }

      } else if (disk1_ == 1 && disk2_ == 2) {
        if (settings_.debugTracklet())
          edm::LogVerbatim("Tracklet") << getName() << " Disk-disk pair";

        int lookupbits = firstvmstub.vmbits().value() & 511;
        bool negdisk = firstvmstub.stub()->disk().value() < 0;
        int rdiffmax = (lookupbits >> 6);
        int newbin = (lookupbits & 63);
        int bin = newbin / 8;

        int rbinfirst = newbin & 7;

        int start = (bin >> 1);
        if (negdisk)
          start += 4;
        int last = start + (bin & 1);
        assert(last < 8);
        for (int ibin = start; ibin <= last; ibin++) {
          if (settings_.debugTracklet()) {
            edm::LogVerbatim("Tracklet") << getName() << " looking for matching stub in " << secondvmstubs_->getName()
                                         << " in bin = " << ibin << " with " << secondvmstubs_->nVMStubsBinned(ibin)
                                         << " stubs";
          }
          for (unsigned int j = 0; j < secondvmstubs_->nVMStubsBinned(ibin); j++) {
            if (countall >= settings_.maxStep("TE"))
              break;
            countall++;

            const VMStubTE& secondvmstub = secondvmstubs_->getVMStubTEBinned(ibin, j);

            int rbin = (secondvmstub.vmbits().value() & 7);
            if (start != ibin)
              rbin += 8;
            if (rbin < rbinfirst)
              continue;
            if (rbin - rbinfirst > rdiffmax)
              continue;

            unsigned int irsecondbin = secondvmstub.vmbits().value() >> 2;

            FPGAWord iphifirstbin = firstvmstub.finephi();
            FPGAWord iphisecondbin = secondvmstub.finephi();

            unsigned int index = (irsecondbin << (secondphibits_ + firstphibits_)) +
                                 (iphifirstbin.value() << secondphibits_) + iphisecondbin.value();

            FPGAWord firstbend = firstvmstub.bend();
            FPGAWord secondbend = secondvmstub.bend();

            index = (index << firstbend.nbits()) + firstbend.value();
            index = (index << secondbend.nbits()) + secondbend.value();

            if ((settings_.enableTripletTables() && !settings_.writeTripletTables()) &&
                (index >= table_.size() || table_.at(index).empty())) {
              if (settings_.debugTracklet()) {
                edm::LogVerbatim("Tracklet")
                    << "Stub pair rejected because of stub pt cut bends : "
                    << settings_.benddecode(firstvmstub.bend().value(), disk1_ + 5, firstvmstub.isPSmodule()) << " "
                    << settings_.benddecode(secondvmstub.bend().value(), disk2_ + 5, secondvmstub.isPSmodule());
              }
              continue;
            }

            if (settings_.debugTracklet())
              edm::LogVerbatim("Tracklet") << "Adding disk-disk pair in " << getName();

            for (unsigned int isp = 0; isp < stubpairs_.size(); ++isp) {
              if ((!settings_.enableTripletTables() || settings_.writeTripletTables()) ||
                  (index < table_.size() && table_.at(index).count(isp))) {
                if (settings_.writeMonitorData("Seeds")) {
                  ofstream fout("seeds.txt", ofstream::app);
                  fout << __FILE__ << ":" << __LINE__ << " " << name_ << " " << iSeed_ << endl;
                  fout.close();
                }
                stubpairs_.at(isp)->addStubPair(firstvmstub, secondvmstub, index, getName());
              }
            }
            countpass++;
          }
        }
      }
    }
  }
  if (countall > 5000) {
    edm::LogVerbatim("Tracklet") << "In TrackletEngineDisplaced::execute : " << getName() << " " << nInnerStubs << " "
                                 << secondvmstubs_->nVMStubs() << " " << countall << " " << countpass;
    for (auto& iInnerMem : firstvmstubs_) {
      for (unsigned int i = 0; i < iInnerMem->nVMStubs(); i++) {
        const VMStubTE& firstvmstub = iInnerMem->getVMStubTE(i);
        edm::LogVerbatim("Tracklet") << "In TrackletEngineDisplaced::execute first stub : "
                                     << firstvmstub.stub()->l1tstub()->r() << " "
                                     << firstvmstub.stub()->l1tstub()->phi() << " "
                                     << firstvmstub.stub()->l1tstub()->r() * firstvmstub.stub()->l1tstub()->phi() << " "
                                     << firstvmstub.stub()->l1tstub()->z();
      }
    }
    for (unsigned int i = 0; i < secondvmstubs_->nVMStubs(); i++) {
      const VMStubTE& secondvmstub = secondvmstubs_->getVMStubTE(i);
      edm::LogVerbatim("Tracklet") << "In TrackletEngineDisplaced::execute second stub : "
                                   << secondvmstub.stub()->l1tstub()->r() << " "
                                   << secondvmstub.stub()->l1tstub()->phi() << " "
                                   << secondvmstub.stub()->l1tstub()->r() * secondvmstub.stub()->l1tstub()->phi() << " "
                                   << secondvmstub.stub()->l1tstub()->z();
    }
  }

  if (settings_.writeMonitorData("TED")) {
    globals_->ofstream("trackletenginedisplaces.txt") << getName() << " " << countall << " " << countpass << endl;
  }
}

memNameToIndex()

short TrackletEngineDisplaced::memNameToIndex

(

const std::string &

name

)

Definition at line 417 of file TrackletEngineDisplaced.cc.

References Skims_PA_cff::name, and stubpairs_.

Referenced by readTables().

                                                                {
  for (unsigned int isp = 0; isp < stubpairs_.size(); ++isp)
    if (stubpairs_.at(isp)->getName() == name)
      return isp;
  return -1;
}

readTables()

void TrackletEngineDisplaced::readTables

(

)

Definition at line 394 of file TrackletEngineDisplaced.cc.

References Exception, groupFilesInBlocks::fin, recoMuon::in, mps_splice::line, memNameToIndex(), trklet::ProcessBase::name_, trklet::ProcessBase::settings_, table_, HLT_2024v14_cff::tableName, and trklet::Settings::tableTEDFile().

Referenced by execute().

                                         {
  ifstream fin;
  string tableName, line, word;

  string tablePath = settings_.tableTEDFile();
  unsigned int finddir = tablePath.find("table_TED_");
  tableName = tablePath.substr(0, finddir) + "table_" + name_ + ".txt";

  fin.open(tableName, ifstream::in);
  if (!fin) {
    throw cms::Exception("BadConfig") << "TripletEngine::readTables, file " << tableName << " not known";
  }

  while (getline(fin, line)) {
    istringstream iss(line);
    table_.resize(table_.size() + 1);

    while (iss >> word)
      table_[table_.size() - 1].insert(memNameToIndex(word));
  }
  fin.close();
}

Member Data Documentation

disk1_

int trklet::TrackletEngineDisplaced::disk1_

private

Definition at line 38 of file TrackletEngineDisplaced.h.

Referenced by execute(), and TrackletEngineDisplaced().

disk2_

int trklet::TrackletEngineDisplaced::disk2_

private

Definition at line 39 of file TrackletEngineDisplaced.h.

Referenced by execute(), and TrackletEngineDisplaced().

firstphibits_

int trklet::TrackletEngineDisplaced::firstphibits_

private

Definition at line 48 of file TrackletEngineDisplaced.h.

Referenced by execute(), and TrackletEngineDisplaced().

firstvmstubs_

std::vector<VMStubsTEMemory*> trklet::TrackletEngineDisplaced::firstvmstubs_

private

Definition at line 41 of file TrackletEngineDisplaced.h.

Referenced by addInput(), execute(), and TrackletEngineDisplaced().

iSeed_

int trklet::TrackletEngineDisplaced::iSeed_

private

Definition at line 51 of file TrackletEngineDisplaced.h.

Referenced by execute(), and TrackletEngineDisplaced().

layer1_

int trklet::TrackletEngineDisplaced::layer1_

private

Definition at line 36 of file TrackletEngineDisplaced.h.

Referenced by execute(), and TrackletEngineDisplaced().

layer2_

int trklet::TrackletEngineDisplaced::layer2_

private

Definition at line 37 of file TrackletEngineDisplaced.h.

Referenced by execute(), and TrackletEngineDisplaced().

secondphibits_

int trklet::TrackletEngineDisplaced::secondphibits_

private

Definition at line 49 of file TrackletEngineDisplaced.h.

Referenced by execute(), and TrackletEngineDisplaced().

secondvmstubs_

VMStubsTEMemory* trklet::TrackletEngineDisplaced::secondvmstubs_

private

Definition at line 42 of file TrackletEngineDisplaced.h.

Referenced by addInput(), execute(), and TrackletEngineDisplaced().

stubpairs_

std::vector<StubPairsMemory*> trklet::TrackletEngineDisplaced::stubpairs_

private

Definition at line 44 of file TrackletEngineDisplaced.h.

Referenced by addOutput(), execute(), memNameToIndex(), and TrackletEngineDisplaced().

table_

std::vector<std::set<short> > trklet::TrackletEngineDisplaced::table_

private

Definition at line 46 of file TrackletEngineDisplaced.h.

Referenced by execute(), readTables(), and ~TrackletEngineDisplaced().