FairRootFileSink.h

Go to the documentation of this file.

/********************************************************************************
 *    Copyright (C) 2014 GSI Helmholtzzentrum fuer Schwerionenforschung GmbH    *
 *                                                                              *
 *              This software is distributed under the terms of the             *
 *         GNU Lesser General Public Licence version 3 (LGPL) version 3,        *
 *                  copied verbatim in the file "LICENSE"                       *
 ********************************************************************************/
// -----------------------------------------------------------------------------
// -----                                                                   -----
// -----                             FairRootFileSink                      -----
// -----                    Created 22.12.2017 by R. Karabowicz            -----
// -----                                                                   -----
// -----------------------------------------------------------------------------

#ifndef __FAIRROOT__FairRootFileSink__
#define __FAIRROOT__FairRootFileSink__

#include "FairSink.h"

#include <Rtypes.h>
#include <TFile.h>
#include <TString.h>
#include <typeinfo>

class FairEventHeader;
class FairFileHeader;
class TBranch;
class TObjArray;
class TObject;
class TTree;
class TFolder;

class FairRootFileSink : public FairSink
{
  public:
    FairRootFileSink(TFile* f, const char* Title = "OutputRootFile");
    FairRootFileSink(const TString* RootFileName, const char* Title = "OutputRootFile");
    FairRootFileSink(const TString RootFileName, const char* Title = "OutputRootFile");
    //  FairRootFileSink(const FairRootFileSink& file);
    virtual ~FairRootFileSink();

    virtual Bool_t InitSink();
    virtual void Close();
    virtual void Reset();

    virtual Sink_Type GetSinkType() { return kFILESINK; }

    virtual void FillEventHeader(FairEventHeader* feh);

    virtual TFile* OpenRootFile(TString fileName = "");
    TFile* GetRootFile() { return fRootFile; }
    virtual TString GetFileName() { return (fRootFile ? fRootFile->GetName() : ""); }

    virtual void SetOutTree(TTree* fTree) { fOutTree = fTree; }
    TTree* GetOutTree() { return fOutTree; }

    virtual void Fill();

    virtual Int_t Write(const char* name = 0, Int_t option = 0, Int_t bufsize = 0);

    virtual void RegisterImpl(const char*, const char*, void*);
    virtual void RegisterAny(const char* brname, const std::type_info& oi, const std::type_info& pi, void* obj);

    virtual void WriteFolder();
    virtual bool CreatePersistentBranchesAny();

    virtual void WriteObject(TObject* f, const char*, Int_t option = 0);
    virtual void WriteGeometry();

    virtual FairSink* CloneSink();

  private:
    TString fOutputTitle;
    TFile* fRootFile;
    TTree* fOutTree;
    TObjArray* fListFolder;   

    TFolder* fCbmout;
    Bool_t fIsInitialized;

    FairRootFileSink(const FairRootFileSink&);
    FairRootFileSink operator=(const FairRootFileSink&);

    void TruncateBranchNames();
    void TruncateBranchNames(TBranch* b, TString ffn);
    // bool CreatePersistentBranchesAny();

    FairFileHeader* fFileHeader;   

    ClassDef(FairRootFileSink, 1);
};

#endif /* defined(__FAIRROOT__FairRootFileSink__) */