---

V0Fitter.h

Go to the documentation of this file.

// -*- C++ -*-
//
// Package:    V0Producer
// Class:      V0Fitter
// 
//
// Original Author:  Brian Drell
//         Created:  Fri May 18 22:57:40 CEST 2007
// $Id: V0Fitter.h,v 1.12 2008/04/28 23:32:42 drell Exp $
//
//

#ifndef RECOVERTEX__V0_FITTER_H
#define RECOVERTEX__V0_FITTER_H

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

#include "DataFormats/Common/interface/Ref.h"

#include "DataFormats/VertexReco/interface/Vertex.h"
#include "DataFormats/TrackReco/interface/Track.h"
#include "RecoVertex/VertexPrimitives/interface/TransientVertex.h"
#include "TrackingTools/TransientTrack/interface/TransientTrack.h"
#include "RecoVertex/KalmanVertexFit/interface/KalmanVertexFitter.h"

#include "MagneticField/Records/interface/IdealMagneticFieldRecord.h"
#include "MagneticField/VolumeBasedEngine/interface/VolumeBasedMagneticField.h"

#include "DataFormats/Candidate/interface/VertexCompositeCandidate.h"
#include "DataFormats/RecoCandidate/interface/RecoChargedCandidate.h"

#include "FWCore/ParameterSet/interface/ParameterSet.h"

#include "Geometry/CommonDetUnit/interface/TrackingGeometry.h"
#include "Geometry/CommonDetUnit/interface/GeomDetUnit.h"
#include "Geometry/TrackerGeometryBuilder/interface/TrackerGeometry.h"
#include "Geometry/Records/interface/TrackerDigiGeometryRecord.h"
#include "Geometry/CommonDetUnit/interface/GeomDet.h"
#include "Geometry/TrackerGeometryBuilder/interface/GluedGeomDet.h"

#include "DataFormats/TrackingRecHit/interface/TrackingRecHit.h"

#include <string>
#include <fstream>


class V0Fitter {
 public:
  V0Fitter(const edm::ParameterSet& theParams,
           const edm::Event& iEvent, const edm::EventSetup& iSetup);
  ~V0Fitter();

  // Get methods for the VertexCollections
  //reco::VertexCollection getKshortCollection() const;
  //reco::VertexCollection getLambdaCollection() const;
  //reco::VertexCollection getLambdaBarCollection() const;

  // Switching to L. Lista's reco::Candidate infrastructure for V0 storage
  const reco::VertexCompositeCandidateCollection& getKshorts() const;
  const reco::VertexCompositeCandidateCollection& getLambdas() const;
  const reco::VertexCompositeCandidateCollection& getLambdaBars() const;

 private:
  // STL vector of VertexCompositeCandidate that will be filled with VertexCompositeCandidates by fitAll()
  reco::VertexCompositeCandidateCollection theKshorts;
  reco::VertexCompositeCandidateCollection theLambdas;
  reco::VertexCompositeCandidateCollection theLambdaBars;

  // Vector used to temporarily hold candidates before cuts and selection
  reco::VertexCompositeCandidateCollection preCutCands;

  // Tracker geometry for discerning hit positions
  const TrackerGeometry* trackerGeom;

  const MagneticField* magField;

  std::string recoAlg;
  bool useRefTrax;
  bool storeRefTrax;
  bool doKshorts;
  bool doLambdas;

  bool doPostFitCuts;
  bool doTkQualCuts;

  // Cuts
  double chi2Cut;
  double tkChi2Cut;
  int tkNhitsCut;
  double rVtxCut;
  double vtxSigCut;
  double collinCut;
  double kShortMassCut;
  double lambdaMassCut;

  // Helper method that does the actual fitting using the KalmanVertexFitter
  void fitAll(const edm::Event& iEvent, const edm::EventSetup& iSetup);

  // Applies cuts to the VertexCompositeCandidates after they are fitted/created.
  void applyPostFitCuts();

  // Stuff for debug file output.
  std::ofstream mPiPiMassOut;

  inline void initFileOutput() {
    mPiPiMassOut.open("mPiPi.txt", ios::app);
  }
  inline void cleanupFileOutput() {
    mPiPiMassOut.close();
  }
};

#endif

---