STEER/AliLegoGeneratorXYZ.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 //-------------------------------------------------------------------------
  19 //      Lego generator in x - y - z bins
  20 //    Uses geantino rays to check the material distributions and detector's
  21 //    geometry
  22 //    Author: A.Morsch
  23 //-------------------------------------------------------------------------
  24
  25 #include "AliLegoGeneratorXYZ.h"
  26 #include "AliRun.h"
  27
  28 ClassImp(AliLegoGeneratorXYZ)
  29
  30
  31 //___________________________________________
  32
  33
  34 AliLegoGeneratorXYZ::AliLegoGeneratorXYZ()
  35 {
  36 // Default Constructor
  37     fDir1[0]=1.; fDir1[1]=0.; fDir1[2]=0.;
  38     fDir2[0]=0.; fDir2[1]=1.; fDir2[2]=0.;
  39     fDir3[0]=0.; fDir3[1]=0.; fDir3[2]=1.;
  40 }
  41
  42 AliLegoGeneratorXYZ::AliLegoGeneratorXYZ(char* axis)
  43 {
  44 // Constructor
  45     if (!strcmp(axis,"x") || !strcmp(axis,"X"))
  46     {
  47         fDir1[0]=0.; fDir1[1]=1.; fDir1[2]=0.;
  48         fDir2[0]=0.; fDir2[1]=0.; fDir2[2]=1.;
  49         fDir3[0]=1.; fDir3[1]=0.; fDir3[2]=0.;
  50     }
  51     else if (!strcmp(axis,"y") || !strcmp(axis,"Y"))
  52     {
  53         fDir1[0]=1.; fDir1[1]=0.; fDir1[2]=0.;
  54         fDir2[0]=0.; fDir2[1]=0.; fDir2[2]=1;
  55         fDir3[0]=0.; fDir3[1]=1.; fDir3[2]=0.;
  56     }
  57     else if (!strcmp(axis,"z") || !strcmp(axis,"Z"))
  58     {
  59         fDir1[0]=1.; fDir1[1]=0.; fDir1[2]=0.;
  60         fDir2[0]=0.; fDir2[1]=1.; fDir2[2]=0.;
  61         fDir3[0]=0.; fDir3[1]=0.; fDir3[2]=1.;
  62     }
  63     else
  64     {
  65         fDir1[0]=1.; fDir1[1]=0.; fDir1[2]=0.;
  66         fDir2[0]=0.; fDir2[1]=1.; fDir2[2]=0.;
  67         fDir3[0]=0.; fDir3[1]=0.; fDir3[2]=1.;
  68     }
  69 }
  70
  71
  72 AliLegoGeneratorXYZ::AliLegoGeneratorXYZ(Int_t nc1, Float_t c1min,
  73                                          Float_t c1max, Int_t nc2,
  74                                          Float_t c2min, Float_t c2max,
  75                                          Float_t rmin, Float_t rmax, Float_t zmax) :
  76     AliLegoGenerator(nc1, c1min, c1max, nc2, c2min, c2max,
  77                      rmin, rmax, zmax)
  78 {
  79 //  Constructor
  80     fDir1[0]=1.; fDir1[1]=0.; fDir1[2]=0.;
  81     fDir2[0]=0.; fDir2[1]=1.; fDir2[2]=0.;
  82     fDir3[0]=0.; fDir3[1]=0.; fDir3[2]=1.;
  83 }
  84
  85
  86 //___________________________________________
  87 void AliLegoGeneratorXYZ::Generate()
  88 {
  89 // Create a geantino with kinematics corresponding to the current bins
  90 // Here: Coor1 =  x
  91 //       Coor2 =  z
  92
  93   //
  94   // Rootinos are 0
  95    const Int_t kMpart = 0;
  96    Float_t orig[3], pmom[3];
  97
  98    // Prepare for next step
  99    if(fCoor1Bin>=fNCoor1-1)
 100      if(fCoor2Bin>=fNCoor2-1) {
 101        Warning("Generate","End of Lego Generation");
 102        return;
 103      } else {
 104        fCoor2Bin++;
 105        printf("Generating rays in Coordinate 2 bin:%d\n",fCoor2Bin);
 106        fCoor1Bin=0;
 107      } else fCoor1Bin++;
 108
 109
 110    fCurCoor1 = (fCoor1Min+(fCoor1Bin+0.5)*(fCoor1Max-fCoor1Min)/fNCoor1);
 111    fCurCoor2 = (fCoor2Min+(fCoor2Bin+0.5)*(fCoor2Max-fCoor2Min)/fNCoor2);
 112
 113 // Origin and direction
 114    Int_t i;
 115    for (i=0; i<3; i++) {
 116        pmom[i]=fDir3[i];
 117        orig[i]=fCurCoor1*fDir1[i]+fCurCoor2*fDir2[i];
 118    }
 119
 120    Float_t dalicz = 3000;
 121    if (fRadMin > 0) {
 122        Float_t t = PropagateCylinder(orig,pmom,fRadMin,dalicz);
 123        orig[0] = pmom[0]*t;
 124        orig[1] = pmom[1]*t;
 125        orig[2] = pmom[2]*t;
 126        if (TMath::Abs(orig[2]) > fZMax) return;
 127    }
 128
 129    Float_t polar[3]={0.,0.,0.};
 130    Int_t ntr;
 131    gAlice->SetTrack(1, -1, kMpart, pmom, orig, polar, 0, kPPrimary, ntr);
 132
 133 }
 134
 135
 136
 137