/* This file is part of MAUS: http://micewww.pp.rl.ac.uk:8080/projects/maus
 *
 * MAUS is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * MAUS is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with MAUS.  If not, see <http://www.gnu.org/licenses/>.
 *
 */

#include "Utils/ReduceCppTools.hh"

namespace MAUS {

CanvasWrapper* ReduceCppTools::get_canvas_wrapper(TCanvas *canv,
                                                  TH1* hist,
                                                  std::string name,
                                                  std::string description,
						  Option_t *draw_option) {

  CanvasWrapper *wrap = new CanvasWrapper();
  wrap->SetDescription(description);
  wrap->SetFileTag(name);
  wrap->SetCanvas(canv);

  canv->cd();
  hist->Draw(draw_option);
  gPad->Update();
  TPaveStats *pave_stats = static_cast<TPaveStats*>(hist->FindObject("stats"));
  if ( pave_stats )
    pave_stats->SetOptStat(10);

  return wrap;
}

CanvasWrapper* ReduceCppTools::get_canvas_multi_wrapper(TCanvas *canv,
                                 	       	        std::vector<TObject*> objarray,
                                                  	std::string name,
                                                  	std::string description,
							Option_t *draw_option) {

  CanvasWrapper *wrap = new CanvasWrapper();
  wrap->SetDescription(description);
  wrap->SetFileTag(name);
  wrap->SetCanvas(canv);

  canv->cd();
  objarray[0]->Draw(draw_option);
  for (size_t i = 1; i < objarray.size(); i++)
    objarray[i]->Draw("SAME");

  return wrap;
}

CanvasWrapper* ReduceCppTools::get_canvas_emr_wrapper(TCanvas *canv,
                                 	              TH2EMR *hemr,
                                                      std::string name,
                                                      std::string description) {

  CanvasWrapper *wrap = new CanvasWrapper();
  wrap->SetDescription(description);
  wrap->SetFileTag(name);
  wrap->SetCanvas(canv);

  canv->cd();
  hemr->GetHistogram()->Draw("COLZ L");
  gPad->Update();
  TPaveStats *pave_stats = static_cast<TPaveStats*>(hemr->GetHistogram()->FindObject("stats"));
  if ( pave_stats )
    pave_stats->SetOptStat(0);

  return wrap;
}

CanvasWrapper* ReduceCppTools::get_canvas_divide_wrapper(TCanvas *canv,
                                                         int div_x, int div_y, bool log,
                                                         std::vector<TH1 *> histos,
                                                         std::string name,
                                                         std::string description,
                                                         Option_t *draw_option) {

  CanvasWrapper *wrap = new CanvasWrapper();
  wrap->SetDescription(description);
  wrap->SetFileTag(name);
  wrap->SetCanvas(canv);

  canv->Divide(div_x, div_y);
  int xPad = 0;
  for (auto &h:histos) {
    canv->cd(++xPad);
    h->Draw(draw_option);
    gPad->SetLogy(log);
//     gPad->Update();
//     TPaveStats *pave_stats = static_cast<TPaveStats*>(h->FindObject("stats"));
//     if ( pave_stats )
//       pave_stats->SetOptStat(10);
  }

  return wrap;
}
}