--- /dev/null
+#include "TFileMerger.h"
+#include "TSystem.h"
+
+#include <map>
+#include <iostream>
+#include <fstream>
+#include <string>
+
+std::map<int, int> runToFill; // maps: run -> fill
+std::map<int, TFileMerger*> fillToMerger;
+
+void makeFillMap()
+{
+}
+
+void FillMerge(const TString filelist="filelist.txt",
+ const TString runlist="runlist.txt",
+ const TString runFillFile = "run_fill.txt")
+{
+ // Use to merge run files to fill files
+ // For some reason, it does not work without compiling:
+ // .L FillMerge.C+g
+ // FillMerge()
+ //
+ // run_fill.txt must be a two column txt file,
+ // column 1 being run and column 2 being fill.
+
+ //TGrid::Connect("alien://");
+
+
+ // Make runToFill map
+ std::ifstream rfin;
+ rfin.open(runFillFile.Data());
+
+ while (1) {
+ int run=0, fill=0;
+ rfin >> run >> fill;
+ if (!rfin.good()) break;
+ runToFill[run] = fill;
+ }
+
+
+ // Make FillMergers and add run files to them.
+ std::ifstream in;
+ in.open(filelist.Data());
+ std::ifstream inRuns;
+ inRuns.open(runlist.Data());
+ char rootFileName[256];
+ Int_t runNumber=0;
+ while (1) {
+ in >> rootFileName;
+ if (!in.good()) break;
+ inRuns >> runNumber;
+ if (!inRuns.good()) break;
+
+ if(169094 == runNumber )
+ continue;
+
+ if( ! runToFill.count(runNumber) ) {
+ Printf("can't map run %d to fill, not in file %s", runNumber, runFillFile.Data());
+ continue;
+ }
+
+ int fill = runToFill[runNumber];
+
+ if( ! fillToMerger.count( fill ) ) { // if no merger for fill
+ fillToMerger[fill] = new TFileMerger(); // create merger
+ gSystem->mkdir("fillMerge");
+ fillToMerger[fill]->OutputFile(Form("fillMerge/%d.root", fill));
+ }
+
+ fillToMerger[fill]->AddFile(rootFileName);
+ }
+
+ std::ofstream fls("fillList.txt");
+ std::ofstream ffls("fillFileList.txt");
+
+
+ // Merge:
+ std::map<int, TFileMerger*>::iterator it;
+ for( it = fillToMerger.begin(); it != fillToMerger.end(); ++it ) {
+ TFileMerger* fm = (*it).second;
+ Printf("merging %s", fm->GetOutputFileName());
+ fm->Merge();
+ fls << (*it).first << std::endl;
+ ffls << fm->GetOutputFileName() << std::endl;
+ delete fm;
+ }
+
+}
git://git.uio.no / u / mrichter / AliRoot.git / commitdiff