#ifndef REMNANTS_Tools_Kinematics_Generator_H
#define REMNANTS_Tools_Kinematics_Generator_H

#include "REMNANTS/Main/Remnant_Base.H"
#include "REMNANTS/Tools/Primordial_KPerp.H"
#include "ATOOLS/Phys/Blob_List.H"
#include "ATOOLS/Phys/Momenta_Stretcher.H"

namespace REMNANTS {
  struct kin_type {
    enum code {
      none   = 0,
      intact = 1,
      coll   = 2,
      DIS1   = 4,
      DIS2   = 5,
      hh     = 8
    };
  };
  class Remnant_Handler;
  
  class Kinematics_Generator {
  private:
    Remnant_Handler   * p_rhandler;
    Remnant_Base      * p_remnants[2];
    ATOOLS::Part_List * p_extracted[2], * p_spectators[2];
    ATOOLS::Blob      * p_softblob;
    
    kin_type::code  m_kintype;
    
    Primordial_KPerp          m_kperpGenerator; 
    ATOOLS::Momenta_Stretcher m_stretcher;

    ATOOLS::Poincare          m_oldcmsboost, m_newcmsboost, m_rotate;
    std::set<ATOOLS::Blob *>  m_boostedblobs;

    double                 m_mass_sum;
    ATOOLS::Vec4D          m_inmom[2], m_checkmom[2];
    ATOOLS::ParticleMomMap m_ktmap[2], m_shuffledmap;

    void SetKinType(Remnant_Handler * const rhandler);
    bool CollinearKinematics();
    bool TransverseKinematics();
    bool TransverseKinematicsHH();
    bool TransverseKinematicsDIS(const size_t & beam);
    bool AdjustShowerInitiators();
    bool ConstructMomentaAndBoosts(ATOOLS::Particle * part[2]);
    bool BoostConnectedBlob(ATOOLS::Blob * blob,size_t & catcher);
    bool AdjustRemnants();
    bool AdjustFinalStateDIS(const size_t & beam);
    
    bool CheckHH();
    bool CheckScatter(ATOOLS::Particle * p[2]);
    bool CheckRemnants();
    bool CheckDIS(const size_t & beam);
    const ATOOLS::Vec4D ExtractColourfulFS(const size_t & beam,
					   std::vector<ATOOLS::Vec4D> & moms,
					   std::vector<double> & masses,
					   std::vector<ATOOLS::Particle *> & parts);
    const ATOOLS::Vec4D ExtractSpectators(const size_t & beam,
					  std::vector<ATOOLS::Vec4D> & moms,
					  std::vector<double> & masses,
					  std::vector<ATOOLS::Particle *> & parts);
    
    void InitKTMaps();
    const ATOOLS::Vec4D & ShuffledMomentum(ATOOLS::Particle *const part);
  public:
    Kinematics_Generator();
    ~Kinematics_Generator();

    void Initialize(Remnant_Handler * const rhandler);
    ATOOLS::Blob * MakeSoftBlob();
    bool FillBlobs(ATOOLS::Blob_List *const bloblist);
      
    inline Primordial_KPerp * GetKPerp() { return &m_kperpGenerator; }
    void Reset();
  };
}

#endif