[u/mrichter/AliRoot.git] / TRD / MakeTRDFullMisAlignment.C

void MakeTRDFullMisAlignment(){
  // Create TClonesArray of full misalignment objects for TRD
  // Expects to read objects for FRAME
  // 
  TClonesArray *array = new TClonesArray("AliAlignObjParams",1000);
  const char* macroname = "MakeTRDFullMisAlignment.C";

  // Activate CDB storage and load geometry from CDB
  AliCDBManager* cdb = AliCDBManager::Instance();
  if(!cdb->IsDefaultStorageSet()) cdb->SetDefaultStorage("local://$ALICE_ROOT");
  cdb->SetRun(0);
  
  AliCDBStorage* storage;
  TString Storage;
  
  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
  }    
		  
  // load FRAME full misalignment objects (if needed, the macro
  // for FRAME has to be ran in advance) and apply them to geometry
  AliCDBPath fpath("GRP","Align","Data");
  AliCDBEntry *eFrame;
  if( TString(gSystem->Getenv("TOCDB")) == TString("kTRUE") ){
    Info(macroname,"Loading FRAME alignment objects from CDB storage %s",
      Storage.Data());
    eFrame = storage->Get(fpath.GetPath(),cdb->GetRun());
  }else{
    eFrame = cdb->Get(fpath.GetPath());
  }
  if(!eFrame) Fatal(macroname,"Could not get the specified CDB entry!");
  TClonesArray* arFrame = (TClonesArray*) eFrame->GetObject();
  arFrame->Sort();
  Int_t nvols = arFrame->GetEntriesFast();
  Bool_t flag = kTRUE;
  for(Int_t j=0; j<nvols; j++)
  {
    AliAlignObj* alobj = (AliAlignObj*) arFrame->UncheckedAt(j);
    if (alobj->ApplyToGeometry() == kFALSE) flag = kFALSE;
  }
  if(!flag) Fatal(macroname,"Error in the application of FRAME objects");

  // Sigmas for the chambers
  Double_t smdx    = 0.3; // 3 mm
  Double_t smdy    = 0.3; // 3 mm
  Double_t smdz    = 0.3; // 3 mm
  Double_t smrx    = 0.4 / 1000.0 / TMath::Pi()*180; // 0.4 mrad
  Double_t smry    = 2.0 / 1000.0 / TMath::Pi()*180; // 2.0 mrad
  Double_t smrz    = 0.4 / 1000.0 / TMath::Pi()*180; // 0.4 mrad
  // Truncation for the chambers
  Double_t cutSmdx = 3.0 * smdx;
  Double_t cutSmdy = 3.0 * smdy;
  Double_t cutSmdz = 3.0 * smdz;

  // Sigmas for the chambers
  Double_t chdx    = 0.05;  // 0.5 mm
  Double_t chdy    = 0.1;   // 1.0 mm
  Double_t chdz    = 0.007; // 70  microns
  Double_t chrx    = 0.0005 / 1000.0 / TMath::Pi()*180; // 0 mrad
  Double_t chry    = 0.0005 / 1000.0 / TMath::Pi()*180; // 0 mrad
  Double_t chrz    = 0.3    / 1000.0 / TMath::Pi()*180; // 0.3 mrad
  // Truncation for the chambers
  Double_t cutChdx = 1.0  * chdx;
  Double_t cutChdy = 1.0  * chdy;
  Double_t cutChdz = 0.14 * chdz;

  Int_t sActive[18]={1,0,0,0,0,0,0,0,1,1,0,0,0,0,0,0,0,1};
  Double_t dx,dy,dz,rx,ry,rz;

  Int_t j=0;
  TRandom *ran = new TRandom(4357);
  UShort_t volid;
  const char *symname;

  // create the supermodules' alignment objects
  for (int iSect; iSect<18; iSect++) {
    TString sm_symname(Form("TRD/sm%02d",iSect));
    dx = AliMathBase::TruncatedGaus(0.0,smdx,cutSmdx); 
    dy = AliMathBase::TruncatedGaus(0.0,smdy,cutSmdy); 
    dz = AliMathBase::TruncatedGaus(0.0,smdz,cutSmdz); 
    rx = ran->Rndm() * 2.0*smrx - smrx;
    ry = ran->Rndm() * 2.0*smry - smry;
    rz = ran->Rndm() * 2.0*smrz - smrz;
    if( (TString(gSystem->Getenv("REALSETUP")) == TString("kTRUE")) && !sActive[iSect] ) continue;
    new((*array)[j++]) AliAlignObjParams(sm_symname.Data(),0,dx,dy,dz,rx,ry,rz,kFALSE);
  }
  // apply supermodules' alignment objects
  Int_t smCounter=0;
  for(Int_t iSect=0; iSect<18; iSect++){
    if( (TString(gSystem->Getenv("REALSETUP")) == TString("kTRUE")) && !sActive[iSect] ) continue;
    AliAlignObjParams* smobj =
      (AliAlignObjParams*)array->UncheckedAt(smCounter++);
    if(!smobj->ApplyToGeometry()){
      Fatal(macroname,Form("application of full misalignment object for sector %d failed!",iSect));
      return;
    }
  }

  // create the chambers' alignment objects
  ran = new TRandom(4357);
  Int_t chId;
  for (Int_t iLayer = AliGeomManager::kTRD1; iLayer <= AliGeomManager::kTRD6; iLayer++) {
    chId=-1;
    for (Int_t iSect = 0; iSect < 18; iSect++){
      for (Int_t iCh = 0; iCh < 5; iCh++) {
        dx = AliMathBase::TruncatedGaus(0.0,chdx,cutChdx); 
        dy = AliMathBase::TruncatedGaus(0.0,chdy,cutChdy); 
        dz = AliMathBase::TruncatedGaus(0.0,chdz,cutChdz); 
        rx = ran->Rndm() * 2.0*chrx - chrx;
        ry = ran->Rndm() * 2.0*chry - chry;
        rz = ran->Rndm() * 2.0*chrz - chrz;
        chId++;
        if ((iSect==13 || iSect==14 || iSect==15) && iCh==2) continue;
        volid = AliGeomManager::LayerToVolUID(iLayer,chId);
        if( (TString(gSystem->Getenv("REALSETUP")) == TString("kTRUE")) && !sActive[iSect] ) continue;
        symname = AliGeomManager::SymName(volid);
        new((*array)[j++]) AliAlignObjParams(symname,volid,dx,dy,dz,rx,ry,rz,kFALSE);
      }
    }
  }

  if( TString(gSystem->Getenv("TOCDB")) != TString("kTRUE") ){
    // save on file
    const char* filename = "TRDfullMisalignment.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,"TRDAlignObjs","kSingleKey");
    f.Close();
  }else{
    // save in CDB storage
    Info(macroname,"Saving alignment objects in CDB storage %s",
	 Storage.Data());
    AliCDBMetaData* md = new AliCDBMetaData();
    md->SetResponsible("Dariusz Miskowiec");
    md->SetComment("Full misalignment for TRD");
    md->SetAliRootVersion(gSystem->Getenv("ARVERSION"));
    AliCDBId id("TRD/Align/Data",0,AliCDBRunRange::Infinity());
    storage->Put(array,id,md);
  }

  array->Delete();
}
Commit	Line	Data
1895a097	1	void MakeTRDFullMisAlignment(){
1895a097	2	// Create TClonesArray of full misalignment objects for TRD
6fce62af	3	// Expects to read objects for FRAME
1895a097	4	//
90dbf5fb	5	TClonesArray *array = new TClonesArray("AliAlignObjParams",1000);
6fce62af	6	const char* macroname = "MakeTRDFullMisAlignment.C";
	7
	8	// Activate CDB storage and load geometry from CDB
	9	AliCDBManager* cdb = AliCDBManager::Instance();
	10	if(!cdb->IsDefaultStorageSet()) cdb->SetDefaultStorage("local://$ALICE_ROOT");
	11	cdb->SetRun(0);
	12
	13	AliCDBStorage* storage;
a24be56b	14	TString Storage;
6fce62af	15
a24be56b	16	if( TString(gSystem->Getenv("TOCDB")) == TString("kTRUE") ){
a24be56b	17	Storage = gSystem->Getenv("STORAGE");
6fce62af	18	if(!Storage.BeginsWith("local://") && !Storage.BeginsWith("alien://")) {
	19	Error(macroname,"STORAGE variable set to %s is not valid. Exiting\n",Storage.Data());
	20	return;
	21	}
	22	storage = cdb->GetStorage(Storage.Data());
	23	if(!storage){
	24	Error(macroname,"Unable to open storage %s\n",Storage.Data());
	25	return;
	26	}
	27	AliCDBPath path("GRP","Geometry","Data");
	28	AliCDBEntry *entry = storage->Get(path.GetPath(),cdb->GetRun());
	29	if(!entry) Fatal(macroname,"Could not get the specified CDB entry!");
	30	entry->SetOwner(0);
	31	TGeoManager* geom = (TGeoManager*) entry->GetObject();
	32	AliGeomManager::SetGeometry(geom);
	33	}else{
	34	AliGeomManager::LoadGeometry(); //load geom from default CDB storage
	35	}
	36
	37	// load FRAME full misalignment objects (if needed, the macro
	38	// for FRAME has to be ran in advance) and apply them to geometry
6fce62af	39	AliCDBPath fpath("GRP","Align","Data");
a24be56b	40	AliCDBEntry *eFrame;
	41	if( TString(gSystem->Getenv("TOCDB")) == TString("kTRUE") ){
	42	Info(macroname,"Loading FRAME alignment objects from CDB storage %s",
	43	Storage.Data());
	44	eFrame = storage->Get(fpath.GetPath(),cdb->GetRun());
6fce62af	45	}else{
a24be56b	46	eFrame = cdb->Get(fpath.GetPath());
6fce62af	47	}
a24be56b	48	if(!eFrame) Fatal(macroname,"Could not get the specified CDB entry!");
	49	TClonesArray* arFrame = (TClonesArray*) eFrame->GetObject();
	50	arFrame->Sort();
	51	Int_t nvols = arFrame->GetEntriesFast();
	52	Bool_t flag = kTRUE;
	53	for(Int_t j=0; j<nvols; j++)
	54	{
	55	AliAlignObj* alobj = (AliAlignObj*) arFrame->UncheckedAt(j);
	56	if (alobj->ApplyToGeometry() == kFALSE) flag = kFALSE;
	57	}
	58	if(!flag) Fatal(macroname,"Error in the application of FRAME objects");
1895a097	59
476c024c	60	// Sigmas for the chambers
	61	Double_t smdx = 0.3; // 3 mm
	62	Double_t smdy = 0.3; // 3 mm
	63	Double_t smdz = 0.3; // 3 mm
	64	Double_t smrx = 0.4 / 1000.0 / TMath::Pi()*180; // 0.4 mrad
	65	Double_t smry = 2.0 / 1000.0 / TMath::Pi()*180; // 2.0 mrad
	66	Double_t smrz = 0.4 / 1000.0 / TMath::Pi()*180; // 0.4 mrad
	67	// Truncation for the chambers
	68	Double_t cutSmdx = 3.0 * smdx;
	69	Double_t cutSmdy = 3.0 * smdy;
	70	Double_t cutSmdz = 3.0 * smdz;
d3a9c52a	71
476c024c	72	// Sigmas for the chambers
	73	Double_t chdx = 0.05; // 0.5 mm
	74	Double_t chdy = 0.1; // 1.0 mm
	75	Double_t chdz = 0.007; // 70 microns
	76	Double_t chrx = 0.0005 / 1000.0 / TMath::Pi()*180; // 0 mrad
	77	Double_t chry = 0.0005 / 1000.0 / TMath::Pi()*180; // 0 mrad
	78	Double_t chrz = 0.3 / 1000.0 / TMath::Pi()*180; // 0.3 mrad
	79	// Truncation for the chambers
	80	Double_t cutChdx = 1.0 * chdx;
	81	Double_t cutChdy = 1.0 * chdy;
	82	Double_t cutChdz = 0.14 * chdz;
1895a097	83
05b6df61	84	Int_t sActive[18]={1,0,0,0,0,0,0,0,1,1,0,0,0,0,0,0,0,1};
1895a097	85	Double_t dx,dy,dz,rx,ry,rz;
	86
	87	Int_t j=0;
	88	TRandom *ran = new TRandom(4357);
	89	UShort_t volid;
d3a9c52a	90	const char *symname;
d3a9c52a	91
d3a9c52a	92	// create the supermodules' alignment objects
a24be56b	93	for (int iSect; iSect<18; iSect++) {
a24be56b	94	TString sm_symname(Form("TRD/sm%02d",iSect));
476c024c	95	dx = AliMathBase::TruncatedGaus(0.0,smdx,cutSmdx);
	96	dy = AliMathBase::TruncatedGaus(0.0,smdy,cutSmdy);
	97	dz = AliMathBase::TruncatedGaus(0.0,smdz,cutSmdz);
	98	rx = ran->Rndm() * 2.0*smrx - smrx;
	99	ry = ran->Rndm() * 2.0*smry - smry;
	100	rz = ran->Rndm() * 2.0*smrz - smrz;
05b6df61	101	if( (TString(gSystem->Getenv("REALSETUP")) == TString("kTRUE")) && !sActive[iSect] ) continue;
6fce62af	102	new((*array)[j++]) AliAlignObjParams(sm_symname.Data(),0,dx,dy,dz,rx,ry,rz,kFALSE);
d3a9c52a	103	}
a24be56b	104	// apply supermodules' alignment objects
	105	Int_t smCounter=0;
	106	for(Int_t iSect=0; iSect<18; iSect++){
05b6df61	107	if( (TString(gSystem->Getenv("REALSETUP")) == TString("kTRUE")) && !sActive[iSect] ) continue;
a24be56b	108	AliAlignObjParams* smobj =
a24be56b	109	(AliAlignObjParams*)array->UncheckedAt(smCounter++);
d3a9c52a	110	if(!smobj->ApplyToGeometry()){
a24be56b	111	Fatal(macroname,Form("application of full misalignment object for sector %d failed!",iSect));
d3a9c52a	112	return;
	113	}
	114	}
6fce62af	115
1895a097	116	// create the chambers' alignment objects
d3a9c52a	117	ran = new TRandom(4357);
a24be56b	118	Int_t chId;
ae079791	119	for (Int_t iLayer = AliGeomManager::kTRD1; iLayer <= AliGeomManager::kTRD6; iLayer++) {
a24be56b	120	chId=-1;
	121	for (Int_t iSect = 0; iSect < 18; iSect++){
	122	for (Int_t iCh = 0; iCh < 5; iCh++) {
476c024c	123	dx = AliMathBase::TruncatedGaus(0.0,chdx,cutChdx);
	124	dy = AliMathBase::TruncatedGaus(0.0,chdy,cutChdy);
	125	dz = AliMathBase::TruncatedGaus(0.0,chdz,cutChdz);
	126	rx = ran->Rndm() * 2.0*chrx - chrx;
	127	ry = ran->Rndm() * 2.0*chry - chry;
	128	rz = ran->Rndm() * 2.0*chrz - chrz;
	129	chId++;
	130	if ((iSect==13 \|\| iSect==14 \|\| iSect==15) && iCh==2) continue;
	131	volid = AliGeomManager::LayerToVolUID(iLayer,chId);
	132	if( (TString(gSystem->Getenv("REALSETUP")) == TString("kTRUE")) && !sActive[iSect] ) continue;
	133	symname = AliGeomManager::SymName(volid);
	134	new((*array)[j++]) AliAlignObjParams(symname,volid,dx,dy,dz,rx,ry,rz,kFALSE);
	135	}
1895a097	136	}
	137	}
	138
a24be56b	139	if( TString(gSystem->Getenv("TOCDB")) != TString("kTRUE") ){
1895a097	140	// save on file
dfe9c69d	141	const char* filename = "TRDfullMisalignment.root";
	142	TFile f(filename,"RECREATE");
	143	if(!f){
	144	Error(macroname,"cannot open file for output\n");
	145	return;
	146	}
	147	Info(macroname,"Saving alignment objects to the file %s", filename);
1895a097	148	f.cd();
	149	f.WriteObject(array,"TRDAlignObjs","kSingleKey");
	150	f.Close();
	151	}else{
	152	// save in CDB storage
dfe9c69d	153	Info(macroname,"Saving alignment objects in CDB storage %s",
dfe9c69d	154	Storage.Data());
1895a097	155	AliCDBMetaData* md = new AliCDBMetaData();
	156	md->SetResponsible("Dariusz Miskowiec");
	157	md->SetComment("Full misalignment for TRD");
5bd470e1	158	md->SetAliRootVersion(gSystem->Getenv("ARVERSION"));
dfe9c69d	159	AliCDBId id("TRD/Align/Data",0,AliCDBRunRange::Infinity());
1895a097	160	storage->Put(array,id,md);
	161	}
	162
	163	array->Delete();
1895a097	164	}
1895a097	165