Adding StageCastor (Marian)

[u/mrichter/AliRoot.git] / TPC / macros / testTPC / AlienToolkit.cxx

diff --git a/TPC/macros/testTPC/AlienToolkit.cxx b/TPC/macros/testTPC/AlienToolkit.cxx
index ea7e9c0145a0519884b225640771456bdf5466d4..ecb5a3ee24c5eabb5372b64d05241a7e780672cb 100644 (file)
--- a/TPC/macros/testTPC/AlienToolkit.cxx
+++ b/TPC/macros/testTPC/AlienToolkit.cxx
@@ -9,24 +9,35 @@
 #include "TAlienResult.h"
 #include <fstream>
 
+/*
+  Simple toolkit for Alien
+  Use ROOT C++ interfaces classes to access information from ALIEN
+
+  Important functionality:
+  MakeJobList - prepared the list which is later used by agent.sh
+  to analyze data on batch system 
+
 
 
+  See example bellow and class heeader for details.
+  !!! The functionality not expelicitly mentioned in example can be not working
+  Code is still under development
+
+*/
+
 
  
 /*
   gSystem->Load("libXrdClient.so");
   gSystem->Load("libNetx.so");
   .L $ALICE_ROOT/TPC/macros/testTPC/AlienToolkit.cxx+
-  //Reconstructed data example  
-  char *mask = "v14*";
-  char *path = "/alice/cern.ch/user/m/miranov/test2007/"
-  // raw data
-  char *mask = "0000191*RecPo*.root";
-  char *path = "/alice/data/2008/LHC08a_TPC/"
- 
+  //Raw data example
+  char *mask = "20225";
+  char *path = "/alice/data/2008/"
+  //
   AlienToolkit toolkit;
-  toolkit.MakeCollection(path,mask);
-
+  toolkit.MakeCollection(path,mask); // make a list of the registerd data
+  toolkit.StageCastor();    // stage files on castor
   //
   toolkit.MakeJobList("job.list","", "", "rec");
 
@@ -39,6 +50,7 @@ public:
   AlienToolkit();
   TGridCollection* MakeCollection(const char *path,  char *mask);
   void             Stage();
+  void             StageCastor();
   void             LocalCopy(const char* destination);
   void             RemoteCopy(const char* destination="root://gsiaf.gsi.de:1094/", Int_t maxfiles=20);
   void             PrintPFN();
@@ -157,6 +169,32 @@ void AlienToolkit::Stage(){
   gSystem->Exec("aliensh file:stage.txt");
 }
 
+void AlienToolkit::StageCastor(){
+  //
+  // Stage selected alien files
+  //
+  Int_t entries = fInfoArray.GetEntries();
+  ofstream aout("stage.sh");
+  aout<<"#!/usr/local/bin/bash"<<endl;
+  for (Int_t i=0; i<entries;i++){
+    TMap &map = *((TMap*)fInfoArray.At(i));
+    TObjString *pfn = (TObjString*)map("alienSURL");
+    if (!pfn) continue;
+    if (!pfn->String().Contains("//castor")) continue;
+    Char_t * cstr = &( pfn->String()[pfn->String().Index("/castor")]);
+    if (!cstr) continue;
+    printf("Staging submitfor\t%s\n",cstr);
+    char command[1000];
+    sprintf(command,"stager_qry -M %s | grep /castor | gawk  \'{print $3;}\'", cstr);
+    gSystem->Exec(command);    
+    sprintf(command,"stager_get -M %s", cstr);
+    aout<<command<<endl;
+    gSystem->Exec(command);
+  }
+  aout.close();
+  gSystem->Exec("source stage.sh");
+}
+
 void AlienToolkit::LocalCopy(const char *destination){
   //
   // Copy selected files to the destination directory