Back to home page

Project CMSSW displayed by LXR

 
 

    


File indexing completed on 2021-07-07 22:33:39

0001 #include "L1Trigger/TrackFindingTracklet/interface/TrackletEngineDisplaced.h"
0002 #include "L1Trigger/TrackFindingTracklet/interface/Settings.h"
0003 #include "L1Trigger/TrackFindingTracklet/interface/Globals.h"
0004 #include "L1Trigger/TrackFindingTracklet/interface/VMStubsTEMemory.h"
0005 #include "L1Trigger/TrackFindingTracklet/interface/StubPairsMemory.h"
0006 #include "L1Trigger/TrackFindingTracklet/interface/MemoryBase.h"
0007 #include "L1Trigger/TrackFindingTracklet/interface/FPGAWord.h"
0008 
0009 #include "FWCore/MessageLogger/interface/MessageLogger.h"
0010 #include "FWCore/Utilities/interface/Exception.h"
0011 
0012 using namespace std;
0013 using namespace trklet;
0014 
0015 TrackletEngineDisplaced::TrackletEngineDisplaced(string name, Settings const& settings, Globals* global)
0016     : ProcessBase(name, settings, global) {
0017   stubpairs_.clear();
0018   firstvmstubs_.clear();
0019   secondvmstubs_ = nullptr;
0020   layer1_ = 0;
0021   layer2_ = 0;
0022   disk1_ = 0;
0023   disk2_ = 0;
0024   string name1 = name.substr(1);  //this is to correct for "TED" having one more letter then "TE"
0025   if (name1[3] == 'L') {
0026     layer1_ = name1[4] - '0';
0027   }
0028   if (name1[3] == 'D') {
0029     disk1_ = name1[4] - '0';
0030   }
0031   if (name1[11] == 'L') {
0032     layer2_ = name1[12] - '0';
0033   }
0034   if (name1[11] == 'D') {
0035     disk2_ = name1[12] - '0';
0036   }
0037   if (name1[12] == 'L') {
0038     layer2_ = name1[13] - '0';
0039   }
0040   if (name1[12] == 'D') {
0041     disk2_ = name1[13] - '0';
0042   }
0043 
0044   iSeed_ = -1;
0045   if (layer1_ == 3 && layer2_ == 4)
0046     iSeed_ = 8;
0047   if (layer1_ == 5 && layer2_ == 6)
0048     iSeed_ = 9;
0049   if (layer1_ == 2 && layer2_ == 3)
0050     iSeed_ = 10;
0051   if (disk1_ == 1 && disk2_ == 2)
0052     iSeed_ = 11;
0053 
0054   firstphibits_ = settings_.nfinephi(0, iSeed_);
0055   secondphibits_ = settings_.nfinephi(1, iSeed_);
0056 }
0057 
0058 TrackletEngineDisplaced::~TrackletEngineDisplaced() { table_.clear(); }
0059 
0060 void TrackletEngineDisplaced::addOutput(MemoryBase* memory, string output) {
0061   if (settings_.writetrace()) {
0062     edm::LogVerbatim("Tracklet") << "In " << name_ << " adding output to " << memory->getName() << " to output "
0063                                  << output;
0064   }
0065   if (output == "stubpairout") {
0066     StubPairsMemory* tmp = dynamic_cast<StubPairsMemory*>(memory);
0067     assert(tmp != nullptr);
0068     stubpairs_.push_back(tmp);
0069     return;
0070   }
0071   throw cms::Exception("BadConfig") << __FILE__ << " " << __LINE__ << " Could not find output : " << output;
0072 }
0073 
0074 void TrackletEngineDisplaced::addInput(MemoryBase* memory, string input) {
0075   if (settings_.writetrace()) {
0076     edm::LogVerbatim("Tracklet") << "In " << name_ << " adding input from " << memory->getName() << " to input "
0077                                  << input;
0078   }
0079   if (input == "firstvmstubin") {
0080     VMStubsTEMemory* tmp = dynamic_cast<VMStubsTEMemory*>(memory);
0081     assert(tmp != nullptr);
0082     firstvmstubs_.push_back(tmp);
0083     return;
0084   }
0085   if (input == "secondvmstubin") {
0086     VMStubsTEMemory* tmp = dynamic_cast<VMStubsTEMemory*>(memory);
0087     assert(tmp != nullptr);
0088     secondvmstubs_ = tmp;
0089     return;
0090   }
0091   throw cms::Exception("BadConfig") << __FILE__ << " " << __LINE__ << " Could not find input : " << input;
0092 }
0093 
0094 void TrackletEngineDisplaced::execute() {
0095   if (!settings_.useSeed(iSeed_))
0096     return;
0097 
0098   if (table_.empty() && (settings_.enableTripletTables() && !settings_.writeTripletTables()))
0099     readTables();
0100 
0101   unsigned int countall = 0;
0102   unsigned int countpass = 0;
0103   unsigned int nInnerStubs = 0;
0104 
0105   for (unsigned int iInnerMem = 0; iInnerMem < firstvmstubs_.size();
0106        nInnerStubs += firstvmstubs_.at(iInnerMem)->nVMStubs(), iInnerMem++)
0107     ;
0108 
0109   assert(!firstvmstubs_.empty());
0110   assert(secondvmstubs_ != nullptr);
0111 
0112   for (auto& iInnerMem : firstvmstubs_) {
0113     assert(iInnerMem->nVMStubs() == iInnerMem->nVMStubs());
0114     for (unsigned int i = 0; i < iInnerMem->nVMStubs(); i++) {
0115       const VMStubTE& firstvmstub = iInnerMem->getVMStubTE(i);
0116       if (settings_.debugTracklet()) {
0117         edm::LogVerbatim("Tracklet") << "In " << getName() << " have first stub";
0118       }
0119 
0120       if ((layer1_ == 3 && layer2_ == 4) || (layer1_ == 5 && layer2_ == 6)) {
0121         int lookupbits = firstvmstub.vmbits().value() & 1023;
0122         int zdiffmax = (lookupbits >> 7);
0123         int newbin = (lookupbits & 127);
0124         int bin = newbin / 8;
0125 
0126         int zbinfirst = newbin & 7;
0127 
0128         int start = (bin >> 1);
0129         int last = start + (bin & 1);
0130 
0131         assert(last < 8);
0132 
0133         if (settings_.debugTracklet()) {
0134           edm::LogVerbatim("Tracklet") << "Will look in zbins " << start << " to " << last;
0135         }
0136         for (int ibin = start; ibin <= last; ibin++) {
0137           for (unsigned int j = 0; j < secondvmstubs_->nVMStubsBinned(ibin); j++) {
0138             if (settings_.debugTracklet()) {
0139               edm::LogVerbatim("Tracklet") << "In " << getName() << " have second stub(1) " << ibin << " " << j;
0140             }
0141 
0142             if (countall >= settings_.maxStep("TE"))
0143               break;
0144             countall++;
0145             const VMStubTE& secondvmstub = secondvmstubs_->getVMStubTEBinned(ibin, j);
0146 
0147             int zbin = (secondvmstub.vmbits().value() & 7);
0148             if (start != ibin)
0149               zbin += 8;
0150             if (zbin < zbinfirst || zbin - zbinfirst > zdiffmax) {
0151               if (settings_.debugTracklet()) {
0152                 edm::LogVerbatim("Tracklet") << "Stubpair rejected because of wrong zbin";
0153               }
0154               continue;
0155             }
0156 
0157             assert(firstphibits_ != -1);
0158             assert(secondphibits_ != -1);
0159 
0160             FPGAWord iphifirstbin = firstvmstub.finephi();
0161             FPGAWord iphisecondbin = secondvmstub.finephi();
0162 
0163             unsigned int index = (iphifirstbin.value() << secondphibits_) + iphisecondbin.value();
0164 
0165             FPGAWord firstbend = firstvmstub.bend();
0166             FPGAWord secondbend = secondvmstub.bend();
0167 
0168             index = (index << firstbend.nbits()) + firstbend.value();
0169             index = (index << secondbend.nbits()) + secondbend.value();
0170 
0171             if ((settings_.enableTripletTables() && !settings_.writeTripletTables()) &&
0172                 (index >= table_.size() || table_.at(index).empty())) {
0173               if (settings_.debugTracklet()) {
0174                 edm::LogVerbatim("Tracklet")
0175                     << "Stub pair rejected because of stub pt cut bends : "
0176                     << settings_.benddecode(firstvmstub.bend().value(), layer1_ - 1, firstvmstub.isPSmodule()) << " "
0177                     << settings_.benddecode(secondvmstub.bend().value(), layer2_ - 1, secondvmstub.isPSmodule());
0178               }
0179 
0180               //FIXME temporarily commented out until stub bend table fixed
0181               //if (!settings_.writeTripletTables())
0182               //  continue;
0183             }
0184 
0185             if (settings_.debugTracklet())
0186               edm::LogVerbatim("Tracklet") << "Adding layer-layer pair in " << getName();
0187             for (unsigned int isp = 0; isp < stubpairs_.size(); ++isp) {
0188               if (!settings_.enableTripletTables() || settings_.writeTripletTables() || table_.at(index).count(isp)) {
0189                 if (settings_.writeMonitorData("Seeds")) {
0190                   ofstream fout("seeds.txt", ofstream::app);
0191                   fout << __FILE__ << ":" << __LINE__ << " " << name_ << " " << iSeed_ << endl;
0192                   fout.close();
0193                 }
0194                 stubpairs_.at(isp)->addStubPair(firstvmstub, secondvmstub, index, getName());
0195               }
0196             }
0197 
0198             countpass++;
0199           }
0200         }
0201 
0202       } else if (layer1_ == 2 && layer2_ == 3) {
0203         int lookupbits = firstvmstub.vmbits().value() & 1023;
0204         int zdiffmax = (lookupbits >> 7);
0205         int newbin = (lookupbits & 127);
0206         int bin = newbin / 8;
0207 
0208         int zbinfirst = newbin & 7;
0209 
0210         int start = (bin >> 1);
0211         int last = start + (bin & 1);
0212 
0213         assert(last < 8);
0214 
0215         if (settings_.debugTracklet()) {
0216           edm::LogVerbatim("Tracklet") << "Will look in zbins " << start << " to " << last;
0217         }
0218         for (int ibin = start; ibin <= last; ibin++) {
0219           for (unsigned int j = 0; j < secondvmstubs_->nVMStubsBinned(ibin); j++) {
0220             if (settings_.debugTracklet()) {
0221               edm::LogVerbatim("Tracklet") << "In " << getName() << " have second stub(2) ";
0222             }
0223 
0224             if (countall >= settings_.maxStep("TE"))
0225               break;
0226             countall++;
0227 
0228             const VMStubTE& secondvmstub = secondvmstubs_->getVMStubTEBinned(ibin, j);
0229 
0230             int zbin = (secondvmstub.vmbits().value() & 7);
0231             if (start != ibin)
0232               zbin += 8;
0233             if (zbin < zbinfirst || zbin - zbinfirst > zdiffmax) {
0234               if (settings_.debugTracklet()) {
0235                 edm::LogVerbatim("Tracklet") << "Stubpair rejected because of wrong zbin";
0236               }
0237               continue;
0238             }
0239 
0240             assert(firstphibits_ != -1);
0241             assert(secondphibits_ != -1);
0242 
0243             FPGAWord iphifirstbin = firstvmstub.finephi();
0244             FPGAWord iphisecondbin = secondvmstub.finephi();
0245 
0246             unsigned int index = (iphifirstbin.value() << secondphibits_) + iphisecondbin.value();
0247 
0248             FPGAWord firstbend = firstvmstub.bend();
0249             FPGAWord secondbend = secondvmstub.bend();
0250 
0251             index = (index << firstbend.nbits()) + firstbend.value();
0252             index = (index << secondbend.nbits()) + secondbend.value();
0253 
0254             if ((settings_.enableTripletTables() && !settings_.writeTripletTables()) &&
0255                 (index >= table_.size() || table_.at(index).empty())) {
0256               if (settings_.debugTracklet()) {
0257                 edm::LogVerbatim("Tracklet")
0258                     << "Stub pair rejected because of stub pt cut bends : "
0259                     << settings_.benddecode(firstvmstub.bend().value(), layer1_ - 1, firstvmstub.isPSmodule()) << " "
0260                     << settings_.benddecode(secondvmstub.bend().value(), layer2_ - 1, secondvmstub.isPSmodule());
0261               }
0262               continue;
0263             }
0264 
0265             if (settings_.debugTracklet())
0266               edm::LogVerbatim("Tracklet") << "Adding layer-layer pair in " << getName();
0267             for (unsigned int isp = 0; isp < stubpairs_.size(); ++isp) {
0268               if ((!settings_.enableTripletTables() || settings_.writeTripletTables()) ||
0269                   (index < table_.size() && table_.at(index).count(isp))) {
0270                 if (settings_.writeMonitorData("Seeds")) {
0271                   ofstream fout("seeds.txt", ofstream::app);
0272                   fout << __FILE__ << ":" << __LINE__ << " " << name_ << " " << iSeed_ << endl;
0273                   fout.close();
0274                 }
0275                 stubpairs_.at(isp)->addStubPair(firstvmstub, secondvmstub, index, getName());
0276               }
0277             }
0278 
0279             countpass++;
0280           }
0281         }
0282 
0283       } else if (disk1_ == 1 && disk2_ == 2) {
0284         if (settings_.debugTracklet())
0285           edm::LogVerbatim("Tracklet") << getName() << " Disk-disk pair";
0286 
0287         int lookupbits = firstvmstub.vmbits().value() & 511;
0288         bool negdisk = firstvmstub.stub()->disk().value() < 0;
0289         int rdiffmax = (lookupbits >> 6);
0290         int newbin = (lookupbits & 63);
0291         int bin = newbin / 8;
0292 
0293         int rbinfirst = newbin & 7;
0294 
0295         int start = (bin >> 1);
0296         if (negdisk)
0297           start += 4;
0298         int last = start + (bin & 1);
0299         assert(last < 8);
0300         for (int ibin = start; ibin <= last; ibin++) {
0301           if (settings_.debugTracklet()) {
0302             edm::LogVerbatim("Tracklet") << getName() << " looking for matching stub in " << secondvmstubs_->getName()
0303                                          << " in bin = " << ibin << " with " << secondvmstubs_->nVMStubsBinned(ibin)
0304                                          << " stubs";
0305           }
0306           for (unsigned int j = 0; j < secondvmstubs_->nVMStubsBinned(ibin); j++) {
0307             if (countall >= settings_.maxStep("TE"))
0308               break;
0309             countall++;
0310 
0311             const VMStubTE& secondvmstub = secondvmstubs_->getVMStubTEBinned(ibin, j);
0312 
0313             int rbin = (secondvmstub.vmbits().value() & 7);
0314             if (start != ibin)
0315               rbin += 8;
0316             if (rbin < rbinfirst)
0317               continue;
0318             if (rbin - rbinfirst > rdiffmax)
0319               continue;
0320 
0321             unsigned int irsecondbin = secondvmstub.vmbits().value() >> 2;
0322 
0323             FPGAWord iphifirstbin = firstvmstub.finephi();
0324             FPGAWord iphisecondbin = secondvmstub.finephi();
0325 
0326             unsigned int index = (irsecondbin << (secondphibits_ + firstphibits_)) +
0327                                  (iphifirstbin.value() << secondphibits_) + iphisecondbin.value();
0328 
0329             FPGAWord firstbend = firstvmstub.bend();
0330             FPGAWord secondbend = secondvmstub.bend();
0331 
0332             index = (index << firstbend.nbits()) + firstbend.value();
0333             index = (index << secondbend.nbits()) + secondbend.value();
0334 
0335             if ((settings_.enableTripletTables() && !settings_.writeTripletTables()) &&
0336                 (index >= table_.size() || table_.at(index).empty())) {
0337               if (settings_.debugTracklet()) {
0338                 edm::LogVerbatim("Tracklet")
0339                     << "Stub pair rejected because of stub pt cut bends : "
0340                     << settings_.benddecode(firstvmstub.bend().value(), disk1_ + 5, firstvmstub.isPSmodule()) << " "
0341                     << settings_.benddecode(secondvmstub.bend().value(), disk2_ + 5, secondvmstub.isPSmodule());
0342               }
0343               continue;
0344             }
0345 
0346             if (settings_.debugTracklet())
0347               edm::LogVerbatim("Tracklet") << "Adding disk-disk pair in " << getName();
0348 
0349             for (unsigned int isp = 0; isp < stubpairs_.size(); ++isp) {
0350               if ((!settings_.enableTripletTables() || settings_.writeTripletTables()) ||
0351                   (index < table_.size() && table_.at(index).count(isp))) {
0352                 if (settings_.writeMonitorData("Seeds")) {
0353                   ofstream fout("seeds.txt", ofstream::app);
0354                   fout << __FILE__ << ":" << __LINE__ << " " << name_ << " " << iSeed_ << endl;
0355                   fout.close();
0356                 }
0357                 stubpairs_.at(isp)->addStubPair(firstvmstub, secondvmstub, index, getName());
0358               }
0359             }
0360             countpass++;
0361           }
0362         }
0363       }
0364     }
0365   }
0366   if (countall > 5000) {
0367     edm::LogVerbatim("Tracklet") << "In TrackletEngineDisplaced::execute : " << getName() << " " << nInnerStubs << " "
0368                                  << secondvmstubs_->nVMStubs() << " " << countall << " " << countpass;
0369     for (auto& iInnerMem : firstvmstubs_) {
0370       for (unsigned int i = 0; i < iInnerMem->nVMStubs(); i++) {
0371         const VMStubTE& firstvmstub = iInnerMem->getVMStubTE(i);
0372         edm::LogVerbatim("Tracklet") << "In TrackletEngineDisplaced::execute first stub : "
0373                                      << firstvmstub.stub()->l1tstub()->r() << " "
0374                                      << firstvmstub.stub()->l1tstub()->phi() << " "
0375                                      << firstvmstub.stub()->l1tstub()->r() * firstvmstub.stub()->l1tstub()->phi() << " "
0376                                      << firstvmstub.stub()->l1tstub()->z();
0377       }
0378     }
0379     for (unsigned int i = 0; i < secondvmstubs_->nVMStubs(); i++) {
0380       const VMStubTE& secondvmstub = secondvmstubs_->getVMStubTE(i);
0381       edm::LogVerbatim("Tracklet") << "In TrackletEngineDisplaced::execute second stub : "
0382                                    << secondvmstub.stub()->l1tstub()->r() << " "
0383                                    << secondvmstub.stub()->l1tstub()->phi() << " "
0384                                    << secondvmstub.stub()->l1tstub()->r() * secondvmstub.stub()->l1tstub()->phi() << " "
0385                                    << secondvmstub.stub()->l1tstub()->z();
0386     }
0387   }
0388 
0389   if (settings_.writeMonitorData("TED")) {
0390     globals_->ofstream("trackletenginedisplaces.txt") << getName() << " " << countall << " " << countpass << endl;
0391   }
0392 }
0393 
0394 void TrackletEngineDisplaced::readTables() {
0395   ifstream fin;
0396   string tableName, line, word;
0397 
0398   string tablePath = settings_.tableTEDFile();
0399   unsigned int finddir = tablePath.find("table_TED_");
0400   tableName = tablePath.substr(0, finddir) + "table_" + name_ + ".txt";
0401 
0402   fin.open(tableName, ifstream::in);
0403   if (!fin) {
0404     throw cms::Exception("BadConfig") << "TripletEngine::readTables, file " << tableName << " not known";
0405   }
0406 
0407   while (getline(fin, line)) {
0408     istringstream iss(line);
0409     table_.resize(table_.size() + 1);
0410 
0411     while (iss >> word)
0412       table_[table_.size() - 1].insert(memNameToIndex(word));
0413   }
0414   fin.close();
0415 }
0416 
0417 short TrackletEngineDisplaced::memNameToIndex(const string& name) {
0418   for (unsigned int isp = 0; isp < stubpairs_.size(); ++isp)
0419     if (stubpairs_.at(isp)->getName() == name)
0420       return isp;
0421   return -1;
0422 }