mergeGraphs.C

Go to the documentation of this file.

#include "TGraphErrors.h"
#include "TFile.h"
#include "TKey.h"
#include "TObjArray.h"
#include <iostream>
#include <vector>

using namespace std;

TGraphErrors* merge(const TGraphErrors& g1, const TGraphErrors& g2)
{
  vector<double> x;
  vector<double> y;
  vector<double> xerr;
  vector<double> yerr;

  for ( int i = 0; i < g1.GetN(); ++i )
    {
      x.push_back(g1.GetX()[i]);
      xerr.push_back(g1.GetEX()[i]);
      y.push_back(g1.GetY()[i]);
      yerr.push_back(g1.GetEY()[i]);
    }

  for ( int i = 0; i < g2.GetN(); ++i )
    {
      x.push_back(g2.GetX()[i]);
      xerr.push_back(g2.GetEX()[i]);
      y.push_back(g2.GetY()[i]);
      yerr.push_back(g2.GetEY()[i]);
    }

  TGraphErrors* g = new TGraphErrors(x.size(),
                                     &(x[0]),&(y[0]),
                                     &(xerr[0]),&(yerr[0]));
  g->SetName(g1.GetName());
  return g;
  
}

void mergeGraphs(const char* in1, const char* in2,
                 const char* out)
{
  TFile f2(in2);
  TFile f1(in1);
  TFile fout(out,"RECREATE");
  TIter nextkey(f1.GetListOfKeys());
  TKey* key1;
  while ( (key1=(TKey*)nextkey()) )
    {
      f1.cd();
      cout << key1->GetName() << endl;
      TGraphErrors* g1 = (TGraphErrors*)key1->ReadObj();
      TGraphErrors* g2 = (TGraphErrors*)f2.Get(key1->GetName());
      TGraphErrors* g = merge(*g1,*g2);

//       cout << g1->GetN() << " "
//         << g1->GetX()[0] << ":" << g1->GetX()[g1->GetN()-1] << endl;
//       cout << g2->GetN() << " " 
//         << g2->GetX()[0] << ":" << g2->GetX()[g2->GetN()-1] << endl;
//       cout << g->GetN() << " " 
//         << g->GetX()[0] << ":" << g->GetX()[g->GetN()-1] << endl;

      delete g1;
      delete g2;
      fout.cd();
      g->Write();
      fout.Write();
      delete g;
    }
}