RSSpinorBarWaveFunction.h

// -*- C++ -*-
//
// RSSpinorBarWaveFunction.h is a part of ThePEG - Toolkit for HEP Event Generation
// Copyright (C) 2003-2019 Peter Richardson, Leif Lonnblad
//
// ThePEG is licenced under version 3 of the GPL, see COPYING for details.
// Please respect the MCnet academic guidelines, see GUIDELINES for details.
//
#ifndef ThePEG_RSSpinorBarWaveFunction_H
#define ThePEG_RSSpinorBarWaveFunction_H
//
// This is the declaration of the RSSpinorBarWaveFunction class.
//
#include "WaveFunctionBase.h"
#include <ThePEG/Helicity/LorentzRSSpinorBar.h>
#include <ThePEG/Helicity/RSFermionSpinInfo.h>
#include <ThePEG/EventRecord/Particle.h>
#include <ThePEG/EventRecord/RhoDMatrix.h>

namespace ThePEG {

namespace Helicity {

class RSSpinorBarWaveFunction: public WaveFunctionBase {

public:

  RSSpinorBarWaveFunction(const Lorentz5Momentum & p,tcPDPtr part,
                          complex<double> xs1, complex<double> xs2,
                          complex<double> xs3, complex<double> xs4,
                          complex<double> ys1, complex<double> ys2,
                          complex<double> ys3, complex<double> ys4,
                          complex<double> zs1, complex<double> zs2,
                          complex<double> zs3, complex<double> zs4,
                          complex<double> ts1, complex<double> ts2,
                          complex<double> ts3, complex<double> ts4)
    : WaveFunctionBase(p,part), _wf(xs1,xs2,xs3,xs4,
                                    ys1,ys2,ys3,ys4,
                                    zs1,zs2,zs3,zs4,
                                    ts1,ts2,ts3,ts4)
  {
    assert(iSpin()==4);
  }
  
  RSSpinorBarWaveFunction(const Lorentz5Momentum & p,tcPDPtr part,
                          const LorentzRSSpinorBar<double> & wave,
                          Direction dir=intermediate) 
    : WaveFunctionBase(p,part,dir), _wf(wave)
  {
    assert(iSpin()==4);
  }
  
  RSSpinorBarWaveFunction(const tPPtr & p,
                          const LorentzRSSpinorBar<SqrtEnergy> & wave,
                          Direction dir=intermediate) 
    : WaveFunctionBase(p->momentum(),p->dataPtr(),dir), _wf(wave.Type())
  {
    assert(iSpin()==4);
    for (unsigned int i=0; i<4; ++i)
      for(unsigned int j=0; j<4; ++j)
        _wf(i,j)=Complex(wave(i,j)*UnitRemoval::InvSqrtE);
  }

  RSSpinorBarWaveFunction(const Lorentz5Momentum & p,tcPDPtr part,
                          unsigned int ihel,Direction dir)
    : WaveFunctionBase(p,part,dir)
  {
    assert(iSpin()==4);
    calculateWaveFunction(ihel);
  }
  
  RSSpinorBarWaveFunction(Lorentz5Momentum p,tcPDPtr part,Direction dir)
    : WaveFunctionBase(p,part,dir), _wf()
  {
    assert(iSpin()==4);
  }
  
  RSSpinorBarWaveFunction() 
    : WaveFunctionBase(), _wf()
  {}

  RSSpinorBarWaveFunction(vector<RSSpinorBarWaveFunction> & wave,
                          tPPtr part,Direction dir,bool time,bool=true) {
    calculateWaveFunctions(wave,part,dir);
    constructSpinInfo(wave,part,dir,time);
  }

  complex<double> operator ()(int i, int j) const {
    assert( i>=0 && i<=3  && j>=0 && j<=3 );
    return _wf(i,j);
  }
    
  const LorentzRSSpinorBar<double> & wave() const {return _wf;}

  LorentzRSSpinorBar<SqrtEnergy> dimensionedWf() const {
    LorentzRSSpinorBar<SqrtEnergy> temp(_wf.Type());
    for (unsigned int i=0; i<4; ++i)
      for (unsigned int j=0; j<4; ++j)
        temp(i,j) = _wf(i,j)*UnitRemoval::SqrtE;
    return temp;
  }
  
  complex<double> xs1() const {return _wf.xs1();}

  complex<double> xs2() const {return _wf.xs2();}

  complex<double> xs3() const {return _wf.xs3();}

  complex<double> xs4() const {return _wf.xs4();}

  complex<double> ys1() const {return _wf.ys1();}

  complex<double> ys2() const {return _wf.ys2();}
  
  complex<double> ys3() const {return _wf.ys3();}
  
  complex<double> ys4() const {return _wf.ys4();}
  
  complex<double> zs1() const {return _wf.zs1();}
  
  complex<double> zs2() const {return _wf.zs2();}
  
  complex<double> zs3() const {return _wf.zs3();}
  
  complex<double> zs4() const {return _wf.zs4();}
  
  complex<double> ts1() const {return _wf.ts1();}
  
  complex<double> ts2() const {return _wf.ts2();}
  
  complex<double> ts3() const {return _wf.ts3();}
  
  complex<double> ts4() const {return _wf.ts4();}

  void reset(unsigned int ihel) {
    calculateWaveFunction(ihel);
  }

public:

  void transform(const LorentzRotation & r) {
    _wf.transform(r);
    transformMomentum(r);
  }

public:

  static void calculateWaveFunctions(vector<LorentzRSSpinorBar<SqrtEnergy> > & waves,
                                     tPPtr particle,Direction);

  static void calculateWaveFunctions(vector<RSSpinorBarWaveFunction> & waves,
                                     tPPtr particle,Direction);
  
  static void calculateWaveFunctions(vector<RSSpinorBarWaveFunction> & waves,
                                     const Lorentz5Momentum & momentum,
                                     tcPDPtr parton,Direction);

  static void calculateWaveFunctions(vector<LorentzRSSpinorBar<SqrtEnergy> > & waves,
                                     RhoDMatrix & rho,
                                     tPPtr particle,Direction);

  static void calculateWaveFunctions(vector<RSSpinorBarWaveFunction> & waves,
                                     RhoDMatrix & rho,
                                     tPPtr particle,Direction);

  static void constructSpinInfo(const vector<LorentzRSSpinorBar<SqrtEnergy> > & waves,
                                tPPtr part,Direction dir, bool time);

  static void constructSpinInfo(const vector<RSSpinorBarWaveFunction> & waves,
                                tPPtr part,Direction dir, bool time);

private:

  void calculateWaveFunction(unsigned int ihel);
  
 private:
  
  LorentzRSSpinorBar<double> _wf;
};

}
}

#endif /* ThePEG_RSSpinorBarWaveFunction_H */