HeatMap.cxx

Go to the documentation of this file.

#include "larreco/QuadVtx/HeatMap.h"

#include "TH2.h"

namespace quad
{
  // -------------------------------------------------------------------------
  HeatMap::HeatMap(int _Nz, double _minz, double _maxz,
                   int _Nx, double _minx, double _maxx)
    : minz(_minz), minx(_minx), maxz(_maxz), maxx(_maxx), Nx(_Nx), Nz(_Nz),
      map(Nz*Nx, 0)
  {
  }

  // -------------------------------------------------------------------------
  std::unique_ptr<TH2F> HeatMap::AsTH2() const
  {
    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;
  }
}