CosmicTracker_module.cc File Reference

#include <math.h>
#include <algorithm>
#include <iostream>
#include <iomanip>
#include <fstream>
#include <vector>
#include <string>
#include "art/Framework/Core/EDProducer.h"
#include "art/Framework/Core/ModuleMacros.h"
#include "canvas/Persistency/Common/FindManyP.h"
#include "art/Framework/Principal/Event.h"
#include "fhiclcpp/ParameterSet.h"
#include "art/Framework/Principal/Handle.h"
#include "canvas/Persistency/Common/Ptr.h"
#include "canvas/Persistency/Common/PtrVector.h"
#include "art/Framework/Services/Registry/ServiceHandle.h"
#include "art/Framework/Services/Optional/TFileService.h"
#include "art/Framework/Services/Optional/TFileDirectory.h"
#include "messagefacility/MessageLogger/MessageLogger.h"
#include "larcore/Geometry/Geometry.h"
#include "larcorealg/Geometry/PlaneGeo.h"
#include "larcorealg/Geometry/WireGeo.h"
#include "lardataobj/RecoBase/Hit.h"
#include "lardataobj/RecoBase/Cluster.h"
#include "lardataobj/RecoBase/Track.h"
#include "lardataobj/RecoBase/SpacePoint.h"
#include "lardata/Utilities/AssociationUtil.h"
#include "larreco/RecoAlg/ClusterMatchTQ.h"
#include "larreco/RecoAlg/CosmicTrackerAlg.h"
#include "TVectorD.h"
#include "TF1.h"
#include "TGraph.h"
#include "TMath.h"
#include "TH1D.h"
#include "TVirtualFitter.h"

Go to the source code of this file.

Classes
struct	trkPoint
class	trkf::CosmicTracker

Namespaces
	trkf

Functions
bool	AnglesConsistent (const TVector3 &p1, const TVector3 &p2, const TVector3 &a1, const TVector3 &a2, double angcut)
bool	MatchTrack (const std::vector< trkPoint > &trkpts1, const std::vector< trkPoint > &trkpts2, double discut, double angcut)
bool	SortByWire (art::Ptr< recob::Hit > const &h1, art::Ptr< recob::Hit > const &h2)
bool	sp_sort_x0 (const trkPoint &tp1, const trkPoint &tp2)
bool	sp_sort_x1 (const trkPoint &tp1, const trkPoint &tp2)
bool	sp_sort_y0 (const trkPoint &tp1, const trkPoint &tp2)
bool	sp_sort_y1 (const trkPoint &tp1, const trkPoint &tp2)
bool	sp_sort_z0 (const trkPoint &tp1, const trkPoint &tp2)
bool	sp_sort_z1 (const trkPoint &tp1, const trkPoint &tp2)
bool	spt_sort_x0 (const recob::SpacePoint h1, const recob::SpacePoint h2)
bool	spt_sort_x1 (const recob::SpacePoint h1, const recob::SpacePoint h2)
bool	spt_sort_y0 (const recob::SpacePoint h1, const recob::SpacePoint h2)
bool	spt_sort_y1 (const recob::SpacePoint h1, const recob::SpacePoint h2)
bool	spt_sort_z0 (const recob::SpacePoint h1, const recob::SpacePoint h2)
bool	spt_sort_z1 (const recob::SpacePoint h1, const recob::SpacePoint h2)

Function Documentation

bool AnglesConsistent	(	const TVector3 &	p1,
		const TVector3 &	p2,
		const TVector3 &	a1,
		const TVector3 &	a2,
		double	angcut
	)

Definition at line 70 of file CosmicTracker_module.cc.

Referenced by MatchTrack().

                                                                                                                    {
  double angle1 = a1.Angle(a2);
  if (angle1>TMath::PiOver2()) angle1 = TMath::Pi() - angle1;
  double angle2 = a1.Angle(p1-p2);
  if (angle2>TMath::PiOver2()) angle2 = TMath::Pi() - angle2;
  if (angle1<angcut&&angle2<angcut) return true;
  else return false;
}

bool MatchTrack	(	const std::vector< trkPoint > &	trkpts1,
		const std::vector< trkPoint > &	trkpts2,
		double	discut,
		double	angcut
	)

Definition at line 80 of file CosmicTracker_module.cc.

References AnglesConsistent(), trkPoint::dir, and trkPoint::pos.

Referenced by trkf::CosmicTracker::produce().

                                                                                                                     {
  bool match = false;
  if (!trkpts1.size()) return match;
  if (!trkpts2.size()) return match;
  if ((trkpts1[0].hit)->WireID().Cryostat == (trkpts2[0].hit)->WireID().Cryostat &&
      (trkpts1[0].hit)->WireID().TPC == (trkpts2[0].hit)->WireID().TPC) return match; 
//  art::ServiceHandle<geo::Geometry> geom;
//  const geo::TPCGeo &thetpc1 = geom->TPC((trkpts1[0].hit)->WireID().TPC, (trkpts1[0].hit)->WireID().Cryostat);
//  const geo::TPCGeo &thetpc2 = geom->TPC((trkpts2[0].hit)->WireID().TPC, (trkpts2[0].hit)->WireID().Cryostat);

  //std::cout<<trkpts1[0].pos.Y()<<" "<<trkpts1.back().pos.Y()<<" "<<trkpts1[0].pos.Z()<<" "<<trkpts1.back().pos.Z()<<std::endl;
  //std::cout<<trkpts2[0].pos.Y()<<" "<<trkpts2.back().pos.Y()<<" "<<trkpts2[0].pos.Z()<<" "<<trkpts2.back().pos.Z()<<std::endl;

//  if (thetpc1.InFiducialY(trkpts1[0].pos.Y(),5)&&
//      thetpc1.InFiducialY(trkpts1.back().pos.Y(),5)&&
//      thetpc1.InFiducialZ(trkpts1[0].pos.Z(),5)&&
//      thetpc1.InFiducialZ(trkpts1.back().pos.Z(),5)) return match;
//  //std::cout<<"pass 1"<<std::endl;
//  if (thetpc2.InFiducialY(trkpts2[0].pos.Y(),5)&&
//      thetpc2.InFiducialY(trkpts2.back().pos.Y(),5)&&
//      thetpc2.InFiducialZ(trkpts2[0].pos.Z(),5)&&
//      thetpc2.InFiducialZ(trkpts2.back().pos.Z(),5)) return match;
//  //std::cout<<"pass 2"<<std::endl;

  if (AnglesConsistent(trkpts1[0].pos, trkpts2[0].pos, 
                       trkpts1[0].dir, trkpts2[0].dir, angcut)) match = true;
  if (AnglesConsistent(trkpts1.back().pos, trkpts2[0].pos, 
                       trkpts1.back().dir, trkpts2[0].dir, angcut)) match = true;
  if (AnglesConsistent(trkpts1[0].pos, trkpts2.back().pos, 
                       trkpts1[0].dir, trkpts2.back().dir, angcut)) match = true;
  if (AnglesConsistent(trkpts1.back().pos, trkpts2.back().pos, 
                       trkpts1.back().dir, trkpts2.back().dir, angcut)) match = true;

  return match;
}

bool SortByWire	(	art::Ptr< recob::Hit > const &	h1,
		art::Ptr< recob::Hit > const &	h2
	)

Definition at line 116 of file CosmicTracker_module.cc.

References geo::WireID::Wire, and recob::Hit::WireID().

                                                                         { 
  return h1->WireID().Wire < h2->WireID().Wire;
}

bool sp_sort_x0	(	const trkPoint &	tp1,
		const trkPoint &	tp2
	)

Definition at line 120 of file CosmicTracker_module.cc.

References trkPoint::pos.

Referenced by trkf::CosmicTracker::produce().

{
  return tp1.pos.X() < tp2.pos.X();
}

bool sp_sort_x1	(	const trkPoint &	tp1,
		const trkPoint &	tp2
	)

Definition at line 125 of file CosmicTracker_module.cc.

References trkPoint::pos.

Referenced by trkf::CosmicTracker::produce().

{
  return tp1.pos.X() > tp2.pos.X();
}

bool sp_sort_y0	(	const trkPoint &	tp1,
		const trkPoint &	tp2
	)

Definition at line 130 of file CosmicTracker_module.cc.

References trkPoint::pos.

Referenced by trkf::CosmicTracker::produce().

{
  return tp1.pos.Y() < tp2.pos.Y();
}

bool sp_sort_y1	(	const trkPoint &	tp1,
		const trkPoint &	tp2
	)

Definition at line 135 of file CosmicTracker_module.cc.

References trkPoint::pos.

Referenced by trkf::CosmicTracker::produce().

{
  return tp1.pos.Y() > tp2.pos.Y();
}

bool sp_sort_z0	(	const trkPoint &	tp1,
		const trkPoint &	tp2
	)

Definition at line 140 of file CosmicTracker_module.cc.

References trkPoint::pos.

Referenced by trkf::CosmicTracker::produce().

{
  return tp1.pos.Z() < tp2.pos.Z();
}

bool sp_sort_z1	(	const trkPoint &	tp1,
		const trkPoint &	tp2
	)

Definition at line 145 of file CosmicTracker_module.cc.

References trkPoint::pos.

Referenced by trkf::CosmicTracker::produce().

{
  return tp1.pos.Z() > tp2.pos.Z();
}

bool spt_sort_x0	(	const recob::SpacePoint	h1,
		const recob::SpacePoint	h2
	)

Definition at line 150 of file CosmicTracker_module.cc.

References recob::SpacePoint::XYZ().

Referenced by trkf::CosmicTracker::produce().

{
  const double* xyz1 = h1.XYZ();
  const double* xyz2 = h2.XYZ();
  return xyz1[0] < xyz2[0];
}

bool spt_sort_x1	(	const recob::SpacePoint	h1,
		const recob::SpacePoint	h2
	)

Definition at line 157 of file CosmicTracker_module.cc.

References recob::SpacePoint::XYZ().

Referenced by trkf::CosmicTracker::produce().

{
  const double* xyz1 = h1.XYZ();
  const double* xyz2 = h2.XYZ();
  return xyz1[0] > xyz2[0];
}

bool spt_sort_y0	(	const recob::SpacePoint	h1,
		const recob::SpacePoint	h2
	)

Definition at line 164 of file CosmicTracker_module.cc.

References recob::SpacePoint::XYZ().

Referenced by trkf::CosmicTracker::produce().

{
  const double* xyz1 = h1.XYZ();
  const double* xyz2 = h2.XYZ();
  return xyz1[1] < xyz2[1];
}

bool spt_sort_y1	(	const recob::SpacePoint	h1,
		const recob::SpacePoint	h2
	)

Definition at line 171 of file CosmicTracker_module.cc.

References recob::SpacePoint::XYZ().

Referenced by trkf::CosmicTracker::produce().

{
  const double* xyz1 = h1.XYZ();
  const double* xyz2 = h2.XYZ();
  return xyz1[1] > xyz2[1];
}

bool spt_sort_z0	(	const recob::SpacePoint	h1,
		const recob::SpacePoint	h2
	)

Definition at line 178 of file CosmicTracker_module.cc.

References recob::SpacePoint::XYZ().

Referenced by trkf::CosmicTracker::produce().

{
  const double* xyz1 = h1.XYZ();
  const double* xyz2 = h2.XYZ();
  return xyz1[2] < xyz2[2];
}

bool spt_sort_z1	(	const recob::SpacePoint	h1,
		const recob::SpacePoint	h2
	)

Definition at line 185 of file CosmicTracker_module.cc.

References recob::SpacePoint::XYZ().

Referenced by trkf::CosmicTracker::produce().

{
  const double* xyz1 = h1.XYZ();
  const double* xyz2 = h2.XYZ();
  return xyz1[2] > xyz2[2];
}