SHUTTLE/TestShuttle/TestPreprocessor.C

   1 /* $Id$ */
   2
   3 // This class runs the test preprocessor
   4 // It uses AliTestShuttle to simulate a full Shuttle process
   5
   6 // The input data is created in the functions
   7 //   CreateDCSAliasMap() creates input that would in the same way come from DCS
   8 //   ReadDCSAliasMap() reads from a file
   9 //   CreateInputFilesMap() creates a list of local files, that can be accessed by the shuttle
  10
  11 void TestPreprocessor()
  12 {
  13   // load library
  14   gSystem->Load("libTestShuttle.so");
  15
  16   // initialize location of CDB
  17   AliCDBManager::Instance()->SetDefaultStorage("local://$ALICE_ROOT/SHUTTLE/TestShuttle/TestCDB");
  18
  19   // create AliTestShuttle instance
  20   // The parameters are run, startTime, endTime
  21   AliTestShuttle* shuttle = new AliTestShuttle(7, 0, 1);
  22
  23   // TODO(1)
  24   //
  25   // The shuttle can read DCS data, if the preprocessor should be tested to process DCS data,
  26   // some fake data has to be created.
  27   //
  28   // The "fake" input data can be taken using either (a) or (b):
  29   // (a) data from a file: Use ReadDCSAliasMap()
  30   //     the format of the file is explained in ReadDCSAliasMap()
  31   //     To use it uncomment the following line:
  32   //
  33   //TMap* dcsAliasMap = ReadDCSAliasMap();
  34   //
  35   // (b) generated in this macro: Use CreateDCSAliasMap() and its documentation
  36   //     To use it uncomment the following line:
  37   //
  38   TMap* dcsAliasMap = CreateDCSAliasMap();
  39
  40   // now give the alias map to the shuttle
  41   shuttle->SetDCSInput(dcsAliasMap);
  42
  43   // TODO(2)
  44   //
  45   // The shuttle can also process files that originate from DCS, DAQ and HLT.
  46   // To test it, we provide some local files and locations where these would be found when
  47   // the online machinery would be there.
  48   // In real life this functions would be produces by the sub-detectors
  49   // calibration programs in DCS, DAQ or HLT. These files can then be retrieved using the Shuttle.
  50   //
  51   // Files are added with the function AliTestShuttle::AddInputFile. The syntax is:
  52   // AddInputFile(<system>, <detector>, <id>, <source>, <local-file>)
  53   // In this example we add a file originating from the GDC with the id PEDESTALS
  54   // Three files originating from different LDCs but with the same id are also added
  55   shuttle->AddInputFile(AliTestShuttle::kDAQ, "DET", "PEDESTALS", "GDC", "file1.root");
  56   shuttle->AddInputFile(AliTestShuttle::kDAQ, "DET", "DRIFTVELOCITY", "LDC0", "file2a.root");
  57   shuttle->AddInputFile(AliTestShuttle::kDAQ, "DET", "DRIFTVELOCITY", "LDC1", "file2b.root");
  58   shuttle->AddInputFile(AliTestShuttle::kDAQ, "DET", "DRIFTVELOCITY", "LDC2", "file2b.root");
  59
  60   // TODO(3)
  61   // Create the preprocessor that should be tested, it registers itself automatically to the shuttle
  62   AliPreprocessor* pp = new AliTestPreprocessor("DET", shuttle);
  63
  64   // Test the preprocessor
  65   shuttle->Process();
  66
  67   // TODO(4)
  68   // In the preprocessor AliShuttleInterface::Store should be called to put the final
  69   // data to the CDB. To check if all went fine have a look at the files produced in
  70   // $ALICE_ROOT/SHUTTLE/TestShuttle/TestCDB/<detector>/SHUTTLE/Data
  71   //
  72   // Check the file which should have been created
  73   AliCDBEntry* entry = AliCDBManager::Instance()->Get("DET/SHUTTLE/Data", 0);
  74   if (!entry)
  75   {
  76     printf("The file is not there. Something went wrong.\n");
  77     return;
  78   }
  79
  80   AliTestDataDCS* output = dynamic_cast<AliTestDataDCS*> (entry->GetObject());
  81   // If everything went fine, draw the result
  82   if (output)
  83     output->Draw();
  84 }
  85
  86 TMap* CreateDCSAliasMap()
  87 {
  88   // Creates a DCS structure
  89   // The structure is the following:
  90   //   TMap (key --> value)
  91   //     <DCSAlias> --> <valueList>
  92   //     <DCSAlias> is a string
  93   //     <valueList> is a TObjArray of AliDCSValue
  94   //     An AliDCSValue consists of timestamp and a value in form of a AliSimpleValue
  95
  96   // In this example 6 aliases exists: DCSAlias1 ... DCSAlias6
  97   // Each contains 1000 values randomly generated by TRandom::Gaus + 5*nAlias
  98
  99   TMap* aliasMap = new TMap;
 100   aliasMap->SetOwner(1);
 101
 102   TRandom random;
 103
 104   for(int nAlias=0;nAlias<6;nAlias++)
 105   {
 106     TObjArray* valueSet = new TObjArray;
 107     valueSet->SetOwner(1);
 108
 109     TString aliasName="DCSAlias";
 110     aliasName += nAlias;
 111     //printf("\n\n alias: %s\n\n",aliasName.Data());
 112
 113     for (int timeStamp=0;timeStamp<1000;timeStamp+=10)
 114     {
 115       AliDCSValue* dcsVal = new AliDCSValue((Float_t) (random.Gaus()+5*nAlias), timeStamp);
 116       //printf("%s\n",dcsVal->ToString().Data());
 117       valueSet->Add(dcsVal);
 118     }
 119     aliasMap->Add(new TObjString(aliasName), valueSet);
 120   }
 121
 122   return aliasMap;
 123 }
 124
 125 TMap* ReadDCSAliasMap()
 126 {
 127   // Open a file that contains DCS input data
 128   // The CDB framework is used to open the file, this means the file is located
 129   // in $ALICE_ROOT/SHUTTLE/TestShuttle/TestCDB/<detector>/DCS/Data
 130   // The file contains an AliCDBEntry that contains a TMap with the DCS structure.
 131   // An explanation of the structure can be found in CreateDCSAliasMap()
 132
 133   AliCDBEntry *entry = AliCDBManager::Instance()->Get("DET/DCS/Data", 0);
 134   return dynamic_cast<TMap*> (entry->GetObject());
 135 }
 136
 137 void WriteDCSAliasMap()
 138 {
 139   // This writes the output from CreateDCSAliasMap to a CDB file
 140
 141   TMap* dcsAliasMap = CreateDCSAliasMap();
 142
 143   AliCDBMetaData metaData;
 144         metaData.SetBeamPeriod(0);
 145         metaData.SetResponsible("Responsible person");
 146         metaData.SetComment("Test object for TestPreprocessor.C");
 147
 148   AliCDBId id("DET/DCS/Data", 0, 0);
 149
 150   // initialize location of CDB
 151   AliCDBManager::Instance()->SetDefaultStorage("local://$ALICE_ROOT/SHUTTLE/TestShuttle/TestCDB");
 152
 153   AliCDBManager::Instance()->Put(dcsAliasMap, id, &metaData);
 154 }