#pragma once

#include <iostream>
#include <fstream>
#include <string>
#include <vector>
#include <complex>
#include <boost/shared_ptr.hpp>
#include <boost/function.hpp>

#include "qft++/topincludes/relativistic-quantum-mechanics.hh"

#include "PwaUtils/AbsLhNew.hh"
#include "Examples/D0ToKsPipPim/D0ToKsPipPimStates.hh"
#include "PwaUtils/DataUtils.hh"
#include "Minuit2/MnUserParameters.h"
#include "PwaDynamics/FVector.hh"
#include "PwaDynamics/FVectorPiPiS.hh"


class D0ToKsPipPimLh : public AbsLhNew {

public:
  D0ToKsPipPimLh(boost::shared_ptr<const EvtDataBaseListNew>, 
		   const std::vector<std::string>& hypVec, 
		   boost::shared_ptr<D0ToKsPipPimStates> theStates);
  D0ToKsPipPimLh(boost::shared_ptr<AbsLhNew>, const std::vector<std::string>& hypVec, 
		   boost::shared_ptr<D0ToKsPipPimStates> theStates);
  virtual ~D0ToKsPipPimLh();

  virtual AbsLhNew* clone_() const{
    return new D0ToKsPipPimLh(_evtListPtr, _hypVec, _theStatesPtr);
  }
  virtual double calcEvtIntensity( EvtDataNew* theData, fitParamsNew& theParamVal);
  
  virtual void getDefaultParams(fitParamsNew& fitVal, fitParamsNew& fitErr);
  virtual void print(std::ostream& os) const;

protected:
  const std::vector<std::string> _hypVec;
  boost::shared_ptr<D0ToKsPipPimStates> _theStatesPtr;
 
  bool _usePhasespace;
  bool _pipiSHyp;  
  const std::string _pipiSKey;
  const std::string _phasespaceKey;
  bool _recalculatepipiS;
 
  virtual complex<double> D0ToPiPiSAmp(EvtDataNew* theData);
  
  virtual void checkParamVariation(fitParamsNew& theParamVal);
  virtual void updateFitParams(fitParamsNew& theParamVal);
private:
  void initializeHypothesis();
  virtual void cacheTheAmps();

  boost::shared_ptr<FVectorPiPiS> _pipiSFVec;

  std::vector<std::string> _enabledAmpKeys;
  std::vector<std::string> _enabledMassKeys;
  std::vector<std::string> _enabledWidthKeys;
  std::vector<std::string> _enabledFactorKeys;
  std::map<std::string, double> _currentMassMap;
  std::map<std::string, double> _currentWidthMap;
  std::map<std::string, std::map< boost::shared_ptr<const JPCLS>, double, pawian::Collection::SharedPtrLess > > _currentParamMagMap;
  std::map<std::string, std::map< boost::shared_ptr<const JPCLS>, double, pawian::Collection::SharedPtrLess > > _currentParamPhiMap;
  std::map<std::string, double> _currentgFactorMap;
  std::map<std::string, double> _currentbFactorMap;
  std::map<std::string, double> _currentfProdFactorMap; 
  double _currentS0Val;  
};