StripListKG.cxx

Go to the documentation of this file.

// Cluster3D package
//
// StripListKG.cxx
//
// StripListKG - list of strips
//
// Author:  Katarzyna Grzelak K.Grzelak1@physics.ox.ac.uk
//
#include <algorithm>
#include <cstdlib>

#include "Cluster3D/StripListKG.h"
#include "Algorithm/AlgConfig.h"
#include "MessageService/MsgService.h" // needed for CVS
#include "Cluster3D/StripKG.h"
#include "Cluster3D/PlaneKG.h"
#include "RecoBase/CandStripHandle.h"     
#include "RecoBase/CandSliceHandle.h"     
#include "UgliGeometry/UgliGeomHandle.h"   // UgliGeom handling

ClassImp(StripListKG)

CVSID("$Id: StripListKG.cxx,v 1.4 2009/02/28 21:46:11 gmieg Exp $");

//............................................................................

Bool_t StripSortCriterion (const StripKG &s1, const StripKG &s2){
//
//  Purpose:  Rules how to sort STL vector of strips  
//
//  Arguments: 
//            in   s1 - first strip
//            in   s2 - second strip
//
//  Return:  boolean variable
//
//  Contact:   K.Grzelak1@physics.ox.ac.uk
//
//
//  Program Notes:-
//  =============
//
// Strip is "less" than another strip if number of plane is smaller
// 
  
      if(s1.plane == s2.plane){
         return (s1.strip < s2.strip);
      }
      else{
         return (s1.plane < s2.plane); 
      } 
} 

//............................................................................

StripListKG::StripListKG():energyRaw(0.),energyPe(0.),energyGeV(0.){
//
//  Purpose:  Default constructor
//
}

//............................................................................

void StripListKG::GetStrips(vector<StripKG> &allstp) const{
//
//  Purpose:  Return STL vector of strips
//
//  Arguments: 
//
//  Return:  n/a
//
//  Contact:   K.Grzelak1@physics.ox.ac.uk
//
//
//  Program Notes:-
//  =============
//

    allstp=allstps;
}

//............................................................................

UInt_t StripListKG::GetNStrips() const{
//
//  Purpose:  Return number of strips
//
//  Arguments: 
//              
//  Return:  
//
//  Contact:   K.Grzelak1@physics.ox.ac.uk
//
//
//  Program Notes:-
//  =============
//

  return allstps.size();
}

//............................................................................

void StripListKG::SetEnergy(){
//
//  Purpose:  Sets energy
//
//  Arguments: 
//              
//  Return:  n/a
//
//  Contact:   K.Grzelak1@physics.ox.ac.uk
//
//
//  Program Notes:-
//  =============
//
  for(UInt_t i=0; i<allstps.size(); ++i){
    energyPe=energyPe+allstps[i].peE+allstps[i].peW;
    energyRaw=energyRaw+allstps[i].rawE+allstps[i].rawW; 
  } 
}

//............................................................................

Double_t StripListKG::GetEnergy(Int_t whatKind) const{
//
//  Purpose:  Return energy
//
//  Arguments: 
//              
//  Return:  
//
//  Contact:   K.Grzelak1@physics.ox.ac.uk
//
//
//  Program Notes:-
//  =============
//

  if(whatKind==0){
    return energyRaw;
  }
  else if(whatKind==1){
    return energyPe;
  }
  else if(whatKind==2){
    return energyGeV;
  }
  else{
    return energyPe;
  }

  return energyPe;
}

//............................................................................

Bool_t StripListKG::OppViewNeighbours(Int_t planeNbInp1, Int_t planeNbInp2, Int_t cdist){
//
//  Purpose:  Return 
//
//  Arguments: 
//              
//  Return:  n/a
//
//  Contact:   K.Grzelak1@physics.ox.ac.uk
//
//
//  Program Notes:-
//  =============
//

  Int_t dist = abs(planeNbInp1-planeNbInp2);

  if(dist%2!=0 && dist <= cdist){
    return true;
  }
  else{
    return false;
  }
}

//............................................................................

Int_t StripListKG::NbStpInPlane(Int_t planeNbInp, const std::vector<PlaneKG> &p){
//
//  Purpose:  Return 
//
//  Arguments: 
//              
//  Return:  
//
//  Contact:   K.Grzelak1@physics.ox.ac.uk
//
//
//  Program Notes:-
//  =============
//

  Int_t flag=0;
  Int_t ile=0;
  for(UInt_t j=0; j<p.size(); ++j){
    if(planeNbInp==p[j].number){
      flag=1;
      ile = p[j].nstppplane;
      break;
    }
    else{
      continue;
    }
  }
  if(flag==1){
    return ile;
  }
  else{
    cout << "NbStpInPlane : Strip belongs to an unknown plane " << endl;
    return 0;
  }
}

//............................................................................

void StripListKG::CreateStrips(AlgConfig &ac, const CandSliceHandle* slice, UgliGeomHandle *ugh){
//
//  Purpose:  Create strips
//
//  Arguments: 
//              
//  Return:  
//
//  Contact:   K.Grzelak1@physics.ox.ac.uk
//
//
//  Program Notes:-
//  =============
//

  const Int_t xtalkmod = ac.GetInt("xTalkMode");


  CandStripHandleItr stripItr(slice->GetDaughterIterator());
  //while (CandStripHandle *strip = stripItr()) {
  Int_t nstrips = slice->GetNStrip();

  MSG("Cluster3D", Msg::kDebug)
                        << "Create strip list"
                        << "nstrips = " << nstrips 
                        << endl;

  Int_t i=0;
  StripKG s;
  
  while (stripItr.IsValid()) {

     CandStripHandle *strip = stripItr.Ptr();

     s.Set(strip,ugh,i);    
     allstps.push_back(s); 

     ++i;  
     stripItr.Next();  
  }
  stable_sort(allstps.begin(),allstps.end(),StripSortCriterion);

  for(UInt_t ii=0;ii<allstps.size();++ii){

    if(xtalkmod==1){
        allstps[i].XTalk(ac);
    }
    else{
      if(FirstInPlane(ii) && LastInPlane(ii)){
         allstps[ii].XTalk(ac);
      }
      else if(FirstInPlane(ii)){
         Int_t dist1 = allstps[ii+1].strip-allstps[ii].strip;
         if(dist1>5){
            allstps[ii].XTalk(ac);
         }         
      }
      else if(LastInPlane(ii)){
         Int_t dist2 = allstps[ii].strip-allstps[ii-1].strip;
         if(dist2>5){
            allstps[ii].XTalk(ac);
         }
      }
      else{
         Int_t dist1 = allstps[ii+1].strip-allstps[ii].strip;
         Int_t dist2 = allstps[ii].strip-allstps[ii-1].strip;
         if(dist1>5 && dist2>5){
            allstps[ii].XTalk(ac);
         }
      } 
      MSG("Cluster3D", Msg::kDebug)
                        << " allstps[ii].plane= "
                        << allstps[ii].plane 
                        << " allstps[ii].strip= "
                        << allstps[ii].strip 
                        << " allstps[ii].xtalk= "
                        << allstps[ii].xtalk 
                        << endl;    
    }
  }
}

//............................................................................

CandStripHandle* StripListKG::GetStripHandle(UInt_t index) const{
//
//  Purpose:  Return handle to a strip  
//
//  Arguments:
//           index        in  -  index of a strip in STL vector     
//              
//  Return: CandStripHandle* handle
//
//  Contact:   K.Grzelak1@physics.ox.ac.uk
//
//
//  Program Notes:-
//  =============
//

  return allstps[index].handle;
}

//............................................................................

Int_t StripListKG::GetStripIndex(UInt_t index) const{
//
//  Purpose:  Return strip index  
//
//  Arguments:
//           index        in  -  index of a strip in STL vector     
//              
//  Return: Strip number 
//
//  Contact:   K.Grzelak1@physics.ox.ac.uk
//
//
//  Program Notes:-
//  =============
//

  return allstps[index].index;
}

//............................................................................

Int_t StripListKG::GetStripStrip(UInt_t index) const{
//
//  Purpose:  Return strip number  
//
//  Arguments:
//           index        in  -  index of a strip in STL vector     
//              
//  Return: Strip number 
//
//  Contact:   K.Grzelak1@physics.ox.ac.uk
//
//
//  Program Notes:-
//  =============
//

  return allstps[index].strip;
}

//............................................................................

Int_t StripListKG::GetStripPlane(UInt_t index) const{
//
//  Purpose:  Return plane number of a strip 
//
//  Arguments:
//           index        in  -  index of a strip in STL vector     
//              
//  Return: Plane number 
//
//  Contact:   K.Grzelak1@physics.ox.ac.uk
//
//
//  Program Notes:-
//  =============
//

  return allstps[index].plane;
}

//............................................................................

Float_t StripListKG::GetStripTPos(UInt_t index) const{
//
//  Purpose:  Return transverse position of a strip 
//
//  Arguments:
//           index        in  -  index of a strip in STL vector     
//              
//  Return: Transverse position  
//
//  Contact:   K.Grzelak1@physics.ox.ac.uk
//
//
//  Program Notes:-
//  =============
//

  return allstps[index].tpos;
}

//............................................................................

Float_t StripListKG::GetStripZPos(UInt_t index) const{
//
//  Purpose:  Return z position of a strip 
//
//  Arguments:
//           index        in  -  index of a strip in STL vector     
//              
//  Return: Z position  
//
//  Contact:   K.Grzelak1@physics.ox.ac.uk
//
//
//  Program Notes:-
//  =============
//

  return allstps[index].zpos;
}

//............................................................................

Int_t StripListKG::GetStripView(UInt_t index) const{
//
//  Purpose:  Return plane view 
//
//  Arguments:
//           index        in  -  index of a strip in STL vector     
//              
//  Return: Plane view  
//
//  Contact:   K.Grzelak1@physics.ox.ac.uk
//
//
//  Program Notes:-
//  =============
//

  return allstps[index].planeview;
}

//............................................................................

Double_t StripListKG::GetStripRawE(UInt_t index) const{
//
//  Purpose:  Return energy in ADC 
//
//  Arguments:
//           index        in  -  index of a strip in STL vector     
//              
//  Return: Energy in ADC for E strip end
//
//  Contact:   K.Grzelak1@physics.ox.ac.uk
//
//
//  Program Notes:-
//  =============
//

  return allstps[index].rawE;
}

//............................................................................

Double_t StripListKG::GetStripRawW(UInt_t index) const{
//
//  Purpose:  Return energy in ADC 
//
//  Arguments:
//           index        in  -  index of a strip in STL vector     
//              
//  Return: Energy in ADC for W strip end
//
//  Contact:   K.Grzelak1@physics.ox.ac.uk
//
//
//  Program Notes:-
//  =============
//

  return allstps[index].rawW;
}

//............................................................................

Double_t StripListKG::GetStripPeE(UInt_t index) const{
//
//  Purpose:  Return energy in pe 
//
//  Arguments:
//           index        in  -  index of a strip in STL vector     
//              
//  Return: Energy in pe for E end
//
//  Contact:   K.Grzelak1@physics.ox.ac.uk
//
//
//  Program Notes:-
//  =============
//

  return allstps[index].peE;
}

//............................................................................

Double_t StripListKG::GetStripPeW(UInt_t index) const{
//
//  Purpose:  Return energy in pe 
//
//  Arguments:
//           index        in  -  index of a strip in STL vector     
//              
//  Return: Energy in pe for W end
//
//  Contact:   K.Grzelak1@physics.ox.ac.uk
//
//
//  Program Notes:-
//  =============
//

  return allstps[index].peW;
}

//............................................................................

Int_t StripListKG::GetStripXTalk(UInt_t index) const{
//
//  Purpose:  Return crosstalk flag for a strip 
//
//  Arguments:
//           index        in  -  index of a strip in STL vector     
//              
//  Return:  XTalk flag 
//
//  Contact:   K.Grzelak1@physics.ox.ac.uk
//
//
//  Program Notes:-
//  =============
//

  return allstps[index].xtalk;
}

//............................................................................

void StripListKG::GetTimePos(UInt_t index, AlgConfig &ac, Bool_t oneStp, Float_t &dt, Float_t &timpos, Float_t &halflength){
//
//  Purpose:  Return hit position from timing information for a strip 
//
//  Arguments:
//           ac           in  -  AlgConfig 
//           index        in  -  index of a strip in STL vector     
//           oneStp       in  -  true = one hit per plane => use timing info
//                               false = more than one hit => do not use timing
//              
//  Return:  
//
//  Contact:   K.Grzelak1@physics.ox.ac.uk
//
//
//  Program Notes:-
//  =============
//

  allstps[index].TimePos(ac, oneStp, dt, timpos, halflength);  
}

//............................................................................

Bool_t StripListKG::FirstInPlane(UInt_t index){
//
//  Purpose:  Is this strip the first in plane ?
//
//  Arguments:
//           index        in  - position in vector allstps     
//              
//  Return:  
//
//  Contact:   K.Grzelak1@physics.ox.ac.uk
//
//
//  Program Notes:-
//  =============
//
  if(index==0){
    return true;
  }
  else{
    if(allstps[index-1].plane != allstps[index].plane){
      return true;  
    }
    else{
      return false;
    }
  }
}

//............................................................................

Bool_t StripListKG::LastInPlane(UInt_t index){
//
//  Purpose:  Is this strip the last in plane ?
//
//  Arguments:
//           index        in  - position in vector allstps     
//              
//  Return:  
//
//  Contact:   K.Grzelak1@physics.ox.ac.uk
//
//
//  Program Notes:-
//  =============
//
  if(index == allstps.size()-1){
     return true; 
  }
  else{
    if(allstps[index].plane != allstps[index+1].plane){
      return true;  
    }
    else{
      return false;
    }
  }
}

//............................................................................

void StripListKG::SetStripAmbiguity(UInt_t index){
//
//  Purpose:  to how many clusters strip belongs
//
//  Arguments: 
//            index    in  - position in vector allstps     
//              
//  Return:  n/a
//
//  Contact:   K.Grzelak1@physics.ox.ac.uk
//
//
//  Program Notes:-
//  =============
//

  allstps[index].ambig=allstps[index].ambig+1;  

}

//............................................................................

Int_t StripListKG::GetStripAmbiguity(UInt_t index) const{
//
//  Purpose:  to how many clusters strip belongs
//
//  Arguments:             
//           index        in  -  index of a strip in STL vector     
//              
//  Return:  
//
//  Contact:   K.Grzelak1@physics.ox.ac.uk
//
//
//  Program Notes:-
//  =============
//

  return allstps[index].ambig;  

}

---