]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - TOF/MakeTOFResMisAlignment.C
Enabled option for selecting tree name in trneding macro
[u/mrichter/AliRoot.git] / TOF / MakeTOFResMisAlignment.C
index 5b3150dce089f877856dd61827c71b09edaf3068..fa04750662bceb803da42e464f6670c03dda27c4 100644 (file)
-void MakeTOFResMisAlignment(){
+void MakeTOFResMisAlignment() {
+  //
   // Create TClonesArray of residual misalignment objects for TOF
   //
-  TClonesArray *array = new TClonesArray("AliAlignObjAngles",2000);
+
+  const char* macroname = "MakeTOFResMisAlignment.C";
+
+  TClonesArray *array = new TClonesArray("AliAlignObjParams",2000);
   TClonesArray &alobj = *array;
    
-  if(!gGeoManager) TGeoManager::Import("/home/rgrosso/Prove/AliRoot/geometry.root");
-  // needed for the constructors with local coordinates not to fail
+  // Activate CDB storage and load geometry from CDB
+  AliCDBManager* cdb = AliCDBManager::Instance();
+  if (!cdb->IsDefaultStorageSet())
+    cdb->SetDefaultStorage("local://$ALICE_ROOT/OCDB");
+  cdb->SetRun(0);
+
+  AliCDBStorage* storage;
+  TString Storage;
 
-  AliAlignObjAngles a;
+  if( TString(gSystem->Getenv("TOCDB")) == TString("kTRUE") ){
+    Storage = gSystem->Getenv("STORAGE");
+    if (!Storage.BeginsWith("local://") && !Storage.BeginsWith("alien://")) {
+      Error(macroname,"STORAGE variable set to %s is not valid. Exiting\n",Storage.Data());
+      return;
+    }
+    storage = cdb->GetStorage(Storage.Data());
+    if (!storage) {
+      Error(macroname,"Unable to open storage %s\n",Storage.Data());
+      return;
+    }
+    AliCDBPath path("GRP","Geometry","Data");
+    AliCDBEntry *entry = storage->Get(path.GetPath(),cdb->GetRun());
+    if (!entry)
+      Fatal(macroname,"Could not get the specified CDB entry!");
+
+    entry->SetOwner(0);
+    TGeoManager* geom = (TGeoManager*) entry->GetObject();
+    AliGeomManager::SetGeometry(geom);
+  } else
+    AliGeomManager::LoadGeometry(); //load geom from default CDB storage
 
   AliGeomManager::ELayerID idTOF = AliGeomManager::kTOF;
-  Int_t i;
   Int_t j=0;
-  Double_t dx=0.; 
-  Double_t dy=0.; 
-  Double_t dz=0.;
-  Double_t  dpsi, dtheta, dphi;
-  TRandom *rnd   = new TRandom(4357);
-  Double_t sigmatr = 0.1; // max shift in cm w.r.t. local ideal RS
-
-  for(i=0; i<AliGeomManager::LayerSize(idTOF); i++) {
-    dx = 0;
-    dy = rnd->Gaus(0.,sigmatr);
-    dz = rnd->Gaus(0.,sigmatr);
-    dpsi = 0.;
-    dtheta = 0.;
-    dphi = 0.;
-    new(alobj[j]) AliAlignObjAngles(AliGeomManager::SymName(idTOF,i), AliGeomManager::LayerToVolUID(idTOF,i), dx, dy, dz, dpsi, dtheta, dphi, kFALSE);
-    j++;
+  Int_t nSectors=18;
+
+  //Produce objects for TOF supermodules
+  Int_t iIndex=0; //let all modules have index=0 in a layer with no LUT
+  AliGeomManager::ELayerID iLayer = AliGeomManager::kInvalidLayer;
+  UShort_t dvoluid = AliGeomManager::LayerToVolUID(iLayer,iIndex); //dummy vol id 
+  Double_t smdx, smdy, smdz=0., dpsi=0., dtheta, dphi=0.;
+  for(Int_t isect=0; isect<nSectors; isect++) {
+    TString symname(Form("TOF/sm%02d",isect));
+    new(alobj[j++]) AliAlignObjParams(symname.Data(),
+                                     dvoluid,
+                                     smdx, smdy, smdz, dpsi, dtheta, dphi, kFALSE);
   }
 
-  if( gSystem->Getenv("TOCDB") != TString("kTRUE") ){
+  Int_t strId=-1;
+  Double_t dx=0., dy=0., dz=0., dpsi=0., dtheta=0., dphi=0.;
+  //TRandom *rnd   = new TRandom(4357);
+  Double_t sigmatr = 0.1; // sigma (in cm) for shift w.r.t. local ideal RS
+
+  Int_t nstrA=15;
+  Int_t nstrB=19;
+  Int_t nstrC=19;
+  Int_t nStrips=nstrA+2*nstrB+2*nstrC;
+
+  Double_t cuty=0., cutz=0., cut=3*sigmatr;
+  for (Int_t isect = 0; isect < nSectors; isect++) {
+    for (Int_t istr = 1; istr <= nStrips; istr++) {
+      //dy = rnd->Gaus(0.,sigmatr);
+      //dz = rnd->Gaus(0.,sigmatr);
+      //strId++;
+
+      switch (istr) {
+      case 25:
+      case 29:
+      case 63:
+      case 67:
+       cuty = sigmatr*0.6;
+       dy  = AliMathBase::TruncatedGaus(0., sigmatr, cut, cuty);
+       dz  = AliMathBase::TruncatedGaus(0., sigmatr, cut);
+       strId++;
+       break;
+       /*
+      case 38:
+       cuty = sigmatr*2.5;
+       cutz = sigmatr*2.5;
+       dy  = AliMathBase::TruncatedGaus(0., sigmatr, cut, cuty);
+       dz  = AliMathBase::TruncatedGaus(0., sigmatr, cut, cutz);
+       strId++;
+       break;
+      case 54:
+       cuty = sigmatr*2.5;
+       cutz = sigmatr*2.5;
+       dy  = AliMathBase::TruncatedGaus(0., sigmatr, cut, cuty);
+       dz  = AliMathBase::TruncatedGaus(0., sigmatr, cutz, cut);
+       strId++;
+       break;
+       */
+      default:
+       dy = AliMathBase::TruncatedGaus(0., sigmatr, cut);
+       dz = AliMathBase::TruncatedGaus(0., sigmatr, cut);
+       strId++;
+       break;
+      }
+
+      if ((isect==13 || isect==14 || isect==15) && (istr >= 39 && istr <= 53)) continue;
+      new(alobj[j++]) AliAlignObjParams(AliGeomManager::SymName(idTOF,strId),
+                                       AliGeomManager::LayerToVolUID(idTOF,strId),
+                                       dx, dy, dz, dpsi, dtheta, dphi, kFALSE);
+    }
+  }
+
+  if( TString(gSystem->Getenv("TOCDB")) != TString("kTRUE") ) {
     // save on file
-    TFile f("TOFresidualMisalignment.root","RECREATE");
-    if(!f) cerr<<"cannot open file for output\n";
+    const char* filename = "TOFresidualMisalignment.root";
+    TFile f(filename,"RECREATE");
+    if(!f){
+      Error(macroname,"cannot open file for output\n");
+      return;
+    }
+    Info(macroname,"Saving alignment objects to the file %s", filename);
     f.cd();
     f.WriteObject(array,"TOFAlignObjs","kSingleKey");
     f.Close();
-  }else{
+  } else {
     // save in CDB storage
-    const char* Storage = gSystem->Getenv("STORAGE");
-    AliCDBManager* cdb = AliCDBManager::Instance();
-    AliCDBStorage* storage = cdb->GetStorage(Storage);
     AliCDBMetaData* md = new AliCDBMetaData();
     md->SetResponsible("Silvia Arcelli");
     md->SetComment("Residual misalignment for TOF, sigmatr=1mm in the local RS");
     md->SetAliRootVersion(gSystem->Getenv("ARVERSION"));
-    AliCDBId id("TOF/Align/Data",0,9999999);
+    AliCDBId id("TOF/Align/Data",0,AliCDBRunRange::Infinity());
     storage->Put(array,id,md);
   }
 
@@ -54,4 +138,15 @@ void MakeTOFResMisAlignment(){
 
 }
 
-
+/*
+Double_t LocalTruncatedGaus(Double_t mean, Double_t sigma, Double_t cutatL, Double_t cutatR)
+{
+  // return number generated according to a gaussian distribution N(mean,sigma) truncated at cutat
+  //
+  Double_t value;
+  do{
+    value=gRandom->Gaus(mean,sigma);
+  }while(value-mean<-cutatL || value-mean>cutatR);
+  return value;
+}
+*/