quad::HeatMap Class Reference

#include <HeatMap.h>

Public Member Functions
	HeatMap (int _Nz, double _minz, double _maxz, int _Nx, double _minx, double _maxx)
std::unique_ptr< TH2F >	AsTH2 () const
double	ZBinCenter (int iz) const
double	XBinCenter (int ix) const
int	ZToBin (double z) const
int	XToBin (double x) const

Public Attributes
const double	minz
const double	minx
const double	maxz
const double	maxx
const int	Nx
const int	Nz
std::vector< float >	map

Private Member Functions
int	fast_floor (double x) const

Detailed Description

Definition at line 11 of file HeatMap.h.

Constructor & Destructor Documentation

quad::HeatMap::HeatMap	(	int	_Nz,
		double	_minz,
		double	_maxz,
		int	_Nx,
		double	_minx,
		double	_maxx
	)

Definition at line 8 of file HeatMap.cxx.

    : minz(_minz), minx(_minx), maxz(_maxz), maxx(_maxx), Nx(_Nx), Nz(_Nz),
      map(Nz*Nx, 0)
  {
  }

Member Function Documentation

std::unique_ptr< TH2F > quad::HeatMap::AsTH2

(

)

const

Definition at line 16 of file HeatMap.cxx.

  {
    auto h = std::make_unique<TH2F>("", "", Nz, minz, maxz, Nx, minx, maxx);
    for(unsigned int i = 0; i < map.size(); ++i){
      h->SetBinContent((i/Nx)+1, (i%Nx)+1, map[i]);
    }

    double maxz = 0;
    for(int i = 0; i < (h->GetNbinsX()+2)*(h->GetNbinsY()+2); ++i){
      maxz = std::max(maxz, h->GetBinContent(i));
    }

    // Slim out bins with low population (reduces output file size)
    for(int i = 0; i < (h->GetNbinsX()+2)*(h->GetNbinsY()+2); ++i){
      if(h->GetBinContent(i) < .05*maxz) h->SetBinContent(i, 0);
    }

    h->Sumw2(false); // Drop all the errors

    return h;
  }

int quad::HeatMap::fast_floor ( double  x ) const

inlineprivate

Definition at line 31 of file HeatMap.h.

{return int(x+100000)-100000;}

double quad::HeatMap::XBinCenter ( int  ix ) const

inline

Definition at line 20 of file HeatMap.h.

{return minx + (ix+.5)*(maxx-minx)/Nx;}

int quad::HeatMap::XToBin ( double  x ) const

inline

Definition at line 22 of file HeatMap.h.

{return fast_floor((x-minx)/(maxx-minx)*Nx);}

double quad::HeatMap::ZBinCenter ( int  iz ) const

inline

Definition at line 19 of file HeatMap.h.

{return minz + (iz+.5)*(maxz-minz)/Nz;}

int quad::HeatMap::ZToBin ( double  z ) const

inline

Definition at line 21 of file HeatMap.h.

{return fast_floor((z-minz)/(maxz-minz)*Nz);}

Member Data Documentation

std::vector<float> quad::HeatMap::map

Definition at line 27 of file HeatMap.h.

const double quad::HeatMap::maxx

Definition at line 24 of file HeatMap.h.

const double quad::HeatMap::maxz

Definition at line 24 of file HeatMap.h.

const double quad::HeatMap::minx

Definition at line 24 of file HeatMap.h.

const double quad::HeatMap::minz

Definition at line 24 of file HeatMap.h.

const int quad::HeatMap::Nx

Definition at line 25 of file HeatMap.h.

const int quad::HeatMap::Nz

Definition at line 25 of file HeatMap.h.

---

The documentation for this class was generated from the following files:

• HeatMap.h
• HeatMap.cxx