new option for tests in alifakedisplay

[u/mrichter/AliRoot.git] / RAW / AliRawCastorDB.cxx

// @(#) $Id$
// Author: Fons Rademakers  26/11/99

/**************************************************************************
 * Copyright(c) 1998-2003, ALICE Experiment at CERN, All rights reserved. *
 *                                                                        *
 * Author: The ALICE Off-line Project.                                    *
 * Contributors are mentioned in the code where appropriate.              *
 *                                                                        *
 * Permission to use, copy, modify and distribute this software and its   *
 * documentation strictly for non-commercial purposes is hereby granted   *
 * without fee, provided that the above copyright notice appears in all   *
 * copies and that both the copyright notice and this permission notice   *
 * appear in the supporting documentation. The authors make no claims     *
 * about the suitability of this software for any purpose. It is          *
 * provided "as is" without express or implied warranty.                  *
 **************************************************************************/

//////////////////////////////////////////////////////////////////////////
//                                                                      //
// AliRawCastorDB                                                       //
//                                                                      //
//////////////////////////////////////////////////////////////////////////

#include <TSystem.h>
#include <TUrl.h>

#include "AliRawCastorDB.h"


ClassImp(AliRawCastorDB)


//______________________________________________________________________________
AliRawCastorDB::AliRawCastorDB(AliRawEventV2 *event,
			       AliESDEvent *esd,
			       Int_t compress,
			       const char* fileName,Int_t basketsize)
   : AliRawDB(event, esd, compress, fileName, basketsize)
{
   // Create a new raw DB that will be accessed via CASTOR and rootd.

   static int init = 0;
   if (!init) {
      // THESE ENVIRONMENT VARIABLES ARE IN PRINCIPLE HARDCODED IN
      // THE CASTOR CLIENT LIBRARY
      // however for sanity we check if they are set by the user
      if (!gSystem->Getenv("RH_HOST"))
         Error("AliRawRFIODB", "RH_HOST not set");
      if (!gSystem->Getenv("SVCCLASS"))
         Error("AliRawRFIODB", "SVCCLASS not set");
      init = 1;
   }

#if ROOT_VERSION_CODE < ROOT_VERSION(5,99,0)
   if (fRawDB) fRawDB->UseCache(50, 0x200000);  //0x100000 = 1MB)
#endif
}

//______________________________________________________________________________
const char *AliRawCastorDB::GetFileName() const
{
   // Return filename based on hostname and date and time. This will make
   // each file unique. Also the directory will be made unique for each
   // day by adding the date to the fs. Assumes there is always enough
   // space on the device.

   static TString fname;

   TString fs  = fFS1;
   TString fsr = fs;
   fsr.ReplaceAll("castor:", "rfio:");
   TDatime dt;

   // make a new subdirectory for each day
   fs += "/adc-";
   fs += dt.GetDate();

   fsr += "/adc-";
   fsr += dt.GetDate();

   Long_t id, size, flags, time;
   if (gSystem->GetPathInfo(fsr, &id, &size, &flags, &time) == 1) {
      // directory does not exist, create it
      if (gSystem->mkdir(fsr, kTRUE) == -1) {
         Error("GetFileName", "cannot create dir %s, using %s", fsr.Data(),
               fFS1.Data());
         fs = fFS1;
      }
   }
   // FIXME: should check if fs is a directory

   TString hostname = gSystem->HostName();
   Int_t pos;
   if ((pos = hostname.Index(".")) != kNPOS)
      hostname.Remove(pos);

   fname = fs + "/" + hostname + "_";
   fname += dt.GetDate();
   fname += "_";
   fname += dt.GetTime();
   fname += ".root";

   return fname;
}

//______________________________________________________________________________
Long64_t AliRawCastorDB::Close()
{
   // Close raw CASTOR/rootd DB.

   if (!fRawDB) return 0;

   if (!fRawDB->IsOpen()) return 0;

   fRawDB->cd();

   // Write the tree.
   Bool_t error = kFALSE;
   if (fTree)
     if (fTree->Write() == 0)
       error = kTRUE;
   if (fESDTree)
     if (fESDTree->Write() == 0)
       error = kTRUE;

   // Close DB, this also deletes the fTree
   fRawDB->Close();

   fTree = NULL;

   Long64_t filesize = fRawDB->GetEND();

   if (fDeleteFiles) {
      TUrl u(fRawDB->GetName());
      gSystem->Exec(Form("rfrm %s", u.GetFile()));
   }

   delete fRawDB;
   fRawDB = 0;

   if(!error)
     return filesize;
   else
     return -1;
}