Example macros for using AliAODv0 class

[u/mrichter/AliRoot.git] / TFluka / usdraw.cxx

diff --git a/TFluka/usdraw.cxx b/TFluka/usdraw.cxx
index db5728eca41d2b4409190a3f26ee84919f188a22..3074d480d427fe12fee7e4bdfbaf3c0a6519d932 100644 (file)
--- a/TFluka/usdraw.cxx
+++ b/TFluka/usdraw.cxx
@@ -1,14 +1,12 @@
 #include <Riostream.h>
 #include "TVirtualMCApplication.h"
-
-#ifndef WITH_ROOT
 #include "TFluka.h"
-#else
-#include "TFlukaGeo.h"
-#endif
-
+#include "TFlukaCodes.h"
+#include <TLorentzVector.h>
 #include "Fdimpar.h"  //(DIMPAR) fluka include
 #include "Ftrackr.h"  //(TRACKR) fluka common
+#include "Fltclcm.h"  //(LTCLCM) fluka common
+#include "Femfstk.h"  //(EMFSTK) fluka common
 #ifndef WIN32
 # define usdraw usdraw_
 #else
@@ -19,15 +17,49 @@ void usdraw(Int_t& icode, Int_t& mreg,
             Double_t& xsco, Double_t& ysco, Double_t& zsco)
 {
   TFluka *fluka = (TFluka*)gMC;
-  fluka->SetCaller(6);
-  fluka->SetIcode(icode);
-  fluka->SetMreg(mreg);
+  // nothing to do if particle inside dummy region
+  if (mreg == fluka->GetDummyRegion()) return;
+  Int_t verbosityLevel = fluka->GetVerbosityLevel();
+  Bool_t debug = (verbosityLevel >= 3)? kTRUE : kFALSE;
+  fluka->SetCaller(kUSDRAW);
+  fluka->SetIcode((FlukaProcessCode_t) icode);
+
+  if (icode/100 == kEMFSCO) {
+      for (Int_t npnw = EMFSTK.npstrt-1; npnw <= EMFSTK.npemf-1; npnw++) {
+         if (EMFSTK.iespak[npnw][mkbmx2-1] ==  TRACKR.ispusr[mkbmx2 - 1] ) {
+             EMFSTK.iespak[npnw][mkbmx2 - 2] = 1;
+// Save properties at point where particle disappears in case this is only an interruption
+             TLorentzVector p;
+             gMC->TrackMomentum(p);
+             EMFSTK.espark[npnw][0] = xsco;               // x
+             EMFSTK.espark[npnw][1] = ysco;               // y
+             EMFSTK.espark[npnw][2] = zsco;               // z
+             EMFSTK.espark[npnw][3] = gMC->TrackTime();   // t
+             EMFSTK.espark[npnw][4] = p[0];               // px
+             EMFSTK.espark[npnw][5] = p[1];               // py
+             EMFSTK.espark[npnw][6] = p[2];               // pz
+             EMFSTK.espark[npnw][7] = p[3];               // e
+             EMFSTK.espark[npnw][8] = gMC->TrackLength(); // Length 
+         } // Track found in stack
+      } // Loop over emf stack 
+  } // Electromagnetic process
+
+  Int_t mlttc = LTCLCM.mlatm1;
+  fluka->SetMreg(mreg, mlttc);
   fluka->SetXsco(xsco);
   fluka->SetYsco(ysco);
   fluka->SetZsco(zsco);
+
+  if (debug) printf("USDRAW: Number of track segments:%6d %6d icode=%d tof=%10.3e track=%d pdg=%d\n",
+  TRACKR.ntrack, TRACKR.mtrack, icode, TRACKR.atrack, TRACKR.ispusr[mkbmx2-1], fluka->PDGFromId(TRACKR.jtrack) );
+
+  TVirtualMCStack* cppstack = fluka->GetStack();
+  cppstack->SetCurrentTrack( TRACKR.ispusr[mkbmx2-1] );
+
   (TVirtualMCApplication::Instance())->Stepping();
   fluka->SetTrackIsNew(kFALSE);
-  
+
+ 
 } // end of usdraw
 } // end of extern "C"