ProductionMatrixElement.h

// -*- C++ -*-
//
// ProductionMatrixElement.h is a part of Herwig - A multi-purpose Monte Carlo event generator
// Copyright (C) 2002-2019 The Herwig Collaboration
//
// Herwig is licenced under version 3 of the GPL, see COPYING for details.
// Please respect the MCnet academic guidelines, see GUIDELINES for details.
//
#ifndef HERWIG_ProductionMatrixElement_H
#define HERWIG_ProductionMatrixElement_H
//
// This is the declaration of the ProductionMatrixElement class.
 
#include <ThePEG/EventRecord/RhoDMatrix.h>
 
namespace Herwig {
 
 
using namespace ThePEG;
 
class ProductionMatrixElement {  
      
public:
 
  ProductionMatrixElement(PDT::Spin in1,PDT::Spin in2,PDT::Spin out);
 
  ProductionMatrixElement(PDT::Spin in1,PDT::Spin in2,PDT::Spin out1,
                          PDT::Spin out2);
  
  ProductionMatrixElement(PDT::Spin in1,PDT::Spin in2,PDT::Spin out1,
                          PDT::Spin out2,PDT::Spin out3);
 
  ProductionMatrixElement(PDT::Spin in1,PDT::Spin in2,PDT::Spin out1,
                          PDT::Spin out2,PDT::Spin out3, PDT::Spin out4);
 
  ProductionMatrixElement(PDT::Spin in1,PDT::Spin in2,PDT::Spin out1,
                          PDT::Spin out2,PDT::Spin out3, PDT::Spin out4,
                          PDT::Spin out5);
 
  ProductionMatrixElement(PDT::Spin in1,PDT::Spin in2,PDT::Spin out1,
                          PDT::Spin out2,PDT::Spin out3, PDT::Spin out4,
                          PDT::Spin out5, PDT::Spin out6);
  
  ProductionMatrixElement(PDT::Spin in1,PDT::Spin in2,vector<PDT::Spin> out);
  
  ProductionMatrixElement() : _nout(0) {}
 
public:
     
  vector<PDT::Spin> inspin() {return _inspin;}
 
  vector<PDT::Spin> outspin() {return _outspin;}
 
public:
 
  Complex   operator () (unsigned int in1,unsigned int in2,
                         unsigned int out) const;
  
  Complex & operator () (unsigned int in1,unsigned int in2,
                         unsigned int out);
 
  Complex   operator () (unsigned int in1,unsigned int in2,
                         unsigned int out1,unsigned int out2) const;
 
  Complex & operator () (unsigned int in1,unsigned int in2,
                         unsigned int out1,unsigned int out2);
 
  Complex   operator () (unsigned int in1,unsigned int in2,
                         unsigned int out1,unsigned int out2,
                         unsigned int out3) const;
 
  Complex & operator () (unsigned int in1,unsigned int in2,
                         unsigned int out1,unsigned int out2,
                         unsigned int out3);
 
  Complex   operator () (unsigned int in1,unsigned int in2,
                         unsigned int out1,unsigned int out2,
                         unsigned int out3,unsigned int out4) const;
  
  Complex & operator () (unsigned int in1,unsigned int in2,
                         unsigned int out1,unsigned int out2,
                         unsigned int out3, unsigned int out4);
 
  Complex   operator () (unsigned int in1,unsigned int in2,
                         unsigned int out1,unsigned int out2,
                         unsigned int out3,unsigned int out4,
                         unsigned int out5) const;
  
  Complex & operator () (unsigned int in1,unsigned int in2,
                         unsigned int out1,unsigned int out2,
                         unsigned int out3, unsigned int out4,
                         unsigned int out5);
 
  Complex   operator () (unsigned int in1,unsigned int in2,
                         unsigned int out1,unsigned int out2,
                         unsigned int out3,unsigned int out4,
                         unsigned int out5,unsigned int out6) const;
 
  Complex & operator () (unsigned int in1,unsigned int in2,
                         unsigned int out1,unsigned int out2,
                         unsigned int out3, unsigned int out4,
                         unsigned int out5, unsigned int out6);
 
  Complex   operator () (vector<unsigned int> hel) const;
  
  Complex & operator () (vector<unsigned int> hel);
 
public:
 
  RhoDMatrix calculateDMatrix(int,const RhoDMatrix &,
                              const vector<RhoDMatrix> &) const;
 
  RhoDMatrix calculateRhoMatrix(int,const RhoDMatrix &,
                                const RhoDMatrix &,
                                const vector<RhoDMatrix> &) const;
 
  double average() const;
 
  double average(const RhoDMatrix & in1, 
                 const RhoDMatrix & in2) const;
 
  Complex average(const ProductionMatrixElement & me2,
                  const RhoDMatrix & in1, 
                  const RhoDMatrix & in2) const;
  
public:
 
  void reset(const ProductionMatrixElement & x) const;
  
  static void Init();
  
private:
  
  void setMESize();
  
private:
  
  mutable unsigned int _nout;
 
  mutable vector<PDT::Spin> _inspin;
 
  mutable vector<PDT::Spin> _outspin;
 
  mutable vector<Complex> _matrixelement;
 
  mutable vector<int> _constants;
 
};
 
}
 
#endif /* HERWIG_ProductionMatrixElement_H */