c_cpp ROOT Plot直方图

Posted 2021-05-14

void plotHistograms(const char *filename, string extention) {
  TFile *f1 = TFile::Open(filename);
  if(!f1) return;

  TIter next(f1->GetListOfKeys());
  TKey *key;
  TCanvas c1;
  while ((key = (TKey*)next())) {
    TClass *cl = gROOT->GetClass(key->GetClassName());
    if (!cl->InheritsFrom("TH1")) continue;
    TH1 *h = (TH1*)key->ReadObj();
    h->Draw("colz");
    gStyle->SetOptStat(kFALSE);
    auto hName = h->GetName();
    string hNameString = string(hName);
    auto hFileName = "plots/" + hNameString + extention + ".png";
    std::cout << hName << '\t' << hFileName << std::endl;
    c1.Print(hFileName.c_str());
  }

}