00001
00002
00003
00004
00005
00006
00007
00008
00009
00010
00011
00012
00013
00014
00015
00016
00017
00018
00019
00020
00021
00022
00023
00024
00025
00026
00027
00028
00029
00030
00031
00032
00033
00034
00035
00036
00037
00038
00039
00040
00041
00042
00043
00044
00045
00047
00048
00049
00051 #include "TTree.h"
00052 #include "StEvent/StEvent.h"
00053 #include "StMcEventMaker/StMcEventMaker.h"
00054 #include "StAssociationMaker/StAssociationMaker.h"
00055 #include "StAssociationMaker/StTrackPairInfo.hh"
00056 #include "StTrack.h"
00057 #include "StGlobalTrack.h"
00058 #include "StV0Vertex.h"
00059 #include "StV0MuDst.hh"
00060 #include "StV0Mc.hh"
00061 #include "StMcEventTypes.hh"
00062 #include "StParticleDefinition.hh"
00063 #include "StTrackDetectorInfo.h"
00064
00065 #include "StStrangeControllerInclude.h"
00066
00067 class StStrangeEvMuDst;
00068
00069
00070 StV0Controller::StV0Controller() : StStrangeControllerBase(v0T) {
00071 }
00072
00073 StV0Controller::~StV0Controller() {
00074 }
00075
00076 Int_t StV0Controller::MakeReadDst() {
00077
00078 Int_t j;
00079 StStrangeEvMuDst* ev = masterMaker->GetEvent();
00080 entries = GetN();
00081 for (j = 0; j<entries; j++) {
00082 StV0MuDst* v0 = (StV0MuDst*) (*dataArray)[j];
00083 v0->SetEvent(ev);
00084 }
00085 PrintNumCand("read",entries);
00086 nEntries += entries;
00087
00088 if (doMc) {
00089 ev = masterMaker->GetMcEvent();
00090 Int_t mc_entries = GetNMc();
00091 for (j = 0; j<mc_entries; j++) {
00092 StV0Mc* mc_v0 = (StV0Mc*) (*mcArray)[j];
00093 mc_v0->SetEvent(ev);
00094 }
00095 }
00096
00097 return kStOK;
00098 }
00099
00100 Int_t StV0Controller::MakeCreateDst(StEvent& event) {
00101
00102
00103 StSPtrVecV0Vertex& v0Vertices = event.v0Vertices();
00104 entries = v0Vertices.size();
00105 Int_t asize = dataArray->GetSize();
00106 if (entries > asize) dataArray->Expand(entries+increment);
00107 StStrangeEvMuDst* ev = masterMaker->GetEvent();
00108 Int_t j=0;
00109 for (Int_t i=0; i<entries; i++) {
00110 StV0Vertex* v0Vertex = v0Vertices[i];
00111 if ((v0Vertex) && (v0Vertex->dcaParentToPrimaryVertex() >= 0))
00112 new((*dataArray)[j++]) StV0MuDst(v0Vertex,ev);
00113 }
00114 entries = j;
00115 PrintNumCand("found",entries);
00116 nEntries += entries;
00117
00118 return kStOK;
00119 }
00120
00121 Int_t StV0Controller::MakeCreateMcDst(StMcVertex* mcVert) {
00122
00123 mcV0MapType* theMcV0Map = 0;
00124 mcTrackMapType* theMcTrackMap = 0;
00125 if (assocMaker) {
00126 theMcV0Map = assocMaker->mcV0Map();
00127 theMcTrackMap = assocMaker->mcTrackMap();
00128 }
00129 if (!((assocMaker)&&(theMcV0Map)&&(theMcTrackMap))) return kStOk;
00130 StStrangeEvMuDst* ev = masterMaker->GetMcEvent();
00131 StMcTrack *Pos = 0;
00132 StMcTrack *Neg = 0;
00133 const StV0Vertex* rcV0Partner = 0;
00134 Int_t indexRecoArray = -1;
00135 Int_t count = theMcV0Map->count(mcVert);
00136
00137 if (count>0) {
00138 pair<mcV0MapIter,mcV0MapIter> mcV0Bounds = theMcV0Map->equal_range(mcVert);
00139 rcV0Partner = (*mcV0Bounds.first).second;
00140 float x, y, z, delta, xd, yd, zd;
00141 x = mcVert->position().x();
00142 y = mcVert->position().y();
00143 z = mcVert->position().z();
00144 xd = x - rcV0Partner->position().x();
00145 yd = y - rcV0Partner->position().y();
00146 zd = z - rcV0Partner->position().z();
00147 delta = xd*xd + yd*yd + zd*zd;
00148
00149
00150 for(mcV0MapIter mcV0MapIt = mcV0Bounds.first;
00151 mcV0MapIt != mcV0Bounds.second; ++mcV0MapIt) {
00152 const StV0Vertex *temp = (*mcV0MapIt).second;
00153 if (temp != rcV0Partner) {
00154 xd = x - temp->position().x();
00155 yd = y - temp->position().y();
00156 zd = z - temp->position().z();
00157 float delta2 = xd*xd + yd*yd + zd*zd;
00158 if (delta2 < delta) { rcV0Partner = temp; delta = delta2; }
00159 }
00160 }
00161 x = rcV0Partner->position().x();
00162 y = rcV0Partner->position().y();
00163 z = rcV0Partner->position().z();
00164
00165 for(Int_t i = 0; i < GetN(); i++) {
00166 StV0MuDst* tmpV0 = (StV0MuDst*) dataArray->At(i);
00167 if( fabs(x - tmpV0->decayVertexV0X()) < 0.00001 &&
00168 fabs(y - tmpV0->decayVertexV0Y()) < 0.00001 &&
00169 fabs(z - tmpV0->decayVertexV0Z()) < 0.00001 )
00170 { indexRecoArray = i; break; }
00171 }
00172 }
00173
00174 StSPtrVecMcTrack& Daughters = mcVert->daughters();
00175 for (StMcTrackIterator DTrackIt = Daughters.begin();
00176 DTrackIt != Daughters.end(); DTrackIt++) {
00177 switch ((Int_t)(*DTrackIt)->particleDefinition()->charge()) {
00178 case ( 1) :
00179 Pos = (*DTrackIt); break;
00180 case (-1) :
00181 Neg = (*DTrackIt); break;
00182 }
00183 }
00184
00185 if ((Pos)&&(Neg)) {
00186 StV0Mc* v0Mc = new((*mcArray)[mcEntries++]) StV0Mc(mcVert,Pos,Neg,ev);
00187 if((assocMaker)&&(indexRecoArray!=-1)) {
00188 new((*assocArray)[assocEntries++])
00189 StStrangeAssoc(indexRecoArray,mcEntries-1);
00190
00191 pair<mcTrackMapIter,mcTrackMapIter> mcTrackBounds =
00192 theMcTrackMap->equal_range(Pos);
00193 StTrackPairInfo* bestPairInfo = (*mcTrackBounds.first).second;
00194 {for(mcTrackMapIter mcMapIt = mcTrackBounds.first;
00195 mcMapIt != mcTrackBounds.second; ++mcMapIt) {
00196 if ((*mcMapIt).second->commonTpcHits() > bestPairInfo->commonTpcHits())
00197 bestPairInfo = (*mcMapIt).second;
00198 }}
00199 if (mcTrackBounds.first != mcTrackBounds.second) {
00200 v0Mc->SetHitInfoPositive(bestPairInfo->commonTpcHits());
00201 }
00202
00203
00204 mcTrackBounds = theMcTrackMap->equal_range(Neg);
00205
00206 bestPairInfo = (*mcTrackBounds.first).second;
00207
00208 {for(mcTrackMapIter mcMapIt = mcTrackBounds.first;
00209 mcMapIt != mcTrackBounds.second; ++mcMapIt) {
00210 if ((*mcMapIt).second->commonTpcHits() > bestPairInfo->commonTpcHits())
00211 bestPairInfo = (*mcMapIt).second;
00212 }}
00213 if (mcTrackBounds.first != mcTrackBounds.second) {
00214 v0Mc->SetHitInfoNegative(bestPairInfo->commonTpcHits());
00215 }
00216 }
00217 }
00218
00219 return kStOK;
00220 }
00221
00222 ClassImp(StV0Controller)
