L1TMuonEndCapTrackProducer.cc

Go to the documentation of this file.

// Upgraded Encdap Muon Track Finding Algorithm             //
//                                                          //
// Info: A human-readable version of the firmware based     //
//       track finding algorithm which will be implemented  //
//       in the upgraded endcaps of CMS. DT and RPC inputs  //
//       are not considered in this algorithm.              //
//                                                          //
// Author: M. Carver (UF)                                   //

#define NUM_SECTORS 12

#include "L1Trigger/L1TMuonEndCap/plugins/L1TMuonEndCapTrackProducer.h"
#include "L1Trigger/CSCCommonTrigger/interface/CSCPatternLUT.h"
#include "L1Trigger/CSCTrackFinder/test/src/RefTrack.h"

#include <stdio.h>
#include <stdlib.h>
#include <math.h>

#include "L1Trigger/L1TMuonEndCap/interface/PrimitiveConverter.h"
#include "L1Trigger/L1TMuonEndCap/interface/BXAnalyzer.h"
#include "L1Trigger/L1TMuonEndCap/interface/ZoneCreation.h"
#include "L1Trigger/L1TMuonEndCap/interface/PatternRecognition.h"
#include "L1Trigger/L1TMuonEndCap/interface/SortSector.h"
#include "L1Trigger/L1TMuonEndCap/interface/Matching.h"
#include "L1Trigger/L1TMuonEndCap/interface/Deltas.h"
#include "L1Trigger/L1TMuonEndCap/interface/BestTracks.h"
#include "L1Trigger/L1TMuonEndCap/interface/PtAssignment.h"
#include "L1Trigger/L1TMuonEndCap/interface/MakeRegionalCand.h"


using namespace L1TMuon;


L1TMuonEndCapTrackProducer::L1TMuonEndCapTrackProducer(const PSet& p) {

  inputTokenCSC = consumes<CSCCorrelatedLCTDigiCollection>(p.getParameter<edm::InputTag>("CSCInput"));
  
  produces<l1t::RegionalMuonCandBxCollection >("EMTF");
}


void L1TMuonEndCapTrackProducer::produce(edm::Event& ev,
                   const edm::EventSetup& es) {

  //bool verbose = false;


  //std::cout<<"Start Upgraded Track Finder Producer::::: event = "<<ev.id().event()<<"\n\n";

  //fprintf (write,"12345\n"); //<-- part of printing text file to send verilog code, not needed if George's package is included


  //std::auto_ptr<L1TMuon::InternalTrackCollection> FoundTracks (new L1TMuon::InternalTrackCollection);
  std::auto_ptr<l1t::RegionalMuonCandBxCollection > OutputCands (new l1t::RegionalMuonCandBxCollection);

  std::vector<BTrack> PTracks[NUM_SECTORS];

  std::vector<TriggerPrimitive> tester;
  //std::vector<InternalTrack> FoundTracks;
  
  
  
  edm::Handle<CSCCorrelatedLCTDigiCollection> MDC;
  ev.getByToken(inputTokenCSC, MDC);
  std::vector<TriggerPrimitive> out;
  
  auto chamber = MDC->begin();
  auto chend  = MDC->end();
  for( ; chamber != chend; ++chamber ) {
    auto digi = (*chamber).second.first;
    auto dend = (*chamber).second.second;
    for( ; digi != dend; ++digi ) {
      out.push_back(TriggerPrimitive((*chamber).first,*digi));
    }
  }
  


 // auto tpsrc = _tpinputs.cbegin();
  //auto tpend = _tpinputs.cend();
 // for( ; tpsrc != tpend; ++tpsrc ) {
   // edm::Handle<TriggerPrimitiveCollection> tps;
   // ev.getByLabel(*tpsrc,tps);
    auto tp = out.cbegin();
    auto tpend = out.cend();

    for( ; tp != tpend; ++tp ) {
      if(tp->subsystem() == 1)
      {
        //TriggerPrimitiveRef tpref(out,tp - out.cbegin());

        tester.push_back(*tp);

        //if(verbose) std::cout<<"\ntrigger prim found station:"<<tp->detId<CSCDetId>().station()<<std::endl;
      }

     }
   //}
  std::vector<ConvertedHit> CHits[NUM_SECTORS];
  MatchingOutput MO[NUM_SECTORS];

for(int SectIndex=0;SectIndex<NUM_SECTORS;SectIndex++){//perform TF on all 12 sectors





    std::vector<ConvertedHit> ConvHits = PrimConv(tester,SectIndex);
    CHits[SectIndex] = ConvHits;


    //for(std::vector<ConvertedHit>::iterator h = ConvHits.begin();h != ConvHits.end();h++){

        //if((h->Id()) > 9){h->SetId(h->Id() - 9);h->SetStrip(h->Strip() + 128);}
        //fprintf (write,"0 1   1   %d  %d\n",h->Sub(),h->Station());
        //fprintf (write,"1 %d  %d  %d\n",h->Quality(),h->Pattern(),h->Wire());
        //fprintf (write,"%d    0   %d\n",h->Id(),h->Strip());
    //}





 std::vector<std::vector<ConvertedHit>> GroupedHits = GroupBX(ConvHits);



 std::vector<ZonesOutput> Zout = Zones(GroupedHits);




  std::vector<PatternOutput> Pout = Patterns(Zout);

  PatternOutput Test = DeleteDuplicatePatterns(Pout);

  //PrintQuality(Test.detected);




  SortingOutput Sout = SortSect(Test);




  MatchingOutput Mout = PhiMatching(Sout);
  MO[SectIndex] = Mout;



 std::vector<std::vector<DeltaOutput>> Dout = CalcDeltas(Mout);




  std::vector<BTrack> Bout = BestTracks(Dout);
   PTracks[SectIndex] = Bout;



  }



 BTrack FourBest[4];//ok
 std::vector<BTrack> PTemp[NUM_SECTORS];
 for (int i=0; i<NUM_SECTORS; i++) PTemp[i] = PTracks[i];

 int windex[4] = {-1,-1,-1,-1};



 for(int i=0;i<4;i++){

    for(int j=0;j<36;j++){


            if(!PTemp[j/3][j%3].phi)//no track
                continue;

            if((windex[0] == j) || (windex[1] == j) || (windex[2] == j) || (windex[3] == j))//already picked
                continue;

            if(PTracks[j/3][j%3].winner.Rank() > FourBest[i].winner.Rank()){

                FourBest[i] = PTemp[j/3][j%3];
                windex[i] = j;

            }

    }
}


  for(int fbest=0;fbest<4;fbest++){

    if(FourBest[fbest].phi){


        InternalTrack tempTrack;
        tempTrack.setType(2);
        tempTrack.phi = FourBest[fbest].phi;
        tempTrack.theta = FourBest[fbest].theta;
        tempTrack.rank = FourBest[fbest].winner.Rank();
        tempTrack.deltas = FourBest[fbest].deltas;
        std::vector<int> ps, ts;


        int sector = -1;
        bool ME13 = false;
        int me1address = 0, me2address = 0, CombAddress = 0, mode = 0;

        for(std::vector<ConvertedHit>::iterator A = FourBest[fbest].AHits.begin();A != FourBest[fbest].AHits.end();A++){

            if(A->Phi() != -999){

                int station = A->TP().detId<CSCDetId>().station();
                int id = A->TP().getCSCData().cscID;
                int trknm = A->TP().getCSCData().trknmb;

                tempTrack.addStub(A->TP());
                ps.push_back(A->Phi());
                ts.push_back(A->Theta());
                sector = (A->TP().detId<CSCDetId>().endcap() -1)*6 + A->TP().detId<CSCDetId>().triggerSector() - 1;
                //std::cout<<"Q: "<<A->Quality()<<", keywire: "<<A->Wire()<<", strip: "<<A->Strip()<<std::endl;

                switch(station){
                    case 1: mode |= 8;break;
                    case 2: mode |= 4;break;
                    case 3: mode |= 2;break;
                    case 4: mode |= 1;break;
                    default: mode |= 0;
                }


                if(A->TP().detId<CSCDetId>().station() == 1 && A->TP().detId<CSCDetId>().ring() == 3)
                    ME13 = true;

                if(station == 1 && id > 3 && id < 7){

                    int sub = 2;
                    if(A->TP().detId<CSCDetId>().chamber()%6 > 2)
                        sub = 1;

                    me1address = id;
                    me1address -= 3;
                    me1address += 3*(sub - 1);
                    me1address = id<<1;//
                    me1address |= trknm-1;

                }

                if(station == 2 && id > 3){

                    me2address = id;
                    me2address -= 3;
                    me2address = me2address<<1;
                    me2address |= trknm-1;

                }


            }

        }
        tempTrack.phis = ps;
        tempTrack.thetas = ts;

        float xmlpt = CalculatePt(tempTrack,es);
        tempTrack.pt = xmlpt*1.4;
        //FoundTracks->push_back(tempTrack);

        CombAddress = (me2address<<4) | me1address;


        l1t::RegionalMuonCand outCand = MakeRegionalCand(xmlpt*1.4,FourBest[fbest].phi,FourBest[fbest].theta,
                                                                 CombAddress,mode,1,sector);
        // NOTE: assuming that all candidates come from the central BX:
        int bx = 0;
        float theta_angle = (FourBest[fbest].theta*0.2851562 + 8.5)*(3.14159265359/180);
        float eta = (-1)*log(tan(theta_angle/2));
        if(!ME13 && fabs(eta) > 1.1)
            OutputCands->push_back(bx, outCand);
    }
  }


//ev.put( FoundTracks, "DataITC");
ev.put( OutputCands, "EMTF");
  //std::cout<<"End Upgraded Track Finder Prducer:::::::::::::::::::::::::::\n:::::::::::::::::::::::::::::::::::::::::::::::::\n\n";

}//analyzer

void L1TMuonEndCapTrackProducer::beginJob()
{

}
void L1TMuonEndCapTrackProducer::endJob()
{

}
#include "FWCore/Framework/interface/MakerMacros.h"
DEFINE_FWK_MODULE(L1TMuonEndCapTrackProducer);