HistDisp.cxx

Go to the documentation of this file.

#include "HistDisp.h"

#include <DataUtil/GetDetectorBins.h>
using namespace DataUtil;

#include <TCanvas.h>
#include <TH2D.h>
#include <TView.h>
#include <TProfile.h>

#include <iostream>
using namespace std;

HistDisp::HistDisp(UgliGeomHandle ugh, TPad* pad)
{
    int nzbins=0, nwbins=0;
    double zmin=0,zmax=0,wmin=0,wmax=0;

    GetDetectorBinsZ(ugh,nzbins,zmin,zmax);
    cerr << nzbins << " " << zmin << " " << zmax << endl;

    GetDetectorBinsUV(ugh,PlaneView::kU,nwbins,wmin,wmax);
    cerr << nwbins << " " << wmin << " " << wmax << endl;
    fHquz[0] = new TH2D("quz","Charge/strip in U vs. Z",
                        nzbins,zmin,zmax,nwbins,wmin,wmax);

    GetDetectorBinsUV(ugh,PlaneView::kV,nwbins,wmin,wmax);
    cerr << nwbins << " " << wmin << " " << wmax << endl;
    fHqvz[0] = new TH2D("qvz","Charge/strip in V vs. Z",
                        nzbins,zmin,zmax,nwbins,wmin,wmax);

    if (pad) fPad = pad;
    else fPad = new TCanvas("hdc","HistDisp Canvas",800,800);
    this->Init();
}

HistDisp::HistDisp(int nubins, double umin, double umax,
                   int nvbins, double vmin, double vmax, 
                   int nzbins, double zmin, double zmax)
{
    fHquz[0] = new TH2D("quz","Charge in U vs. Z",
                        nzbins,zmin,zmax,nubins,umin,umax);
    fHquz[0]->SetXTitle("Z (meter)");
    fHquz[0]->SetYTitle("U (meter)");
    fHquz[0]->SetZTitle("Charge");
    fHquz[0]->SetStats(false);

    fHqvz[0] = new TH2D("qvz","Charge in V vs. Z",
                        nzbins,zmin,zmax,nvbins,vmin,vmax);
    fHqvz[0]->SetXTitle("Z (meter)");
    fHqvz[0]->SetYTitle("U (meter)");
    fHqvz[0]->SetZTitle("Charge");
    fHqvz[0]->SetStats(false);

    fPad = new TCanvas("hdc","HistDisp Canvas",800,800);
    this->Init();
}

void HistDisp::Init()
{
    fPad->Divide(2,2);
    this->Clear();
}

HistDisp::~HistDisp()
{
    if (fHquz[0]) delete fHquz[0]; fHquz[0]=0;
    if (fHqvz[0]) delete fHqvz[0]; fHqvz[0]=0;
}


void HistDisp::FillU(double u, double z, double q, double t)
{
    t=0;                        // unused warning
    for (int ind=1;ind<=2; ++ind) fHquz[ind]->Fill(z,u,q);
}
void HistDisp::FillV(double v, double z, double q, double t)
{
    t=0;                        // unused warning
    for (int ind=1;ind<=2; ++ind) fHqvz[ind]->Fill(z,v,q);
}

void HistDisp::Clear()
{
    fPad->cd(1);
    fHquz[1] = (TH2D*)fHquz[0]->DrawCopy("lego2");
    fPad->cd(2);
    fHqvz[1] = (TH2D*)fHqvz[0]->DrawCopy("lego2");
    fPad->cd(3);
    fHquz[2] = (TH2D*)fHquz[0]->DrawCopy("colz");
    fPad->cd(4);
    fHqvz[2] = (TH2D*)fHqvz[0]->DrawCopy("colz");
}

void HistDisp::Update()
{
    for (int ind=1; ind<=4; ++ind) fPad->cd(ind)->Modified();
    fPad->cd();
    fPad->Modified();
    fPad->Update();
}

void HistDisp::SetRotation(double deg_polar, double deg_azimuth)
{
    for (int ind=1;ind<=2;++ind) {
        fPad->cd(ind);          // ROOT has 
        TVirtualPad* p = gPad;  // the weirdest 
        TView* v = p->GetView(); // fuckin API
        v->RotateView(deg_polar,deg_azimuth,p);
    }
}


static bool find_bounds(TH2D* h, int &min_bin, int &max_bin, bool isx)
{
    min_bin=max_bin=0;

    TProfile* p = 0;
    if (isx) 
        p = h->ProfileX();
    else
        p = h->ProfileY();

    int bin=0, nbins = p->GetNbinsX();

    for(bin = 1; bin < nbins; ++bin) {
        if (0.0 != p->GetBinContent(bin)) break;
    }
    if (bin == nbins) return false;
    for( ; nbins > bin; --nbins) {
        if (0.0 != p->GetBinContent(nbins)) break;
    }
    if (bin == nbins) return false;
    min_bin = bin;
    max_bin = nbins;
    cerr << "["<<min_bin<<","<<max_bin<<"] isx? " << isx << endl;
    return true;
}

void HistDisp::AutoZoom()
{
    int umin=0,umax=0,zumin=0,zumax=0;
    find_bounds(fHquz[1],umin,umax,false);
    find_bounds(fHquz[1],zumin,zumax,true);

    int vmin=0,vmax=0,zvmin=0,zvmax=0;
    find_bounds(fHqvz[1],vmin,vmax,false);
    find_bounds(fHqvz[1],zvmin,zvmax,true);

    int zmin=zumin, zmax=zumax;
    if (zvmin < zmin) zmin = zvmin;
    if (zvmax > zmax) zmax = zvmax;

    for (int ind=1; ind<=2; ++ind) {
        fHquz[ind]->GetYaxis()->SetRange(umin,umax);
        fHqvz[ind]->GetYaxis()->SetRange(vmin,vmax);

        fHquz[ind]->GetXaxis()->SetRange(zmin,zmax);
        fHqvz[ind]->GetXaxis()->SetRange(zmin,zmax);
    }
        
}

void HistDisp::Zoom(double umin, double umax,
                    double vmin, double vmax, 
                    double zmin, double zmax)
{
    

    for (int ind=1; ind<=2; ++ind) {
        fHquz[ind]->SetAxisRange(umin,umax,"Y");
        fHqvz[ind]->SetAxisRange(vmin,vmax,"Y");

        fHquz[ind]->SetAxisRange(zmin,zmax,"X");
        fHqvz[ind]->SetAxisRange(zmin,zmax,"X");
    }
}

void HistDisp::Unzoom()
{
    for (int ind=1; ind<=2; ++ind) {
        fHquz[ind]->GetYaxis()->SetRange(1,fHquz[ind]->GetNbinsY());
        fHqvz[ind]->GetYaxis()->SetRange(1,fHqvz[ind]->GetNbinsY());

        fHquz[ind]->GetXaxis()->SetRange(1,fHquz[ind]->GetNbinsX());
        fHqvz[ind]->GetXaxis()->SetRange(1,fHqvz[ind]->GetNbinsX());
    }
}

---