UtilCanvas.cxx

Go to the documentation of this file.

#include "TCanvas.h"
#include "TLegend.h"
#include "TPaveText.h"
#include "TH1.h"
#include "TMath.h"
#include "TList.h"

#include "MessageService/MsgService.h"

#include "AtNuUtils/UtilHist.h"
#include "AtNuUtils/UtilCanvas.h"

CVSID("$Id: UtilCanvas.cxx,v 1.4 2007/02/01 21:40:17 rhatcher Exp $");

using namespace std;

TCanvas* UtilCanvas::NewCanvas(string cn, string ti) {
  TCanvas *tc = 0;
  if(cn.size()==0) tc = new TCanvas();
  else {
    if(ti.size()==0) tc = new TCanvas(cn.c_str(),cn.c_str());
    else tc = new TCanvas(cn.c_str(),ti.c_str());
  }
  tc->SetBorderSize(2);
  tc->Range(0.0, 0.0, 100.0, 100.0);
  tc->SetFrameFillColor(0);
  return tc;
}

TCanvas* UtilCanvas::NewLetterCanvas(string cn, string ti) {
  TCanvas *tc = NewCanvas(cn,ti);
  tc->SetWindowSize(930, 750);
  return(tc);
}

TCanvas* UtilCanvas::NewNHighCanvas(int NHigh, string cn, string ti) {
  TCanvas *tc = NewLetterCanvas(cn,ti);
  tc->Divide(1, NHigh);
  return(tc);
}

TCanvas* UtilCanvas::NewNWideCanvas(int NWide, string cn, string ti) {
  TCanvas *tc = NewLetterCanvas(cn,ti);
  tc->Divide(NWide, 1);
  return(tc);
}

TCanvas* UtilCanvas::NewFourByCanvas(string cn, string ti) {
  TCanvas *tc = NewLetterCanvas(cn,ti);
  tc->Divide(2, 2);
  return(tc);
}

TCanvas* UtilCanvas::NewSixByCanvas(string cn, string ti) {
  TCanvas *tc = NewLetterCanvas(cn,ti);
  tc->Divide(3, 2);
  return(tc);
}

void UtilCanvas::FixPad(TPad *pad) {
  MSG("Util",Msg::kSynopsis) << "UtilCanvas::FixPad" << endl;
  if(!pad) return;
  static vector<TH1*> histos; histos.clear();

  pad->Modified();
  pad->Update();

  TObject *obj = 0;
  TIterator *itr2 = pad->GetListOfPrimitives()->MakeIterator();
  while ( (obj = itr2->Next()) ) {
    TH1* h1 = dynamic_cast<TH1*>(obj);
    if(h1) histos.push_back(h1);
    TLegend *leg = dynamic_cast<TLegend*>(obj);
    if(leg) leg->SetBorderSize(0);
    TPaveText *pav = dynamic_cast<TPaveText*>(obj);
    if(pav) pav->SetFillStyle(0);
    TPad *inpad = dynamic_cast<TPad*>(obj);
    if(inpad) UtilCanvas::FixPad(inpad);
  }
  pad->Modified();

  //Only concerned with 2 or more histograms on same pad
  if (histos.size() < 2) {
    MSG("Util",Msg::kSynopsis) << "This pad only has " << histos.size()
           << " hists, no need to fix" << endl;
    return;
  }

  double Max = UtilHist::HMax(histos);
  double Min = UtilHist::HMin(histos);
  double Min0 = UtilHist::HMin(histos, 0);

  //Save a handful of later divide by zero errors
  if((Max==0 && Min==0) || Max==Min) return;

  if (pad->GetLogy()) {
    MSG("Util",Msg::kSynopsis) << "Correcting for Y Axis Log Scale" << endl;

    //For LogY, a Min<=0 causes ugliness, fix here
    if(Min <= 0.0)  Min = Min0;

    double Buf = TMath::Power(Max/Min, 0.05);

    MSG("Util",Msg::kSynopsis) << "For Max = " << Max << ", Min = " << Min
           << ", Using Buf = " << Buf << endl;

    Max *= Buf;
    Min /= Buf;
  }
  else {
    MSG("Util",Msg::kSynopsis) << "Correcting for Y Axis Linear Scale" << endl;
    double Buf = (Max-Min) * 0.05;

    MSG("Util",Msg::kSynopsis) << "For Max = " << Max << ", Min = " << Min
           << ", Using Buf = " << Buf << endl;

    Max += Buf;
    if(Min<0 || Min>Buf) Min -= Buf;
    else Min = 0.0;
  }

  histos[0]->GetYaxis()->SetRangeUser(Min, Max);
  pad->Modified();
}

void UtilCanvas::UpdatePad(TPad *pad) {
  if(!pad) return;
  TList *tl = pad->GetListOfPrimitives();
  TIterator *itr2 = tl->MakeIterator();
  TObject *obj = 0;
  
  while ( (obj=itr2->Next()) ) {
    TPad* p1 = dynamic_cast<TPad*>(obj);
    if(p1) p1->Modified();
  }
  pad->Modified();
  pad->Update();
}

bool UtilCanvas::PadFilled(TPad *pad) {
  if(!pad) return false;
  TList *tl = pad->GetListOfPrimitives();
  TIterator *itr2 = tl->MakeIterator();
  TObject *obj = 0;
  bool IsFilled = false;
  
  while ( (obj=itr2->Next()) ) {
    TPad* p1 = dynamic_cast<TPad*>(obj);
    if(p1) IsFilled = IsFilled || UtilCanvas::PadFilled(p1);
    if((dynamic_cast<TH1*>(obj))) IsFilled = true;;
  }
  return IsFilled;
}

---