ScalarWaveFunction.h

// -*- C++ -*-
//
// ScalarWaveFunction.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_ScalarWaveFunction_H
#define ThePEG_ScalarWaveFunction_H
//
// This is the declaration of the ScalarWaveFunction class.

#include "WaveFunctionBase.h"
#include <ThePEG/Helicity/ScalarSpinInfo.h>
#include <ThePEG/EventRecord/Particle.h>
#include <ThePEG/EventRecord/RhoDMatrix.h>

namespace ThePEG {
namespace Helicity {

class ScalarWaveFunction : public WaveFunctionBase {

public:


  ScalarWaveFunction(const Lorentz5Momentum & p,tcPDPtr part,
                     Complex wave,Direction dir=intermediate) 
    : WaveFunctionBase(p,part,dir), _wf(wave)
  {
    assert(iSpin()==1);
  }

  ScalarWaveFunction(const Lorentz5Momentum & p,tcPDPtr part,Direction dir) 
    : WaveFunctionBase(p,part,dir), _wf(1.0)
  {
    assert(iSpin()==1);
  }

  static void calculateWaveFunctions(RhoDMatrix & rho,
                                     tPPtr, Direction) {
    rho=RhoDMatrix(PDT::Spin0);
  }

  static void constructSpinInfo(tPPtr part,Direction, bool time) {
    tScalarSpinPtr inspin;
    if(part->spinInfo()) inspin=dynamic_ptr_cast<tScalarSpinPtr>(part->spinInfo());
    if(inspin) return;
    assert(!part->spinInfo());
    ScalarSpinPtr temp = new_ptr(ScalarSpinInfo(part->momentum(),time));
    part->spinInfo(temp);
  }

  ScalarWaveFunction() : WaveFunctionBase(), _wf(1.0) {}

  ScalarWaveFunction(tPPtr p,Direction dir,bool time) 
    : WaveFunctionBase(p->momentum(), p->dataPtr(), dir), _wf(1.0)
  {
    assert(iSpin()==1);
    constructSpinInfo(p,dir,time);
  }

  const Complex & wave() const {return _wf;}

public:

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

private:

  Complex _wf;

};
}
}

#endif