L1TMuonBarrelKalmanRegionModule Class Reference

#include <L1TMuonBarrelKalmanRegionModule.h>

Classes
class	SeedSorter

Public Member Functions
	L1TMuonBarrelKalmanRegionModule (const edm::ParameterSet &, int wheel, int sector)
L1MuKBMTrackCollection	process (L1TMuonBarrelKalmanAlgo *, const L1MuKBMTCombinedStubRefVector &stubs, int bx)
const int	wheel () const
	~L1TMuonBarrelKalmanRegionModule ()

Private Member Functions
L1MuKBMTrackCollection	cleanHigher (const L1MuKBMTrackCollection &tracks1, const L1MuKBMTrackCollection &tracks2)
L1MuKBMTrackCollection	cleanLower (const L1MuKBMTrackCollection &tracks1, const L1MuKBMTrackCollection &tracks2)
L1MuKBMTrackCollection	cleanRegion (const L1MuKBMTrackCollection &, const L1MuKBMTrackCollection &, const L1MuKBMTrackCollection &)
L1MuKBMTrackCollection	selfClean (const L1MuKBMTrackCollection &tracks)
L1MuKBMTrackCollection	sort4 (const L1MuKBMTrackCollection &in)

Private Attributes
int	nextSector_
int	nextWheel_
int	previousSector_
int	sector_
int	verbose_
int	wheel_

Detailed Description

Definition at line 8 of file L1TMuonBarrelKalmanRegionModule.h.

Constructor & Destructor Documentation

L1TMuonBarrelKalmanRegionModule::L1TMuonBarrelKalmanRegionModule	(	const edm::ParameterSet &	iConfig,
		int	wheel,
		int	sector
	)

Definition at line 5 of file L1TMuonBarrelKalmanRegionModule.cc.

References nextSector_, nextWheel_, and previousSector_.

                                                                                                                   : 
  verbose_(iConfig.getParameter<int>("verbose")),
  sector_(sector),
  wheel_(wheel)
{

  if (sector==11) {
    nextSector_=0;
    previousSector_ = 10;
  }
  else if (sector==0) {
    nextSector_=1;
    previousSector_ = 11;
  }
  else {
    nextSector_= sector+1;
    previousSector_ = sector-1;
  }

  switch(wheel) {

  case -2:
    nextWheel_=-1;
    break;

  case -1:
    nextWheel_=0;
    break;

  case 0:
    nextWheel_=999;
    break;

  case 1:
    nextWheel_=0;
    break;

  case 2:
    nextWheel_=1;
    break;

  default:
    nextWheel_=999;
    break;
  }
}

L1TMuonBarrelKalmanRegionModule::~L1TMuonBarrelKalmanRegionModule

(

)

Definition at line 54 of file L1TMuonBarrelKalmanRegionModule.cc.

{}

Member Function Documentation

L1MuKBMTrackCollection L1TMuonBarrelKalmanRegionModule::cleanHigher ( const L1MuKBMTrackCollection &  tracks1, const L1MuKBMTrackCollection &  tracks2  )

private

Definition at line 154 of file L1TMuonBarrelKalmanRegionModule.cc.

References mps_fire::i, keep, MillePedeFileConverter_cfg::out, and parallelization::uint().

Referenced by cleanRegion().

                                                                                                                                               {
  L1MuKBMTrackCollection out;

  for (uint i=0;i<tracks1.size();++i) {
    bool keep=true;
        
    for (uint j=0;j<tracks2.size();++j) {
      if (tracks1[i].overlapTrack(tracks2[j])) {
    if (tracks1[i].rank()<=tracks2[j].rank()) {
      keep=false;
    }
      }
    }
    if (keep)
      out.push_back(tracks1[i]);
  }
 
  return out;
}

L1MuKBMTrackCollection L1TMuonBarrelKalmanRegionModule::cleanLower ( const L1MuKBMTrackCollection &  tracks1, const L1MuKBMTrackCollection &  tracks2  )

private

Definition at line 174 of file L1TMuonBarrelKalmanRegionModule.cc.

References mps_fire::i, keep, MillePedeFileConverter_cfg::out, and parallelization::uint().

Referenced by cleanRegion().

                                                                                                                                              {
  L1MuKBMTrackCollection out;

  for (uint i=0;i<tracks1.size();++i) {
    bool keep=true;
        
    for (uint j=0;j<tracks2.size();++j) {
      if (tracks1[i].overlapTrack(tracks2[j])) {
    if (tracks1[i].rank()<tracks2[j].rank()) {
      keep=false;
    }
      }
    }
    if (keep)
      out.push_back(tracks1[i]);
  }
 
  return out;
}

L1MuKBMTrackCollection L1TMuonBarrelKalmanRegionModule::cleanRegion ( const L1MuKBMTrackCollection &  tracks2, const L1MuKBMTrackCollection &  tracks3, const L1MuKBMTrackCollection &  tracks4  )

private

Definition at line 294 of file L1TMuonBarrelKalmanRegionModule.cc.

References cleanHigher(), cleanLower(), MillePedeFileConverter_cfg::out, selfClean(), and sort4().

Referenced by process().

                                                                                                                                                                                     {

   
  L1MuKBMTrackCollection cleaned2 = selfClean(tracks2);
  L1MuKBMTrackCollection cleaned3 = selfClean(tracks3);

  L1MuKBMTrackCollection cleaned23 = cleanHigher(cleaned2,tracks3);
  L1MuKBMTrackCollection cleaned32 = cleanLower(cleaned3,tracks2);

  //printf("Cleaned sizes = 2=%d 3=%d 23=%d 32=%d \n",int(cleaned2.size()),int(cleaned3.size()),int(cleaned23.size()),int(cleaned32.size()));

  //merge 2,3 
  L1MuKBMTrackCollection step1;
  
  if (!cleaned32.empty())
    step1.insert(step1.end(),cleaned32.begin(),cleaned32.end());
  if (!cleaned23.empty())
    step1.insert(step1.end(),cleaned23.begin(),cleaned23.end());
    

  //take the best 2
  L1MuKBMTrackCollection sorted23=sort4(step1);
  //  printf("Sorted 23 =%d\n",int(sorted23.size()));

  //Now clean the tracks 4 between them
  L1MuKBMTrackCollection cleaned4 = selfClean(tracks4);
  //  printf("Cleaned4 =%d\n",int(cleaned4.size()));

  //Now clean the 23 tracks from tracks4
  L1MuKBMTrackCollection cleanedSorted23 = cleanHigher(sorted23,tracks4);

  //  printf("Cleaned23 from 4  =%d\n",int(cleanedSorted23.size()));

  //Now clean the  tracks4 from sorted 23
  L1MuKBMTrackCollection cleanedSorted4 = cleanLower(cleaned4,sorted23);
  //  printf("Cleaned4 from 23  =%d\n",int(cleanedSorted4.size()));


  //Now merge all of those
  L1MuKBMTrackCollection step2;

  if (!cleanedSorted4.empty())
    step2.insert(step2.end(),cleanedSorted4.begin(),cleanedSorted4.end());
  if (!cleanedSorted23.empty())
    step2.insert(step2.end(),cleanedSorted23.begin(),cleanedSorted23.end());

  L1MuKBMTrackCollection out=sort4(step2);
  return out;


}

L1MuKBMTrackCollection L1TMuonBarrelKalmanRegionModule::process	(	L1TMuonBarrelKalmanAlgo *	trackMaker,
		const L1MuKBMTCombinedStubRefVector &	stubs,
		int	bx
	)

Definition at line 56 of file L1TMuonBarrelKalmanRegionModule.cc.

References L1TMuonBarrelKalmanAlgo::chain(), cleanRegion(), createfilelist::int, nextSector_, nextWheel_, MillePedeFileConverter_cfg::out, previousSector_, sector_, SurveyInfoScenario_cff::seed, MCScenario_CRAFT1_22X::sorter(), verbose_, and wheel_.

Referenced by wheel().

                                                                                                                                                         {
  L1MuKBMTCombinedStubRefVector stubs;
  L1MuKBMTCombinedStubRefVector seeds;
  L1MuKBMTrackCollection pretracks2;
  L1MuKBMTrackCollection pretracks3;
  L1MuKBMTrackCollection pretracks4;
  for (const auto& stub : stubsAll) {
    if (stub->bxNum()!=bx)
      continue;

    if ((stub->scNum()==nextSector_ && stub->phi()>=-112)||(stub->scNum()==previousSector_ && stub->phi()<=111))
      continue;
    
    if (stub->whNum()==wheel_  && stub->scNum()==sector_) {
      seeds.push_back(stub);
      stubs.push_back(stub);
    }
    else if (stub->whNum()==wheel_  && (stub->scNum()==nextSector_||stub->scNum()==previousSector_ )) {
      stubs.push_back(stub);
    }
    else if (stub->whNum()==nextWheel_  && (stub->scNum()==nextSector_||stub->scNum()==previousSector_||stub->scNum()==sector_) ) {
      stubs.push_back(stub);
    }
  }


  //Sort the seeds by tag so that the emulator is aligned like the firmware
  
  SeedSorter sorter;
  if (seeds.size()>1) {
    std::sort(seeds.begin(),seeds.end(),sorter);
  }


  for (const auto seed : seeds) {
    std::pair<bool,L1MuKBMTrack> trackInfo = trackMaker->chain(seed,stubs);
    if (trackInfo.first) {
      if (seed->stNum()==2)
    pretracks2.push_back(trackInfo.second);
      if (seed->stNum()==3)
    pretracks3.push_back(trackInfo.second);
      if (seed->stNum()==4)
    pretracks4.push_back(trackInfo.second);
    }
  } 
  //for (const auto& track1 :pretracks2)
      //      printf("SEED=2 Kalman Track charge=%d pt=%f eta=%f phi=%f curvature=%d curvature STA =%d stubs=%d chi2=%d pts=%f %f pattern=%d\n",track1.charge(),track1.pt(),track1.eta(),track1.phi(),track1.curvatureAtVertex(),track1.curvatureAtMuon(),int(track1.stubs().size()),track1.approxChi2(),track1.pt(),track1.ptUnconstrained(),track1.hitPattern()); 

  //for (const auto& track1 :pretracks3)
      //      printf("SEED=3 Kalman Track charge=%d pt=%f eta=%f phi=%f curvature=%d curvature STA =%d stubs=%d chi2=%d pts=%f %f pattern=%d\n",track1.charge(),track1.pt(),track1.eta(),track1.phi(),track1.curvatureAtVertex(),track1.curvatureAtMuon(),int(track1.stubs().size()),track1.approxChi2(),track1.pt(),track1.ptUnconstrained(),track1.hitPattern()); 


  //for (const auto& track1 :pretracks4)
  //  printf("SEED=4 Kalman Track charge=%d pt=%f eta=%f phi=%f curvature=%d curvature STA =%d stubs=%d chi2=%d pts=%f %f pattern=%d\n",track1.charge(),track1.pt(),track1.eta(),track1.phi(),track1.curvatureAtVertex(),track1.curvatureAtMuon(),int(track1.stubs().size()),track1.approxChi2(),track1.pt(),track1.ptUnconstrained(),track1.hitPattern()); 



  //  trackMaker->resolveEtaUnit(pretracks);
  L1MuKBMTrackCollection out =cleanRegion(pretracks2,pretracks3,pretracks4);
  if (verbose_) {
    printf(" -----Sector Processor Kalman Tracks-----\n");
    for (const auto& track1 :out)
      printf("Kalman Track charge=%d pt=%f eta=%f phi=%f curvature=%d curvature STA =%d stubs=%d chi2=%d pts=%f %f\n",track1.charge(),track1.pt(),track1.eta(),track1.phi(),track1.curvatureAtVertex(),track1.curvatureAtMuon(),int(track1.stubs().size()),track1.approxChi2(),track1.pt(),track1.ptUnconstrained()); 
  }

  return out;
}

L1MuKBMTrackCollection L1TMuonBarrelKalmanRegionModule::selfClean ( const L1MuKBMTrackCollection &  tracks )

private

Definition at line 125 of file L1TMuonBarrelKalmanRegionModule.cc.

References mps_fire::i, keep, MillePedeFileConverter_cfg::out, and parallelization::uint().

Referenced by cleanRegion().

                                                                                                      {
  L1MuKBMTrackCollection out;

  for (uint i=0;i<tracks.size();++i) {
    bool keep=true;
        
    for (uint j=0;j<tracks.size();++j) {
      if (i==j)
    continue;
     
      if (tracks[i].overlapTrack(tracks[j])) {
    if (tracks[i].rank()<tracks[j].rank()) {
      keep=false;
    }
    else if (tracks[i].rank()==tracks[j].rank()) { //if same rank prefer seed that is high
      if (!tracks[j].stubs()[0]->tag())
        keep=false;
    }
      }
    }

    if (keep)
      out.push_back(tracks[i]);
  }
 
  return out;
}

L1MuKBMTrackCollection L1TMuonBarrelKalmanRegionModule::sort4 ( const L1MuKBMTrackCollection &  in )

private

Definition at line 195 of file L1TMuonBarrelKalmanRegionModule.cc.

References MillePedeFileConverter_cfg::out, EnergyCorrector::pt, and reco::LeafCandidate::pt().

Referenced by cleanRegion().

                                                                                              {

  L1MuKBMTrackCollection out;
  //partial sort like in firmwarE (bitonic)
  if (in.size()<=2)
    return in;
  else if (in.size()==3) {
    //Step 1 
    L1MuKBMTrack s2_1;
    L1MuKBMTrack s2_3;
    if (in[2].pt()>=in[0].pt()) {
      s2_1=in[2];
      s2_3=in[0];
    }
    else {
      s2_1=in[0];
      s2_3=in[2];
    }

    L1MuKBMTrack s2_2=in[1];
    //Step 2;
    L1MuKBMTrack s3_1=s2_1;
    L1MuKBMTrack s3_2;
    L1MuKBMTrack s3_3;

    if (s2_3.pt()>=s2_2.pt()) {
      s3_2=s2_3;
      s3_3=s2_2;
    }
    else {
      s3_2=s2_2;
      s3_3=s2_3;
    }

    out.push_back(s3_1);
    out.push_back(s3_2);

  }
  else {
    //Step 1 
    L1MuKBMTrack s2_1;
    L1MuKBMTrack s2_2;
    L1MuKBMTrack s2_3;
    L1MuKBMTrack s2_4;
    
    if (in[2].pt()>=in[0].pt()) {
      s2_1=in[2];
      s2_3=in[0];
    }
    else {
      s2_1=in[0];
      s2_3=in[2];
    }
    if (in[3].pt()>=in[1].pt()) {
      s2_2=in[3];
      s2_4=in[1];
    }
    else {
      s2_2=in[1];
      s2_4=in[3];
    }
    //Step 2 
    L1MuKBMTrack s3_1;
    L1MuKBMTrack s3_2;
    L1MuKBMTrack s3_3;
    L1MuKBMTrack s3_4;


    if (s2_4.pt()>=s2_1.pt()) {
      s3_1=s2_4;
      s3_4=s2_1;
    }
    else {
      s3_1=s2_1;
      s3_4=s2_4;
    }

    if (s2_3.pt()>=s2_2.pt()) {
      s3_2=s2_3;
      s3_3=s2_2;
    }
    else {
      s3_2=s2_2;
      s3_3=s2_3;
    }


    
    out.push_back(s3_1);
    out.push_back(s3_2);
  }

  return out;
  



}

const int L1TMuonBarrelKalmanRegionModule::wheel ( ) const

inline

Definition at line 13 of file L1TMuonBarrelKalmanRegionModule.h.

References process(), and wheel_.

Referenced by geometryXMLparser.DTAlignable::index().

                          {
    return wheel_;
  }

Member Data Documentation

int L1TMuonBarrelKalmanRegionModule::nextSector_

private

Definition at line 22 of file L1TMuonBarrelKalmanRegionModule.h.

Referenced by L1TMuonBarrelKalmanRegionModule(), and process().

int L1TMuonBarrelKalmanRegionModule::nextWheel_

private

Definition at line 24 of file L1TMuonBarrelKalmanRegionModule.h.

Referenced by L1TMuonBarrelKalmanRegionModule(), and process().

int L1TMuonBarrelKalmanRegionModule::previousSector_

private

Definition at line 23 of file L1TMuonBarrelKalmanRegionModule.h.

Referenced by L1TMuonBarrelKalmanRegionModule(), and process().

int L1TMuonBarrelKalmanRegionModule::sector_

private

Definition at line 20 of file L1TMuonBarrelKalmanRegionModule.h.

Referenced by process().

int L1TMuonBarrelKalmanRegionModule::verbose_

private

Definition at line 19 of file L1TMuonBarrelKalmanRegionModule.h.

Referenced by process().

int L1TMuonBarrelKalmanRegionModule::wheel_

private

Definition at line 21 of file L1TMuonBarrelKalmanRegionModule.h.

Referenced by process(), and wheel().