ANtpEventManipulator.cxx

Go to the documentation of this file.

//$Id: ANtpEventManipulator.cxx,v 1.6 2007/03/21 21:50:15 brebel Exp $
//
//ANtpEventManipulator.cxx
//
//B. Rebel 10/2004

#include "AnalysisNtuples/Module/ANtpEventManipulator.h"
#include "MessageService/MsgService.h"

#include <cassert>
#include <algorithm>

ClassImp(ANtpEventManipulator)

CVSID("$Id: ANtpEventManipulator.cxx,v 1.6 2007/03/21 21:50:15 brebel Exp $");

//-------------------------------------------------------------------
ANtpEventManipulator::ANtpEventManipulator() :
  fNtpSREvent(0),
  fEventArray(0),
  fTrackArray(0),
  fShowerArray(0),
  fStripArray(0),
  fUseLengthTrack(0),
  fUsePHShower(0),
  fUseNPlanesShower(0),
  fUsePHTrack(0),
  fUseNPlanesTrack(0)
{
    
  MSG("ANtpEventManipulator", Msg::kDebug) << "ANtpEventManipulator::Constructor" << endl;
  
}

//-------------------------------------------------------------------
void ANtpEventManipulator::Initialize(TClonesArray  *eventArray, TClonesArray *trackArray, 
                                      TClonesArray *showerArray, TClonesArray *stripArray)
{
  fNtpSREvent = 0;
  fEventArray = eventArray;
  fTrackArray = trackArray;
  fShowerArray = showerArray;
  fStripArray =stripArray;
  MSG("ANtpEventManipulator", Msg::kDebug) << "ANtpEventManipulator::Constructor" << endl;    
}


//----------------------------------------------------------------------
ANtpEventManipulator::~ANtpEventManipulator()
{ 
  MSG("ANtpEventManipulator", Msg::kDebug) << "ANtpEventManipulator::Destructor" << endl; 
}

//------------------------------------------------------------------------
NtpSREvent *ANtpEventManipulator::GetEvent()
{
  return fNtpSREvent;
}

//------------------------------------------------------------------------
//get the track corresponding to entry index in the NtpSREvent trk array
NtpSRTrack *ANtpEventManipulator::GetTrack(Int_t index) 
{
  NtpSRTrack *ntpTrack = 0;
  
  if(!fNtpSREvent){
    MSG("ANtpEventManipulator", Msg::kWarning) << "No NtpSREvent to get a NtpSRTrack from"
                                               << " - returning empty pointer" << endl;
    return ntpTrack;
  } 
    
  if(index < fNtpSREvent->ntrack)
    ntpTrack = dynamic_cast<NtpSRTrack *>(fTrackArray->At(fNtpSREvent->trk[index]));
  
  return ntpTrack;
}

//------------------------------------------------------------------------
//get the shower corresponding to entry index in the NtpSREvent shw array
NtpSRShower *ANtpEventManipulator::GetShower(Int_t index) 
{
  NtpSRShower *ntpShower = 0;

  if(!fNtpSREvent){
    MSG("ANtpEventManipulator", Msg::kWarning) << "No NtpSREvent to get a NtpSRShower from"
                                               << " - returning empty pointer" << endl;
    return ntpShower;
  }
  
  if(index < fNtpSREvent->nshower)
    ntpShower = dynamic_cast<NtpSRShower *>(fShowerArray->At(fNtpSREvent->shw[index]));
  
  return ntpShower;
}

//-------------------------------------------------------------------------
NtpSRTrack *ANtpEventManipulator::GetPrimaryTrack() 
{
  NtpSRTrack *primaryTrack = 0;
  NtpSRTrack *ntpTrack = 0;

  Int_t index = 0;
  
  if(!fNtpSREvent){
    MSG("ANtpEventManipulator", Msg::kWarning) << "No NtpSREvent to get a NtpSRTrack from"
                                               << " - returning empty pointer" << endl;
    return primaryTrack;
  } 
  
  for(Int_t i = 0; i < fNtpSREvent->ntrack; ++i){
    
    MSG("ANtpEventManipulator", Msg::kDebug) << "current track = " << i 
                                             << "/" << fNtpSREvent->ntrack << endl;
    
    index = fNtpSREvent->trk[i];
    
    MSG("ANtpEventManipulator", Msg::kDebug) << "current track index = " << index 
                                             << "/" << fNtpSREvent->ntrack << endl;
    
    ntpTrack = dynamic_cast<NtpSRTrack *>(fTrackArray->At(index));
    
    //if you dont already have a candidate, make this track it
    if(!primaryTrack) primaryTrack = ntpTrack;
    
    //check to see if this track is better than the best so far.  the fUseXXX
    //variables are set to 1. or 0. depending on if they should be used to 
    //determine which is the primary track
    if(ntpTrack->ds > primaryTrack->ds*fUseLengthTrack
       && ntpTrack->plane.n > primaryTrack->plane.n*fUseNPlanesTrack
       && ntpTrack->ph.sigmap > primaryTrack->ph.sigmap*fUsePHTrack)
      primaryTrack = ntpTrack;
    
  }//end loop over tracks in the event
  
  return primaryTrack;
  
}

//------------------------------------------------------------------------------------------------------

NtpSRTrack *ANtpEventManipulator::GetPrimaryTrackNS() 
{
  NtpSRTrack *primaryTrack = 0;
  NtpSRTrack *ntpTrack = 0;

  Int_t    index  = 0;
  Double_t lenmax =-999;
  
  if(!fNtpSREvent){
    MSG("ANtpEventManipulator", Msg::kWarning) << "No NtpSREvent to get a NtpSRTrack from"
                                               << " - returning empty pointer" << endl;
    return primaryTrack;
  } 
  
  for(Int_t i = 0; i < fNtpSREvent->ntrack; ++i){
    
//     MSG("ANtpEventManipulator", Msg::kInfo) << "current track = " << i 
//                                           << "/" << fNtpSREvent->ntrack << endl;
    
    index = fNtpSREvent->trk[i];
    
//     MSG("ANtpEventManipulator", Msg::kInfo) << "current track index = " << index 
//                                           << "/" << fNtpSREvent->ntrack << endl;
    
    ntpTrack = dynamic_cast<NtpSRTrack *>(fTrackArray->At(index));
    Double_t len = ntpTrack->plane.end-ntpTrack->plane.beg+1;
        
    if(len>=lenmax) {
     lenmax=len;
     primaryTrack = ntpTrack;
    }
       
  }//end loop over tracks in the event
  
  return primaryTrack;
  
}

//--------------------------------------------------------------------------------------------------------
NtpSRShower *ANtpEventManipulator::GetPrimaryShower() 
{
  
  NtpSRShower *ntpShower = 0;
  NtpSRShower *primaryShower = 0;
  
  if(!fNtpSREvent){
    MSG("ANtpEventManipulator", Msg::kWarning) << "No NtpSREvent to get a NtpSRShower from"
                                               << " - returning empty pointer" << endl;
    return ntpShower;
  } 
  
  for(Int_t i = 0; i < fNtpSREvent->nshower; ++i){
    
    ntpShower = dynamic_cast<NtpSRShower *>(fShowerArray->At(fNtpSREvent->shw[i]));
    
    //if you dont already have a candidate, make this Shower it
    if(!primaryShower) primaryShower = ntpShower;
    
    //check to see if this Shower is better than the best so far.  the fUseXXX
    //variables are set to 1. or 0. depending on if they should be used to 
    //determine which is the primary Shower
//     if(ntpShower->plane.n > primaryShower->plane.n*fUseNPlanesShower
//        && ntpShower->ph.sigmap > primaryShower->ph.sigmap*fUsePHShower)
//       primaryShower = ntpShower;
    
  }//end loop over Showers in the event
  
  return primaryShower;
  
}

//------------------------------------------------------------------------
//get the strip corresponding to entry index in the NtpSREvent stp array
NtpSRStrip *ANtpEventManipulator::GetStrip(Int_t index) 
{
  NtpSRStrip *ntpStrip = 0;

  if(!fNtpSREvent){
    MSG("ANtpEventManipulator", Msg::kWarning) << "No NtpSREvent to get a NtpSRStrip from"
                                               << " - returning empty pointer" << endl;
    return ntpStrip;
  }
  
  if(index < fNtpSREvent->nstrip)
    ntpStrip = dynamic_cast<NtpSRStrip *>(fStripArray->At(fNtpSREvent->stp[index]));
  
  return ntpStrip;
}

//--------------------------------------------------------------------------




void ANtpEventManipulator::SetPrimaryTrackCriteria(Int_t useNPlanes, Int_t useLength,
                                                   Int_t usePH)
{
  fUseLengthTrack = useLength;
  fUseNPlanesTrack = useNPlanes;
  fUsePHTrack = usePH;
  
  return;
}

//--------------------------------------------------------------------------
void ANtpEventManipulator::SetPrimaryShowerCriteria(Int_t useNPlanes, Int_t usePH)
{
  fUseNPlanesShower = useNPlanes;
  fUsePHShower = usePH;
  
  return;
}

//--------------------------------------------------------------------------
void ANtpEventManipulator::SetEventInSnarl(Int_t eventIndex)
{
  //check that such an event exists
  if(eventIndex < fEventArray->GetLast()+1) 
    fNtpSREvent = dynamic_cast<NtpSREvent *>(fEventArray->At(eventIndex));
  else{
    MSG("ANtpEventManipulator", Msg::kWarning) << "invalid event index, "
                                               << eventIndex << "/" 
                                               << fEventArray->GetLast()+1 
                                               << " set to first event in array" 
                                               << endl;
    fNtpSREvent = dynamic_cast<NtpSREvent *>(fEventArray->At(0));
  }

  return;
}

---