StRoot  1
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Groups Pages
StMuMomentumShiftMaker.h
1 /***************************************************************************
2  *
3  * $Id: StMuMomentumShiftMaker.h,v 1.2 2011/04/08 01:25:51 fisyak Exp $
4  * Author: Marco van Leeuwen, LBNL
5  ***************************************************************************/
6 #ifndef __StMuMomentumShiftmaker_hh__
7 #define __StMuMomentumShiftmaker_hh__
8 
9 //#include "StMuDstMaker.h"
10 #include "StMaker.h"
11 #include "TString.h"
12 
13 class TFile;
14 class TTree;
15 class StMuTrack;
16 #ifndef __NO_STRANGE_MUDST__
17 class StKinkMuDst;
18 class StV0MuDst;
19 class StXiMuDst;
20 #endif
22 
23 public:
25  StMuMomentumShiftMaker(const char *outDir);
27 
28  void setScaleFactor(Float_t scale) { mMomentumScale=scale; }
29  void setWriteMuDst(Bool_t writeFlag) { mWriteMuDst=writeFlag; }
30  void ScaleMomentum(StMuTrack *track);
31 #ifndef __NO_STRANGE_MUDST__
32  void ScaleMomentum(StKinkMuDst *kink);
33  void ScaleMomentum(StV0MuDst *kink);
34  void ScaleMomentum(StXiMuDst *kink);
35 #endif
36  int Make();
37  int Finish();
38  private:
39  Float_t mMomentumScale;
40  Bool_t mWriteMuDst;
41  TString mOutDir;
42  TFile *mOutFile;
43  TTree *mOutTree;
44 
45  ClassDef(StMuMomentumShiftMaker,0)
46 };
47 
48 #endif