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