EvtSpinAmp.hh

Go to the documentation of this file.

#ifndef __EVTSPINAMP_HH__
#define __EVTSPINAMP_HH__
 
#include "EvtComplex.hh"
#include "EvtSpinType.hh"
 
#include <vector>
using std::vector;
 
#include <cstdarg>
 
class EvtSpinAmp;
EvtSpinAmp operator*( const EvtComplex&, const EvtSpinAmp& );
EvtSpinAmp operator*( const EvtSpinAmp&, const EvtComplex& );
EvtSpinAmp operator/( const EvtSpinAmp&, const EvtComplex& );
 
class EvtSpinAmp {
 
  friend EvtSpinAmp    operator*( const EvtComplex&, const EvtSpinAmp& );
  friend EvtSpinAmp    operator*( const EvtSpinAmp&, const EvtComplex& );
  friend EvtSpinAmp    operator/( const EvtSpinAmp&, const EvtComplex& );
  friend std::ostream& operator<<( std::ostream&, const EvtSpinAmp& );
 
public:
  EvtSpinAmp(){};
  EvtSpinAmp( const vector<EvtSpinType::spintype>& twospin );
  EvtSpinAmp( const vector<EvtSpinType::spintype>& twospin, const EvtComplex& val );
  EvtSpinAmp( const vector<EvtSpinType::spintype>& twospin, const vector<EvtComplex>& elem );
  EvtSpinAmp( const EvtSpinAmp& );
 
  ~EvtSpinAmp(){};
 
  // Input to the index functions are twice the magnetic quantum number
  EvtComplex&       operator()( const vector<int>& );
  const EvtComplex& operator()( const vector<int>& ) const;
  EvtComplex&       operator()( int, ... );
  const EvtComplex& operator()( int, ... ) const;
 
  EvtSpinAmp& operator=( const EvtSpinAmp& );
 
  EvtSpinAmp  operator+( const EvtSpinAmp& ) const;
  EvtSpinAmp& operator+=( const EvtSpinAmp& );
 
  EvtSpinAmp  operator-( const EvtSpinAmp& ) const;
  EvtSpinAmp& operator-=( const EvtSpinAmp& );
 
  // Direct Product
  EvtSpinAmp  operator*( const EvtSpinAmp& ) const;
  EvtSpinAmp& operator*=( const EvtSpinAmp& );
 
  EvtSpinAmp& operator*=( const EvtComplex& );
  EvtSpinAmp& operator/=( const EvtComplex& );
 
  // Contraction of amplitudes
  void intcont( int, int );
  void extcont( const EvtSpinAmp&, int, int );
 
  // assign this value to every member in the container
  void assign( const EvtComplex& val ) { _elem.assign( _elem.size(), val ); }
 
  // get the order of the container
  int rank() const { return _twospin.size(); }
 
  // get the dimension vector of the container
  const vector<int>& dims() const { return _twospin; }
 
  // set the elements and the dimensions of the vector - useful for something
  // things eventough it is usually not the cleanest solution
  void addspin( int twospin ) { _twospin.push_back( twospin ); }
  void setelem( const vector<EvtComplex>& elem ) { _elem = elem; }
 
  bool        iterate( vector<int>& index ) const;
  vector<int> iterinit() const;
 
  bool        allowed( const vector<int>& index ) const;
  bool        iterateallowed( vector<int>& index ) const;
  vector<int> iterallowedinit() const;
 
private:
  void        checkindexargs( const vector<int>& index ) const;
  void        checktwospin( const vector<int>& twospin ) const;
  int         findtrueindex( const vector<int>& index ) const;
  vector<int> calctwospin( const vector<EvtSpinType::spintype>& type ) const;
 
  vector<EvtSpinType::spintype> _type;
  vector<int>                   _twospin;
  vector<EvtComplex>            _elem;
};
 
#endif // __EVTSPINAMP__