StRoot: StHbtMaker/Reader/StHbtStrangeMuDstEventReader.cxx Source File

00001 /***************************************************************************
00002  *
00003  * $Id: StHbtStrangeMuDstEventReader.cxx,v 1.2 2001/06/21 19:18:42 laue Exp $
00004  *
00005  * Author: Frank Laue, Ohio State, laue@mps.ohio-state.edu
00006  ***************************************************************************
00007  *
00008  * Description: part of STAR HBT Framework: StHbtMaker package
00009  * This is the HbtEventReader class to be used when reading the
00010  * StStrangeMuDsts produced by the STAR Strangeness group
00011  *
00012  ***************************************************************************
00013  *
00014  * $Log: StHbtStrangeMuDstEventReader.cxx,v $
00015  * Revision 1.2  2001/06/21 19:18:42  laue
00016  * Modified Files: (to match the changed base classes)
00017  *      StHbtAsciiReader.cxx StHbtAsciiReader.h
00018  *      StHbtAssociationReader.cxx StHbtAssociationReader.h
00019  *      StHbtBinaryReader.cxx StHbtBinaryReader.h
00020  *      StHbtGstarTxtReader.cxx StHbtGstarTxtReader.h
00021  *      StHbtStrangeMuDstEventReader.cxx
00022  *      StHbtStrangeMuDstEventReader.h StStandardHbtEventReader.cxx
00023  * Added Files: new reader
00024  *      StHbtTTreeReader.cxx StHbtTTreeReader.h
00025  *
00026  * Revision 1.1  2000/12/13 20:45:00  laue
00027  * New reader to read directly from the StStrangeMuDstMaker's V0 files
00028  *
00029  *
00030  **************************************************************************/
00031 #include "StHbtMaker/Reader/StHbtStrangeMuDstEventReader.h"
00032 
00033 #include "StChain.h"
00034 
00035 #include <math.h>
00036 
00037 
00038 #include "StHbtMaker/Reader/StHbtGstarTxtReader.h"
00039 #include "StHbtMaker/Base/StHbtEventCut.h"
00040 #include "StHbtMaker/Base/StHbtTrackCut.h"
00041 #include "StHbtMaker/Base/StHbtV0Cut.h"
00042 #include "StHbtMaker/Base/StHbtKinkCut.h"
00043 
00044 #include "SystemOfUnits.h"   // has "tesla" in it
00045 #include "StHbtMaker/Infrastructure/StHbtV0Collection.hh"
00046 #include "StStrangeMuDstMaker/StStrangeMuDstMaker.h"  
00047 #include "StStrangeMuDstMaker/StStrangeEvMuDst.hh"
00048 #include "StStrangeMuDstMaker/StV0MuDst.hh"
00049  
00050 #ifdef __ROOT__
00051 ClassImp(StHbtStrangeMuDstEventReader)
00052 #endif
00053 
00054 #if !(ST_NO_NAMESPACES)
00055   using namespace units;
00056 #endif
00057 
00058 
00059 //__________________
00060 StHbtStrangeMuDstEventReader::StHbtStrangeMuDstEventReader(){
00061   mReaderStatus = 0;  // "good"
00062   mStrangeMuDstMaker =0;
00063 }
00064 //__________________
00065 StHbtStrangeMuDstEventReader::StHbtStrangeMuDstEventReader(StStrangeMuDstMaker* maker) : mStrangeMuDstMaker(maker) {
00066   mReaderStatus = 0;  // "good"
00067 }
00068 //__________________
00069 StHbtStrangeMuDstEventReader::~StHbtStrangeMuDstEventReader(){
00070   if (mEventCut) delete mEventCut;
00071   if (mV0Cut) delete mV0Cut;
00072 }
00073 //__________________
00074 StHbtString StHbtStrangeMuDstEventReader::Report(){
00075   StHbtString temp = "\n This is the StHbtStrangeMuDstEventReader\n";
00076   temp += "---> EventCuts in Reader: ";
00077   if (mEventCut) {
00078     temp += mEventCut->Report();
00079   }
00080   else {
00081     temp += "NONE";
00082   }
00083   temp += "\n---> V0Cuts in Reader: ";
00084   if (mV0Cut) {
00085     temp += mV0Cut->Report();
00086   }
00087   else {
00088     temp += "NONE";
00089   }
00090   temp += "\n";
00091   return temp;
00092 }
00093 //__________________
00094 StHbtEvent* StHbtStrangeMuDstEventReader::ReturnHbtEvent(){
00095 #ifdef STHBTDEBUG
00096     cout << " StHbtStrangeMuDstEventReader::ReturnHbtEvent()" << endl;
00097 #endif
00098   StStrangeEvMuDst* strangeEvMuDst = mStrangeMuDstMaker->GetEvent();
00099   if (!strangeEvMuDst){
00100     cout << " StHbtStrangeMuDstEventReader::ReturnHbtEvent() - No StrangeEvMuDst !!! " << endl;
00101     return 0;
00102   }
00103 
00104   StHbtEvent* hbtEvent = new StHbtEvent;
00105   StHbtThreeVector vp = StHbtThreeVector(strangeEvMuDst->primaryVertexX(),strangeEvMuDst->primaryVertexY(),strangeEvMuDst->primaryVertexZ());
00106   cout << " StHbtStrangeMuDstEventReader::ReturnHbtEvent() - primary vertex : " << vp << endl;
00107   hbtEvent->SetPrimVertPos(vp);
00108   hbtEvent->SetNumberOfTracks(strangeEvMuDst->primaryTracks());
00109   hbtEvent->SetEventNumber(strangeEvMuDst->event());
00110 
00111   // By now, all event-wise information has been extracted and stored in hbtEvent
00112   // see if it passes any front-loaded event cut
00113   if (mEventCut){
00114     if (!(mEventCut->Pass(hbtEvent))){    // event failed! - return null pointer (but leave Reader status flag as "good")
00115       delete hbtEvent;
00116       return 0;
00117     }
00118   }
00119 
00120   for( int i= 0; i < mStrangeMuDstMaker->GetNV0(); i++){
00121     StV0MuDst* v0FromMuDst = mStrangeMuDstMaker->GetV0(i);
00122     //v0FromMuDst->UpdateV0();
00123     StHbtV0* hbtV0 = new StHbtV0;
00124     hbtV0->SetdecayLengthV0(v0FromMuDst->decayLengthV0());
00125     hbtV0->SetdecayVertexV0X(v0FromMuDst->decayVertexV0X());
00126     hbtV0->SetdecayVertexV0Y(v0FromMuDst->decayVertexV0Y());
00127     hbtV0->SetdecayVertexV0Z(v0FromMuDst->decayVertexV0Z());
00128     hbtV0->SetdcaV0Daughters(v0FromMuDst->dcaV0Daughters());
00129     hbtV0->SetdcaV0ToPrimVertex(v0FromMuDst->dcaV0ToPrimVertex());
00130     hbtV0->SetdcaPosToPrimVertex(v0FromMuDst->dcaPosToPrimVertex());
00131     hbtV0->SetdcaNegToPrimVertex(v0FromMuDst->dcaNegToPrimVertex());
00132     hbtV0->SetmomPosX(v0FromMuDst->momPosX());
00133     hbtV0->SetmomPosY(v0FromMuDst->momPosY());
00134     hbtV0->SetmomPosZ(v0FromMuDst->momPosZ());
00135     hbtV0->SetmomNegX(v0FromMuDst->momNegX());
00136     hbtV0->SetmomNegY(v0FromMuDst->momNegY());
00137     hbtV0->SetmomNegZ(v0FromMuDst->momNegZ());
00138 #ifdef STHBTDEBUG
00139     cout << " hist pos ";
00140     cout << v0FromMuDst->topologyMapPos().numberOfHits(kTpcId); 
00141     cout << " hist neg ";
00142     cout << v0FromMuDst->topologyMapNeg().numberOfHits(kTpcId) << endl;
00143 #endif
00144     hbtV0->SettpcHitsPos(v0FromMuDst->topologyMapPos().numberOfHits(kTpcId));
00145     hbtV0->SettpcHitsNeg(v0FromMuDst->topologyMapNeg().numberOfHits(kTpcId));
00146     hbtV0->SetTrackTopologyMapPos(0,v0FromMuDst->topologyMapPos().data(0));
00147     hbtV0->SetTrackTopologyMapPos(1,v0FromMuDst->topologyMapPos().data(1));
00148     hbtV0->SetTrackTopologyMapNeg(0,v0FromMuDst->topologyMapNeg().data(0));
00149     hbtV0->SetTrackTopologyMapNeg(1,v0FromMuDst->topologyMapNeg().data(1));
00150     hbtV0->SetkeyPos(v0FromMuDst->keyPos());
00151     hbtV0->SetkeyNeg(v0FromMuDst->keyNeg());
00152 #ifdef STHBTDEBUG
00153     cout << " keyPos " << v0FromMuDst->keyPos() << endl;
00154     cout << " keyNeg " << v0FromMuDst->keyNeg() << endl;
00155 #endif
00156     hbtV0->SetmomV0X(v0FromMuDst->momV0X());
00157     hbtV0->SetmomV0Y(v0FromMuDst->momV0Y());
00158     hbtV0->SetmomV0Z(v0FromMuDst->momV0Z());
00159 #ifdef STHBTDEBUG
00160     cout << " alpha  ";
00161     cout << v0FromMuDst->alphaV0();
00162     cout << " ptArm  ";
00163     cout << v0FromMuDst->ptArmV0() << endl;
00164 #endif
00165     hbtV0->SetalphaV0(v0FromMuDst->alphaV0());
00166     hbtV0->SetptArmV0(v0FromMuDst->ptArmV0());
00167     hbtV0->SeteLambda(v0FromMuDst->eLambda());
00168     hbtV0->SeteK0Short(v0FromMuDst->eK0Short());
00169     hbtV0->SetePosProton(v0FromMuDst->ePosProton());
00170     hbtV0->SetePosPion(v0FromMuDst->ePosPion());
00171     hbtV0->SeteNegPion(v0FromMuDst->eNegPion());
00172     hbtV0->SeteNegProton(v0FromMuDst->eNegProton());
00173     hbtV0->SetmassLambda(v0FromMuDst->massLambda());
00174     hbtV0->SetmassAntiLambda(v0FromMuDst->massAntiLambda());
00175     hbtV0->SetmassK0Short(v0FromMuDst->massK0Short());
00176     hbtV0->SetrapLambda(v0FromMuDst->rapLambda());
00177     hbtV0->SetrapK0Short(v0FromMuDst->rapK0Short());
00178     hbtV0->SetcTauLambda(v0FromMuDst->cTauLambda());
00179     hbtV0->SetcTauK0Short(v0FromMuDst->cTauK0Short());
00180     hbtV0->SetptV0(v0FromMuDst->ptV0());
00181     hbtV0->SetptotV0(v0FromMuDst->ptotV0());
00182     hbtV0->SetptPos(v0FromMuDst->ptPos());
00183     hbtV0->SetptotPos(v0FromMuDst->ptotPos());
00184     hbtV0->SetptNeg(v0FromMuDst->ptNeg());
00185     hbtV0->SetptotNeg(v0FromMuDst->ptotNeg());
00186     hbtV0->SetdedxPos(v0FromMuDst->dedxPos());
00187     hbtV0->SetdedxNeg(v0FromMuDst->dedxNeg());
00188 
00189     
00190     // By now, all track-wise information has been extracted and stored in hbtTrack
00191     // see if it passes any front-loaded event cut
00192     if (mV0Cut){
00193       if (!(mV0Cut->Pass(hbtV0))){                  // track failed - delete it and skip the push_back
00194         delete hbtV0;
00195         continue;
00196       }
00197     }
00198     
00199     
00200     hbtEvent->V0Collection()->push_back(hbtV0);
00201   } // end of loop over strangeness groups v0's
00202   //Store total number of v0s in v0Mudst so can start from there next time
00203 #ifdef STHBTDEBUG
00204   cout << " StHbtStrangeMuDstEventReader::ReturnHbtEvent() - " << hbtEvent->V0Collection()->size();
00205   cout << " V0s pushed in collection " << endl;
00206 #endif
00207   printf(" StHbtStrangeMuDstEventReader::ReturnHbtEvent() - %8i(%i) V0s pushed into collection \n",
00208          hbtEvent->V0Collection()->size(),
00209          mStrangeMuDstMaker->GetNV0());
00210 
00211   // There might be event cuts that modify the collections of Tracks or V0 in the event.
00212   // These cuts have to be done after the event is built. That's why we have the event cut
00213   // at this point for the second time.
00214   // An example of this kind of cuts will be an cut that removes spit tracks from the event.
00215   if (mEventCut){
00216     if (!(mEventCut->Pass(hbtEvent))){    // event failed! - return null pointer (but leave Reader status flag as "good")
00217       delete hbtEvent;
00218       return 0;
00219     }
00220   }
00221   
00222   return hbtEvent;
00223 }
00224 
00225 
00226 
00227 
00228 
00229 
00230 
00231 
00232 
00233 
00234 
00235 
00236 
00237 
00238 
00239 
00240 
00241 
00242 
00243 
00244