#include <iostream>
#include <fstream>

#include <TROOT.h>
#include <TCanvas.h>
#include <TStyle.h>

#include "MetroProMap.h"
#include "Map2D.h"
#include "colz.h"

void colz(TH2D &h)
{
   TCanvas *c1 = (TCanvas*)gROOT->FindObject("c1");
   if (c1 == 0) {
      c1 = new TCanvas("c1","c1",5,5,560,500);
   }
   c1->SetRightMargin(0.2);
   gStyle->SetPalette(1);
   h.SetContour(150);
   h.SetStats(0);
   h.Draw("colz");
}

void colz(char *filename)
{
   ifstream ifs(filename);
   if (!ifs.is_open()) {
      std::cerr << "Error opening input file " << filename
                << " - cannot continue." << std::endl;
      return;
   }
   MetroProMap mmap;
   ifs >> mmap;
   TObject *h2 = gROOT->FindObject("h2");
   if (h2) {
      delete h2;
   }
   TH2D *h = mmap.makeHist2D("h2",filename);
   colz(*h);
}