HLT/programs/gettransform.cxx

   1 //#define DOTESTS /* undefine this for getting the parameters */
   2
   3 #ifdef DOTESTS
   4 #define use_root
   5 #endif
   6
   7 #include <stream.h>
   8 #include <libgen.h>
   9 #include <math.h>
  10 #include <AliL3RootTypes.h>
  11 #include <AliL3Defs.h>
  12 #include <AliL3Transform.h>
  13 #include <AliL3Logging.h>
  14 #include <AliL3Logger.h>
  15 #include <AliL3MemHandler.h>
  16
  17 #ifdef use_root
  18 #include <TApplication.h>
  19 #include <TCanvas.h>
  20 #include <TPad.h>
  21 #include <TH1.h>
  22 #include <TMath.h>
  23 #endif
  24
  25 /*
  26   This program extracts parameters and lookup tables needed for the
  27   vhdl implementation of the Hough transform.
  28  */
  29
  30 int main(int argc,char **argv)
  31 {
  32   Int_t patch=0;
  33   Int_t slice=0;
  34   Char_t path[1000];
  35   //cout<<"Usage: transform [slice] [patch] [path]"<<endl;
  36
  37   if (argc>1) {
  38     slice=atoi(argv[1]);
  39   }
  40   if (argc>2) {
  41     patch=atoi(argv[2]);
  42   }  if (argc>3) {
  43     strcpy(path,argv[3]);
  44   } else strcpy(path,"/tmp/data/RawData/slice0");
  45
  46   AliL3Logger l;
  47   l.UseStdout();
  48   //l.UseStream();
  49
  50   AliL3Transform::Init(path);
  51   //cout << "Transform version: " << AliL3Transform::GetVersion() << endl;
  52
  53   Int_t npads=0;
  54   Int_t sector=0;
  55   Int_t sector_row=0;
  56
  57   Float_t Xt[200];
  58   Float_t Yt[200];
  59
  60   Float_t xbins[126];
  61   for(int t=0;t<126;t++){
  62     Float_t eta=exp(2.*1./125*t);
  63     xbins[t]=(eta-1)/(1+eta);
  64     //cout << eta << " " << xbins[t] << endl;
  65   }
  66
  67 #ifdef DOTESTS /* do some tests in root histos */
  68 #ifdef use_root
  69   TApplication theApp("App", &argc, argv);
  70
  71   TCanvas *c = new TCanvas("c", "The Histogram Canvas", 800, 800);
  72   TPad *pad1 = new TPad("pad1","Pad for Eta",0.05,0.50,0.95,0.95,21);
  73   TPad *pad2 = new TPad("pad2","Pad for Tests",0.05,0.05,0.95,0.45,21);
  74   //  TH1F *h1f = new TH1F("h1f","Eta Distribution",125,0,1.);
  75   TH1F *h1f = new TH1F("h1f","Test Error",100,90,110);
  76   TH1F *h2f = new TH1F("h2f","Test Distribution",1000,0,10);
  77   //  TH1F *h2f = new TH1F("h2f","Test Distribution",125,xbins);
  78   pad1->Draw();
  79   pad2->Draw();
  80   pad1->cd();
  81 #endif
  82
  83   Int_t ntime=AliL3Transform::GetNTimeBins();
  84   Float_t xyz[3]={0,0,0};
  85   Float_t rpe[3]={0,0,0};
  86   int rebinned=0;
  87   for(Int_t rr=NRows[patch][0];rr<=NRows[patch][1];rr++){
  88     npads=AliL3Transform::GetNPads(rr);
  89     AliL3Transform::Slice2Sector(slice,rr,sector,sector_row);
  90     for(Int_t pp=0;pp<npads;pp++){
  91       for (Int_t tt=0;tt<ntime;tt++){
  92         AliL3Transform::Raw2Local(xyz,sector,sector_row,pp,tt);
  93         AliL3Transform::XYZtoRPhiEta(rpe,xyz);
  94 #ifdef use_root
  95         if(!rebinned) {
  96           h2f->SetBins(500,0,252.*252/xyz[0]/xyz[0]);
  97           rebinned=1;
  98         }
  99         //h1f->Fill(rpe[2]);
 100         //h2f->Fill(xyz[2]/rpe[0]);
 101         Float_t A2=xyz[2]*xyz[2]/(xyz[0]*xyz[0]+xyz[1]*xyz[1]);
 102         Float_t Sroot=1./TMath::Sqrt(1.+A2);
 103         Int_t bin=h2f->FindBin(A2);
 104         Double_t binCenter = h2f->GetBinCenter(bin);
 105         Double_t binCenterval = 1./TMath::Sqrt(1.+binCenter);
 106         Float_t err=(Sroot*100./binCenterval);
 107         h1f->Fill(err);
 108         h2f->Fill(A2);
 109         //printf("%.3f %.3f %.3f\n",Sroot,binCenter,err);
 110 #endif
 111         //printf("%d %d %d %f %f %f %f %f %f\n",rr,pp,tt,xyz[0],xyz[1],xyz[2],rpe[0],rpe[1],rpe[2]);
 112       }
 113     }
 114   }
 115 #else /* do the extraction of the parameters */
 116   AliL3Transform::Slice2Sector(slice,NRows[patch][0],sector,sector_row);
 117
 118   Float_t ytabval=0;
 119   Int_t maxrow=NRows[patch][1]-NRows[patch][0];
 120   Float_t padpitch=0;
 121   if(sector<AliL3Transform::GetNSectorLow())
 122     padpitch=AliL3Transform::GetPadPitchWidthLow();
 123   else
 124     padpitch=AliL3Transform::GetPadPitchWidthUp();
 125
 126   printf("MinRow: %d\nMaxRow: %d\n",0,maxrow);
 127   printf("YPadWidth: %.2f\n",padpitch);
 128   printf("ZSign: %d\n",slice < 18 ? 1:-1);
 129   printf("ZWidth: %.2f\n",AliL3Transform::GetZWidth());
 130   printf("TimeWidth: %.2f\n\n",AliL3Transform::GetZWidth());
 131
 132   //calculating lookup tables for slice and patch!
 133   for(Int_t rr=NRows[patch][0];rr<=NRows[patch][1];rr++){
 134     npads=AliL3Transform::GetNPads(rr);
 135
 136     //Y(row,pad)=pad*padpitch-ytabval(row);
 137     ytabval=0.5*(npads-1)*padpitch;
 138
 139     Xt[rr-NRows[patch][0]]=AliL3Transform::Row2X(rr);
 140     Yt[rr-NRows[patch][0]]=ytabval;
 141     //row in patch: X(row) Y_part(row)
 142     printf("Row: %d X: %.2f Y: %.2f\n",rr-NRows[patch][0],AliL3Transform::Row2X(rr),ytabval);
 143   }
 144   printf("\n\nVHDL-Output for LUT:\nX_table := (");
 145   for(int i=0;i<maxrow-1;i++){
 146     printf("%.2f, ",Xt[i]);
 147   }
 148   printf("%.2f);\n",Xt[maxrow-1]);
 149   printf("Y_table := (");
 150   for(int i=0;i<maxrow-1;i++){
 151     printf("%.2f, ",Yt[i]);
 152   }
 153   printf("%.2f);\n",Yt[maxrow-1]);
 154 #endif
 155
 156 #ifdef DOTESTS
 157 #ifdef use_root
 158   h1f->Draw();
 159   pad2->cd();
 160   h2f->Draw();
 161   c->Update();
 162
 163   theApp.Run();
 164 #endif
 165 #endif
 166
 167   exit(0);
 168 }
 169
 170 //#############################################################
 171
 172 #if 0
 173   AliL3MemHandler file; //Does all the file/data handling
 174
 175   //Open the data file:
 176   if(!file.SetBinaryInput(digitfile))
 177     {
 178       cerr<<"Error opening file "<< digitfile <<endl;
 179       return -1;
 180     }
 181
 182   //Store the data in memory, and get the pointer to it:
 183   unsigned int ndigits=0;
 184   AliL3DigitRowData *digits=0;
 185   digits=(AliL3DigitRowData*)file.CompBinary2Memory(ndigits);
 186   file.CloseBinaryInput();
 187
 188   /*
 189   for(Int_t r=0; r<31; r++) {
 190     UInt_t padrow=digits->fRow;
 191     AliL3DigitData *dPt = (AliL3DigitData*)digits->fDigitData;
 192     cout<<"padrow "<<padrow<<" ndigits "<<digits->fNDigit<<endl;
 193
 194     for(Int_t d=0; d<digits->fNDigit; d++) {
 195       cout<<" padrow "<<padrow<<" pad "<<(int)dPt[d]->fPad<<" time " <<(int)dPt[d]->fTime<<" charge "<<(int)dPt[d]->fCharge<<endl;
 196     }
 197     file->UpdateRowPointer(digits);
 198   }
 199   return 100;
 200   */
 201
 202   digits=(AliL3DigitRowData*)file->GetDataPointer(ndigits);
 203   AliL3Transform transform; //Storing all detector-spesific quantities, needed by the clusterfinder.
 204   AliL3ClustFinderNew cf(&transform); //The cluster finder itself.
 205
 206   //Switch off deconvolution:
 207   cf.SetDeconv(false);
 208   cf.SetXYError(0.2);
 209   cf.SetZError(0.3);
 210
 211   //Init cluster finder
 212   cf.InitSlice(0,0,0,20,10000);
 213
 214   //Give the data pointer to the cluster finder
 215   cf.Read(ndigits,digits);
 216   cout << digits << endl;
 217   //Start processing:
 218   cf.ProcessDigits();
 219
 220 /*
 221  int slice=1;
 222   a = new AliL3Hough(digitfile,kTRUE,n_eta_segments);
 223
 224   a->ReadData(slice);
 225
 226   a->Transform();
 227   a->AddAllHistograms();
 228
 229   c1 = new TCanvas("c1","",2);
 230   a->GetTransformer(0)->GetHistogram(0)->Draw("box");
 231
 232
 233   a->FindTrackCandidates();
 234   //a->Evaluate(2);
 235  */
 236   /*
 237   //Display the track:
 238   dhist = new AliL3Histogram("dhist","",250,0,250,250,-125,125);
 239   for(int i=0; i<6; i++)
 240     a->GetEval(i)->DisplayEtaSlice(0,dhist);
 241   c2 = new TCanvas("c2","",2);
 242   dhist->Draw();
 243   */
 244
 245   //a->WriteTracks();//"../compression/");
 246
 247 /*
 248   tracks = (AliL3TrackArray*)a->GetTracks(0);
 249
 250   float xyz[3];
 251   int row=0;
 252   cout<<"Found "<<tracks->GetNTracks()<<" tracks"<<endl;
 253   for(int i=0; i<tracks->GetNTracks(); i++)
 254     {
 255       AliL3HoughTrack *track = (AliL3HoughTrack*)tracks->GetCheckedTrack(i);
 256       if(!track) continue;
 257       //track->GetCrossingPoint(row,xyz);
 258       //transform->Local2Raw(xyz,1,row);
 259       //cout<<"Recon. pad "<<(int)xyz[1]<<" time "<<(int)xyz[2]<<endl;
 260       cout<<"Pt "<<track->GetPt()<<" phi "<<track->GetPhi0()<<" charge "<<track->GetCharge()<<" weigth "<<track->GetWeight()<<endl;
 261     }
 262 */
 263
 264   return;
 265 }
 266
 267 #endif