trklet::MatchEngine Class Reference

#include <MatchEngine.h>

Inheritance diagram for trklet::MatchEngine:

Public Member Functions
void	addInput (MemoryBase *memory, std::string input) override
void	addOutput (MemoryBase *memory, std::string output) override
void	execute ()
	MatchEngine (std::string name, Settings const &settings, Globals *global, unsigned int iSector)
	~MatchEngine () override=default
Public Member Functions inherited from trklet::ProcessBase
unsigned int	getISeed (std::string name)
std::string const &	getName () const
unsigned int	initLayerDisk (unsigned int pos)
void	initLayerDisk (unsigned int pos, int &layer, int &disk)
void	initLayerDisk (unsigned int pos, int &layer, int &disk, int &layerdisk)
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, unsigned int iSector)
virtual	~ProcessBase ()=default

Private Attributes
CandidateMatchMemory *	candmatches_
int	disk_
int	layer_
std::vector< bool >	table2S_
std::vector< bool >	table_
std::vector< bool >	tablePS_
VMProjectionsMemory *	vmprojs_
VMStubsMEMemory *	vmstubs_

Additional Inherited Members
Protected Attributes inherited from trklet::ProcessBase
Globals *	globals_
unsigned int	iSector_
std::string	name_
double	phimax_
double	phimin_
Settings const &	settings_

Detailed Description

Definition at line 16 of file MatchEngine.h.

Constructor & Destructor Documentation

MatchEngine()

MatchEngine::MatchEngine	(	std::string	name,
		Settings const &	settings,
		Globals *	global,
		unsigned int	iSector
	)

Definition at line 16 of file MatchEngine.cc.

    : ProcessBase(name, settings, global, iSector) {
  layer_ = 0;
  disk_ = 0;
  string subname = name.substr(3, 2);
  if (subname.substr(0, 1) == "L")
    layer_ = stoi(subname.substr(1, 1));
  else if (subname.substr(0, 1) == "D")
    disk_ = stoi(subname.substr(1, 1));
  else
    throw cms::Exception("BadConfig") << __FILE__ << " " << __LINE__ << " " << name << " subname = " << subname << " "
                                      << layer_ << " " << disk_;
 
  if (layer_ > 0) {
    unsigned int nbits = 3;
    if (layer_ >= 4)
      nbits = 4;
 
    for (unsigned int irinv = 0; irinv < 32; irinv++) {
      double rinv = (irinv - 15.5) * (1 << (settings_.nbitsrinv() - 5)) * settings_.krinvpars();
 
      double stripPitch =
          (settings_.rmean(layer_ - 1) < settings_.rcrit()) ? settings_.stripPitch(true) : settings_.stripPitch(false);
      double projbend = bend(settings_.rmean(layer_ - 1), rinv, stripPitch);
      for (unsigned int ibend = 0; ibend < (unsigned int)(1 << nbits); ibend++) {
        double stubbend = benddecode(ibend, layer_ <= 3);
        bool pass = std::abs(stubbend - projbend) < settings_.bendcutme(layer_ - 1);
        table_.push_back(pass);
      }
    }
 
    if (settings_.writeTable()) {
      ofstream out;
      char layer = '0' + layer_;
      string fname = "METable_L";
      fname += layer;
      fname += ".tab";
      out.open(fname.c_str());
      out << "{" << endl;
      for (unsigned int i = 0; i < table_.size(); i++) {
        if (i != 0) {
          out << "," << endl;
        }
        out << table_[i];
      }
      out << "};" << endl;
      out.close();
    }
  }
 
  if (disk_ > 0) {
    for (unsigned int iprojbend = 0; iprojbend < 32; iprojbend++) {
      double projbend = 0.5 * (iprojbend - 15.0);
      for (unsigned int ibend = 0; ibend < 8; ibend++) {
        double stubbend = benddecode(ibend, true);
        bool pass = std::abs(stubbend - projbend) < settings_.bendcutme(disk_ + 5);
        tablePS_.push_back(pass);
      }
      for (unsigned int ibend = 0; ibend < 16; ibend++) {
        double stubbend = benddecode(ibend, false);
        bool pass = std::abs(stubbend - projbend) < settings_.bendcutme(disk_ + 5);
        table2S_.push_back(pass);
      }
    }
  }
}

References funct::abs(), trklet::bend(), trklet::Settings::bendcutme(), trklet::benddecode(), disk_, Exception, alignmentValidation::fname, mps_fire::i, createfilelist::int, trklet::Settings::krinvpars(), layer_, Skims_PA_cff::name, trklet::ProcessBase::nbits(), trklet::Settings::nbitsrinv(), MillePedeFileConverter_cfg::out, trklet::Settings::rcrit(), trklet::rinv(), trklet::Settings::rmean(), trklet::ProcessBase::settings_, trklet::Settings::stripPitch(), table2S_, table_, tablePS_, and trklet::Settings::writeTable().

~MatchEngine()

trklet::MatchEngine::~MatchEngine ( )

overridedefault

Member Function Documentation

addInput()

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

overridevirtual

Implements trklet::ProcessBase.

Definition at line 97 of file MatchEngine.cc.

                                                           {
  if (settings_.writetrace()) {
    edm::LogVerbatim("Tracklet") << "In " << name_ << " adding input from " << memory->getName() << " to input "
                                 << input;
  }
  if (input == "vmstubin") {
    auto* tmp = dynamic_cast<VMStubsMEMemory*>(memory);
    assert(tmp != nullptr);
    vmstubs_ = tmp;
    return;
  }
  if (input == "vmprojin") {
    auto* tmp = dynamic_cast<VMProjectionsMemory*>(memory);
    assert(tmp != nullptr);
    vmprojs_ = tmp;
    return;
  }
  throw cms::Exception("BadConfig") << __FILE__ << " " << __LINE__ << " could not find input: " << input;
}

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

addOutput()

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

overridevirtual

Implements trklet::ProcessBase.

Definition at line 83 of file MatchEngine.cc.

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

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

execute()

void MatchEngine::execute

(

void

)

Definition at line 117 of file MatchEngine.cc.

                          {
  bool barrel = layer_ > 0;
 
  unsigned int countall = 0;
  unsigned int countpass = 0;
 
  constexpr unsigned int kNBitsBuffer = 3;
 
  int writeindex = 0;
  int readindex = 0;
  std::pair<int, int> projbuffer[1 << kNBitsBuffer];  //iproj zbin
 
  //The next projection to read, the number of projections and flag if we have more projections to read
  int iproj = 0;
  int nproj = vmprojs_->nTracklets();
  bool moreproj = iproj < nproj;
 
  //Projection that is read from the buffer and compared to stubs
  int rzbin = 0;
  int projfinerz = 0;
  int projfinerzadj = 0;
 
  int projindex = 0;
  int projrinv = 0;
  bool isPSseed = false;
 
  //Number of stubs for current zbin and the stub being processed on this clock
  int nstubs = 0;
  int istub = 0;
 
  //Main processing loops starts here
  for (unsigned int istep = 0; istep < settings_.maxStep("ME"); istep++) {
    countall++;
 
    int writeindexplus = (writeindex + 1) % (1 << kNBitsBuffer);
    int writeindexplusplus = (writeindex + 2) % (1 << kNBitsBuffer);
 
    //Determine if buffer is full - or near full as a projection
    //can point to two z bins we might fill two slots in the buffer
    bool bufferfull = (writeindexplus == readindex) || (writeindexplusplus == readindex);
 
    //Determin if buffer is empty
    bool buffernotempty = (writeindex != readindex);
 
    //If we have more projections and the buffer is not full we read
    //next projection and put in buffer if there are stubs in the
    //memory the projection points to
 
    if ((!moreproj) && (!buffernotempty))
      break;
 
    if (moreproj && (!bufferfull)) {
      Tracklet* proj = vmprojs_->getTracklet(iproj);
 
      int iprojtmp = iproj;
 
      iproj++;
      moreproj = iproj < nproj;
 
      unsigned int rzfirst = barrel ? proj->zbin1projvm(layer_) : proj->rbin1projvm(disk_);
      unsigned int rzlast = rzfirst;
      bool second = (barrel ? proj->zbin2projvm(layer_) : proj->rbin2projvm(disk_)) == 1;
      if (second)
        rzlast += 1;
 
      //Check if there are stubs in the memory
      int nstubfirst = vmstubs_->nStubsBin(rzfirst);
      int nstublast = vmstubs_->nStubsBin(rzlast);
      bool savefirst = nstubfirst != 0;
      bool savelast = second && (nstublast != 0);
 
      int writeindextmp = writeindex;
      int writeindextmpplus = (writeindex + 1) % (1 << kNBitsBuffer);
 
      if (savefirst && savelast) {
        writeindex = writeindexplusplus;
      } else if (savefirst || savelast) {
        writeindex = writeindexplus;
      }
 
      if (savefirst) {  //TODO for HLS (make code logic simpler)
        std::pair<int, int> tmp(iprojtmp, rzfirst);
        projbuffer[writeindextmp] = tmp;
      }
      if (savelast) {
        std::pair<int, int> tmp(iprojtmp, rzlast + 100);  //TODO for HLS (fix flagging that this is second bin)
        if (savefirst) {
          projbuffer[writeindextmpplus] = tmp;
        } else {
          projbuffer[writeindextmp] = tmp;
        }
      }
    }
 
    //If the buffer is not empty we have a projection that we need to process.
 
    if (buffernotempty) {
      int istubtmp = istub;
 
      //New projection
      if (istub == 0) {
        projindex = projbuffer[readindex].first;
        rzbin = projbuffer[readindex].second;
        bool second = false;
        if (rzbin >= 100) {
          rzbin -= 100;
          second = true;
        }
 
        Tracklet* proj = vmprojs_->getTracklet(projindex);
 
        nstubs = vmstubs_->nStubsBin(rzbin);
 
        projfinerz = barrel ? proj->finezvm(layer_) : proj->finervm(disk_);
 
        projrinv =
            barrel
                ? (16 + (((-2) * proj->fpgaphiprojder(layer_).value()) >> (proj->fpgaphiprojder(layer_).nbits() - 4)))
                : proj->getBendIndex(disk_).value();
        assert(projrinv >= 0);
        if (settings_.extended() && projrinv == 32) {
          if (settings_.debugTracklet()) {
            edm::LogVerbatim("Tracklet") << "Extended tracking, projrinv:" << projrinv;
          }
          projrinv = 31;
        }
        assert(projrinv < 32);
 
        isPSseed = proj->PSseed() == 1;
 
        //Calculate fine z position
        if (second) {
          projfinerzadj = projfinerz - 8;
        } else {
          projfinerzadj = projfinerz;
        }
        if (nstubs == 1) {
          istub = 0;
          readindex = (readindex + 1) % (1 << kNBitsBuffer);
        } else {
          istub++;
        }
      } else {
        //Check if last stub, if so, go to next buffer entry
        if (istub + 1 >= nstubs) {
          istub = 0;
          readindex = (readindex + 1) % (1 << kNBitsBuffer);
        } else {
          istub++;
        }
      }
 
      //Read vmstub memory and extract data fields
      const VMStubME& vmstub = vmstubs_->getVMStubMEBin(rzbin, istubtmp);
 
      bool isPSmodule = vmstub.isPSmodule();
 
      int stubfinerz = vmstub.finerz().value();
 
      int nbits = isPSmodule ? 3 : 4;
 
      //TODO - should use finephi information to reduce combinatorics
 
      unsigned int index = (projrinv << nbits) + vmstub.bend().value();
 
      //Check if stub z position consistent
      int idrz = stubfinerz - projfinerzadj;
      bool pass;
 
      if (barrel) {
        if (isPSseed) {
          pass = idrz >= -2 && idrz <= 2;
        } else {
          pass = idrz >= -5 && idrz <= 5;
        }
      } else {
        if (isPSmodule) {
          pass = idrz >= -1 && idrz <= 1;
        } else {
          pass = idrz >= -5 && idrz <= 5;
        }
      }
 
      //Check if stub bend and proj rinv consistent
      if (pass) {
        if (barrel ? table_[index] : (isPSmodule ? tablePS_[index] : table2S_[index])) {
          Tracklet* proj = vmprojs_->getTracklet(projindex);
          std::pair<Tracklet*, int> tmp(proj, vmprojs_->getAllProjIndex(projindex));
          if (settings_.writeMonitorData("Seeds")) {
            ofstream fout("seeds.txt", ofstream::app);
            fout << __FILE__ << ":" << __LINE__ << " " << name_ << "_" << iSector_ << " " << proj->getISeed() << endl;
            fout.close();
          }
          candmatches_->addMatch(tmp, vmstub.stub());
          countpass++;
        }
      }
    }
  }
 
  if (settings_.writeMonitorData("ME")) {
    globals_->ofstream("matchengine.txt") << getName() << " " << countall << " " << countpass << endl;
  }
}

References trklet::CandidateMatchMemory::addMatch(), cms::cuda::assert(), Reference_intrackfit_cff::barrel, trklet::VMStubME::bend(), candmatches_, trklet::Settings::debugTracklet(), disk_, trklet::Settings::extended(), trklet::VMStubME::finerz(), groupFilesInBlocks::fout, trklet::VMProjectionsMemory::getAllProjIndex(), trklet::ProcessBase::getName(), trklet::VMProjectionsMemory::getTracklet(), trklet::VMStubsMEMemory::getVMStubMEBin(), trklet::ProcessBase::globals_, trklet::ProcessBase::iSector_, trklet::VMStubME::isPSmodule(), layer_, trklet::Settings::maxStep(), trklet::ProcessBase::name_, trklet::ProcessBase::nbits(), trklet::VMStubsMEMemory::nStubsBin(), trklet::VMProjectionsMemory::nTracklets(), trklet::Globals::ofstream(), amptDefault_cfi::proj, edm::second(), trklet::ProcessBase::settings_, trklet::VMStubME::stub(), table2S_, table_, tablePS_, createJobs::tmp, trklet::FPGAWord::value(), vmprojs_, vmstubs_, and trklet::Settings::writeMonitorData().

Member Data Documentation

candmatches_

CandidateMatchMemory* trklet::MatchEngine::candmatches_

private

Definition at line 31 of file MatchEngine.h.

Referenced by addOutput(), and execute().

disk_

int trklet::MatchEngine::disk_

private

Definition at line 34 of file MatchEngine.h.

Referenced by execute(), and MatchEngine().

layer_

int trklet::MatchEngine::layer_

private

Definition at line 33 of file MatchEngine.h.

Referenced by execute(), and MatchEngine().

table2S_

std::vector<bool> trklet::MatchEngine::table2S_

private

Definition at line 41 of file MatchEngine.h.

Referenced by execute(), and MatchEngine().

table_

std::vector<bool> trklet::MatchEngine::table_

private

Definition at line 37 of file MatchEngine.h.

Referenced by execute(), and MatchEngine().

tablePS_

std::vector<bool> trklet::MatchEngine::tablePS_

private

Definition at line 40 of file MatchEngine.h.

Referenced by execute(), and MatchEngine().

vmprojs_

VMProjectionsMemory* trklet::MatchEngine::vmprojs_

private

Definition at line 29 of file MatchEngine.h.

Referenced by addInput(), and execute().

vmstubs_

VMStubsMEMemory* trklet::MatchEngine::vmstubs_

private

Definition at line 28 of file MatchEngine.h.

Referenced by addInput(), and execute().