rpr::TrackReco Class Reference

#include <TrackReco.h>

Inheritance diagram for rpr::TrackReco:

List of all members.

Public Member Functions
	TrackReco (const char *version)
void	Update (const cfg::Config &c)
virtual	~TrackReco ()
jobc::Result	Reco (edm::EventHandle &evt)
void	Connection (std::vector< recobase::TrackBase > &track, geo::View_t xychoice)
void	Matching (std::vector< recobase::TrackBase > &track, std::vector< recobase::TrackBase > &track3d)
void	CalcChi2Ndof (recobase::TrackBase &trk, double &chi2, int &ndof, geo::View_t xychoice)
Private Attributes
short	fDetGeom
geo::Geometry *	fGeo
float	fInterval
int	fGap

---

Constructor & Destructor Documentation

TrackReco::TrackReco (  const char *  version  )

Definition at line 26 of file TrackReco.cxx. References jobc::Module::SetCfgVersion(). : jobc::Module("TrackReco"), fDetGeom(1), fGeo(0), fInterval(13.27), fGap(2) { this->SetCfgVersion(version); }

TrackReco::~TrackReco (   )  [virtual]

Definition at line 35 of file TrackReco.cxx. { }

---

Member Function Documentation

void TrackReco::CalcChi2Ndof (  recobase::TrackBase &  trk, double &  chi2, int &  ndof, geo::View_t  xychoice )

Definition at line 380 of file TrackReco.cxx. References recobase::PlaneCluster::dT(), fGeo, recobase::RecoPoint::GetPC(), recobase::TrackBase::NRecoPt(), geo::Geometry::Plane(), recobase::PlaneCluster::Plane(), recobase::TrackBase::RecoPt(), recobase::RecoPoint::T(), recobase::PlaneCluster::Tpos(), geo::PlaneGeo::View(), and geo::View_t. Referenced by Connection(). { int pln0 = trk.RecoPt(0).GetPC()->Plane(); int pln1 = trk.RecoPt(trk.NRecoPt()-1).GetPC()->Plane(); chi2 = 0.; ndof = 0; for(int ipl=pln0; ipl<=pln1; ipl++){ if(fGeo->Plane(ipl).View() != xychoice) continue; ndof++; bool blank = true; unsigned int i=0; while(i<trk.NRecoPt()&&blank) { if(trk.RecoPt(i).GetPC()->Plane() == ipl) { blank = false; chi2 += pow((trk.RecoPt(i).GetPC()->Tpos() - trk.RecoPt(i).T(xychoice)) / trk.RecoPt(i).GetPC()->dT(), 2); } i++; } if(blank) chi2 += pow(2.,2); } }

void TrackReco::Connection (  std::vector< recobase::TrackBase > &  track, geo::View_t  xychoice )

Definition at line 88 of file TrackReco.cxx. References CalcChi2Ndof(), recobase::TrackBase::Clear(), recobase::PlaneCluster::dT(), recobase::TrackBase::EndZ(), fGap, fGeo, fInterval, recobase::RecoPoint::GetPC(), geo::PlaneGeo::LocalToWorld(), recobase::TrackBase::NRecoPt(), geo::Geometry::Plane(), recobase::PlaneCluster::Plane(), recobase::TrackBase::Put(), recobase::TrackBase::RecoPt(), recobase::RecoPoint::SetPosT(), recobase::TrackBase::SetXY(), recobase::TrackBase::StartT(), recobase::TrackBase::StartZ(), recobase::TrackBase::T(), recobase::PlaneCluster::Tpos(), and geo::View_t. Referenced by Reco(). { double local[3] = {0.,0.,0.}; double world[3] = {0.,0.,0.}; double chi2 = 0.; int ndof = 0; int ndof_single = 0; vector<TrackBase> trkseg; vector<TrackBase>::iterator itr = track.begin(); while(itr!=track.end()){ if(itr->XY()==xychoice) { trkseg.push_back(*itr); track.erase(itr); } else { ++itr; } } // connection of tracks while(trkseg.size()>0){ // find track segment at downstream double max_start_z = 0.; int isel_post = 0; for (unsigned int i=0; i<trkseg.size(); ++i) { if( trkseg[i].StartZ() > max_start_z ) { max_start_z = trkseg[i].StartZ(); isel_post = i; } } TrackBase posttrk = trkseg[isel_post]; itr = trkseg.begin() + isel_post; trkseg.erase(itr); double z_min_post = posttrk.StartZ(); double z_max_post = posttrk.EndZ(); double z_min_pre = 0.; double z_max_pre = 0.; float ex_min_dist=0.; while(1) { float min_dist=10000; int isel_pre = 0; for (unsigned int i=0; i<trkseg.size(); ++i) { float distance = sqrt(pow(trkseg[i].EndZ() - posttrk.StartZ(),2) + pow(trkseg[i].EndT(xychoice) - posttrk.StartT(xychoice),2)); if(distance<min_dist && distance>ex_min_dist) { isel_pre = i; min_dist = distance; } } ex_min_dist = min_dist; // Mimimum distance condition. 1 plane gap is allowed, 2 and more are not. if(min_dist>fInterval*(fGap+0.5)) break; TrackBase pretrk = trkseg[isel_pre]; z_min_pre = pretrk.StartZ(); z_max_pre = pretrk.EndZ(); // Sum of chi^2 from two tracks before connection float chi2_sep = 0.; int ndof_sep = 0; CalcChi2Ndof(pretrk, chi2, ndof, xychoice); chi2_sep += chi2; ndof_sep += ndof; ndof_single = ndof; CalcChi2Ndof(posttrk, chi2, ndof, xychoice); chi2_sep += chi2; ndof_sep += ndof; if( ndof > ndof_single ) ndof_single = ndof; // Slide joint point along both tracks and calculate minimum chi^2. double chi2_min = 100000.; double t_cross = 0., z_cross = 0.; TrackBase combtrk; TrackBase combtrk_save; for(int nloop=0; nloop<2; nloop++) { // slide along (nloop==0)? pretrk: posttrk; double z_min = (nloop==0)? z_min_pre : z_min_post; double z_max = (nloop==0)? z_max_pre : z_max_post; for(double zc = z_min; zc<z_max; zc+=fInterval/2.){ // Crossing point of 2 tracks: (zc, tc) double tc = (nloop==0)? pretrk.T(zc, xychoice) : posttrk.T(zc, xychoice); double dtdz0 = (zc != pretrk.StartZ() && zc != z_min_pre)? (tc - pretrk.T(z_min_pre, xychoice))/(zc - z_min_pre) : 0.; double dtdz1 = (zc != posttrk.EndZ() && z_max_post != zc)? (posttrk.T(z_max_post, xychoice) - tc)/(z_max_post - zc) : 0.; // Make combined track combtrk.Clear(); for(int ipl = pretrk.RecoPt(0).GetPC()->Plane(); ipl < posttrk.RecoPt(posttrk.NRecoPt()-1).GetPC()->Plane(); ipl++) { fGeo->Plane(ipl).LocalToWorld(local, world); double t = (world[2] <= zc)? pretrk.StartT(xychoice) + dtdz0 * (world[2] - z_min_pre) : tc + dtdz1 * (world[2] - zc); int ipre = -1; for(unsigned int i=0; i<pretrk.NRecoPt(); i++) { if(pretrk.RecoPt(i).GetPC()->Plane() == ipl) ipre = i; } int ipost = -1; for(unsigned int i=0; i<posttrk.NRecoPt(); i++) { if(posttrk.RecoPt(i).GetPC()->Plane() == ipl) ipost = i; } if(ipre<0 && ipost<0) continue; RecoPoint recopt; if(ipre>=0 && ipost<0) { // Hit exists only in pre track recopt = pretrk.RecoPt(ipre); } else if (ipre<0 && ipost>=0) { // Hit exists only in post track recopt = posttrk.RecoPt(ipre); } else if (ipre>=0 && ipost>=0) { // Both in this plane recopt = (abs(pretrk.RecoPt(ipre).GetPC()->Tpos()-t) < abs(posttrk.RecoPt(ipost).GetPC()->Tpos()-t)) ? pretrk.RecoPt(ipre) : posttrk.RecoPt(ipost); } recopt.SetPosT(t, xychoice); if(abs(recopt.GetPC()->Tpos()-t)<2.*recopt.GetPC()->dT()) combtrk.Put(recopt); } if(combtrk.NRecoPt()==0) continue; CalcChi2Ndof(combtrk, chi2, ndof, xychoice); double scat_angle = abs(atan(dtdz0)-atan(dtdz1))*180./3.1415; chi2 += pow(scat_angle/10.,2); // contribution from scattering angle: 10 deg as 1-sigma chi2 -= min(ndof - ndof_single, 4); // magic number: longer track is favored if(ndof>0 && chi2/ndof<chi2_min) { combtrk_save = combtrk; chi2_min = chi2/ndof; t_cross = tc; z_cross = zc; } } } if(chi2_min<1.2 || chi2_min<chi2_sep/ndof_sep){ posttrk = combtrk_save; itr = trkseg.begin() + isel_pre; trkseg.erase(itr); } } posttrk.SetXY(xychoice); track.push_back(posttrk); } }

void TrackReco::Matching (  std::vector< recobase::TrackBase > &  track, std::vector< recobase::TrackBase > &  track3d )

Definition at line 263 of file TrackReco.cxx. References recobase::TrackBase::Clear(), fGap, fInterval, recobase::TrackBase::Put(), recobase::RecoPoint::SetPos(), and recobase::RecoPoint::Z(). Referenced by Reco(). { vector<bool> trkused; trkused.resize(track.size()); for(unsigned int i=0; i<trkused.size(); i++) { trkused[i] = false; } TrackBase trk; // Construct 3D track from 2D tracks in X and Y view while(1){ double min_mean_dist_frac = 10000; double min_mean_dist = 10000; int mean_frac_x = 0, mean_frac_y = 0; int mean_x = 0, mean_y = 0; for (unsigned int ix=0; ix<track.size(); ++ix) { for (unsigned int iy=0; iy<track.size(); ++iy) { if(ix!=iy&&track[ix].XY()==geo::kX&&track[iy].XY()==geo::kY) { bool used = false; if(trkused[ix]||trkused[iy]) used = true; if(used) continue; double range_x = track[ix].EndZ() - track[ix].StartZ(); double range_y = track[iy].EndZ() - track[iy].StartZ(); double vtx_dist = abs(track[ix].StartZ() - track[iy].StartZ()); double end_dist = abs(track[ix].EndZ() - track[iy].EndZ()); double mean_dist = (vtx_dist+end_dist)/2.; double mean_dist_frac = mean_dist/((range_x+range_y)/2.); if(mean_dist_frac<min_mean_dist_frac) { min_mean_dist_frac = mean_dist_frac; mean_frac_x = ix; mean_frac_y = iy; } if(mean_dist<min_mean_dist) { min_mean_dist = mean_dist; mean_x = ix; mean_y = iy; } } } } // Conditions of matching x and y tracks: int matched_x = 0; int matched_y = 0; if(min_mean_dist_frac<0.1) { // 1. Mean distance of vertex and end positions relative to the range matched_x = mean_frac_x; matched_y = mean_frac_y; } else if(min_mean_dist<fInterval*(fGap+0.25)) { // 2. Mean distance of vertex and end position (3+5) plane is OK, (5+5) is not. matched_x = mean_x; matched_y = mean_y; } else if(min_mean_dist_frac<0.3) { // 3. Mean distance of vertex and end position relative to the range -> Vertex or end position. double vtx_dist = fabs(track[mean_frac_x].StartZ() - track[mean_frac_y].StartZ()); double end_dist = fabs(track[mean_frac_x].EndZ() - track[mean_frac_y].EndZ()); if(vtx_dist<fInterval*(fGap+0.25)){ matched_x = mean_frac_x; matched_y = mean_frac_y; } else if(end_dist<fInterval*(fGap+0.25)){ matched_x = mean_frac_x; matched_y = mean_frac_y; } else { break; } } else { // If no condition is satisfied. break; } trkused[matched_x] = true; trkused[matched_y] = true; trk.Clear(); RecoPoint recopt; TVector3 xyz; for(unsigned int i=0; i<track[matched_x].NRecoPt(); i++) { recopt = track[matched_x].RecoPt(i); xyz.SetXYZ(track[matched_x].X(recopt.Z()), track[matched_y].Y(recopt.Z()), recopt.Z()); recopt.SetPos(xyz); trk.Put(recopt); } for(unsigned int i=0; i<track[matched_y].NRecoPt(); i++) { recopt = track[matched_y].RecoPt(i); xyz.SetXYZ(track[matched_x].X(recopt.Z()), track[matched_y].Y(recopt.Z()), recopt.Z()); recopt.SetPos(xyz); trk.Put(recopt); } track3d.push_back(trk); } // While there are tracks which have closed vertex and/or end position. }

jobc::Result TrackReco::Reco (  edm::EventHandle &  evt  )  [virtual]

Reimplemented from jobc::Module. Definition at line 49 of file TrackReco.cxx. References geo::PlaneGeo::Cell(), Connection(), fDetGeom, fGeo, fInterval, geo::CellGeo::GetCenter(), edm::EventHandle::Header(), geo::Geometry::Instance(), Matching(), geo::Geometry::Plane(), edm::EventHandle::Reco(), and jobc::Result. { if (!fGeo) fGeo = &geo::Geometry::Instance(evt.Header().Run(), fDetGeom); // Gap between two planes double x1[3], x2[3]; fGeo->Plane(0).Cell(0).GetCenter(x1); fGeo->Plane(2).Cell(0).GetCenter(x2); fInterval = x2[2] - x1[2]; // Get Track Sengemnts vector<const recobase::TrackBase*> trackseg(0); try{ evt.Reco().Get("./Segment",trackseg); } catch(edm::Exception e) { return jobc::kFailed; } vector<recobase::TrackBase> track; for(unsigned int itrk=0; itrk<trackseg.size(); itrk++) track.push_back(*trackseg[itrk]); this->Connection(track, geo::kX); this->Connection(track, geo::kY); vector<recobase::TrackBase> track3d; this->Matching(track, track3d); evt.Reco().MakeFolder("./Final"); for (unsigned int i=0; i<track3d.size(); ++i) { evt.Reco().Put(track3d[i],"./Final"); } return jobc::kPassed; }

void TrackReco::Update (  const cfg::Config &  c  )  [virtual]

Implements cfg::Observer. Definition at line 41 of file TrackReco.cxx. References fDetGeom, and fGap. { c("DetGeom"). Get(fDetGeom); c("Gap"). Get(fGap); }

---

Member Data Documentation

short rpr::TrackReco::fDetGeom [private]

Definition at line 45 of file TrackReco.h. Referenced by Reco(), and Update().

int rpr::TrackReco::fGap [private]

Definition at line 48 of file TrackReco.h. Referenced by Connection(), Matching(), and Update().

geo::Geometry* rpr::TrackReco::fGeo [private]

Definition at line 46 of file TrackReco.h. Referenced by CalcChi2Ndof(), Connection(), and Reco().

float rpr::TrackReco::fInterval [private]

Definition at line 47 of file TrackReco.h. Referenced by Connection(), Matching(), and Reco().

---

The documentation for this class was generated from the following files:

• TrackReco.h
• TrackReco.cxx

---