DTSegment2DQuality.cc

Go to the documentation of this file.

/*
 *  See header file for a description of this class.
 *
 *  $Date: 2008/10/21 10:52:20 $
 *  $Revision: 1.2 $
 *  \author S. Bolognesi and G. Cerminara - INFN Torino
 */

#include "DTSegment2DQuality.h"
#include "Validation/DTRecHits/interface/DTHitQualityUtils.h"

#include "Geometry/DTGeometry/interface/DTSuperLayer.h"
#include "DataFormats/MuonDetId/interface/DTWireId.h"
#include "Geometry/DTGeometry/interface/DTGeometry.h"
#include "Geometry/Records/interface/MuonGeometryRecord.h"

#include "SimDataFormats/TrackingHit/interface/PSimHitContainer.h"
#include "DataFormats/DTRecHit/interface/DTRecHitCollection.h"
#include "DataFormats/DTRecHit/interface/DTRecSegment2DCollection.h"

#include "FWCore/Framework/interface/MakerMacros.h"
#include "FWCore/Framework/interface/Frameworkfwd.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/ESHandle.h"

#include "Histograms.h"

#include "TFile.h"

#include <iostream>
#include <map>

using namespace std;
using namespace edm;

// Constructor
DTSegment2DQuality::DTSegment2DQuality(const ParameterSet& pset)  {
   // Get the debug parameter for verbose output
  debug = pset.getUntrackedParameter<bool>("debug");
  DTHitQualityUtils::debug = debug;
  rootFileName = pset.getUntrackedParameter<string>("rootFileName");
  // the name of the simhit collection
  simHitLabel = pset.getUntrackedParameter<string>("simHitLabel", "SimG4Object");
  // the name of the 2D rec hit collection
  segment2DLabel = pset.getUntrackedParameter<string>("segment2DLabel");

  //sigma resolution on position
  sigmaResPos = pset.getParameter<double>("sigmaResPos");
  //sigma resolution on angle
  sigmaResAngle = pset.getParameter<double>("sigmaResAngle");

  // Create the root file
  theFile = new TFile(rootFileName.c_str(), "RECREATE");
  theFile->cd();

  if(debug)
    cout << "[DTSegment2DQuality] Constructor called " << endl;

  h2DHitRPhi = new HRes2DHit ("RPhi");
  h2DHitRZ= new HRes2DHit ("RZ");
  h2DHitRZ_W0= new HRes2DHit ("RZ_W0");
  h2DHitRZ_W1= new HRes2DHit ("RZ_W1");
  h2DHitRZ_W2= new HRes2DHit ("RZ_W2");

  h2DHitEff_RPhi= new HEff2DHit ("RPhi");
  h2DHitEff_RZ= new HEff2DHit ("RZ");
  h2DHitEff_RZ_W0= new HEff2DHit ("RZ_W0");
  h2DHitEff_RZ_W1= new HEff2DHit ("RZ_W1");
  h2DHitEff_RZ_W2= new HEff2DHit ("RZ_W2");
  if(debug)
    cout << "[DTSegment2DQuality] hitsos created " << endl;
}

// Destructor
DTSegment2DQuality::~DTSegment2DQuality(){

}

void DTSegment2DQuality::endJob() {
  // Write the histos to file
  theFile->cd();

  h2DHitRPhi->Write();
  h2DHitRZ->Write();
  h2DHitRZ_W0->Write();
  h2DHitRZ_W1->Write();
  h2DHitRZ_W2->Write();

  h2DHitEff_RPhi->ComputeEfficiency();
  h2DHitEff_RZ->ComputeEfficiency();
  h2DHitEff_RZ_W0->ComputeEfficiency();
  h2DHitEff_RZ_W1->ComputeEfficiency();
  h2DHitEff_RZ_W2->ComputeEfficiency();

  h2DHitEff_RPhi->Write();
  h2DHitEff_RZ->Write();
  h2DHitEff_RZ_W0->Write();
  h2DHitEff_RZ_W1->Write();
  h2DHitEff_RZ_W2->Write();

  theFile->Close();
} 

// The real analysis
void DTSegment2DQuality::analyze(const Event & event, const EventSetup& eventSetup){
  theFile->cd();

  // Get the DT Geometry
  ESHandle<DTGeometry> dtGeom;
  eventSetup.get<MuonGeometryRecord>().get(dtGeom);

  // Get the SimHit collection from the event
  edm::Handle<PSimHitContainer> simHits;
  event.getByLabel(simHitLabel, "MuonDTHits", simHits); //FIXME: second string to be removed

  //Map simHits by sl
  map<DTSuperLayerId, PSimHitContainer > simHitsPerSl;
  for(PSimHitContainer::const_iterator simHit = simHits->begin();
      simHit != simHits->end(); simHit++){
    // Create the id of the sl (the simHits in the DT known their wireId)
    DTSuperLayerId slId = ((DTWireId(simHit->detUnitId())).layerId()).superlayerId();
    // Fill the map
    simHitsPerSl[slId].push_back(*simHit);
  }

  // Get the 2D rechits from the event
  Handle<DTRecSegment2DCollection> segment2Ds;
  event.getByLabel(segment2DLabel, segment2Ds);
    
  // Loop over all superlayers containing a segment
  DTRecSegment2DCollection::id_iterator slId;
  for (slId = segment2Ds->id_begin();
       slId != segment2Ds->id_end();
       ++slId){

      //------------------------- simHits ---------------------------//
    //Get simHits of each superlayer
    PSimHitContainer simHits =  simHitsPerSl[(*slId)];
       
    // Map simhits per wire
    map<DTWireId, PSimHitContainer > simHitsPerWire = DTHitQualityUtils::mapSimHitsPerWire(simHits);
    map<DTWireId, const PSimHit*> muSimHitPerWire = DTHitQualityUtils::mapMuSimHitsPerWire(simHitsPerWire);
    int nMuSimHit = muSimHitPerWire.size();
    if(nMuSimHit == 0 || nMuSimHit == 1) {
      if(debug && nMuSimHit == 1)
        cout << "[DTSegment2DQuality] Only " << nMuSimHit << " mu SimHit in this SL, skipping!" << endl;
      continue; // If no or only one mu SimHit is found skip this SL
    } 
    if(debug)
      cout << "=== SL " << (*slId) << " has " << nMuSimHit << " SimHits" << endl;
  
    //Find outer and inner mu SimHit to build a segment
    pair<const PSimHit*, const PSimHit*> inAndOutSimHit = DTHitQualityUtils::findMuSimSegment(muSimHitPerWire); 
    //Check that outermost and innermost SimHit are not the same
    if(inAndOutSimHit.first ==inAndOutSimHit.second ) {
      cout << "[DTHitQualityUtils]***Warning: outermost and innermost SimHit are the same!" << endl;
      continue;
    }

    //Find direction and position of the sim Segment in SL RF
    pair<LocalVector, LocalPoint> dirAndPosSimSegm = DTHitQualityUtils::findMuSimSegmentDirAndPos(inAndOutSimHit,
                                                  (*slId),&(*dtGeom));

    LocalVector simSegmLocalDir = dirAndPosSimSegm.first;
    LocalPoint simSegmLocalPos = dirAndPosSimSegm.second;
    if(debug)
      cout<<"  Simulated segment:  local direction "<<simSegmLocalDir<<endl
        <<"                      local position  "<<simSegmLocalPos<<endl;
    const DTSuperLayer* superLayer = dtGeom->superLayer(*slId);
    GlobalPoint simSegmGlobalPos = superLayer->toGlobal(simSegmLocalPos);

    //Atan(x/z) angle and x position in SL RF
    float angleSimSeg = DTHitQualityUtils::findSegmentAlphaAndBeta(simSegmLocalDir).first;
    float posSimSeg = simSegmLocalPos.x(); 
    //Position (in eta,phi coordinates) in the global RF
    float etaSimSeg = simSegmGlobalPos.eta(); 
    float phiSimSeg = simSegmGlobalPos.phi();


    //---------------------------- recHits --------------------------//
    // Get the range of rechit for the corresponding slId
    bool recHitFound = false;
    DTRecSegment2DCollection::range range = segment2Ds->get(*slId);
    int nsegm = distance(range.first, range.second);
    if(debug)
      cout << "   Sl: " << *slId << " has " << nsegm
       << " 2D segments" << endl;

    if (nsegm!=0) {
      // Find the best RecHit: look for the 2D RecHit with the angle closest
      // to that of segment made of SimHits. 
      // RecHits must have delta alpha and delta position within 5 sigma of
      // the residual distribution (we are looking for residuals of segments
      // usefull to the track fit) for efficency purpose
      const DTRecSegment2D* bestRecHit = 0;
      bool bestRecHitFound = false;
      double deltaAlpha = 99999;
  
      // Loop over the recHits of this slId
      for (DTRecSegment2DCollection::const_iterator segment2D = range.first;
       segment2D!=range.second;
       ++segment2D){
    // Check the dimension
    if((*segment2D).dimension() != 2) {
      if(debug) cout << "[DTSegment2DQuality]***Error: This is not 2D segment!!!" << endl;
      abort();
    }
    // Segment Local Direction and position (in SL RF)
    LocalVector recSegDirection = (*segment2D).localDirection();
    LocalPoint recSegPosition = (*segment2D).localPosition();
      
    float recSegAlpha = DTHitQualityUtils::findSegmentAlphaAndBeta(recSegDirection).first;
    if(debug)
      cout << "  RecSegment direction: " << recSegDirection << endl
         << "             position : " << recSegPosition << endl
         << "             alpha    : " << recSegAlpha << endl;
      
    if(fabs(recSegAlpha - angleSimSeg) < deltaAlpha) {
      deltaAlpha = fabs(recSegAlpha - angleSimSeg);
      bestRecHit = &(*segment2D);
      bestRecHitFound = true;
    }
      }  // End of Loop over all 2D RecHits

      if(bestRecHitFound) {
    // Best rechit direction and position in SL RF
    LocalPoint bestRecHitLocalPos = bestRecHit->localPosition();
    LocalVector bestRecHitLocalDir = bestRecHit->localDirection();

    LocalError bestRecHitLocalPosErr = bestRecHit->localPositionError();
    LocalError bestRecHitLocalDirErr = bestRecHit->localDirectionError();

    float angleBestRHit = DTHitQualityUtils::findSegmentAlphaAndBeta(bestRecHitLocalDir).first;

    if(fabs(angleBestRHit - angleSimSeg) < 5*sigmaResAngle &&
       fabs(bestRecHitLocalPos.x() - posSimSeg) < 5*sigmaResPos) {
      recHitFound = true;
    }

    // Fill Residual histos
    HRes2DHit *hRes = 0;
    if((*slId).superlayer() == 1 || (*slId).superlayer() == 3) { //RPhi SL
      hRes = h2DHitRPhi;
    } else if((*slId).superlayer() == 2) { //RZ SL
      h2DHitRZ->Fill(angleSimSeg,
                    angleBestRHit,
                    posSimSeg,
                    bestRecHitLocalPos.x(),
                    etaSimSeg,
                    phiSimSeg,
                    sqrt(bestRecHitLocalPosErr.xx()),
                    sqrt(bestRecHitLocalDirErr.xx())
                    );
      if(abs((*slId).wheel()) == 0)
        hRes = h2DHitRZ_W0;
      else if(abs((*slId).wheel()) == 1)
        hRes = h2DHitRZ_W1;
      else if(abs((*slId).wheel()) == 2)
        hRes = h2DHitRZ_W2;
    }
    hRes->Fill(angleSimSeg,
               angleBestRHit,
               posSimSeg,
               bestRecHitLocalPos.x(),
               etaSimSeg,
               phiSimSeg,
               sqrt(bestRecHitLocalPosErr.xx()),
               sqrt(bestRecHitLocalDirErr.xx())
               );
      }
    } //end of if(nsegm!=0)

      // Fill Efficiency plot
    HEff2DHit *hEff = 0;
    if((*slId).superlayer() == 1 || (*slId).superlayer() == 3) { //RPhi SL
      hEff = h2DHitEff_RPhi;
    } else if((*slId).superlayer() == 2) { //RZ SL
      h2DHitEff_RZ->Fill(etaSimSeg, phiSimSeg, posSimSeg, angleSimSeg, recHitFound);
      if(abs((*slId).wheel()) == 0)
    hEff = h2DHitEff_RZ_W0;
      else if(abs((*slId).wheel()) == 1)
    hEff = h2DHitEff_RZ_W1;
      else if(abs((*slId).wheel()) == 2)
    hEff = h2DHitEff_RZ_W2;
    }
    hEff->Fill(etaSimSeg, phiSimSeg, posSimSeg, angleSimSeg, recHitFound);
  } // End of loop over superlayers
}