SaveCanvas.C

Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file SaveCanvas.C

// $Id: $

#ifndef SaveCanvas_C

#define SaveCanvas_C

#include <TList.h>
#include <TClass.h>
#include <TCanvas.h>
#include <TSystem.h>
#include <TPad.h>
#include <TString.h>
#include <TDirectory.h>
#include <TFile.h>
#include <TStyle.h>
#include <TObject.h>
#include <TH1F.h>

#include <iostream>

using namespace std;

void
SavePad(TPad * p)
{
  if (!p)
    return;

  TList * l = p->GetListOfPrimitives();
//  l->Print();

  TIter next(l);
  TObject *obj = NULL;
  while ((obj = next()))
    {

      if (obj->IsA()->GetBaseClassOffset(TClass::GetClass("TPad")) >= 0)
        {
          if ((TPad *) obj != p)
            SavePad((TPad *) obj);
        }
      else if (obj->IsA()->GetBaseClassOffset(TClass::GetClass("TH1")) >= 0)
        {
          cout << "Save TH1 " << obj->GetName() << endl;
          obj->Clone()->Write(obj->GetName(), TObject::kOverwrite);
        }
      else if (obj->IsA()->GetBaseClassOffset(TClass::GetClass("TF1")) >= 0)
        {
          cout << "Save TF1 " << obj->GetName() << endl;
          obj->Clone()->Write(obj->GetName(), TObject::kOverwrite);
        }
      else if (obj->IsA()->GetBaseClassOffset(TClass::GetClass("TGraph")) >= 0)
        {
          cout << "Save TGraph " << obj->GetName() << endl;
          obj->Clone()->Write(obj->GetName(), TObject::kOverwrite);
        }
    }
}


void
SaveCanvas(TCanvas * c, TString name = "", Bool_t bEPS = kTRUE)
{
  if (name.Length() == 0)
    name = c->GetName();

  c->Print(name + ".png");

  TDirectory * oldd = gDirectory;

  TString rootfilename;

  c->Print(rootfilename = name + ".root");

  TFile f(rootfilename, "update");

  SavePad(c);

  f.Close();

  oldd->cd();

  if (bEPS)
    {
//      c->Print(name + ".pdf");

      float x = 20;
      float y = 20;
      gStyle->GetPaperSize(x, y);

      gStyle->SetPaperSize(c->GetWindowWidth() / 72 * 2.54,
          c->GetWindowHeight() / 72 * 2.54);
//      c->Print(name + ".eps");
      c->Print(name + ".svg");
      gSystem->Exec("rsvg-convert -f pdf -o "+name + ".pdf " + name + ".svg");
      gSystem->Exec("rm -fv " +  name + ".svg");

      gStyle->SetPaperSize(x, y);
    }
  //      c->Print(name+".C");
}


void
example_save_canvas()
{

  TCanvas *c1 = new TCanvas("CanvasTest", "CanvasTest", 800, 900);

  TH1F * h1 = new TH1F("h1", "histo from a gaussian", 100, -3, 3);
  h1->FillRandom("gaus", 10000);

  h1->Draw();

  // single call to save c1 to file RootFileName.*
  SaveCanvas(c1, "RootFileName");

}

#endif