// Header file for the Oxbridge Stransverse Mass Library -- oxbridgekinetics.
// See http://www.hep.phy.cam.ac.uk/~lester/mt2/index.html
// Authors: Christopher Lester and Alan Barr

#ifndef Basic_MCT2_332_Calculator_H
#define Basic_MCT2_332_Calculator_H

#include "Mt2/MCT2_332_Calculator.h"
#include "Minuit2/FCNBase.h"
#include <vector>

namespace Mt2 {

  class Basic_MCT2_332_Calculator : public MCT2_332_Calculator {
  public:
    double mct2_332(const LorentzTransverseVector& visibleA, // 3 d.o.f. 
		    const LorentzTransverseVector& visibleB, // 3 d.o.f.
		    const TwoVector& ptmiss,                 // 2 d.o.f.
		    const double mInvisible);
    double mct2_332_Sq(const LorentzTransverseVector& visibleA, // 3 d.o.f. 
		       const LorentzTransverseVector& visibleB, // 3 d.o.f.
		       const TwoVector& ptmiss,                 // 2 d.o.f.
		       const double mInvisible);
    Basic_MCT2_332_Calculator() : MCT2_332_Calculator("Basic_MCT2_332") {};
    
  private:
    class mCT2Fcn : public ROOT::Minuit2::FCNBase {
    public:
      mCT2Fcn(const double exmiss, 
	      const double eymiss, 
	      const double mchi,
	      const double pT1x,
	      const double pT1y,
	      const double mass1,
	      const double pT2x,
	      const double pT2y,
	      const double mass2) : 
	theExmiss(exmiss),
	theEymiss(eymiss),
	theMchi(mchi),
	thePT1x(pT1x),
	thePT1y(pT1y),
	theMass1(mass1),
	thePT2x(pT2x),
	thePT2y(pT2y),
	theMass2(mass2),
	theErrorDef(1.) {}
      
      ~mCT2Fcn() {}
      
      virtual double Up() const {return theErrorDef;}
      virtual double operator()(const std::vector<double>&) const;
      
      const double exMiss() const {return theExmiss;}
      const double eyMiss() const {return theEymiss;}
      const double mChi() const {return theMchi;}
      const double pT1x() const {return thePT1x;}
      const double pT1y() const {return thePT1y;}
      const double pT2x() const {return thePT2x;}
      const double pT2y() const {return thePT2y;}
      
      void setErrorDef(double def) {theErrorDef = def;}
      
    private:
      
      double theExmiss;
      double theEymiss;
      double theMchi;
      
      double thePT1x;
      double thePT1y;
      double theMass1;
      
      double thePT2x;
      double thePT2y;
      double theMass2;
      
      double theErrorDef;
    };
  };

} //end namespace Mt2

#endif