rtang/ANASEN_analysis
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

modified: scratch/make_prettyplots.C set z min

Browse Source
This commit is contained in:
Vignesh Sitaraman 2026-06-05 08:48:42 -04:00
parent 88cf59bce3
commit 66fd758554
1 changed files with 42 additions and 23 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

39
scratch/make_prettyplots.C
View File

@ -27,7 +27,8 @@
// ---------------------------------------------------------------------------
// Style — called once before anything is drawn
// ---------------------------------------------------------------------------
void SetStyle() {
void SetStyle()
{
gROOT->SetStyle("Plain");
gStyle->SetOptStat(0);
gStyle->SetOptTitle(0);
@ -65,26 +66,31 @@ void SetStyle() {
void make_prettyplots(const char *rootFile,
const char *histName,
const char *xlabel = "",
const char* ylabel = "") {
const char *ylabel = "",
double minZ = -9999.0)
{
SetStyle();
// --- Open file ----------------------------------------------------------
TFile *f = TFile::Open(rootFile, "READ");
if (!f || f->IsZombie()) {
if (!f || f->IsZombie())
{
std::cerr << "ERROR: Cannot open " << rootFile << "\n";
return;
}
TObject *obj = f->Get(histName);
if (!obj) {
if (!obj)
{
std::cerr << "ERROR: '" << histName << "' not found in " << rootFile << "\n";
f->Close();
return;
}
TObject *clone = obj->Clone(Form("%s_clone", histName));
if (!clone) {
if (!clone)
{
std::cerr << "ERROR: Clone failed for '" << histName << "'\n";
f->Close();
return;
@ -104,11 +110,20 @@ void make_prettyplots(const char* rootFile,
c.cd();
// Widen right margin for the colz palette bar
if (is2D) gPad->SetRightMargin(0.13);
if (is2D)
gPad->SetRightMargin(0.13);
if (is2D) {
if (is2D)
{
TH2 *h = (TH2 *)clone;
h->SetStats(0);
// --- Apply Minimum Z if provided ---
if (minZ != -9999.0)
{
h->SetMinimum(minZ);
}
h->GetXaxis()->SetTitle(xlabel);
h->GetYaxis()->SetTitle(ylabel);
h->GetXaxis()->SetTitleOffset(1);
@ -116,7 +131,9 @@ void make_prettyplots(const char* rootFile,
h->GetXaxis()->CenterTitle(true);
h->GetYaxis()->CenterTitle(true);
h->Draw("colz");
} else {
}
else
{
TH1 *h = (TH1 *)clone;
h->SetStats(0);
h->SetLineColor(kAzure - 5);
@ -134,12 +151,14 @@ void make_prettyplots(const char* rootFile,
// Build output path: same directory as input file, named after the histogram
std::string inPath(rootFile);
std::string dir = inPath.substr(0, inPath.find_last_of("/\\"));
if (dir == inPath) dir = "."; // no directory component — use cwd
if (dir == inPath)
dir = "."; // no directory component — use cwd
std::string outPath = dir + "/" + std::string(histName) + ".png";
// Replace any "/" inside histName (e.g. "folder/hist") with "_"
for (char &ch : outPath)
if (ch == '/') ch = '_';
if (ch == '/')
ch = '_';
c.Modified();
c.Update();