Logo ROOT   6.18/05
Reference Guide
TFileMerger.h
Go to the documentation of this file.
1// @(#)root/io:$Id$
2// Author: Andreas Peters + Fons Rademakers 26/5/2005
3
4/*************************************************************************
5 * Copyright (C) 1995-2005, Rene Brun and Fons Rademakers. *
6 * All rights reserved. *
7 * *
8 * For the licensing terms see $ROOTSYS/LICENSE. *
9 * For the list of contributors see $ROOTSYS/README/CREDITS. *
10 *************************************************************************/
11
12#ifndef ROOT_TFileMerger
13#define ROOT_TFileMerger
14
15#include "TList.h"
16#include "TObject.h"
17#include "TString.h"
18#include "TStopwatch.h"
19
20#include <memory>
21
22class TList;
23class TFile;
24class TDirectory;
25
26namespace ROOT {
27class TIOFeatures;
28} // namespace ROOT
29
30class TFileMerger : public TObject {
31private:
33
34 TFileMerger(const TFileMerger&) = delete;
36
37protected:
38 TStopwatch fWatch; ///< Stop watch to measure file copy speed
39 TList fFileList; ///< A list the file (TFile*) which shall be merged
40 TFile *fOutputFile{nullptr}; ///< The outputfile for merging
41 TString fOutputFilename; ///< The name of the outputfile for merging
42 Bool_t fFastMethod{kTRUE}; ///< True if using Fast merging algorithm (default)
43 Bool_t fNoTrees{kFALSE}; ///< True if Trees should not be merged (default is kFALSE)
44 Bool_t fExplicitCompLevel{kFALSE}; ///< True if the user explicitly requested a compressio level change (default kFALSE)
45 Bool_t fCompressionChange{kFALSE}; ///< True if the output and input have different compression level (default kFALSE)
46 Int_t fPrintLevel{0}; ///< How much information to print out at run time
47 TString fMergeOptions; ///< Options (in string format) to be passed down to the Merge functions
48 TIOFeatures *fIOFeatures{nullptr}; ///< IO features to use in the output file.
49 TString fMsgPrefix{"TFileMerger"}; ///< Prefix to be used when printing informational message (default TFileMerger)
50
51 Int_t fMaxOpenedFiles; ///< Maximum number of files opened at the same time by the TFileMerger
52 Bool_t fLocal; ///< Makes local copies of merging files if True (default is kTRUE)
53 Bool_t fHistoOneGo; ///< Merger histos in one go (default is kTRUE)
54 TString fObjectNames; ///< List of object names to be either merged exclusively or skipped
55 TList fMergeList; ///< list of TObjString containing the name of the files need to be merged
56 TList fExcessFiles; ///<! List of TObjString containing the name of the files not yet added to fFileList due to user or system limitiation on the max number of files opened.
57
59 virtual Bool_t AddFile(TFile *source, Bool_t own, Bool_t cpProgress);
60 virtual Bool_t MergeRecursive(TDirectory *target, TList *sourcelist, Int_t type = kRegular | kAll);
61
62public:
63 /// Type of the partial merge
65 kRegular = 0, ///< Normal merge, overwritting the output file.
66 kIncremental = BIT(1), ///< Merge the input file with the content of the output file (if already exising).
67 kResetable = BIT(2), ///< Only the objects with a MergeAfterReset member function.
68 kNonResetable = BIT(3), ///< Only the objects without a MergeAfterReset member function.
69
70 kAll = BIT(2)|BIT(3), ///< Merge all type of objects (default)
71 kAllIncremental = kIncremental | kAll, ///< Merge incrementally all type of objects.
72
73 kOnlyListed = BIT(4), ///< Only the objects specified in fObjectNames list
74 kSkipListed = BIT(5), ///< Skip objects specified in fObjectNames list
75 kKeepCompression= BIT(6) ///< Keep compression level unchanged for each input files
76 };
77
78 TFileMerger(Bool_t isLocal = kTRUE, Bool_t histoOneGo = kTRUE);
79 virtual ~TFileMerger();
80
81 Int_t GetPrintLevel() const { return fPrintLevel; }
82 void SetPrintLevel(Int_t level) { fPrintLevel = level; }
84 const char *GetOutputFileName() const { return fOutputFilename; }
86 TFile *GetOutputFile() const { return fOutputFile; }
88 void SetMaxOpenedFiles(Int_t newmax);
89 const char *GetMsgPrefix() const { return fMsgPrefix; }
90 void SetMsgPrefix(const char *prefix);
91 const char *GetMergeOptions() { return fMergeOptions; }
92 void SetMergeOptions(const TString &options) { fMergeOptions = options; }
93 void SetMergeOptions(const std::string_view &options) { fMergeOptions = options; }
94 void SetIOFeatures(ROOT::TIOFeatures &features) { fIOFeatures = &features; }
95 void AddObjectNames(const char *name) {fObjectNames += name; fObjectNames += " ";}
96 const char *GetObjectNames() const {return fObjectNames.Data();}
98
99 //--- file management interface
100 virtual Bool_t SetCWD(const char * /*path*/) { MayNotUse("SetCWD"); return kFALSE; }
101 virtual const char *GetCWD() { MayNotUse("GetCWD"); return 0; }
102
103 //--- file merging interface
104 virtual void Reset();
105 virtual Bool_t AddFile(const char *url, Bool_t cpProgress = kTRUE);
106 virtual Bool_t AddFile(TFile *source, Bool_t cpProgress = kTRUE);
107 virtual Bool_t AddAdoptFile(TFile *source, Bool_t cpProgress = kTRUE);
108 virtual Bool_t OutputFile(const char *url, Bool_t force);
109 virtual Bool_t OutputFile(const char *url, Bool_t force, Int_t compressionLevel);
110 virtual Bool_t OutputFile(const char *url, const char *mode = "RECREATE");
111 virtual Bool_t OutputFile(const char *url, const char *mode, Int_t compressionLevel);
112 virtual Bool_t OutputFile(std::unique_ptr<TFile> file);
113 virtual void PrintFiles(Option_t *options);
114 virtual Bool_t Merge(Bool_t = kTRUE);
116 virtual void SetFastMethod(Bool_t fast=kTRUE) {fFastMethod = fast;}
117 virtual void SetNotrees(Bool_t notrees=kFALSE) {fNoTrees = notrees;}
118 virtual void RecursiveRemove(TObject *obj);
119
120 ClassDef(TFileMerger, 6) // File copying and merging services
121};
122
123#endif
124
int Int_t
Definition: RtypesCore.h:41
const Bool_t kFALSE
Definition: RtypesCore.h:88
bool Bool_t
Definition: RtypesCore.h:59
const Bool_t kTRUE
Definition: RtypesCore.h:87
const char Option_t
Definition: RtypesCore.h:62
#define ClassDef(name, id)
Definition: Rtypes.h:326
#define BIT(n)
Definition: Rtypes.h:83
char name[80]
Definition: TGX11.cxx:109
int type
Definition: TGX11.cxx:120
TIOFeatures provides the end-user with the ability to change the IO behavior of data written via a TT...
Definition: TIOFeatures.hxx:62
Describe directory structure in memory.
Definition: TDirectory.h:34
This class provides file copy and merging services.
Definition: TFileMerger.h:30
virtual void SetFastMethod(Bool_t fast=kTRUE)
Definition: TFileMerger.h:116
TString fObjectNames
List of object names to be either merged exclusively or skipped.
Definition: TFileMerger.h:54
virtual Bool_t OutputFile(const char *url, Bool_t force)
Open merger output file.
TList fMergeList
list of TObjString containing the name of the files need to be merged
Definition: TFileMerger.h:55
TFile * GetOutputFile() const
Definition: TFileMerger.h:86
virtual Bool_t AddFile(TFile *source, Bool_t own, Bool_t cpProgress)
Add the TFile to this file merger and give ownership of the TFile to this object (unless kFALSE is re...
virtual void PrintFiles(Option_t *options)
Print list of files being merged.
const char * GetMsgPrefix() const
Definition: TFileMerger.h:89
Bool_t fHistoOneGo
Merger histos in one go (default is kTRUE)
Definition: TFileMerger.h:53
virtual Bool_t MergeRecursive(TDirectory *target, TList *sourcelist, Int_t type=kRegular|kAll)
Merge all objects in a directory.
TList fFileList
A list the file (TFile*) which shall be merged.
Definition: TFileMerger.h:39
virtual Bool_t Merge(Bool_t=kTRUE)
Merge the files.
TList * GetMergeList()
Definition: TFileMerger.h:85
TString fOutputFilename
The name of the outputfile for merging.
Definition: TFileMerger.h:41
const char * GetObjectNames() const
Definition: TFileMerger.h:96
virtual void RecursiveRemove(TObject *obj)
Intercept the case where the output TFile is deleted!
const char * GetOutputFileName() const
Definition: TFileMerger.h:84
TString fMsgPrefix
Prefix to be used when printing informational message (default TFileMerger)
Definition: TFileMerger.h:49
void ClearObjectNames()
Definition: TFileMerger.h:97
TIOFeatures * fIOFeatures
IO features to use in the output file.
Definition: TFileMerger.h:48
TFileMerger(const TFileMerger &)=delete
const char * GetMergeOptions()
Definition: TFileMerger.h:91
void SetMsgPrefix(const char *prefix)
Set the prefix to be used when printing informational message.
Int_t GetPrintLevel() const
Definition: TFileMerger.h:81
Bool_t fNoTrees
True if Trees should not be merged (default is kFALSE)
Definition: TFileMerger.h:43
Bool_t HasCompressionChange() const
Definition: TFileMerger.h:83
void SetPrintLevel(Int_t level)
Definition: TFileMerger.h:82
void AddObjectNames(const char *name)
Definition: TFileMerger.h:95
EPartialMergeType
Type of the partial merge.
Definition: TFileMerger.h:64
@ kAll
Merge all type of objects (default)
Definition: TFileMerger.h:70
@ kIncremental
Merge the input file with the content of the output file (if already exising).
Definition: TFileMerger.h:66
@ kKeepCompression
Keep compression level unchanged for each input files.
Definition: TFileMerger.h:75
@ kSkipListed
Skip objects specified in fObjectNames list.
Definition: TFileMerger.h:74
@ kNonResetable
Only the objects without a MergeAfterReset member function.
Definition: TFileMerger.h:68
@ kResetable
Only the objects with a MergeAfterReset member function.
Definition: TFileMerger.h:67
@ kOnlyListed
Only the objects specified in fObjectNames list.
Definition: TFileMerger.h:73
@ kAllIncremental
Merge incrementally all type of objects.
Definition: TFileMerger.h:71
@ kRegular
Normal merge, overwritting the output file.
Definition: TFileMerger.h:65
void SetMergeOptions(const TString &options)
Definition: TFileMerger.h:92
Bool_t fExplicitCompLevel
True if the user explicitly requested a compressio level change (default kFALSE)
Definition: TFileMerger.h:44
Bool_t fCompressionChange
True if the output and input have different compression level (default kFALSE)
Definition: TFileMerger.h:45
virtual ~TFileMerger()
Cleanup.
virtual void SetNotrees(Bool_t notrees=kFALSE)
Definition: TFileMerger.h:117
Int_t fPrintLevel
How much information to print out at run time.
Definition: TFileMerger.h:46
void SetMaxOpenedFiles(Int_t newmax)
Set a limit to the number of files that TFileMerger will open simultaneously.
TString fMergeOptions
Options (in string format) to be passed down to the Merge functions.
Definition: TFileMerger.h:47
Int_t GetMaxOpenedFiles() const
Definition: TFileMerger.h:87
Bool_t OpenExcessFiles()
Open up to fMaxOpenedFiles of the excess files.
void SetIOFeatures(ROOT::TIOFeatures &features)
Definition: TFileMerger.h:94
TList fExcessFiles
! List of TObjString containing the name of the files not yet added to fFileList due to user or syste...
Definition: TFileMerger.h:56
virtual const char * GetCWD()
Definition: TFileMerger.h:101
TStopwatch fWatch
Stop watch to measure file copy speed.
Definition: TFileMerger.h:38
TFile * fOutputFile
The outputfile for merging.
Definition: TFileMerger.h:40
void SetMergeOptions(const std::string_view &options)
Definition: TFileMerger.h:93
virtual Bool_t PartialMerge(Int_t type=kAll|kIncremental)
Merge the files.
Bool_t fLocal
Makes local copies of merging files if True (default is kTRUE)
Definition: TFileMerger.h:52
TFileMerger & operator=(const TFileMerger &)=delete
virtual void Reset()
Reset merger file list.
Int_t fMaxOpenedFiles
Maximum number of files opened at the same time by the TFileMerger.
Definition: TFileMerger.h:51
virtual Bool_t SetCWD(const char *)
Definition: TFileMerger.h:100
virtual Bool_t AddAdoptFile(TFile *source, Bool_t cpProgress=kTRUE)
Add the TFile to this file merger and give ownership of the TFile to this object (unless kFALSE is re...
Bool_t fFastMethod
True if using Fast merging algorithm (default)
Definition: TFileMerger.h:42
A ROOT file is a suite of consecutive data records (TKey instances) with a well defined format.
Definition: TFile.h:48
A doubly linked list.
Definition: TList.h:44
Mother of all ROOT objects.
Definition: TObject.h:37
void MayNotUse(const char *method) const
Use this method to signal that a method (defined in a base class) may not be called in a derived clas...
Definition: TObject.cxx:933
Stopwatch class.
Definition: TStopwatch.h:28
Basic string class.
Definition: TString.h:131
void Clear()
Clear string without changing its capacity.
Definition: TString.cxx:1176
const char * Data() const
Definition: TString.h:364
basic_string_view< char > string_view
Namespace for new ROOT classes and functions.
Definition: StringConv.hxx:21
Definition: file.py:1