RAW/AliRawCastorDB.cxx

   1 // @(#)alimdc:$Name$:$Id$
   2 // Author: Fons Rademakers  26/11/99
   3
   4 /**************************************************************************
   5  * Copyright(c) 1998-2003, ALICE Experiment at CERN, All rights reserved. *
   6  *                                                                        *
   7  * Author: The ALICE Off-line Project.                                    *
   8  * Contributors are mentioned in the code where appropriate.              *
   9  *                                                                        *
  10  * Permission to use, copy, modify and distribute this software and its   *
  11  * documentation strictly for non-commercial purposes is hereby granted   *
  12  * without fee, provided that the above copyright notice appears in all   *
  13  * copies and that both the copyright notice and this permission notice   *
  14  * appear in the supporting documentation. The authors make no claims     *
  15  * about the suitability of this software for any purpose. It is          *
  16  * provided "as is" without express or implied warranty.                  *
  17  **************************************************************************/
  18
  19 //////////////////////////////////////////////////////////////////////////
  20 //                                                                      //
  21 // AliRawCastorDB                                                       //
  22 //                                                                      //
  23 //////////////////////////////////////////////////////////////////////////
  24
  25 #include <TSystem.h>
  26 #include <TUrl.h>
  27
  28 #include "AliRawCastorDB.h"
  29
  30
  31 ClassImp(AliRawCastorDB)
  32
  33
  34 //______________________________________________________________________________
  35 AliRawCastorDB::AliRawCastorDB(AliRawEvent *event,
  36                                AliESD *esd,
  37                                Int_t compress,
  38                                const char* fileName)
  39    : AliRawDB(event, esd, compress, fileName)
  40 {
  41    // Create a new raw DB that will be accessed via CASTOR and rootd.
  42
  43    static int init = 0;
  44    // Set STAGE_POOL environment variable to current host
  45    if (!init) {
  46       // THESE ENVIRONMENT SYMBOLS ARE NOW DEFINED BY THE ALICE DATE SETUP
  47       // THEREFORE WE SHALL NOT USE ANY HARDCODED VALUES BUT RATHER USE
  48       // WHATEVER HAS BEEN SET IN THE DATE SITE
  49       //gSystem->Setenv("STAGE_POOL", "lcg00");
  50       //gSystem->Setenv("STAGE_HOST", "stage013");
  51
  52       // however for sanity we check if they are really set
  53       if (!gSystem->Getenv("STAGE_POOL"))
  54          Error("AliRawRFIODB", "STAGE_POOL not set");
  55       if (!gSystem->Getenv("STAGE_HOST"))
  56          Error("AliRawRFIODB", "STAGE_HOST not set");
  57       init = 1;
  58    }
  59
  60    if (fRawDB) fRawDB->UseCache(50, 0x200000);  //0x100000 = 1MB)
  61 }
  62
  63 //______________________________________________________________________________
  64 const char *AliRawCastorDB::GetFileName() const
  65 {
  66    // Return filename based on hostname and date and time. This will make
  67    // each file unique. Also the directory will be made unique for each
  68    // day by adding the date to the fs. Assumes there is always enough
  69    // space on the device.
  70
  71    static TString fname;
  72
  73    TString fs  = fFS1;
  74    TString fsr = fs;
  75    fsr.ReplaceAll("castor:", "rfio:");
  76    TDatime dt;
  77
  78    // make a new subdirectory for each day
  79    fs += "/adc-";
  80    fs += dt.GetDate();
  81
  82    fsr += "/adc-";
  83    fsr += dt.GetDate();
  84
  85    Long_t id, size, flags, time;
  86    if (gSystem->GetPathInfo(fsr, &id, &size, &flags, &time) == 1) {
  87       // directory does not exist, create it
  88       if (gSystem->mkdir(fsr, kTRUE) == -1) {
  89          Error("GetFileName", "cannot create dir %s, using %s", fsr.Data(),
  90                fFS1.Data());
  91          fs = fFS1;
  92       }
  93    }
  94    // FIXME: should check if fs is a directory
  95
  96    TString hostname = gSystem->HostName();
  97    Int_t pos;
  98    if ((pos = hostname.Index(".")) != kNPOS)
  99       hostname.Remove(pos);
 100
 101    fname = fs + "/" + hostname + "_";
 102    fname += dt.GetDate();
 103    fname += "_";
 104    fname += dt.GetTime();
 105    fname += ".root";
 106
 107    return fname;
 108 }
 109
 110 //______________________________________________________________________________
 111 void AliRawCastorDB::Close()
 112 {
 113    // Close raw CASTOR/rootd DB.
 114
 115    if (!fRawDB) return;
 116
 117    fRawDB->cd();
 118
 119    // Write the tree.
 120    fTree->Write();
 121    if (fESDTree) fESDTree->Write();
 122
 123    // Close DB, this also deletes the fTree
 124    fRawDB->Close();
 125
 126    if (fDeleteFiles) {
 127       TUrl u(fRawDB->GetName());
 128       gSystem->Exec(Form("rfrm %s", u.GetFile()));
 129    }
 130
 131    delete fRawDB;
 132    fRawDB = 0;
 133 }