WaveFunctionBase.h

// -*- C++ -*-
//
// WaveFunctionBase.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_WaveFunctionBase_H
#define ThePEG_WaveFunctionBase_H
//
// This is the declaration of the WaveFunctionBase class.

#include <ThePEG/Vectors/Lorentz5Vector.h>
#include <ThePEG/Vectors/LorentzVector.h>
#include <ThePEG/PDT/ParticleData.h>
#include <ThePEG/Helicity/HelicityDefinitions.h>

namespace ThePEG {

namespace Helicity {



enum Direction 
{
  incoming, 
  outgoing, 
  intermediate 
};

class WaveFunctionBase{

public:



  WaveFunctionBase() 
    : _particle(), _momentum(), _dir(intermediate) 
  {}

  WaveFunctionBase(const Lorentz5Momentum & p,
                   tcPDPtr pd, Direction dir = intermediate) 
    : _particle(pd), _momentum(p), _dir(dir) 
  {
    if(_dir==outgoing) _momentum *= -1.0; 
    if ( dir != outgoing ) {
      tcPDPtr anti = pd->CC();
      if ( anti ) _particle = anti;
    }
  }


  Energy px() const {return _momentum.x();}

  Energy py() const {return _momentum.y();}

  Energy pz() const {return _momentum.z();}

  Energy e()  const {return _momentum.e();}

  Energy mass() const {return _momentum.mass();}

  Energy2 m2() const {return _momentum.m2();}

  const Lorentz5Momentum & momentum() const {return _momentum;}

  long id() const {return _particle->id();}

  PDT::Spin iSpin() const {return _particle->iSpin();}

  tcPDPtr particle() const {return _particle;}

  ThePEG::Helicity::Direction direction() const {return _dir;}

  void direction(ThePEG::Helicity::Direction in) {_dir=in;} 

protected:
  
  void transformMomentum(const LorentzRotation & r) {
    _momentum.transform(r);
  }

private:

  tcPDPtr _particle;

  Lorentz5Momentum _momentum;

  Direction _dir;
};
}
}

#endif