#3 initial skeleton

FlatTree/interface/KFlatTreeAnalyzer.h

@@ -0,0 +1,35 @@
+#ifndef KFlatTreeAnalyzer_h
+#define KFlatTreeAnalyzer_h
+
+#include <iostream>
+
+#include "TFile.h"
+#include "TH1F.h"
+#include "TTree.h"
+#include "TDirectory.h"
+#include "TMath.h"
+
+#include "KrAFT/FlatTree/interface/FlatEvent.h"
+
+#include "Math/GenVector/LorentzVector.h"
+#include "Math/GenVector/VectorUtil.h"
+
+using namespace ROOT::Math::VectorUtil;
+typedef ROOT::Math::LorentzVector<ROOT::Math::PxPyPzE4D<double> > LorentzVector;
+
+void printEntryFraction(int i, int n);
+
+class KFlatTreeAnalyzer
+{
+public:
+  KFlatTreeAnalyzer(const std::string inputFileName, const std::string outputFileName);
+  ~KFlatTreeAnalyzer();
+  void analyze();
+
+private:
+  TFile* inputFile_, * outputFile_;
+  TTree* tree_;
+  FlatEvent* event_;
+};
+
+#endif

FlatTree/src/KFlatTreeAnalyzer.cc

@@ -0,0 +1,35 @@
+#include "KrAFT/FlatTree/interface/KFlatTreeAnalyzer.h"
+#include "TString.h"
+
+using namespace std;
+
+void printEntryFraction(int i, int n)
+{
+  static int digit = 1;
+  if ( i%max(1, digit/10) == 0 )
+  {
+    cout << Form(TString("% ")+Form("%d", int(log10(n))+2)+"d/%d processed\r", i, n);
+    cout.flush();
+  }
+  if ( i%digit == 0 ) digit *= 10;
+
+  if ( i==(n-1) )
+  {
+    cout << Form(TString("% ")+Form("%d", int(log10(n))+2)+"d/%d processed", i, n) << endl;
+    digit = 1;
+  }
+}
+
+KFlatTreeAnalyzer::KFlatTreeAnalyzer(const string inputFileName, const string outputFileName)
+{
+  inputFile_ = TFile::Open(inputFileName.c_str());
+  outputFile_ = TFile::Open(outputFileName.c_str(), "RECREATE");
+}
+
+KFlatTreeAnalyzer::~KFlatTreeAnalyzer()
+{
+  inputFile_->Close();
+  outputFile_->Write();
+  outputFile_->Close();
+}
+