FillTrackGeom.cxx

Go to the documentation of this file.

// $Id: FillTrackGeom.cxx,v 1.9 2008/09/05 15:42:33 rustem Exp $

// C/C++
#include <cassert>
#include <cmath>
#include <sstream>

// Local
#include "PhysicsNtuple/Default.h"
#include "PhysicsNtuple/Factory.h"

// Local
#include "FillTrackGeom.h"

REGISTER_ANP_OBJECT(AlgSnarl,FillTrackGeom)

using namespace std;

//------------------------------------------------------------------------------------------
Anp::FillTrackGeom::FillTrackGeom()
{
}

//------------------------------------------------------------------------------------------
Anp::FillTrackGeom::~FillTrackGeom()
{
}

//------------------------------------------------------------------------------------------
bool Anp::FillTrackGeom::Run(Record &record)
{
   for(TrackIterator it = record.TrackBegIterator(); it != record.TrackEndIterator(); ++it)
   {
      Fill(*it);
   }

   return true;
}

//------------------------------------------------------------------------------------------
void Anp::FillTrackGeom::Fill(Track &track) const
{
   const double x_vtx = track.GetBegVtx().X();
   const double y_vtx = track.GetBegVtx().Y();
   const double z_vtx = track.GetBegVtx().Z();

   const double x_end = track.GetEndVtx().X();
   const double y_end = track.GetEndVtx().Y();
   const double z_end = track.GetEndVtx().Z();
   
   const double cosx_vtx = track.GetBegVtx().CosX();
   const double cosy_vtx = track.GetBegVtx().CosY();

   // transform all x,y coordinates into the new coordinates
   const double xy_angle_vtx = Anp::angle(x_vtx, y_vtx);

   const double rot_x_vtx =  x_vtx * std::cos(xy_angle_vtx) + y_vtx * std::sin(xy_angle_vtx);
   const double rot_y_vtx = -x_vtx * std::sin(xy_angle_vtx) + y_vtx * std::cos(xy_angle_vtx);

   const double rot_x_end =  x_end * std::cos(xy_angle_vtx) + y_end * std::sin(xy_angle_vtx);
   const double rot_y_end = -x_end * std::sin(xy_angle_vtx) + y_end * std::cos(xy_angle_vtx);

   const double x_rel = rot_x_end - rot_x_vtx;
   const double y_rel = rot_y_end - rot_y_vtx;
   const double z_rel = z_end - z_vtx;

   // determine angles of track's end relative to vtx 
   // in new coordinates - xy plane
   
   if(!track.KeyExists(104))
   {
      track.Add(104, Anp::angle(x_rel, y_rel));
   }

   // difference between hit and projected straight line from vtx

   if(track.GetBegVtx().CosZ() > 0.0)
   {
      const double t_end = z_rel/track.GetBegVtx().CosZ();
      
      const double x_proj = cosx_vtx * t_end + x_vtx;
      const double y_proj = cosy_vtx * t_end + y_vtx;

      const double x_proj_rot =  x_proj * std::cos(xy_angle_vtx) + y_proj * std::sin(xy_angle_vtx);
      const double y_proj_rot = -x_proj * std::sin(xy_angle_vtx) + y_proj * std::cos(xy_angle_vtx);

      const double x_proj_diff = rot_x_end - x_proj_rot;
      const double y_proj_diff = rot_y_end - y_proj_rot;

      const double proj_xy_angle = Anp::angle(x_proj_diff, y_proj_diff);
      
      const double proj_distance = std::sqrt(x_proj_diff*x_proj_diff + y_proj_diff*y_proj_diff);

      if(!track.KeyExists(108))
      {      
         track.Add(108, proj_xy_angle);
      }
      if(!track.KeyExists(109))
      {
         track.Add(109, x_proj_diff);
      }
      if(!track.KeyExists(110))
      {
         track.Add(110, y_proj_diff);
      }
      if(!track.KeyExists(111))
      {
         track.Add(111, proj_distance);
      }
   }

   //
   // Add standard track variables for convienience
   //
   if(track.Pass() && track.ErrorQP() > 0.0)
   {
      if(!track.KeyExists(120)) track.Add(120, track.ErrorQP());
      if(!track.KeyExists(121)) track.Add(121, track.QP());
      if(!track.KeyExists(122)) track.Add(122, track.QP()/track.ErrorQP());
   }
}

---