StBTofUtil/StV0TofCorrection.h

Go to the documentation of this file.

// $Id: StV0TofCorrection.h,v 1.1 2011/01/24 21:01:49 geurts Exp $
// ---
// $Log: StV0TofCorrection.h,v $
// Revision 1.1  2011/01/24 21:01:49  geurts
// *** empty log message ***
//
// ---
#ifndef STV0TOFCORRECTION_H
#define STV0TOFCORRECTION_H

#include <stdio.h>
#include <stdarg.h>

#include <TROOT.h>
#include "TMath.h"
#include "StPhysicalHelixD.hh"
#include "StarClassLibrary/StLorentzVectorD.hh"
#include "StarClassLibrary/StThreeVectorD.hh"
#include <iostream> 
#include "TVirtualStreamerInfo.h"

#include <vector>


template < class T, class container = std::vector<T> > class StInlineContainer : public container {
 private:
  Int_t NrArgs; 
  StInlineContainer<T>* pContainer; 
  
 public:
  StInlineContainer() { NrArgs = 0; pContainer = NULL; }
  virtual ~StInlineContainer() {}
  explicit StInlineContainer(const T vec) : container(1,vec) { } 

  StInlineContainer<T> operator()(const T vec) {
    NrArgs++;
    pContainer->setNrArgs(NrArgs);
    pContainer->push_back(vec);
    return *pContainer;
  }
  void  setNrArgs(Int_t c) { NrArgs = c; }
  Int_t getNrArgs()        { return NrArgs; }
  void  setPointer2Container(StInlineContainer<T>* p) { pContainer = p; }
};


class StV0TofCorrection {
 private:
  inline Float_t calcTof(StLorentzVectorD trackAB, Float_t PathAB); 
  Float_t calcPathCorr(StPhysicalHelixD helixA, StThreeVectorD vectorAB, StThreeVectorD vectorTof) ;
  Float_t calcPathOnHelixToPoint(StPhysicalHelixD helixA, StThreeVectorD space_vec) ; 
  inline Bool_t inputOk();
  inline Bool_t cutOnMass2(Float_t Mass2, Int_t pidnr);
  void setNrDecays(Int_t n) { NrDecays = n; } 

  Float_t clight; 
  Bool_t cleared, cleared2; 

  Int_t NrDecays; 
  StInlineContainer<StThreeVectorD>* Vectors3D; 
  StInlineContainer<StLorentzVectorD>* tracks; 

 public:
  StV0TofCorrection() ; 
  virtual ~StV0TofCorrection() ; 

  StInlineContainer<StThreeVectorD> setVectors3D(const StThreeVectorD vec) {
    if ( cleared ) {
      Vectors3D = new StInlineContainer<StThreeVectorD>;
    }
    else {
      Error("StV0TofCorrection::setVectors3D","Make sure to call clearContainers() function every time a correction of a particle is done! Otherwise you're using the same container over and over again!");
    }
    Vectors3D->setNrArgs(0);
    Vectors3D->setPointer2Container(Vectors3D);
    cleared = kFALSE;
    return (*Vectors3D)(vec);
  }

  StInlineContainer<StLorentzVectorD> setMotherTracks(const StLorentzVectorD tr) {
    if ( cleared2 ) {
      tracks  = new StInlineContainer<StLorentzVectorD>;
      tracks->setNrArgs(0);
      tracks->setPointer2Container(tracks);
      cleared2 = kFALSE;
      return (*tracks)(tr);
    }
    else {
      Error("StV0TofCorrection::setMotherTracks","Again: Don't forget clearContainers()!");
      return (*tracks);
    }
  }

  Bool_t correctBeta(StPhysicalHelixD helixA, Float_t TofA, Float_t& BetaCorr, Float_t MomentumA = -999., Int_t pidnr = -1) ;
 
  void clearContainers() {
    delete Vectors3D; delete tracks;
    Vectors3D = NULL; tracks = NULL;
    cleared = kTRUE; cleared2 = kTRUE;
  }

  ClassDef(StV0TofCorrection,1)

};


#endif

---