/data/doxygen/doxygen-1.7.3/gen/CMSSW_4_2_8/src/SimTracker/TrackerFilters/plugins/CosmicTIFTrigFilter.cc

Go to the documentation of this file.

// livio.fano@cern.ch

#include "SimTracker/TrackerFilters/interface/CosmicTIFTrigFilter.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "FWCore/Framework/interface/ESHandle.h"
#include "DataFormats/Common/interface/Handle.h"
#include "MagneticField/Engine/interface/MagneticField.h"
#include "MagneticField/Records/interface/IdealMagneticFieldRecord.h"
#include "HepMC/GenVertex.h"
#include <map>
#include <vector>

using namespace std;
namespace cms

{
  CosmicTIFTrigFilter::CosmicTIFTrigFilter(const edm::ParameterSet& conf):    conf_(conf)
  {
    trigconf  = conf_.getParameter<int>("trig_conf");
    trigS1  = conf_.getParameter<std::vector<double> >("PosScint1");
    trigS2  = conf_.getParameter<std::vector<double> >("PosScint2");
    trigS3  = conf_.getParameter<std::vector<double> >("PosScint3");
    trigS4  = conf_.getParameter<std::vector<double> >("PosScint4");

    /*
    std::cout << "S1 = " << trigS1[0] << ", " <<  trigS1[1] << ", " <<trigS1[2]
              << "\nS2 = " << trigS2[0] << ", " <<  trigS2[1] << ", " <<trigS2[2]
              << "\nS3 = " << trigS3[0] << ", " <<  trigS3[1] << ", " <<trigS3[2]  
              << "\nS4 = " << trigS4[0] << ", " <<  trigS4[1] << ", " <<trigS4[2]  
              << std::endl;  
    */
  }
  
  bool CosmicTIFTrigFilter::filter(edm::Event& iEvent, const edm::EventSetup& iSetup)
  {
    
    
    edm::Handle<edm::HepMCProduct>HepMCEvt;
    iEvent.getByLabel("source","",HepMCEvt);
    const HepMC::GenEvent* MCEvt = HepMCEvt->GetEvent();
    
    bool hit1=false;
    bool hit2=false;
    bool hit3=false;
    bool hit4=false;
    
    
    for(HepMC::GenEvent::particle_const_iterator i=MCEvt->particles_begin(); i != MCEvt->particles_end();++i)
      {
        int myId = (*i)->pdg_id();
        if (abs(myId)==13)
          {
            
            // Get the muon position and momentum
            HepMC::GenVertex* pv = (*i)->production_vertex();
            HepMC::FourVector vertex = pv->position();
            
            HepMC::FourVector  momentum=(*i)->momentum();

            //std::cout << "\t vertex for cut = " << vertex << std::endl; 
            //std::cout << "\t momentum  = " << momentum << std::endl; 
            
            if(trigconf==1){

/*
              HepMC::FourVector S1(350.,1600.,500.,0.);
              HepMC::FourVector S2(350.,-1600.,400.,0.);
              HepMC::FourVector S3(350.,1600.,1600.,0.);
*/
/*
//numbers taken from real data
             HepMC::FourVector S1(150.,1600.,550.,0.);
             HepMC::FourVector S2(400.,-1600.,50.,0.);
             HepMC::FourVector S3(0.,1600.,2400.,0.);
*/
              HepMC::FourVector S1(trigS1[0],trigS1[1],trigS1[2],0.);
              HepMC::FourVector S2(trigS2[0],trigS2[1],trigS2[2],0.);
              HepMC::FourVector S3(trigS3[0],trigS3[1],trigS3[2],0.);
              
              hit1=Sci_trig(vertex, momentum, S1);
              hit2=Sci_trig(vertex, momentum, S2);
              hit3=Sci_trig(vertex, momentum, S3);
              
              // trigger conditions
              
              if((hit1&&hit2) || (hit3&&hit2))
                {
                  /*
                  cout << "\tGot a trigger in configuration A " << endl; 
                  if(hit1)cout << "hit1 " << endl;
                  if(hit2)cout << "hit2 " << endl;
                  if(hit3)cout << "hit3 " << endl;
                  */
                  trig1++;
                  return true;
                }
            }else if(trigconf ==2) {

              /*
              HepMC::FourVector S1(350.,1600.,850.,0.);
              HepMC::FourVector S2(0.,-1550.,-1650.,0.);
              HepMC::FourVector S3(350.,1600.,2300.,0.);
              */

              HepMC::FourVector S1(trigS1[0],trigS1[1],trigS1[2],0.);
              HepMC::FourVector S2(trigS2[0],trigS2[1],trigS2[2],0.);
              HepMC::FourVector S3(trigS3[0],trigS3[1],trigS3[2],0.);

              
              hit1=Sci_trig(vertex, momentum, S1);
              hit2=Sci_trig(vertex, momentum, S2);
              hit3=Sci_trig(vertex, momentum, S3);
              
              // trigger conditions
              
              if((hit1&&hit2) || (hit3&&hit2))
                {
                  /*
                  cout << "\tGot a trigger in configuration B " << endl; 
                  if(hit1)cout << "hit1 " << endl;
                  if(hit2)cout << "hit2 " << endl;
                  if(hit3)cout << "hit3 " << endl;
                  */
                  trig2++;
                  return true;
                }

            }else if(trigconf ==3) {
/*
              HepMC::FourVector S1(350.,1600.,850.,0.);
              HepMC::FourVector S2(350.,-1600.,400.,0.);
              HepMC::FourVector S3(350.,1600.,2300.,0.);
              HepMC::FourVector S4(0.,-1600.,-2000.,0.);
*/
/*
//new configuration from data
              HepMC::FourVector S1(150.,1600.,900.,0.);
              HepMC::FourVector S2(400.,-1600.,100.,0.);
              HepMC::FourVector S3(50.,1600.,2450.,0.);
              HepMC::FourVector S4(-250.,-1600.,-2050.,0.);
*/            

              HepMC::FourVector S1(trigS1[0],trigS1[1],trigS1[2],0.);
              HepMC::FourVector S2(trigS2[0],trigS2[1],trigS2[2],0.);
              HepMC::FourVector S3(trigS3[0],trigS3[1],trigS3[2],0.);
              HepMC::FourVector S4(trigS4[0],trigS4[1],trigS4[2],0.);
              

              /*              std::cout << "S1 = " << S1.x() << "," << S1.y() << ", " << S1.z()  
                        << "\nS2 = " << S2.x() << "," << S2.y() << ", " << S2.z()  
                        << "\nS3 = " << S3.x() << "," << S3.y() << ", " << S3.z()  
                        << "\nS4 = " << S4.x() << "," << S4.y() << ", " << S4.z() 
                        << std::endl;  
              */

              hit1=Sci_trig(vertex, momentum, S1);
              hit2=Sci_trig(vertex, momentum, S2);
              hit3=Sci_trig(vertex, momentum, S3);
              hit4=Sci_trig(vertex, momentum, S4);
              
              // trigger conditions
              if((hit1&&hit2) || (hit3&&hit2) || (hit1&&hit4) || (hit3&&hit4))
                {

                  /*
                  cout << "\tGot a trigger in configuration C " << endl; 
                  if(hit1)cout << "hit1 " << endl;
                  if(hit2)cout << "hit2 " << endl;
                  if(hit3)cout << "hit3 " << endl;
                  if(hit4)cout << "hit4 " << endl;
                  */

                  trig3++;
                  return true;
                }
            }
          }
      }
    
    return false;
  }
  
  bool CosmicTIFTrigFilter::Sci_trig(HepMC::FourVector vertex,  HepMC::FourVector momentum, HepMC::FourVector S)
  {
    float x0= vertex.x();
    float y0= vertex.y();
    float z0= vertex.z();
    float px0=momentum.x();
    float py0=momentum.y();
    float pz0=momentum.z();
    float Sx=S.x();
    float Sy=S.y();
    float Sz=S.z();
    
    //float ys=Sy;
    float zs=(Sy-y0)*(pz0/py0)+z0;
    //    float xs=((Sy-y0)*(pz0/py0)-z0)*(px0/pz0)+x0;
    float xs=(Sy-y0)*(px0/py0)+x0;
    
    //    std::cout << Sx << " " << Sz << " " << xs << " " << zs << std::endl;
    //std::cout << x0 << " " << z0 << " " << px0 << " " << py0 << " " << pz0 << endl;
    
    if((xs<Sx+500 && xs>Sx-500) && (zs<Sz+500 && zs>Sz-500) )
      {
        //std::cout << "PASSED" << std::endl;
        return true;
      }
    else
      {
        return false;
      }
    
  }
  
}