Prompt Namespace Reference

Namespaces
namespace	Def

Classes
class	DecoratePLIT
class	DecoratePromptLeptonImproved
class	DecoratePromptLeptonRNN
struct	FittingInput
class	IRNNTool
class	IVertexFittingTool
class	IVertexMergingTool
struct	MergeResult
struct	MergeResultNotOwner
class	NonPromptLeptonVertexingAlg
class	PrimaryVertexReFitter
class	RNNTool
struct	SortByIDTrackPt
struct	SortObjectByVar
struct	SortTracksByPt
struct	SortTwoTrackVtxByDistToSeed
struct	SortTwoTrackVtxBySumTrackPt
class	TimerScopeHelper
struct	TwoTrackVtx
class	VarEntry
class	VarHolder
class	VertexFittingTool
class	VertexIterativeFitMergingTool
class	VertexMergingTool
struct	VtxCluster

Typedefs
typedef TLorentzVector	FourMom_t
typedef std::vector< Prompt::VarEntry >	VarEntryVec
typedef std::map< Def::Var, std::string >	VarMap

Enumerations
enum	VtxType {   kTwoTrackVtx = 1 , kSimpleMergedVtx = 2 , kDeepMergedVtx = 3 , kIterativeFitVtx = 4 ,   kTwoTrackVtxWithoutLepton = 5 , kIterativeFitVtxWithoutLepton = 6 , kRefittedPriVtx = 7 , kRefittedPriVtxWithoutLep = 8 }

Functions
static const InterfaceID	IID_IRNNTool ("Prompt::IRNNTool", 1, 0)
double	getVertexFitProb (const xAOD::Vertex *vtx)
std::string	vtxAsStr (const xAOD::Vertex *vtx, bool print_tracks=false)
std::string	trkAsStr (const xAOD::TrackParticle *trk)
std::string	truthAsStr (const xAOD::IParticle &particle)
double	getDistance (const xAOD::Vertex *vtx1, const xAOD::Vertex *vtx2)
double	getDistance (const Amg::Vector3D &vtx1, const Amg::Vector3D &vtx2)
double	getNormDist (const Amg::Vector3D &PrimVtx, const Amg::Vector3D &SecVtx, const std::vector< float > &ErrorMatrix, MsgStream &msg)
void	fillTH1 (TH1 *h, double val, double weight=1.0)
std::string	printPromptVertexAsStr (const xAOD::Vertex *vtx, MsgStream &msg)
template<class T1, class T2>
bool	getVar (T1 &obj, T2 &value, const std::string &var_name)
template<class T1, class T2>
bool	GetAuxVar (const T1 &obj, T2 &value, const std::string &var_name)
std::string	PrintResetStopWatch (TStopwatch &watch)
bool	operator== (const VarEntry &lhs, const VarEntry &rhs)
bool	operator< (const VarEntry &lhs, const VarEntry &rhs)
bool	operator== (const VarEntry &var, unsigned key)
bool	operator== (unsigned key, const VarEntry &var)
bool	operator< (const VarEntry &var, unsigned key)
bool	operator< (unsigned key, const VarEntry &var)

Typedef Documentation

FourMom_t

typedef TLorentzVector Prompt::FourMom_t

Definition at line 32 of file DecoratePLIT.h.

VarEntryVec

typedef std::vector<Prompt::VarEntry> Prompt::VarEntryVec

Definition at line 97 of file VarHolder.h.

VarMap

typedef std::map<Def::Var, std::string> Prompt::VarMap

Definition at line 98 of file VarHolder.h.

Enumeration Type Documentation

VtxType

enum Prompt::VtxType

Enumerator
kTwoTrackVtx
kSimpleMergedVtx
kDeepMergedVtx
kIterativeFitVtx
kTwoTrackVtxWithoutLepton
kIterativeFitVtxWithoutLepton
kRefittedPriVtx
kRefittedPriVtxWithoutLep

Definition at line 38 of file IVertexFittingTool.h.

               {
    kTwoTrackVtx                  = 1,
    kSimpleMergedVtx              = 2,
    kDeepMergedVtx                = 3,
    kIterativeFitVtx              = 4,
    kTwoTrackVtxWithoutLepton     = 5,
    kIterativeFitVtxWithoutLepton = 6,
    kRefittedPriVtx               = 7,
    kRefittedPriVtxWithoutLep     = 8
  };

Function Documentation

fillTH1()

void Prompt::fillTH1	(	TH1 *	h,
		double	val,
		double	weight = 1.0 )

Definition at line 104 of file PromptUtils.cxx.

{
  //
  // Read new event entry
  //
  if(!h) {
    return;
  }
 
  const double xmax = h->GetXaxis()->GetXmax();
  const double xmin = h->GetXaxis()->GetXmin();
 
  double x = val;
  if (!(val < xmax)){
    x = h->GetXaxis()->GetBinCenter(h->GetNbinsX());
  } else if (!(val > xmin)) {
    x = h->GetXaxis()->GetBinCenter(1);
  }
 
  h->Fill(x, weight);
}

GetAuxVar()

template<class T1, class T2>

bool Prompt::GetAuxVar	(	const T1 &	obj,
		T2 &	value,
		const std::string &	var_name )

Definition at line 83 of file PromptUtils.h.

  {
    //
    // get the int aux-variable
    //
    typename SG::AuxElement::Accessor<T2> acc(var_name);
 
    if(!acc.isAvailable(obj)) {
      return false;
    }
 
    value = acc(obj);
    return true;
  }

getDistance() [1/2]

double Prompt::getDistance	(	const Amg::Vector3D &	vtx1,
		const Amg::Vector3D &	vtx2 )

Definition at line 51 of file PromptUtils.cxx.

{
  return (vtx2 - vtx1).mag();
}

getDistance() [2/2]

double Prompt::getDistance	(	const xAOD::Vertex *	vtx1,
		const xAOD::Vertex *	vtx2 )

Definition at line 41 of file PromptUtils.cxx.

{
  if((!vtx1) || (!vtx2)) {
    return 9999.0;
  }
 
  return Prompt::getDistance(vtx1->position(), vtx2->position());
}

getNormDist()

double Prompt::getNormDist	(	const Amg::Vector3D &	PrimVtx,
		const Amg::Vector3D &	SecVtx,
		const std::vector< float > &	ErrorMatrix,
		MsgStream &	msg )

Definition at line 57 of file PromptUtils.cxx.

{
  double significance = -10;
 
  Eigen::Matrix<double, 3, 3, 0, 3, 3> PrimCovMtx;  //Create matrix
 
  if(errorMatrix.empty()) {
    msg << "getNormDist - Error matrix is empty" << std::endl;
    return significance;
  }
 
  PrimCovMtx(0,0) = static_cast<double> (errorMatrix[0]);
  PrimCovMtx(0,1) = static_cast<double> (errorMatrix[1]);
  PrimCovMtx(1,0) = static_cast<double> (errorMatrix[1]);
  PrimCovMtx(1,1) = static_cast<double> (errorMatrix[2]);
  PrimCovMtx(0,2) = static_cast<double> (errorMatrix[3]);
  PrimCovMtx(2,0) = static_cast<double> (errorMatrix[3]);
  PrimCovMtx(1,2) = static_cast<double> (errorMatrix[4]);
  PrimCovMtx(2,1) = static_cast<double> (errorMatrix[4]);
  PrimCovMtx(2,2) = static_cast<double> (errorMatrix[5]);
 
  if(PrimCovMtx.determinant() == 0) {
    msg << "getNormDist - Matrix can not be inversed" << std::endl;
    return significance;
  }
 
  Eigen::Matrix<double, 3, 3,0, 3, 3> WgtMtx = PrimCovMtx.inverse();
 
  Eigen::Vector3d dist;
  Amg::Vector3D vtxDiff = vtx1 - vtx2;
  dist[0] = vtxDiff.x();
  dist[1] = vtxDiff.y();
  dist[2] = vtxDiff.z();
 
  significance = dist.transpose() * WgtMtx * dist;
 
  if(significance < 0) {
    msg << "getNormDist - significance is negative" << std::endl;
    return significance;
  }
 
  significance = std::sqrt(significance);
 
  return significance;
}

getVar()

template<class T1, class T2>

bool Prompt::getVar	(	T1 &	obj,
		T2 &	value,
		const std::string &	var_name )

Definition at line 62 of file PromptUtils.h.

  {
    if (!obj) {
      std::cerr << "getVar - received a null object" << std::endl;
      return false;
    }
 
    //
    // get the int aux-variable
    //
    typename SG::AuxElement::Accessor<T2> acc(var_name);
 
    if(!acc.isAvailable(*obj)) {
      return false;
    }
 
    value = acc(*obj);
    return true;
  }

getVertexFitProb()

double Prompt::getVertexFitProb

(

const xAOD::Vertex *

vtx

)

Definition at line 21 of file PromptUtils.cxx.

{
  //
  // Get xAOD::Vertex fit probability
  //
  double fit_prob = -1;
 
  if(!vtx) {
    return fit_prob;
  }
 
  if(vtx->numberDoF() > 0 && vtx->chiSquared() > 0) {
    fit_prob = TMath::Prob(vtx->chiSquared(), vtx->numberDoF());
  }
 
  return fit_prob;
}

IID_IRNNTool()

const InterfaceID Prompt::IID_IRNNTool ( "Prompt::IRNNTool" , 1 , 0  )

static

operator<() [1/3]

bool Prompt::operator< ( const VarEntry & lhs, const VarEntry & rhs )

inline

Definition at line 164 of file VarHolder.h.

  {
    return lhs.getKey() < rhs.getKey();
  }

operator<() [2/3]

bool Prompt::operator< ( const VarEntry & var, unsigned key )

inline

Definition at line 172 of file VarHolder.h.

{ return var.getKey() < key; }

operator<() [3/3]

bool Prompt::operator< ( unsigned key, const VarEntry & var )

inline

Definition at line 173 of file VarHolder.h.

{ return key < var.getKey(); }

operator==() [1/3]

bool Prompt::operator== ( const VarEntry & lhs, const VarEntry & rhs )

inline

Definition at line 160 of file VarHolder.h.

  {
    return lhs.getKey() == rhs.getKey();
  }

operator==() [2/3]

bool Prompt::operator== ( const VarEntry & var, unsigned key )

inline

Definition at line 169 of file VarHolder.h.

{ return var.getKey() == key; }

operator==() [3/3]

bool Prompt::operator== ( unsigned key, const VarEntry & var )

inline

Definition at line 170 of file VarHolder.h.

{ return var.getKey() == key; }

printPromptVertexAsStr()

std::string Prompt::printPromptVertexAsStr	(	const xAOD::Vertex *	vtx,
		MsgStream &	msg )

Definition at line 127 of file PromptUtils.cxx.

{
  std::stringstream str;
 
  str << "xAOD::Vertex pointer = " << vtx<< std::endl;
 
  if(vtx) {
    float chisquared = -9999;
    float numberdof  = -9999;
    int   index      = -999;
 
    if(!getVar(vtx, index, "SecondaryVertexIndex")) {
      msg << "printPromptVertexAsStr -- not valid vtx SecondaryVertexIndex!!!" << std::endl;
    }
 
    if(!getVar(vtx, chisquared, "chiSquared")) {
      msg << "printPromptVertexAsStr -- not valid vtx chiSquared!!!" << std::endl;
    }
 
    if(!getVar(vtx, numberdof, "numberDoF")) {
      msg << "printPromptVertexAsStr -- not valid vtx numberDoF!!!" << std::endl;
    }
 
    str << "   index      " << index             << std::endl;
    str << "   position   " << vtx->position  () << std::endl;
    str << "   x          " << vtx->x         () << std::endl;
    str << "   y          " << vtx->y         () << std::endl;
    str << "   z          " << vtx->z         () << std::endl;
    str << "   chiSquared " << chisquared        << std::endl;
    str << "   numberDoF  " << numberdof         << std::endl;
 
    str << "   covariance.size() = " << vtx->covariance().size() << std::endl;
    str << "   covariance        = [";
 
    for(const float val: vtx->covariance()) {
      str << val << ", ";
    }
 
    str << "]";
  }
 
  return str.str();
}

PrintResetStopWatch()

std::string Prompt::PrintResetStopWatch

(

TStopwatch &

watch

)

Definition at line 244 of file PromptUtils.cxx.

{
  watch.Stop();
 
  double realt = watch.RealTime();
  double cput  = watch.CpuTime();
 
  watch.Reset();
  watch.Start();
 
  const int hours = static_cast<int>(realt/3600.0);
  const int  min  = static_cast<int>(realt/60.0) - 60*hours;
 
  realt -= hours * 3600;
  realt -= min * 60;
 
  if (realt < 0) realt = 0;
  if (cput  < 0) cput  = 0;
 
  const int sec = static_cast<int>(realt);
 
  std::stringstream str;
  str << "Real time "
      << setw(2) << setfill('0') << hours
      << ":" << setw(2) << setfill('0') << min
      << ":" << setw(2) << setfill('0') << sec
      << " CPU time " << setprecision(3) << fixed << cput;
 
  return str.str();
}

trkAsStr()

std::string Prompt::trkAsStr

(

const xAOD::TrackParticle *

trk

)

Definition at line 228 of file PromptUtils.cxx.

{
  if(!trk) {
    return "trkAsStr - null pointer";
  }
 
  stringstream str;
 
  str << "xAOD::TrackParticle - " << trk << ": pT=" << trk->pt()
      << ", eta=" << trk->eta()
      << ", phi=" << trk->phi();
 
  return str.str();
}

truthAsStr()

std::string Prompt::truthAsStr

(

const xAOD::IParticle &

particle

)

Definition at line 213 of file PromptUtils.cxx.

{
  int truthOrigin = -1;
  int truthType   = -1;
 
  Prompt::GetAuxVar(particle, truthOrigin, "truthOrigin");
  Prompt::GetAuxVar(particle, truthType,   "truthType");
 
  std::stringstream str;
  str << "truthType=" << truthType << ", truthOrigin=" << truthOrigin;
 
  return str.str();
}

vtxAsStr()

std::string Prompt::vtxAsStr	(	const xAOD::Vertex *	vtx,
		bool	print_tracks = false )

Definition at line 172 of file PromptUtils.cxx.

{
  if(!vtx) {
    return "vtxAsStr - null pointer";
  }
 
  stringstream str;
 
  float distToPV = -1;
  float sigToPV = -1;
 
  Prompt::GetAuxVar(*vtx, distToPV, "distToPriVtx");
  Prompt::GetAuxVar(*vtx, sigToPV,  "normDistToPriVtx");
 
  str << "xAOD::Vertex - " << vtx << ": ntrack=" << vtx->nTrackParticles()
      << ", chi2/ndof=" << vtx->chiSquared() << "/" << vtx->numberDoF()
      << ", prob=" << Prompt::getVertexFitProb(vtx)
      << ", (x, y, z)=(" << vtx->x() << ", " << vtx->y() << ", " << vtx->z() << ")"
      << ", distToPV=" << distToPV
      << ", sigToPV=" << sigToPV;
 
  if(print_tracks) {
    str << endl;
 
    std::vector<const xAOD::TrackParticle *> tracks;
 
    for(unsigned i = 0; i < vtx->nTrackParticles(); ++i) {
      tracks.push_back(vtx->trackParticle(i));
    }
 
    std::sort(tracks.begin(), tracks.end(), Prompt::SortByIDTrackPt());
 
    for(unsigned i = 0; i < tracks.size(); ++i) {
      str << "     xAOD::Vertex track[" << i << "] " << trkAsStr(tracks.at(i)) << endl;
    }
  }
 
  return str.str();
}