DecayMatrixElement.h

// -*- C++ -*-
#ifndef Herwig_DecayMatrixElement_H
#define Herwig_DecayMatrixElement_H
//
// This is the declaration of the DecayMatrixElement class.
//
 
#include "ThePEG/Pointer/ReferenceCounted.h"
#include <ThePEG/Config/ThePEG.h>
#include <ThePEG/Helicity/HelicityDefinitions.h>
#include <ThePEG/EventRecord/RhoDMatrix.h>
#include "DecayMatrixElement.fh"
 
namespace Herwig {
 
using namespace ThePEG;
 
class DecayMatrixElement: public Pointer::ReferenceCounted {
 
public:
 
  DecayMatrixElement(unsigned int ntemp=999, PDT::Spin spin = PDT::SpinUndefined) 
    : nOut_(ntemp), inSpin_(spin) {}
 
  virtual ~DecayMatrixElement();
 
public: 
      
  PDT::Spin inspin() const {return inSpin_;}
 
  const vector<PDT::Spin> & outspin() const {return outSpin_;}
 
  unsigned int nOut() const {return nOut_;}
 
  virtual RhoDMatrix calculateDMatrix(const vector<RhoDMatrix> & rhoout) const = 0;
 
  virtual RhoDMatrix calculateRhoMatrix(int ipart,const RhoDMatrix & rhoin,
                                        const vector<RhoDMatrix> & rhoout) const = 0;
 
  virtual Complex contract(const RhoDMatrix & rhoin) const = 0;
 
  // /** 
  //  * Contract the matrix element with the \f$\rho\f$ matrix of the 
  //  * incoming particle. The spins of the decay products are summed over.
  //  * @param con The conjugate matrix elemetn for the contraction
  //  * @param rhoin The \f$\rho\f$ matrix for the decaying particle.
  //  */
  // Complex contract(const DecayMatrixElement & con, 
  //               const RhoDMatrix & rhoin);
 
public:
  
  virtual Complex   operator () (unsigned int inhel,unsigned int outhel1,
                                 unsigned int outhel2) const = 0;
  
  virtual Complex & operator () (unsigned int inhel,unsigned int outhel1,
                                 unsigned int outhel2) = 0;
  
  virtual Complex   operator () (unsigned int inhel,unsigned int outhel1,
                                 unsigned int outhel2,unsigned int outhel3) const = 0;
  
  virtual Complex & operator () (unsigned int inhel,unsigned int outhel1,
                                 unsigned int outhel2,unsigned int outhel3) = 0;
  
  virtual Complex   operator () (unsigned int inhel,unsigned int outhel1,
                                 unsigned int outhel2,unsigned int outhel3,
                                 unsigned int outhel4) const = 0;
  
  virtual Complex & operator () (unsigned int inhel,unsigned int outhel1,
                                 unsigned int outhel2,unsigned int outhel3,
                                 unsigned int outhel4) = 0;
  
  virtual Complex   operator () (unsigned int inhel,unsigned int outhel1,
                                 unsigned int outhel2,unsigned int outhel3,
                                 unsigned int outhel4,unsigned int outhel5) const = 0;
  
  virtual Complex & operator () (unsigned int inhel,unsigned int outhel1,
                                 unsigned int outhel2,unsigned int outhel3,
                                 unsigned int outhel4,unsigned int outhel5) = 0;
  
  virtual Complex   operator () (unsigned int inhel,unsigned int outhel1,
                                 unsigned int outhel2,unsigned int outhel3,
                                 unsigned int outhel4,unsigned int outhel5,
                                 unsigned int outhel6) const = 0;
  
  virtual Complex & operator () (unsigned int inhel,unsigned int outhel1,
                                 unsigned int outhel2,unsigned int outhel3,
                                 unsigned int outhel4,unsigned int outhel5,
                                 unsigned int outhel6) = 0;
  
  virtual Complex   operator () (const vector<unsigned int> & in) const = 0;
  
  virtual Complex & operator () (const vector<unsigned int> & in) = 0;
  
  virtual void zero() = 0;
 
private:
 
  DecayMatrixElement & operator=(const DecayMatrixElement &) = delete;
 
protected:
 
  vector<PDT::Spin> & outspin() {return outSpin_;}
  
private:
  
  unsigned int nOut_;
 
  PDT::Spin inSpin_;
 
  vector<PDT::Spin> outSpin_;
 
};
 
}
 
#endif /* Herwig_DecayMatrixElement_H */