EVGEN/AliGenScan.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
  18 // Realisation of AliGenerator that generates particles with
  19 // vertices on a user defined grid.
  20 // The vertex positions can be smeared.
  21 // Momentum vectors are defined through the methods provided by AliGenerator.
  22 // Author: andreas.morsch@cern.ch
  23
  24 #include "AliGenScan.h"
  25
  26  ClassImp(AliGenScan)
  27
  28  AliGenScan::AliGenScan()
  29          :AliGenerator(-1)
  30 {
  31 // Constructor
  32     fXCmin=0;
  33     fXCmax=0;
  34     fNx=1;
  35     fYCmin=0;
  36     fYCmax=0;
  37     fNy=1;
  38     fZmin=0;
  39     fZmax=0;
  40     fNz=1;
  41 //
  42 //  Read all particles
  43     fNpart=-1;
  44 }
  45
  46 AliGenScan::AliGenScan(Int_t npart)
  47     :AliGenerator(npart)
  48 {
  49 // Constructor
  50     fName  = "Scan";
  51     fTitle = "Generator for particles on a grid";
  52
  53
  54     fXCmin=0;
  55     fXCmax=0;
  56     fNx=1;
  57     fYCmin=0;
  58     fYCmax=0;
  59     fNy=1;
  60     fZmin=0;
  61     fZmax=0;
  62     fNz=1;
  63 }
  64
  65 //____________________________________________________________
  66 AliGenScan::~AliGenScan()
  67 {
  68 // Destructor
  69 }
  70
  71 void AliGenScan::SetRange(Int_t nx, Float_t xmin, Float_t xmax,
  72                      Int_t ny, Float_t ymin, Float_t ymax,
  73                      Int_t nz, Float_t zmin, Float_t zmax)
  74 {
  75 // Define the grid
  76     fXCmin=xmin;
  77     fXCmax=xmax;
  78     fNx=nx;
  79     fYCmin=ymin;
  80     fYCmax=ymax;
  81     fNy=ny;
  82     fZmin=zmin;
  83     fZmax=zmax;
  84     fNz=nz;
  85 }
  86
  87 //____________________________________________________________
  88 void AliGenScan::Generate()
  89 {
  90   //
  91   // Generate one trigger
  92   //
  93
  94   Float_t polar[3]= {0,0,0};
  95   //
  96   Float_t origin[3];
  97   Float_t p[3];
  98   Int_t nt;
  99   Float_t pmom, theta, phi;
 100   //
 101   Float_t random[6];
 102   Float_t dx,dy,dz;
 103
 104   //
 105   if (fNx > 0) {
 106       dx=(fXCmax-fXCmin)/fNx;
 107   } else {
 108       dx=1e10;
 109   }
 110
 111   if (fNy > 0) {
 112       dy=(fYCmax-fYCmin)/fNy;
 113   } else {
 114       dy=1e10;
 115   }
 116
 117   if (fNz > 0) {
 118       dz=(fZmax-fZmin)/fNz;
 119   } else {
 120       dz=1e10;
 121   }
 122   for (Int_t ix=0; ix<fNx; ix++) {
 123       for (Int_t iy=0; iy<fNy; iy++) {
 124           for (Int_t iz=0; iz<fNz; iz++){
 125               Rndm(random,6);
 126               origin[0]=fXCmin+ix*dx+2*(random[0]-0.5)*fOsigma[0];
 127               origin[1]=fYCmin+iy*dy+2*(random[1]-0.5)*fOsigma[1];
 128               origin[2]=fZmin+iz*dz+2*(random[2]-0.5)*fOsigma[2];
 129               pmom=fPMin+random[3]*(fPMax-fPMin);
 130               theta=fThetaMin+random[4]*(fThetaMax-fThetaMin);
 131               phi=fPhiMin+random[5]*(fPhiMax-fPhiMin);
 132               p[0] = pmom*TMath::Cos(phi)*TMath::Sin(theta);
 133               p[1] = pmom*TMath::Sin(phi)*TMath::Sin(theta);
 134               p[2] = pmom*TMath::Cos(theta);
 135               PushTrack(fTrackIt,-1,fIpart,p,origin,polar,0,kPPrimary,nt);
 136           }
 137       }
 138   }
 139 }
 140
 141
 142
 143
 144
 145
 146
 147
 148