// -*- C++ -*-
#include "Rivet/Analysis.hh"
#include "Rivet/Projections/FinalState.hh"
#include "Rivet/Projections/ZFinder.hh"
#include "Rivet/Tools/BinnedHistogram.hh"

/// @todo Include more projections as required, e.g. ChargedFinalState, FastJets, ZFinder...

namespace Rivet {


  class ATLAS_2014_I1300647 : public Analysis {
  public:

    /// Constructor
    ATLAS_2014_I1300647()
      : Analysis("ATLAS_2014_I1300647")
    {    }


  public:

    /// @name Analysis methods
    //@{

    /// Book histograms and initialise projections before the run
    void init() {
    
      FinalState fs;
      ZFinder zfinder_dressed_el(-2.4, 2.4, 20*GeV, PID::ELECTRON, 66.0*GeV, 116.0*GeV, 0.1);
      addProjection(zfinder_dressed_el, "ZFinder_dressed_el");
      
      ZFinder zfinder_bare_el(-2.4,    2.4, 20*GeV, PID::ELECTRON, 66.0*GeV, 116.0*GeV, 0.0);
      addProjection(zfinder_bare_el,	"ZFinder_bare_el");
      
      ZFinder zfinder_dressed_mu(-2.4, 2.4, 20*GeV, PID::MUON,     66.0*GeV, 116.0*GeV, 0.1);
      addProjection(zfinder_dressed_mu, "ZFinder_dressed_mu");
      
      ZFinder zfinder_bare_mu(-2.4,    2.4, 20*GeV, PID::MUON,     66.0*GeV, 116.0*GeV, 0.0);
      addProjection(zfinder_bare_mu,	"ZFinder_bare_mu");
      
      // Book histograms
      _hist_zpt_el_dressed = bookHisto1D(1, 1, 1);  // electron "dressed"
      _hist_zpt_mu_dressed = bookHisto1D(1, 1, 2);  // muon "dressed"
      _hist_zpt_el_bare    = bookHisto1D(1, 2, 1);  // electron "bare"
      _hist_zpt_mu_bare    = bookHisto1D(1, 2, 2);  // muon "bare"	 

      //double-differential plots
      _h_zpt_el_mu_dressed.addHistogram(0.0, 1.0, bookHisto1D(3, 1, 2));
      _h_zpt_el_mu_dressed.addHistogram(1.0, 2.0, bookHisto1D(3, 1, 4));
      _h_zpt_el_mu_dressed.addHistogram(2.0, 2.4, bookHisto1D(3, 1, 6));
 
    }


    /// Perform the per-event analysis
    void analyze(const Event& event) {

      const double weight = event.weight();
      
      const ZFinder& zfinder_dressed_el = applyProjection<ZFinder>(event, "ZFinder_dressed_el");
      const ZFinder& zfinder_bare_el    = applyProjection<ZFinder>(event, "ZFinder_bare_el");
      const ZFinder& zfinder_dressed_mu = applyProjection<ZFinder>(event, "ZFinder_dressed_mu");
      const ZFinder& zfinder_bare_mu    = applyProjection<ZFinder>(event, "ZFinder_bare_mu");	
      
      FillPlots1d(zfinder_dressed_el, _hist_zpt_el_dressed, weight);

      FillPlots1d(zfinder_bare_el,    _hist_zpt_el_bare,    weight);
          
      FillPlots1d(zfinder_dressed_mu, _hist_zpt_mu_dressed, weight);
          
      FillPlots1d(zfinder_bare_mu,    _hist_zpt_mu_bare,    weight);  
      
      FillPlots3d(zfinder_dressed_el, _h_zpt_el_mu_dressed, weight);      
      FillPlots3d(zfinder_dressed_mu, _h_zpt_el_mu_dressed, weight);    

    }

    void FillPlots1d(const ZFinder& zfinder, Histo1DPtr hist, double weight) {
      if(zfinder.bosons().size() != 1) return;
      const FourMomentum pZ = zfinder.bosons()[0].momentum();
      hist->fill(pZ.pT()/GeV, weight);
      return; 
    } 
    
    void FillPlots3d(const ZFinder& zfinder, BinnedHistogram<double>& binnedHist, double weight) {
      if(zfinder.bosons().size() != 1) return;
      const FourMomentum pZ = zfinder.bosons()[0].momentum();
      binnedHist.fill(pZ.rapidity(), pZ.pT()/GeV, weight);   
      return; 
    }  

    /// Normalise histograms etc., after the run
    void finalize() {
    
      normalize(_hist_zpt_el_dressed);
      normalize(_hist_zpt_el_bare);
      
      normalize(_hist_zpt_mu_dressed);  
      normalize(_hist_zpt_mu_bare); 

      foreach (Histo1DPtr hist, _h_zpt_el_mu_dressed.getHistograms()) { normalize(hist); }

    }

    //@}


  private:

    // Data members like post-cuts event weight counters go here


  private:

    /// @name Histograms
    //@{
    BinnedHistogram<double> _h_zpt_el_mu_dressed;
  
 
    Histo1DPtr _hist_zpt_el_dressed;
    Histo1DPtr _hist_zpt_el_bare;     
    Histo1DPtr _hist_zpt_mu_dressed;
    Histo1DPtr _hist_zpt_mu_bare;    

    //@}

  };


  // The hook for the plugin system
  DECLARE_RIVET_PLUGIN(ATLAS_2014_I1300647);
}