PVFitter Class Reference

#include <PVFitter.h>

Public Member Functions
void	compressStore ()
	reduce size of primary vertex cache by increasing quality limit More...
void	dumpTxtFile ()
void	FitPerBunchCrossing ()
reco::BeamSpot	getBeamSpot ()
std::map< int, reco::BeamSpot >	getBeamSpotMap ()
int *	getFitLSRange ()
int	getNPVs ()
const std::map< int, int > &	getNPVsperBX ()
std::vector< BeamSpotFitPVData >	getpvStore ()
time_t *	getRefTime ()
double	getWidthX ()
double	getWidthXerr ()
double	getWidthY ()
double	getWidthYerr ()
double	getWidthZ ()
double	getWidthZerr ()
void	initialize (const edm::ParameterSet &iConfig, edm::ConsumesCollector &iColl)
bool	IsFitPerBunchCrossing ()
	PVFitter ()
	PVFitter (const edm::ParameterSet &iConfig, edm::ConsumesCollector &&iColl)
	PVFitter (const edm::ParameterSet &iConfig, edm::ConsumesCollector &iColl)
double	pvQuality (const reco::Vertex &pv) const
	vertex quality measure More...
double	pvQuality (const BeamSpotFitPVData &pv) const
	vertex quality measure More...
void	readEvent (const edm::Event &iEvent)
void	resetAll ()
void	resetLSRange ()
void	resetRefTime ()
void	resizepvStore (unsigned int rmSize)
bool	runBXFitter ()
bool	runFitter ()
void	setFitLSRange (int ls0, int ls1)
void	setRefTime (std::time_t t0, std::time_t t1)
void	setTree (TTree *tree)
virtual	~PVFitter ()

Static Public Member Functions
static void	fillDescription (edm::ParameterSetDescription &)

Private Attributes
std::map< int, std::vector< BeamSpotFitPVData > >	bxMap_
bool	debug_
bool	do3DFit_
double	dynamicQualityCut_
double	errorScale_
std::ofstream	fasciiFile
reco::BeamSpot	fbeamspot
int	fbeginLumiOfFit
std::map< int, reco::BeamSpot >	fbspotMap
int	fendLumiOfFit
bool	fFitPerBunchCrossing
std::time_t	freftime [2]
TTree *	ftree_
double	fwidthX
double	fwidthXerr
double	fwidthY
double	fwidthYerr
double	fwidthZ
double	fwidthZerr
std::unique_ptr< TH2F >	hPVx
std::unique_ptr< TH2F >	hPVy
unsigned int	maxNrVertices_
double	maxVtxNormChi2_
double	maxVtxR_
double	maxVtxZ_
unsigned int	minNrVertices_
double	minSumPt_
double	minVtxNdf_
unsigned int	minVtxTracks_
double	minVtxWgt_
std::map< int, int >	npvsmap_
std::string	outputTxt_
std::vector< double >	pvQualities_
std::vector< BeamSpotFitPVData >	pvStore_
double	sigmaCut_
BeamSpotTreeData	theBeamSpotTreeData_
bool	useOnlyFirstPV_
edm::EDGetTokenT< reco::VertexCollection >	vertexToken_

Detailed Description

Definition at line 45 of file PVFitter.h.

Constructor & Destructor Documentation

PVFitter() [1/3]

PVFitter::PVFitter ( )

inline

Definition at line 47 of file PVFitter.h.

{}

PVFitter() [2/3]

PVFitter::PVFitter	(	const edm::ParameterSet &	iConfig,
		edm::ConsumesCollector &&	iColl
	)

Definition at line 48 of file PVFitter.cc.

References initialize().

                                                                               : ftree_(nullptr) {
  initialize(iConfig, iColl);
}

PVFitter() [3/3]

PVFitter::PVFitter	(	const edm::ParameterSet &	iConfig,
		edm::ConsumesCollector &	iColl
	)

Definition at line 52 of file PVFitter.cc.

References initialize().

                                                                              : ftree_(nullptr) {
  initialize(iConfig, iColl);
}

~PVFitter()

PVFitter::~PVFitter ( )

virtual

Definition at line 95 of file PVFitter.cc.

{}

Member Function Documentation

compressStore()

void PVFitter::compressStore

(

)

reduce size of primary vertex cache by increasing quality limit

Definition at line 487 of file PVFitter.cc.

References dynamicQualityCut_, mps_fire::i, pvQualities_, pvQuality(), pvStore_, and jetUpdater_cfi::sort.

Referenced by readEvent().

                             {
  //
  // fill vertex qualities
  //
  pvQualities_.resize(pvStore_.size());
  for (unsigned int i = 0; i < pvStore_.size(); ++i)
    pvQualities_[i] = pvQuality(pvStore_[i]);
  sort(pvQualities_.begin(), pvQualities_.end());
  //
  // Set new quality cut to median. This cut will be used to reduce the
  // number of vertices in the store and also apply to all new vertices
  // until the next reset
  //
  dynamicQualityCut_ = pvQualities_[pvQualities_.size() / 2];
  //
  // remove all vertices failing the cut from the store
  //   (to be moved to a more efficient memory management!)
  //
  unsigned int iwrite(0);
  for (unsigned int i = 0; i < pvStore_.size(); ++i) {
    if (pvQuality(pvStore_[i]) > dynamicQualityCut_)
      continue;
    if (i != iwrite)
      pvStore_[iwrite] = pvStore_[i];
    ++iwrite;
  }
  pvStore_.resize(iwrite);
  edm::LogInfo("PVFitter") << "Reduced primary vertex store size to " << pvStore_.size()
                           << " ; new dynamic quality cut = " << dynamicQualityCut_ << std::endl;
}

dumpTxtFile()

void PVFitter::dumpTxtFile

(

)

Definition at line 485 of file PVFitter.cc.

{}

fillDescription()

void PVFitter::fillDescription ( edm::ParameterSetDescription &  iDesc )

static

Definition at line 97 of file PVFitter.cc.

References edm::ParameterSetDescription::add(), edm::ParameterSetDescription::addUntracked(), and HLT_2022v15_cff::InputTag.

Referenced by BeamSpotAnalyzer::fillDescriptions(), and AlcaBeamMonitor::fillDescriptions().

                                                                {
  edm::ParameterSetDescription pvFitter;

  pvFitter.addUntracked<bool>("Debug");
  pvFitter.addUntracked<edm::InputTag>("VertexCollection", edm::InputTag("offlinePrimaryVertices"));
  pvFitter.addUntracked<bool>("Apply3DFit");
  pvFitter.addUntracked<unsigned int>("maxNrStoredVertices");
  pvFitter.addUntracked<unsigned int>("minNrVerticesForFit");
  pvFitter.addUntracked<double>("minVertexNdf");
  pvFitter.addUntracked<double>("maxVertexNormChi2");
  pvFitter.addUntracked<unsigned int>("minVertexNTracks");
  pvFitter.addUntracked<double>("minVertexMeanWeight");
  pvFitter.addUntracked<double>("maxVertexR");
  pvFitter.addUntracked<double>("maxVertexZ");
  pvFitter.addUntracked<double>("errorScale");
  pvFitter.addUntracked<double>("nSigmaCut");
  pvFitter.addUntracked<bool>("FitPerBunchCrossing");
  pvFitter.addUntracked<bool>("useOnlyFirstPV");
  pvFitter.addUntracked<double>("minSumPt");

  iDesc.add<edm::ParameterSetDescription>("PVFitter", pvFitter);
}

FitPerBunchCrossing()

void PVFitter::FitPerBunchCrossing ( )

inline

Definition at line 67 of file PVFitter.h.

References fFitPerBunchCrossing.

{ fFitPerBunchCrossing = true; }

getBeamSpot()

reco::BeamSpot PVFitter::getBeamSpot ( )

inline

Definition at line 94 of file PVFitter.h.

References fbeamspot.

Referenced by BeamFitter::runPVandTrkFitter().

{ return fbeamspot; }

getBeamSpotMap()

std::map<int, reco::BeamSpot> PVFitter::getBeamSpotMap ( )

inline

Definition at line 95 of file PVFitter.h.

References fbspotMap.

Referenced by BeamFitter::runPVandTrkFitter().

{ return fbspotMap; }

getFitLSRange()

int* PVFitter::getFitLSRange ( )

inline

Definition at line 97 of file PVFitter.h.

References fbeginLumiOfFit, fendLumiOfFit, and createJobs::tmp.

                       {
    int* tmp = new int[2];
    tmp[0] = fbeginLumiOfFit;
    tmp[1] = fendLumiOfFit;
    return tmp;
  }

getNPVs()

int PVFitter::getNPVs ( )

inline

Definition at line 120 of file PVFitter.h.

References pvStore_.

Referenced by BeamFitter::getNPVs().

{ return pvStore_.size(); }

getNPVsperBX()

const std::map<int, int>& PVFitter::getNPVsperBX ( )

inline

Definition at line 122 of file PVFitter.h.

References bxMap_, genParticles_cff::map, npvsmap_, and findQualityFiles::size.

Referenced by BeamFitter::getNPVsperBX().

                                         {
    npvsmap_.clear();

    for (std::map<int, std::vector<BeamSpotFitPVData> >::const_iterator pvStore = bxMap_.begin();
         pvStore != bxMap_.end();
         ++pvStore) {
      npvsmap_[pvStore->first] = (pvStore->second).size();
    }
    return npvsmap_;
  }

getpvStore()

std::vector<BeamSpotFitPVData> PVFitter::getpvStore ( )

inline

Definition at line 65 of file PVFitter.h.

References pvStore_.

Referenced by BeamFitter::getPVvectorSize().

{ return pvStore_; }

getRefTime()

time_t* PVFitter::getRefTime ( )

inline

Definition at line 104 of file PVFitter.h.

References freftime.

                       {
    time_t* tmptime = new time_t[2];
    tmptime[0] = freftime[0];
    tmptime[1] = freftime[1];
    return tmptime;
  }

getWidthX()

double PVFitter::getWidthX ( )

inline

Definition at line 58 of file PVFitter.h.

References fwidthX.

{ return fwidthX; }

getWidthXerr()

double PVFitter::getWidthXerr ( )

inline

Definition at line 61 of file PVFitter.h.

References fwidthXerr.

Referenced by BeamFitter::runPVandTrkFitter().

{ return fwidthXerr; }

getWidthY()

double PVFitter::getWidthY ( )

inline

Definition at line 59 of file PVFitter.h.

References fwidthY.

{ return fwidthY; }

getWidthYerr()

double PVFitter::getWidthYerr ( )

inline

Definition at line 62 of file PVFitter.h.

References fwidthYerr.

{ return fwidthYerr; }

getWidthZ()

double PVFitter::getWidthZ ( )

inline

Definition at line 60 of file PVFitter.h.

References fwidthZ.

{ return fwidthZ; }

getWidthZerr()

double PVFitter::getWidthZerr ( )

inline

Definition at line 63 of file PVFitter.h.

References fwidthZerr.

{ return fwidthZerr; }

initialize()

void PVFitter::initialize	(	const edm::ParameterSet &	iConfig,
		edm::ConsumesCollector &	iColl
	)

Definition at line 56 of file PVFitter.cc.

References edm::ConsumesCollector::consumes(), debug_, do3DFit_, dynamicQualityCut_, errorScale_, fFitPerBunchCrossing, edm::ParameterSet::getParameter(), hPVx, hPVy, maxNrVertices_, maxVtxNormChi2_, maxVtxR_, maxVtxZ_, minNrVertices_, minSumPt_, minVtxNdf_, minVtxTracks_, minVtxWgt_, sigmaCut_, useOnlyFirstPV_, and vertexToken_.

Referenced by PVFitter().

                                                                                     {
  //In order to make fitting ROOT histograms thread safe
  // one must call this undocumented function
  TMinuitMinimizer::UseStaticMinuit(false);
  debug_ = iConfig.getParameter<edm::ParameterSet>("PVFitter").getUntrackedParameter<bool>("Debug");
  vertexToken_ = iColl.consumes<reco::VertexCollection>(
      iConfig.getParameter<edm::ParameterSet>("PVFitter")
          .getUntrackedParameter<edm::InputTag>("VertexCollection", edm::InputTag("offlinePrimaryVertices")));
  do3DFit_ = iConfig.getParameter<edm::ParameterSet>("PVFitter").getUntrackedParameter<bool>("Apply3DFit");
  //writeTxt_          = iConfig.getParameter<edm::ParameterSet>("PVFitter").getUntrackedParameter<bool>("WriteAscii");
  //outputTxt_         = iConfig.getParameter<edm::ParameterSet>("PVFitter").getUntrackedParameter<std::string>("AsciiFileName");
  maxNrVertices_ =
      iConfig.getParameter<edm::ParameterSet>("PVFitter").getUntrackedParameter<unsigned int>("maxNrStoredVertices");
  minNrVertices_ =
      iConfig.getParameter<edm::ParameterSet>("PVFitter").getUntrackedParameter<unsigned int>("minNrVerticesForFit");
  minVtxNdf_ = iConfig.getParameter<edm::ParameterSet>("PVFitter").getUntrackedParameter<double>("minVertexNdf");
  maxVtxNormChi2_ =
      iConfig.getParameter<edm::ParameterSet>("PVFitter").getUntrackedParameter<double>("maxVertexNormChi2");
  minVtxTracks_ =
      iConfig.getParameter<edm::ParameterSet>("PVFitter").getUntrackedParameter<unsigned int>("minVertexNTracks");
  minVtxWgt_ = iConfig.getParameter<edm::ParameterSet>("PVFitter").getUntrackedParameter<double>("minVertexMeanWeight");
  maxVtxR_ = iConfig.getParameter<edm::ParameterSet>("PVFitter").getUntrackedParameter<double>("maxVertexR");
  maxVtxZ_ = iConfig.getParameter<edm::ParameterSet>("PVFitter").getUntrackedParameter<double>("maxVertexZ");
  errorScale_ = iConfig.getParameter<edm::ParameterSet>("PVFitter").getUntrackedParameter<double>("errorScale");
  sigmaCut_ = iConfig.getParameter<edm::ParameterSet>("PVFitter").getUntrackedParameter<double>("nSigmaCut");
  fFitPerBunchCrossing =
      iConfig.getParameter<edm::ParameterSet>("PVFitter").getUntrackedParameter<bool>("FitPerBunchCrossing");
  useOnlyFirstPV_ = iConfig.getParameter<edm::ParameterSet>("PVFitter").getUntrackedParameter<bool>("useOnlyFirstPV");
  minSumPt_ = iConfig.getParameter<edm::ParameterSet>("PVFitter").getUntrackedParameter<double>("minSumPt");

  // preset quality cut to "infinite"
  dynamicQualityCut_ = 1.e30;

  hPVx = std::make_unique<TH2F>("hPVx", "PVx vs PVz distribution", 200, -maxVtxR_, maxVtxR_, 200, -maxVtxZ_, maxVtxZ_);
  hPVy = std::make_unique<TH2F>("hPVy", "PVy vs PVz distribution", 200, -maxVtxR_, maxVtxR_, 200, -maxVtxZ_, maxVtxZ_);
  hPVx->SetDirectory(nullptr);
  hPVy->SetDirectory(nullptr);
}

IsFitPerBunchCrossing()

bool PVFitter::IsFitPerBunchCrossing ( )

inline

Definition at line 96 of file PVFitter.h.

References fFitPerBunchCrossing.

Referenced by BeamFitter::dumpTxtFile(), and BeamFitter::runPVandTrkFitter().

{ return fFitPerBunchCrossing; }

pvQuality() [1/2]

double PVFitter::pvQuality

(

const reco::Vertex &

pv

)

const

vertex quality measure

Definition at line 518 of file PVFitter.cc.

Referenced by compressStore(), and readEvent().

                                                     {
  //
  // determinant of the transverse part of the PV covariance matrix
  //
  return pv.covariance(0, 0) * pv.covariance(1, 1) - pv.covariance(0, 1) * pv.covariance(0, 1);
}

pvQuality() [2/2]

double PVFitter::pvQuality

(

const BeamSpotFitPVData &

pv

)

const

vertex quality measure

Definition at line 525 of file PVFitter.cc.

                                                            {
  //
  // determinant of the transverse part of the PV covariance matrix
  //
  double ex = pv.posError[0];
  double ey = pv.posError[1];
  return ex * ex * ey * ey * (1 - pv.posCorr[0] * pv.posCorr[0]);
}

readEvent()

void PVFitter::readEvent

(

const edm::Event &

iEvent

)

Definition at line 120 of file PVFitter.cc.

References BeamSpotFitPVData::bunchCrossing, BeamSpotTreeData::bunchCrossing(), simKBmtfDigis_cfi::bx, bxMap_, compressStore(), dynamicQualityCut_, fFitPerBunchCrossing, ftree_, hPVx, hPVy, iEvent, BeamSpotTreeData::lumi(), maxNrVertices_, minSumPt_, minVtxNdf_, minVtxTracks_, minVtxWgt_, BeamSpotFitPVData::posCorr, BeamSpotFitPVData::posError, BeamSpotFitPVData::position, DiDispStaMuonMonitor_cfi::pt, MetAnalyzer::pv(), L1TEGammaOffline_cfi::PVCollection, BeamSpotTreeData::pvData(), pvQuality(), pvStore_, BeamSpotTreeData::run(), TtFullHadEvtBuilder_cfi::sumPt, theBeamSpotTreeData_, useOnlyFirstPV_, and vertexToken_.

Referenced by BeamFitter::readEvent().

                                               {
  //------ Primary Vertices
  edm::Handle<reco::VertexCollection> PVCollection;
  bool hasPVs = false;

  if (iEvent.getByToken(vertexToken_, PVCollection)) {
    hasPVs = true;
  }
  //------

  if (hasPVs) {
    for (reco::VertexCollection::const_iterator pv = PVCollection->begin(); pv != PVCollection->end(); ++pv) {
      if (useOnlyFirstPV_) {
        if (pv != PVCollection->begin())
          break;
      }

      //--- vertex selection
      if (pv->isFake() || pv->tracksSize() == 0)
        continue;
      if (pv->ndof() < minVtxNdf_ || (pv->ndof() + 3.) / pv->tracksSize() < 2 * minVtxWgt_)
        continue;
      //---

      if (pv->tracksSize() < minVtxTracks_)
        continue;

      const auto& testTrack = pv->trackRefAt(0);
      if (testTrack.isNull() || !testTrack.isAvailable()) {
        edm::LogInfo("") << "Track collection not found. Skipping cut on sumPt.";
      } else {
        double sumPt = 0;
        for (auto iTrack = pv->tracks_begin(); iTrack != pv->tracks_end(); ++iTrack) {
          const auto pt = (*iTrack)->pt();
          sumPt += pt;
        }
        if (sumPt < minSumPt_)
          continue;
      }

      hPVx->Fill(pv->x(), pv->z());
      hPVy->Fill(pv->y(), pv->z());

      //
      // 3D fit section
      //
      // apply additional quality cut
      if (pvQuality(*pv) > dynamicQualityCut_)
        continue;
      // if store exceeds max. size: reduce size and apply new quality cut
      if (pvStore_.size() >= maxNrVertices_) {
        compressStore();
        if (pvQuality(*pv) > dynamicQualityCut_)
          continue;
      }
      //
      // copy PV to store
      //
      int bx = iEvent.bunchCrossing();
      BeamSpotFitPVData pvData;
      pvData.bunchCrossing = bx;
      pvData.position[0] = pv->x();
      pvData.position[1] = pv->y();
      pvData.position[2] = pv->z();
      pvData.posError[0] = pv->xError();
      pvData.posError[1] = pv->yError();
      pvData.posError[2] = pv->zError();
      pvData.posCorr[0] = pv->covariance(0, 1) / pv->xError() / pv->yError();
      pvData.posCorr[1] = pv->covariance(0, 2) / pv->xError() / pv->zError();
      pvData.posCorr[2] = pv->covariance(1, 2) / pv->yError() / pv->zError();
      pvStore_.push_back(pvData);

      if (ftree_ != nullptr) {
        theBeamSpotTreeData_.run(iEvent.id().run());
        theBeamSpotTreeData_.lumi(iEvent.luminosityBlock());
        theBeamSpotTreeData_.bunchCrossing(bx);
        theBeamSpotTreeData_.pvData(pvData);
        ftree_->Fill();
      }

      if (fFitPerBunchCrossing)
        bxMap_[bx].push_back(pvData);
    }
  }
}

resetAll()

void PVFitter::resetAll ( )

inline

Definition at line 83 of file PVFitter.h.

References align::BeamSpot, bxMap_, dynamicQualityCut_, fbeamspot, fbspotMap, hPVx, hPVy, pvStore_, resetLSRange(), and resetRefTime().

Referenced by BeamFitter::BeamFitter(), and BeamFitter::resetPVFitter().

                  {
    resetLSRange();
    resetRefTime();
    pvStore_.clear();
    bxMap_.clear();
    dynamicQualityCut_ = 1.e30;
    hPVx->Reset();
    hPVy->Reset();
    fbeamspot = reco::BeamSpot();
    fbspotMap.clear();
  };

resetLSRange()

void PVFitter::resetLSRange ( )

inline

Definition at line 70 of file PVFitter.h.

References fbeginLumiOfFit, and fendLumiOfFit.

Referenced by resetAll().

{ fbeginLumiOfFit = fendLumiOfFit = -1; }

resetRefTime()

void PVFitter::resetRefTime ( )

inline

Definition at line 71 of file PVFitter.h.

References freftime.

Referenced by resetAll().

{ freftime[0] = freftime[1] = 0; }

resizepvStore()

void PVFitter::resizepvStore ( unsigned int  rmSize )

inline

Definition at line 112 of file PVFitter.h.

References pvStore_.

Referenced by BeamFitter::resizePVvector().

{ pvStore_.erase(pvStore_.begin(), pvStore_.begin() + rmSize); }

runBXFitter()

bool PVFitter::runBXFitter

(

)

Definition at line 211 of file PVFitter.cc.

References align::BeamSpot, bxMap_, errorScale_, fbeamspot, fbspotMap, fcn(), fwidthX, fwidthXerr, fwidthY, fwidthYerr, fwidthZ, fwidthZerr, genParticles_cff::map, makeMuonMisalignmentScenario::matrix, minNrVertices_, funct::pow(), reco::BeamSpot::setBeamWidthX(), reco::BeamSpot::setBeamWidthY(), reco::BeamSpot::setType(), sigmaCut_, findQualityFiles::size, and reco::BeamSpot::Tracker.

Referenced by BeamFitter::runPVandTrkFitter().

                           {
  using namespace ROOT::Minuit2;
  edm::LogInfo("PVFitter") << " Number of bunch crossings: " << bxMap_.size() << std::endl;

  bool fit_ok = true;

  for (std::map<int, std::vector<BeamSpotFitPVData> >::const_iterator pvStore = bxMap_.begin(); pvStore != bxMap_.end();
       ++pvStore) {
    // first set null beam spot in case
    // fit fails
    fbspotMap[pvStore->first] = reco::BeamSpot();

    edm::LogInfo("PVFitter") << " Number of PVs collected for PVFitter: " << (pvStore->second).size()
                             << " in bx: " << pvStore->first << std::endl;

    if ((pvStore->second).size() <= minNrVertices_) {
      edm::LogWarning("PVFitter") << " not enough PVs, continue" << std::endl;
      fit_ok = false;
      continue;
    }

    edm::LogInfo("PVFitter") << "Calculating beam spot with PVs ..." << std::endl;

    //
    // LL function and fitter
    //
    FcnBeamSpotFitPV fcn(pvStore->second);
    //
    // fit parameters: positions, widths, x-y correlations, tilts in xz and yz
    //
    MnUserParameters upar;
    upar.Add("x", 0., 0.02, -10., 10.);                                                     // 0
    upar.Add("y", 0., 0.02, -10., 10.);                                                     // 1
    upar.Add("z", 0., 0.20, -30., 30.);                                                     // 2
    upar.Add("ex", 0.015, 0.01, 0., 10.);                                                   // 3
    upar.Add("corrxy", 0., 0.02, -1., 1.);                                                  // 4
    upar.Add("ey", 0.015, 0.01, 0., 10.);                                                   // 5
    upar.Add("dxdz", 0., 0.0002, -0.1, 0.1);                                                // 6
    upar.Add("dydz", 0., 0.0002, -0.1, 0.1);                                                // 7
    upar.Add("ez", 1., 0.1, 0., 30.);                                                       // 8
    upar.Add("scale", errorScale_, errorScale_ / 10., errorScale_ / 2., errorScale_ * 2.);  // 9
    MnMigrad migrad(fcn, upar);

    //
    // first iteration without correlations
    //
    upar.Fix(4);
    upar.Fix(6);
    upar.Fix(7);
    upar.Fix(9);
    FunctionMinimum ierr = migrad(0, 1.);
    if (!ierr.IsValid()) {
      edm::LogInfo("PVFitter") << "3D beam spot fit failed in 1st iteration" << std::endl;
      fit_ok = false;
      continue;
    }
    //
    // refit with harder selection on vertices
    //
    fcn.setLimits(upar.Value(0) - sigmaCut_ * upar.Value(3),
                  upar.Value(0) + sigmaCut_ * upar.Value(3),
                  upar.Value(1) - sigmaCut_ * upar.Value(5),
                  upar.Value(1) + sigmaCut_ * upar.Value(5),
                  upar.Value(2) - sigmaCut_ * upar.Value(8),
                  upar.Value(2) + sigmaCut_ * upar.Value(8));
    ierr = migrad(0, 1.);
    if (!ierr.IsValid()) {
      edm::LogInfo("PVFitter") << "3D beam spot fit failed in 2nd iteration" << std::endl;
      fit_ok = false;
      continue;
    }
    //
    // refit with correlations
    //
    upar.Release(4);
    upar.Release(6);
    upar.Release(7);
    ierr = migrad(0, 1.);
    if (!ierr.IsValid()) {
      edm::LogInfo("PVFitter") << "3D beam spot fit failed in 3rd iteration" << std::endl;
      fit_ok = false;
      continue;
    }

    fwidthX = upar.Value(3);
    fwidthY = upar.Value(5);
    fwidthZ = upar.Value(8);
    fwidthXerr = upar.Error(3);
    fwidthYerr = upar.Error(5);
    fwidthZerr = upar.Error(8);

    reco::BeamSpot::CovarianceMatrix matrix;
    // need to get the full cov matrix
    matrix(0, 0) = pow(upar.Error(0), 2);
    matrix(1, 1) = pow(upar.Error(1), 2);
    matrix(2, 2) = pow(upar.Error(2), 2);
    matrix(3, 3) = fwidthZerr * fwidthZerr;
    matrix(4, 4) = pow(upar.Error(6), 2);
    matrix(5, 5) = pow(upar.Error(7), 2);
    matrix(6, 6) = fwidthXerr * fwidthXerr;

    fbeamspot = reco::BeamSpot(reco::BeamSpot::Point(upar.Value(0), upar.Value(1), upar.Value(2)),
                               fwidthZ,
                               upar.Value(6),
                               upar.Value(7),
                               fwidthX,
                               matrix);
    fbeamspot.setBeamWidthX(fwidthX);
    fbeamspot.setBeamWidthY(fwidthY);
    fbeamspot.setType(reco::BeamSpot::Tracker);

    fbspotMap[pvStore->first] = fbeamspot;
    edm::LogInfo("PVFitter") << "3D PV fit done for this bunch crossing." << std::endl;
    fit_ok = fit_ok & true;
  }

  return fit_ok;
}

runFitter()

bool PVFitter::runFitter

(

)

Definition at line 330 of file PVFitter.cc.

References align::BeamSpot, do3DFit_, debug_messages_cfi::errors, errorScale_, fftjetvertexadder_cfi::errX, fftjetvertexadder_cfi::errY, fftjetvertexadder_cfi::errZ, fbeamspot, fcn(), fwidthX, fwidthXerr, fwidthY, fwidthYerr, fwidthZ, fwidthZerr, hPVx, hPVy, edm::isNotFinite(), makeMuonMisalignmentScenario::matrix, minNrVertices_, funct::pow(), pvStore_, mysort::results, reco::BeamSpot::setBeamWidthX(), reco::BeamSpot::setBeamWidthY(), reco::BeamSpot::setType(), sigmaCut_, str, AlCaHLTBitMon_QueryRunRegistry::string, and reco::BeamSpot::Tracker.

Referenced by BeamFitter::runPVandTrkFitter().

                         {
  using namespace ROOT::Minuit2;
  edm::LogInfo("PVFitter") << " Number of PVs collected for PVFitter: " << pvStore_.size() << std::endl;

  if (pvStore_.size() <= minNrVertices_)
    return false;

  //need to create a unique histogram name to avoid ROOT trying to re-use one
  // also tell ROOT to hide its global state
  TDirectory::TContext guard{nullptr};
  std::ostringstream str;
  str << this;
  std::unique_ptr<TH1D> h1PVx{hPVx->ProjectionX((std::string("h1PVx") + str.str()).c_str(), 0, -1, "e")};
  std::unique_ptr<TH1D> h1PVy{hPVy->ProjectionX((std::string("h1PVy") + str.str()).c_str(), 0, -1, "e")};
  std::unique_ptr<TH1D> h1PVz{hPVx->ProjectionY((std::string("h1PVz") + str.str()).c_str(), 0, -1, "e")};

  //Use our own copy for thread safety
  // also do not add to global list of functions
  TF1 gausx("localGausX", "gaus", 0., 1., TF1::EAddToList::kNo);
  TF1 gausy("localGausY", "gaus", 0., 1., TF1::EAddToList::kNo);
  TF1 gausz("localGausZ", "gaus", 0., 1., TF1::EAddToList::kNo);

  h1PVx->Fit(&gausx, "QLMN0 SERIAL");
  h1PVy->Fit(&gausy, "QLMN0 SERIAL");
  h1PVz->Fit(&gausz, "QLMN0 SERIAL");

  fwidthX = gausx.GetParameter(2);
  fwidthY = gausy.GetParameter(2);
  fwidthZ = gausz.GetParameter(2);
  fwidthXerr = gausx.GetParError(2);
  fwidthYerr = gausy.GetParError(2);
  fwidthZerr = gausz.GetParError(2);

  double estX = gausx.GetParameter(1);
  double estY = gausy.GetParameter(1);
  double estZ = gausz.GetParameter(1);
  double errX = fwidthX * 3.;
  double errY = fwidthY * 3.;
  double errZ = fwidthZ * 3.;

  if (!do3DFit_) {
    reco::BeamSpot::CovarianceMatrix matrix;
    matrix(2, 2) = gausz.GetParError(1) * gausz.GetParError(1);
    matrix(3, 3) = fwidthZerr * fwidthZerr;
    matrix(6, 6) = fwidthXerr * fwidthXerr;

    fbeamspot =
        reco::BeamSpot(reco::BeamSpot::Point(gausx.GetParameter(1), gausy.GetParameter(1), gausz.GetParameter(1)),
                       fwidthZ,
                       0.,
                       0.,
                       fwidthX,
                       matrix);
    fbeamspot.setBeamWidthX(fwidthX);
    fbeamspot.setBeamWidthY(fwidthY);
    fbeamspot.setType(reco::BeamSpot::Tracker);

  } else {  // do 3D fit
    //
    // LL function and fitter
    //
    FcnBeamSpotFitPV fcn(pvStore_);
    //
    // fit parameters: positions, widths, x-y correlations, tilts in xz and yz
    //
    MnUserParameters upar;
    upar.Add("x", estX, errX, -10., 10.);                                                   // 0
    upar.Add("y", estY, errY, -10., 10.);                                                   // 1
    upar.Add("z", estZ, errZ, -30., 30.);                                                   // 2
    upar.Add("ex", 0.015, 0.01, 0., 10.);                                                   // 3
    upar.Add("corrxy", 0., 0.02, -1., 1.);                                                  // 4
    upar.Add("ey", 0.015, 0.01, 0., 10.);                                                   // 5
    upar.Add("dxdz", 0., 0.0002, -0.1, 0.1);                                                // 6
    upar.Add("dydz", 0., 0.0002, -0.1, 0.1);                                                // 7
    upar.Add("ez", 1., 0.1, 0., 30.);                                                       // 8
    upar.Add("scale", errorScale_, errorScale_ / 10., errorScale_ / 2., errorScale_ * 2.);  // 9
    MnMigrad migrad(fcn, upar);
    //
    // first iteration without correlations
    //
    migrad.Fix(4);
    migrad.Fix(6);
    migrad.Fix(7);
    migrad.Fix(9);
    FunctionMinimum ierr = migrad(0, 1.);
    if (!ierr.IsValid()) {
      edm::LogWarning("PVFitter") << "3D beam spot fit failed in 1st iteration" << std::endl;
      return false;
    }
    //
    // refit with harder selection on vertices
    //

    vector<double> results;
    vector<double> errors;
    results = ierr.UserParameters().Params();
    errors = ierr.UserParameters().Errors();

    fcn.setLimits(results[0] - sigmaCut_ * results[3],
                  results[0] + sigmaCut_ * results[3],
                  results[1] - sigmaCut_ * results[5],
                  results[1] + sigmaCut_ * results[5],
                  results[2] - sigmaCut_ * results[8],
                  results[2] + sigmaCut_ * results[8]);
    ierr = migrad(0, 1.);
    if (!ierr.IsValid()) {
      edm::LogWarning("PVFitter") << "3D beam spot fit failed in 2nd iteration" << std::endl;
      return false;
    }
    //
    // refit with correlations
    //
    migrad.Release(4);
    migrad.Release(6);
    migrad.Release(7);
    ierr = migrad(0, 1.);
    if (!ierr.IsValid()) {
      edm::LogWarning("PVFitter") << "3D beam spot fit failed in 3rd iteration" << std::endl;
      return false;
    }

    results = ierr.UserParameters().Params();
    errors = ierr.UserParameters().Errors();

    fwidthX = results[3];
    fwidthY = results[5];
    fwidthZ = results[8];
    fwidthXerr = errors[3];
    fwidthYerr = errors[5];
    fwidthZerr = errors[8];

    // check errors on widths and sigmaZ for nan
    if (edm::isNotFinite(fwidthXerr) || edm::isNotFinite(fwidthYerr) || edm::isNotFinite(fwidthZerr)) {
      edm::LogWarning("PVFitter") << "3D beam spot fit returns nan in 3rd iteration" << std::endl;
      return false;
    }

    reco::BeamSpot::CovarianceMatrix matrix;
    // need to get the full cov matrix
    matrix(0, 0) = pow(errors[0], 2);
    matrix(1, 1) = pow(errors[1], 2);
    matrix(2, 2) = pow(errors[2], 2);
    matrix(3, 3) = fwidthZerr * fwidthZerr;
    matrix(6, 6) = fwidthXerr * fwidthXerr;

    fbeamspot = reco::BeamSpot(
        reco::BeamSpot::Point(results[0], results[1], results[2]), fwidthZ, results[6], results[7], fwidthX, matrix);
    fbeamspot.setBeamWidthX(fwidthX);
    fbeamspot.setBeamWidthY(fwidthY);
    fbeamspot.setType(reco::BeamSpot::Tracker);
  }

  return true;
}

setFitLSRange()

void PVFitter::setFitLSRange ( int  ls0, int  ls1  )

inline

Definition at line 77 of file PVFitter.h.

References fbeginLumiOfFit, and fendLumiOfFit.

                                       {
    fbeginLumiOfFit = ls0;
    fendLumiOfFit = ls1;
  }

setRefTime()

void PVFitter::setRefTime ( std::time_t  t0, std::time_t  t1  )

inline

Definition at line 73 of file PVFitter.h.

References freftime, FrontierCondition_GT_autoExpress_cfi::t0, and RandomServiceHelper::t1.

                                              {
    freftime[0] = t0;
    freftime[1] = t1;
  }

setTree()

void PVFitter::setTree

(

TTree *

tree

)

Definition at line 206 of file PVFitter.cc.

References BeamSpotTreeData::branch(), ftree_, theBeamSpotTreeData_, and MainPageGenerator::tree.

Referenced by BeamFitter::BeamFitter().

                                  {
  ftree_ = tree;
  theBeamSpotTreeData_.branch(ftree_);
}

Member Data Documentation

bxMap_

std::map<int, std::vector<BeamSpotFitPVData> > PVFitter::bxMap_

private

Definition at line 177 of file PVFitter.h.

Referenced by getNPVsperBX(), readEvent(), resetAll(), and runBXFitter().

debug_

bool PVFitter::debug_

private

Definition at line 142 of file PVFitter.h.

Referenced by initialize().

do3DFit_

bool PVFitter::do3DFit_

private

Definition at line 143 of file PVFitter.h.

Referenced by initialize(), and runFitter().

dynamicQualityCut_

double PVFitter::dynamicQualityCut_

private

Definition at line 178 of file PVFitter.h.

Referenced by compressStore(), initialize(), readEvent(), and resetAll().

errorScale_

double PVFitter::errorScale_

private

Definition at line 155 of file PVFitter.h.

Referenced by initialize(), runBXFitter(), and runFitter().

fasciiFile

std::ofstream PVFitter::fasciiFile

private

Definition at line 140 of file PVFitter.h.

fbeamspot

reco::BeamSpot PVFitter::fbeamspot

private

Definition at line 135 of file PVFitter.h.

Referenced by getBeamSpot(), resetAll(), runBXFitter(), and runFitter().

fbeginLumiOfFit

int PVFitter::fbeginLumiOfFit

private

Definition at line 167 of file PVFitter.h.

Referenced by getFitLSRange(), resetLSRange(), and setFitLSRange().

fbspotMap

std::map<int, reco::BeamSpot> PVFitter::fbspotMap

private

Definition at line 136 of file PVFitter.h.

Referenced by getBeamSpotMap(), resetAll(), and runBXFitter().

fendLumiOfFit

int PVFitter::fendLumiOfFit

private

Definition at line 168 of file PVFitter.h.

Referenced by getFitLSRange(), resetLSRange(), and setFitLSRange().

fFitPerBunchCrossing

bool PVFitter::fFitPerBunchCrossing

private

Definition at line 137 of file PVFitter.h.

Referenced by FitPerBunchCrossing(), initialize(), IsFitPerBunchCrossing(), and readEvent().

freftime

std::time_t PVFitter::freftime[2]

private

Definition at line 159 of file PVFitter.h.

Referenced by getRefTime(), resetRefTime(), and setRefTime().

ftree_

TTree* PVFitter::ftree_

private

Definition at line 164 of file PVFitter.h.

Referenced by readEvent(), and setTree().

fwidthX

double PVFitter::fwidthX

private

Definition at line 169 of file PVFitter.h.

Referenced by getWidthX(), runBXFitter(), and runFitter().

fwidthXerr

double PVFitter::fwidthXerr

private

Definition at line 172 of file PVFitter.h.

Referenced by getWidthXerr(), runBXFitter(), and runFitter().

fwidthY

double PVFitter::fwidthY

private

Definition at line 170 of file PVFitter.h.

Referenced by getWidthY(), runBXFitter(), and runFitter().

fwidthYerr

double PVFitter::fwidthYerr

private

Definition at line 173 of file PVFitter.h.

Referenced by getWidthYerr(), runBXFitter(), and runFitter().

fwidthZ

double PVFitter::fwidthZ

private

Definition at line 171 of file PVFitter.h.

Referenced by getWidthZ(), runBXFitter(), and runFitter().

fwidthZerr

double PVFitter::fwidthZerr

private

Definition at line 174 of file PVFitter.h.

Referenced by getWidthZerr(), runBXFitter(), and runFitter().

hPVx

std::unique_ptr<TH2F> PVFitter::hPVx

private

Definition at line 161 of file PVFitter.h.

Referenced by initialize(), readEvent(), resetAll(), and runFitter().

hPVy

std::unique_ptr<TH2F> PVFitter::hPVy

private

Definition at line 162 of file PVFitter.h.

Referenced by initialize(), readEvent(), resetAll(), and runFitter().

maxNrVertices_

unsigned int PVFitter::maxNrVertices_

private

Definition at line 147 of file PVFitter.h.

Referenced by initialize(), and readEvent().

maxVtxNormChi2_

double PVFitter::maxVtxNormChi2_

private

Definition at line 150 of file PVFitter.h.

Referenced by initialize().

maxVtxR_

double PVFitter::maxVtxR_

private

Definition at line 153 of file PVFitter.h.

Referenced by initialize().

maxVtxZ_

double PVFitter::maxVtxZ_

private

Definition at line 154 of file PVFitter.h.

Referenced by initialize().

minNrVertices_

unsigned int PVFitter::minNrVertices_

private

Definition at line 148 of file PVFitter.h.

Referenced by initialize(), runBXFitter(), and runFitter().

minSumPt_

double PVFitter::minSumPt_

private

Definition at line 157 of file PVFitter.h.

Referenced by initialize(), and readEvent().

minVtxNdf_

double PVFitter::minVtxNdf_

private

Definition at line 149 of file PVFitter.h.

Referenced by initialize(), and readEvent().

minVtxTracks_

unsigned int PVFitter::minVtxTracks_

private

Definition at line 151 of file PVFitter.h.

Referenced by initialize(), and readEvent().

minVtxWgt_

double PVFitter::minVtxWgt_

private

Definition at line 152 of file PVFitter.h.

Referenced by initialize(), and readEvent().

npvsmap_

std::map<int, int> PVFitter::npvsmap_

private

Definition at line 134 of file PVFitter.h.

Referenced by getNPVsperBX().

outputTxt_

std::string PVFitter::outputTxt_

private

Definition at line 145 of file PVFitter.h.

pvQualities_

std::vector<double> PVFitter::pvQualities_

private

Definition at line 179 of file PVFitter.h.

Referenced by compressStore().

pvStore_

std::vector<BeamSpotFitPVData> PVFitter::pvStore_

private

Definition at line 176 of file PVFitter.h.

Referenced by compressStore(), getNPVs(), getpvStore(), readEvent(), resetAll(), resizepvStore(), and runFitter().

sigmaCut_

double PVFitter::sigmaCut_

private

Definition at line 156 of file PVFitter.h.

Referenced by initialize(), runBXFitter(), and runFitter().

theBeamSpotTreeData_

BeamSpotTreeData PVFitter::theBeamSpotTreeData_

private

Definition at line 181 of file PVFitter.h.

Referenced by readEvent(), and setTree().

useOnlyFirstPV_

bool PVFitter::useOnlyFirstPV_

private

Definition at line 138 of file PVFitter.h.

Referenced by initialize(), and readEvent().

vertexToken_

edm::EDGetTokenT<reco::VertexCollection> PVFitter::vertexToken_

private

Definition at line 144 of file PVFitter.h.

Referenced by initialize(), and readEvent().