Simple class to export/import data in the HEP data format (import very limited in...

diff --git a/ANALYSIS/AliHEPDataParser.cxx b/ANALYSIS/AliHEPDataParser.cxx
new file mode 100644 (file)
index 0000000..bcb4ae6
--- /dev/null
+++ b/ANALYSIS/AliHEPDataParser.cxx
@@ -0,0 +1,222 @@
+//-------------------------------------------------------------------------
+//           Implementation of Class AliHEPDataParser
+//
+//  This class is used to save the content of hisograms/graphs in the
+//  HEP data format and viceversa. The HEP data format is not strictly
+//  defined and there are many variants, the class only support a few
+//  of them. More will be added as needed.  The input can be a set of
+//  2 TH1, TGraphAsymmErrors or TGraphErrors (one for the stat and one
+//  for the syst error). If the second one is a null pointer, only the
+//  stat error is printed. The class can also import hepdata ascii
+//  file (very preliminary)
+// 
+//  Author: Michele Floris, CERN
+//-------------------------------------------------------------------------
+
+
+#include "AliHEPDataParser.h"
+#include "AliLog.h"
+#include "TGraphAsymmErrors.h"
+#include "TGraph.h"
+#include "TGraphErrors.h"
+#include "TH1.h"
+#include "TObjArray.h"
+#include "TObjString.h"
+#include <fstream>
+#include <iostream>
+using namespace std;
+
+ClassImp(AliHEPDataParser)
+
+AliHEPDataParser::AliHEPDataParser() : TObject(), fHistStat(0),  fHistSyst(0),  fGraphStat(0),  fGraphSyst(0),  fHEPDataFileLines(0), fValueName("")
+{
+  // default ctor
+
+}
+
+AliHEPDataParser::AliHEPDataParser(TH1 * hStat, TH1 * hSyst): TObject(), fHistStat(0),  fHistSyst(0),  fGraphStat(0),  fGraphSyst(0),  fHEPDataFileLines(0), fValueName("y")
+{
+  //ctor
+  fHistStat = hStat;
+  fHistSyst = hSyst;
+  fHEPDataFileLines = new TObjArray;
+
+}
+AliHEPDataParser::AliHEPDataParser(TGraph * grStat, TGraph * grSyst): TObject(), fHistStat(0),  fHistSyst(0),  fGraphStat(0),  fGraphSyst(0),  fHEPDataFileLines(0), fValueName("")
+{
+  // ctor
+  fGraphStat = grStat;
+  fGraphSyst = grSyst;
+  fHEPDataFileLines = new TObjArray;
+}
+
+AliHEPDataParser::AliHEPDataParser(const char * hepfileName): TObject(), fHistStat(0),  fHistSyst(0),  fGraphStat(0),  fGraphSyst(0),  fHEPDataFileLines(0), fValueName("y")
+{
+  //ctor
+  // Put results in graphs
+  fGraphSyst = new TGraphAsymmErrors();
+  fGraphStat = new TGraphAsymmErrors();
+  ifstream infile;
+  infile.open(hepfileName);
+  TString line;
+  Int_t ipoints = 0;
+  while (line.ReadLine(infile)) {
+    TObjArray * tokens = line.Tokenize(" ");
+    if(tokens->GetEntries() < 1) {
+      delete tokens;
+      AliError("not enough columns");
+      return;      
+    }
+    // TODO: Assumes the format
+    // binmin binmax y +-stat +-syst. Try to make it smarter...
+    TObjString * binMin = (TObjString*) tokens->At(0);
+    TObjString * binMax = (TObjString*) tokens->At(1);
+    TObjString * value  = (TObjString*) tokens->At(2);
+    TObjString * stat   = (TObjString*) tokens->At(3);
+    TObjString * syst   = (TObjString*) tokens->At(4);
+    stat->String().ReplaceAll("+-","");
+    if(syst) syst->String().ReplaceAll("+-","");
+    if (!binMin->String().Atof()) continue; // skip headers
+    Float_t binCenter = (binMax->String().Atof() + binMin->String().Atof())/2;
+    Float_t binWidth  =  (binMax->String().Atof() - binMin->String().Atof())/2;
+    cout << line.Data() << endl;//<< " " << binMin->String().Atof() <<" " <<  binCenter << " " << binWidth << endl;
+
+
+    fGraphStat->SetPoint(ipoints, binCenter, value->String().Atof());
+    fGraphSyst->SetPoint(ipoints, binCenter, value->String().Atof());
+    ((TGraphAsymmErrors*)fGraphStat)->SetPointError(ipoints, 
+                                                   binWidth,
+                                                   binWidth,
+                                                   stat->String().Atof(),
+                                                   stat->String().Atof());
+    if(syst) ((TGraphAsymmErrors*)fGraphSyst)->SetPointError(ipoints, 
+                                                            binWidth,
+                                                            binWidth,
+                                                            syst->String().Atof(),
+                                                            syst->String().Atof());
+    ipoints++;
+  }
+  infile.close();
+    
+
+}
+
+AliHEPDataParser::~AliHEPDataParser(){
+  // dtor
+  if(fHistStat) delete fHistStat;
+  if(fHistSyst) delete fHistSyst;
+  if(fGraphStat) delete fGraphStat;
+  if(fGraphSyst) delete fGraphSyst;
+  if(fHEPDataFileLines) delete fHEPDataFileLines;
+}
+  
+void AliHEPDataParser::SaveHEPDataFile(const char * hepfileName, Bool_t trueUseGraphFalesUseHisto) {
+  // Fills fHEPDataFileLines and saves its content to a file
+  if(!fHEPDataFileLines)   fHEPDataFileLines = new TObjArray;
+  if(trueUseGraphFalesUseHisto) {
+    if(!fGraphStat) {
+      AliError("Graph not set");
+      return;
+    }
+    Bool_t asym = kFALSE; // check if this has asymmetric errors
+    if       (!strcmp(fGraphStat->ClassName(), "TGraphErrors")) asym = kFALSE;
+    else if  (!strcmp(fGraphStat->ClassName(), "TGraphAsymmErrors")) asym = kTRUE;
+    else     {AliError("Unsupported graph type"); return;}
+    Int_t npoint = fGraphStat->GetN();
+    if(asym) AliInfo("Assymmetric errors");
+    for(Int_t ipoint = 0; ipoint < npoint; ipoint++){            
+      if(ipoint == 0) {
+       if(fGraphSyst) {
+         if (asym)
+           fHEPDataFileLines->Add(new TObjString(Form("BinCenter %s +stat -stat +syst -syst", fValueName.Data()))); 
+         else 
+           fHEPDataFileLines->Add(new TObjString(Form("BinCenter %s +-stat +-syst", fValueName.Data()))); 
+       }
+       else {
+         if(asym)
+           fHEPDataFileLines->Add(new TObjString(Form("BinCenter %s +stat -stat", fValueName.Data()))); 
+         else 
+           fHEPDataFileLines->Add(new TObjString(Form("BinCenter %s +-stat", fValueName.Data()))); 
+       }
+      }
+      // Skip empty bins
+      if(!fGraphStat->GetY()[ipoint]) continue;
+      TObjString * line = new TObjString;      
+      if(fGraphSyst) {
+       if (asym)
+         line->String().Form("%f %f +%f -%f +%f -%f", 
+                             fGraphStat->GetX()[ipoint], fGraphStat->GetY()[ipoint],
+                             ((TGraphAsymmErrors*)fGraphStat)->GetEYhigh()[ipoint], 
+                             ((TGraphAsymmErrors*)fGraphStat)->GetEYlow()[ipoint], 
+                             ((TGraphAsymmErrors*)fGraphSyst)->GetEYhigh()[ipoint], 
+                             ((TGraphAsymmErrors*)fGraphSyst)->GetEYlow()[ipoint]);
+       else 
+         line->String().Form("%f %f +-%f +-%f", 
+                             fGraphStat->GetX()[ipoint], fGraphStat->GetY()[ipoint],
+                             ((TGraphErrors*)fGraphStat)->GetEY()[ipoint], 
+                             ((TGraphErrors*)fGraphSyst)->GetEY()[ipoint]);
+
+       fHEPDataFileLines->Add(line);
+      } else {
+       if (asym)
+         line->String().Form("%f %f +%f -%f", 
+                             fGraphStat->GetX()[ipoint], fGraphStat->GetY()[ipoint],
+                             ((TGraphAsymmErrors*)fGraphStat)->GetEYhigh()[ipoint], ((TGraphAsymmErrors*)fGraphStat)->GetEYlow()[ipoint]);
+       else { 
+         line->String().Form("%f %f +-%f", 
+                             fGraphStat->GetX()[ipoint], fGraphStat->GetY()[ipoint],
+                             ((TGraphErrors*)fGraphStat)->GetEY()[ipoint]);
+       }
+
+       fHEPDataFileLines->Add(line);
+      }
+    }    
+  }
+  else {
+    if(!fHistStat) {
+      AliError("Hist not set");
+      return;
+    }
+    Int_t nbin = fHistStat->GetNbinsX();
+    
+    for(Int_t ibin = 1; ibin <= nbin; ibin++){
+      if(ibin == 1) {
+       if(fHistSyst)
+         fHEPDataFileLines->Add(new TObjString(Form("BinLow BinHigh %s +-stat +-syst", fValueName.Data()))); 
+       else 
+         fHEPDataFileLines->Add(new TObjString(Form("BinLow BinHigh %s +-stat", fValueName.Data())));          
+      }
+      // Skip empty bins
+      if(!fHistStat->GetBinContent(ibin)) continue;
+      TObjString * line = new TObjString;      
+      if(fHistSyst) {
+       line->String().Form("%f %f %f +-%f +-%f", 
+                           fHistStat->GetBinLowEdge(ibin), 
+                           fHistStat->GetBinLowEdge(ibin)+fHistStat->GetBinWidth(ibin), 
+                           fHistStat->GetBinContent(ibin), fHistStat->GetBinError(ibin), fHistSyst->GetBinError(ibin));
+       fHEPDataFileLines->Add(line);
+      } else {
+       line->String().Form("%f %f %f +-%f", 
+                           fHistStat->GetBinLowEdge(ibin), 
+                           fHistStat->GetBinLowEdge(ibin)+fHistStat->GetBinWidth(ibin), 
+                           fHistStat->GetBinContent(ibin), fHistStat->GetBinError(ibin));
+       fHEPDataFileLines->Add(line);
+      }
+      //      delete line;      
+    }           
+  }
+
+  TIterator * lineIter = fHEPDataFileLines->MakeIterator();
+  TObjString * obj = 0;
+  ofstream outfile;
+  outfile.open (hepfileName);
+  cout << "Saving HEP File " << hepfileName << endl;
+  
+  while ((obj = (TObjString*) lineIter->Next())) {    
+    cout << obj->String().Data() << endl;    
+    outfile << obj->String().Data() << endl;
+  }
+  outfile.close();
+}
+
+

diff --git a/ANALYSIS/AliHEPDataParser.h b/ANALYSIS/AliHEPDataParser.h
new file mode 100644 (file)
index 0000000..df3be38
--- /dev/null
+++ b/ANALYSIS/AliHEPDataParser.h
@@ -0,0 +1,60 @@
+
+#ifndef ALIHEPDATAPARSER_H
+#define ALIHEPDATAPARSER_H
+
+/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
+ * See cxx source for full Copyright notice                               */
+
+//-------------------------------------------------------------------------
+//                      Implementation of   Class AliHEPDataParser
+//
+//  This class is used to save the content of hisograms/graphs in the
+//  HEP data format and viceversa
+//  Author: Michele Floris, CERN
+//-------------------------------------------------------------------------
+
+#include "TObject.h"
+#include "TString.h"
+
+class TH1;
+class TGraph;
+
+class AliHEPDataParser : public TObject{
+
+
+public:
+  AliHEPDataParser();
+  AliHEPDataParser(TH1 * hStat, TH1 * hSyst);
+  AliHEPDataParser(TGraph * grStat, TGraph * grSyst);
+  AliHEPDataParser(const char * hepfileName);
+
+  ~AliHEPDataParser();
+  
+  TH1 * GetHistoStat() { return fHistStat;}
+  TH1 * GetHistoSyst() { return fHistSyst;}
+  TGraph * GetGraphStat() { return fGraphStat;}
+  TGraph * GetGraphSyst() { return fGraphSyst;}
+  void SaveHEPDataFile(const char * hepfileName, Bool_t trueUseGraphFalesUseHisto = 0);
+
+  void SetName(const char * name) { fValueName = name;}
+
+protected:
+
+  TH1 * fHistStat; // statistical errors (hist)
+  TH1 * fHistSyst; // systematic errors (hist) 
+  TGraph * fGraphStat; // statistical errors (hist)
+  TGraph * fGraphSyst; // systematic errors (hist)  
+  TObjArray * fHEPDataFileLines;// TClones array of TObjString
+  TString fValueName; // title for the y axis on the ascii file
+
+
+  ClassDef(AliHEPDataParser, 1);
+    
+private:
+
+
+  AliHEPDataParser(const AliHEPDataParser&);
+  AliHEPDataParser& operator=(const AliHEPDataParser&);
+};
+
+#endif