tkn_root_macros.cpp

#include "tkn_root_macros.h"
#include "tkn_config.h"
 
#include "TImage.h"
#include "TLatex.h"
#include "TPad.h"
 
using namespace tkn;
using namespace std;
 
namespace tkn{
 
inline double XUserToNDC(double x) {
    double umin = gPad->GetUxmin(), umax = gPad->GetUxmax();
    double L = gPad->GetLeftMargin(), R = gPad->GetRightMargin();
    double frac = (x - umin) / (umax - umin);
    return L + frac * (1.0 - L - R);
}
 
inline double YUserToNDC(double y) {
    double vmin = gPad->GetUymin(), vmax = gPad->GetUymax();
    double B = gPad->GetBottomMargin(), T = gPad->GetTopMargin();
    double frac = (y - vmin) / (vmax - vmin);
    return B + frac * (1.0 - B - T);
}
 
void DrawLogoAndCredit(double x1,double y1,double width, double text_size, const char* creditTxt, const char *opt)
{
    TImage* logo = TImage::Open(Form("%s/icons/logo.png", TKN_SYS));
    if (!logo) return;
 
    auto current_pad = gPad;
    if(!current_pad) return;
 
    current_pad->Update();
 
    double x1_ndc, y1_ndc, x2_ndc, y2_ndc;
    double width_ndc;
 
    if(!TString(opt).Contains("NDC")) {
        x1_ndc = XUserToNDC(x1);
        y1_ndc = YUserToNDC(y1);
        double x2_ndc_from_user = XUserToNDC(x1 + width);
        width_ndc = x2_ndc_from_user - x1_ndc;
    }
    else {
        x1_ndc = x1;
        y1_ndc = y1;
        width_ndc = width;
    }
 
    double pw = abs(gPad->XtoAbsPixel(gPad->GetX2()) - gPad->XtoAbsPixel(gPad->GetX1()));
    double ph = abs(gPad->YtoAbsPixel(gPad->GetY2()) - gPad->YtoAbsPixel(gPad->GetY1()));
 
    double px_to_ndc = 1.0 / pw;
    double py_to_ndc = 1.0 / ph;
 
    // Get logo dimensions to preserve aspect ratio
    double imgW = logo->GetWidth() * px_to_ndc;
    double imgH = logo->GetHeight() * py_to_ndc;
 
    if (imgW <= 0 || imgH <= 0) return;
 
    double aspect = double(imgH) / double(imgW);
 
    x2_ndc = x1_ndc + width_ndc;
    y2_ndc = y1_ndc + width_ndc * aspect;             // keep aspect ratio
 
    TPad* padLogo = new TPad("padLogo","padLogo",x1_ndc,y1_ndc,x2_ndc,y2_ndc);
    padLogo->SetMargin(0,0,0,0);
    padLogo->SetFillColor(0);
    padLogo->SetFillStyle(0);
    padLogo->Draw();
 
    padLogo->cd();
    logo->Draw("");
    current_pad->cd();
 
    // Draw credit text just above the logo, aligned to its left
    TLatex credit;
    credit.SetNDC(true);
    credit.SetTextFont(42);
    credit.SetTextSize(text_size);
    credit.SetTextColor(kGray+2);
    credit.SetTextAlign(11); // left, bottom
    credit.DrawLatex(x1_ndc, y2_ndc + 0.01, creditTxt);
}
 
}