Adding functionality to visualize result for defferent setttings (Marian)

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

diff --git a/RAW/AliRawRFIODB.cxx b/RAW/AliRawRFIODB.cxx
index 899eb3346f61c1f674834ee7b4975c622e1e218d..92441bed4b79eac43370b4eefac9aab29866cadc 100644 (file)
--- a/RAW/AliRawRFIODB.cxx
+++ b/RAW/AliRawRFIODB.cxx
@@ -1,4 +1,4 @@
-// @(#)alimdc:$Name$:$Id$
+// @(#) $Id$
 // Author: Fons Rademakers  26/11/99
 
 /**************************************************************************
@@ -33,10 +33,10 @@ ClassImp(AliRawRFIODB)
 
 //______________________________________________________________________________
 AliRawRFIODB::AliRawRFIODB(AliRawEvent *event,
-                          AliESD *esd,
+                          AliESDEvent *esd,
                           Int_t compress,
-                          const char* fileName)
-   : AliRawDB(event, esd, compress, fileName)
+                          const char* fileName,Int_t basketsize)
+   : AliRawDB(event, esd, compress, fileName, basketsize)
 {
    // Create a new raw DB that will be accessed via RFIO.
 
@@ -103,21 +103,29 @@ const char *AliRawRFIODB::GetFileName() const
 }
 
 //______________________________________________________________________________
-void AliRawRFIODB::Close()
+Long64_t AliRawRFIODB::Close()
 {
    // Close raw RFIO DB.
 
-   if (!fRawDB) return;
+   if (!fRawDB) return 0;
+
+   if (!fRawDB->IsOpen()) return 0;
 
    fRawDB->cd();
 
    // Write the tree.
-   fTree->Write();
-   if (fESDTree) fESDTree->Write();
+   Bool_t error = kFALSE;
+   if (fTree->Write() == 0)
+     error = kTRUE;
+   if (fESDTree)
+     if (fESDTree->Write() == 0)
+       error = kTRUE;
 
    // Close DB, this also deletes the fTree
    fRawDB->Close();
 
+   Long64_t filesize = fRawDB->GetEND();
+
    if (fDeleteFiles) {
       TUrl u(fRawDB->GetName());
       gSystem->Exec(Form("rfrm %s", u.GetFile()));
@@ -125,4 +133,9 @@ void AliRawRFIODB::Close()
 
    delete fRawDB;
    fRawDB = 0;
+
+   if(!error)
+     return filesize;
+   else
+     return -1;
 }