try to fix coverity 24179, 24178, 24125, 24124, check cast to pointers not being...
[u/mrichter/AliRoot.git] / STARLIGHT / AliGenStarLight.cxx
1 /**************************************************************************
2  * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
3  *                                                                        *
4  * Author: The ALICE Off-line Project.                                    *
5  * Contributors are mentioned in the code where appropriate.              *
6  *                                                                        *
7  * Permission to use, copy, modify and distribute this software and its   *
8  * documentation strictly for non-commercial purposes is hereby granted   *
9  * without fee, provided that the above copyright notice appears in all   *
10  * copies and that both the copyright notice and this permission notice   *
11  * appear in the supporting documentation. The authors make no claims     *
12  * about the suitability of this software for any purpose. It is          *
13  * provided "as is" without express or implied warranty.                  *
14  **************************************************************************/
15
16 /* $Id:  $ */
17 #include <Riostream.h>
18 #include "AliLog.h"
19 #include "TStarLight.h"
20 #include "starlight.h"
21 #include "upcevent.h"
22 #include "AliGenStarLight.h"
23
24 AliGenStarLight::AliGenStarLight()
25   : AliGenMC()
26   , fTrials(0)
27   , fHeader()
28   , fSLgenerator(NULL) {
29 }
30 //----------------------------------------------------------------------
31 AliGenStarLight::AliGenStarLight(Int_t npart)
32   : AliGenMC(npart)
33   , fTrials(0)
34   , fHeader()
35   , fSLgenerator(NULL) {
36   fSLgenerator = new TStarLight("TStarLight",
37                                 "StarLight UPC Event Generator",
38                                 ""); // no config file name
39 }
40 //----------------------------------------------------------------------
41 AliGenStarLight::~AliGenStarLight() {
42   delete fSLgenerator;
43   fSLgenerator = NULL;
44 }
45 void AliGenStarLight::ImportConfigurationFromFile(const char* filename) {
46   if (NULL == fSLgenerator) {
47     AliFatal("AliGenStarLight class not constructed properly. ");
48     return;
49   }
50   fSLgenerator->ImportConfigurationFromFile(filename);
51 }
52 void AliGenStarLight::SetParameter(const char* line) {
53   if (NULL == fSLgenerator) {
54     AliFatal("AliGenStarLight class not constructed properly. ");
55     return;
56   }
57   fSLgenerator->SetParameter(line);
58 }
59 //----------------------------------------------------------------------
60 void AliGenStarLight::Init() {
61   if (NULL == fSLgenerator) {
62     AliFatal("AliGenStarLight class not constructed properly. ");
63     return;
64   }
65   fSLgenerator->InitStarLight();
66 }
67 //----------------------------------------------------------------------
68 void AliGenStarLight::Generate() {
69   Float_t origin0[4] = {0,0,0,0};
70   if (fVertexSmear == kPerEvent) {
71     Vertex(); // get vertex
72     for (Int_t i=0; i<3; ++i)
73       origin0[i] = fVertex[i];
74     origin0[3] = fTime;
75   }
76
77   Int_t          nt(0); // number of tracks
78   const Int_t    itp(1); // transport all partiles.
79   const Double_t weight(1.0);
80   Bool_t         genOK(kFALSE);
81   fTrials = 0;
82   do { // simulate one event passing all requirements
83     fTrials++;
84     fSLgenerator->GenerateEvent();
85     fSLgenerator->BoostEvent();
86     fSLgenerator->ImportParticles(&fParticles, "ALL");
87     
88     for (size_t i(0), n(fParticles.GetEntries()); i!=n; ++i) {
89       const TParticle *part(dynamic_cast<TParticle*>(fParticles.At(i)));
90       const Int_t iparent(-1); //iparent = part->getParent();
91       //ifDaugh = part->getFirstDaughter();
92       //ilDaugh = part->getLastDaughter();
93       const Double_t pol[3] = {0,0,0};
94       PushTrack(itp, 
95                 iparent, 
96                 part->GetPdgCode(),
97                 part->Px(), part->Py(), part->Pz(), part->Energy(),
98                 origin0[0], origin0[1], origin0[2], origin0[3],
99                 pol[0],     pol[1],     pol[2],
100                 kPNoProcess,
101                 nt,
102                 weight,
103                 part->GetStatusCode());
104       KeepTrack(nt);
105       //cout<<"AliGenStarLight::Generate: fTrials="<<fTrials
106       //  <<" iparent="<<iparent<<" ipdg="<<ipdg
107       //  <<" p="<<px<<","<<py<<","<<pz<<","<<E
108       //  <<" origin="<<origin0[0]<<","<<origin0[1]<<","<<origin0[2]<<","<<origin0[2]
109       //  <<endl;
110     }
111     genOK = kTRUE;
112   } while (!genOK);
113   MakeHeader(nt, origin0);
114   SetHighWaterMark(nt);
115 }
116 //----------------------------------------------------------------------
117 void AliGenStarLight::MakeHeader(Int_t np, Float_t orn[4]) {
118   fHeader.SetInputParameters(fSLgenerator->GetInputParameters());
119   fHeader.SetTrials(fTrials);
120   fHeader.SetEventWeight(1.0);
121   fHeader.SetInteractionTime(orn[3]);
122   fHeader.SetNProduced(np);
123
124   const TArrayF vertexPosition(3, orn);
125   fHeader.SetPrimaryVertex(vertexPosition);
126
127   AddHeader(&fHeader);
128 }
129 //----------------------------------------------------------------------