RAW/AliRawCastorDB.cxx

   1 // @(#) $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(AliRawEventV2 *event,
  36                                AliESDEvent *esd,
  37                                Int_t compress,
  38                                const char* fileName,Int_t basketsize)
  39    : AliRawDB(event, esd, compress, fileName, basketsize)
  40 {
  41    // Create a new raw DB that will be accessed via CASTOR and rootd.
  42
  43    static int init = 0;
  44    if (!init) {
  45       // THESE ENVIRONMENT VARIABLES ARE IN PRINCIPLE HARDCODED IN
  46       // THE CASTOR CLIENT LIBRARY
  47       // however for sanity we check if they are set by the user
  48       if (!gSystem->Getenv("RH_HOST"))
  49          Error("AliRawRFIODB", "RH_HOST not set");
  50       if (!gSystem->Getenv("SVCCLASS"))
  51          Error("AliRawRFIODB", "SVCCLASS not set");
  52       init = 1;
  53    }
  54
  55    if (fRawDB) fRawDB->UseCache(50, 0x200000);  //0x100000 = 1MB)
  56 }
  57
  58 //______________________________________________________________________________
  59 const char *AliRawCastorDB::GetFileName() const
  60 {
  61    // Return filename based on hostname and date and time. This will make
  62    // each file unique. Also the directory will be made unique for each
  63    // day by adding the date to the fs. Assumes there is always enough
  64    // space on the device.
  65
  66    static TString fname;
  67
  68    TString fs  = fFS1;
  69    TString fsr = fs;
  70    fsr.ReplaceAll("castor:", "rfio:");
  71    TDatime dt;
  72
  73    // make a new subdirectory for each day
  74    fs += "/adc-";
  75    fs += dt.GetDate();
  76
  77    fsr += "/adc-";
  78    fsr += dt.GetDate();
  79
  80    Long_t id, size, flags, time;
  81    if (gSystem->GetPathInfo(fsr, &id, &size, &flags, &time) == 1) {
  82       // directory does not exist, create it
  83       if (gSystem->mkdir(fsr, kTRUE) == -1) {
  84          Error("GetFileName", "cannot create dir %s, using %s", fsr.Data(),
  85                fFS1.Data());
  86          fs = fFS1;
  87       }
  88    }
  89    // FIXME: should check if fs is a directory
  90
  91    TString hostname = gSystem->HostName();
  92    Int_t pos;
  93    if ((pos = hostname.Index(".")) != kNPOS)
  94       hostname.Remove(pos);
  95
  96    fname = fs + "/" + hostname + "_";
  97    fname += dt.GetDate();
  98    fname += "_";
  99    fname += dt.GetTime();
 100    fname += ".root";
 101
 102    return fname;
 103 }
 104
 105 //______________________________________________________________________________
 106 Long64_t AliRawCastorDB::Close()
 107 {
 108    // Close raw CASTOR/rootd DB.
 109
 110    if (!fRawDB) return 0;
 111
 112    if (!fRawDB->IsOpen()) return 0;
 113
 114    fRawDB->cd();
 115
 116    // Write the tree.
 117    Bool_t error = kFALSE;
 118    if (fTree)
 119      if (fTree->Write() == 0)
 120        error = kTRUE;
 121    if (fESDTree)
 122      if (fESDTree->Write() == 0)
 123        error = kTRUE;
 124
 125    // Close DB, this also deletes the fTree
 126    fRawDB->Close();
 127
 128    fTree = NULL;
 129
 130    Long64_t filesize = fRawDB->GetEND();
 131
 132    if (fDeleteFiles) {
 133       TUrl u(fRawDB->GetName());
 134       gSystem->Exec(Form("rfrm %s", u.GetFile()));
 135    }
 136
 137    delete fRawDB;
 138    fRawDB = 0;
 139
 140    if(!error)
 141      return filesize;
 142    else
 143      return -1;
 144 }