StateVector.h

Go to the documentation of this file.

// Copyright András Vukics 2006–2014. Distributed under the Boost Software License, Version 1.0. (See accompanying file LICENSE.txt)
// -*- C++ -*-
#ifndef CPPQEDCORE_QUANTUMDATA_STATEVECTOR_H_INCLUDED
#define CPPQEDCORE_QUANTUMDATA_STATEVECTOR_H_INCLUDED

#include "StateVectorFwd.h"

#include "DensityOperatorFwd.h"

#include "ArrayBase.h"
#include "DimensionsBookkeeper.h"
#include "LazyDensityOperator.h"
#include "Types.h"

#include "Operators.h"


namespace quantumdata {

template<int RANK1, int RANK2>
inline
const StateVector<RANK1+RANK2>
operator*(const StateVector<RANK1>&, const StateVector<RANK2>&);


template<int RANK>
const dcomp
braket(const StateVector<RANK>&, const StateVector<RANK>&);



template<int RANK>
class StateVector 
  : public LazyDensityOperator<RANK>, 
    private ArrayBase<RANK>,
    private linalg::VectorSpace<StateVector<RANK> >
{
public:
  static const int N_RANK=RANK;

  typedef LazyDensityOperator<RANK> LDO_Base;
  typedef ArrayBase          <RANK>    ABase;

  typedef typename LDO_Base::Dimensions Dimensions;

  typedef typename ABase::ArrayLow StateVectorLow;

  typedef typename Types<RANK>::DensityOperatorLow DensityOperatorLow;

  typedef typename LDO_Base::Idx Idx;

  using ABase::vectorView; using ABase::getArray;



  StateVector(const StateVectorLow& psi, ByReference) : LDO_Base(psi.shape()), ABase(psi) {}

  explicit StateVector(const Dimensions&, bool init=true); 

  StateVector(const StateVector&); 


  template<int RANK2>
  StateVector(const StateVector<RANK2>&, const StateVector<RANK-RANK2>&);
  

  StateVector& operator=(const StateVector& sv) {ABase::operator=(sv.getArray()); return *this;}


  template<typename OTHER>
  StateVector& operator=(const OTHER& other) {getArray()=other; return *this;}
  

  template<typename... SubscriptPack>
  const dcomp& operator()(int s0, SubscriptPack... subscriptPack) const; 

  template<typename... SubscriptPack>
  dcomp& operator()(int s0, SubscriptPack... subscriptPack) {return const_cast<dcomp&>(static_cast<const StateVector*>(this)->operator()(s0,subscriptPack...));} 





  double   norm() const {return ABase::frobeniusNorm();}
  double renorm()                                     ; 

  


  const DensityOperatorLow dyad(const StateVector&) const;
  const DensityOperatorLow dyad(                  ) const {return dyad(*this);} 



  void addTo(DensityOperator<RANK>& densityOperator) const;


  StateVector& operator+=(const StateVector& psi) {ABase::operator+=(psi); return *this;}
  StateVector& operator-=(const StateVector& psi) {ABase::operator-=(psi); return *this;}

  const StateVector operator-() const {StateVector res(this->getDimensions(),false); res.getArray()=-this->getArray(); return res;} 
  const StateVector operator+() const {return *this;} 



  template<typename OTHER>
  StateVector& operator*=(const OTHER& dc) {ABase::operator*=(dc); return *this;} 

  template<typename OTHER>
  StateVector& operator/=(const OTHER& dc) {ABase::operator/=(dc); return *this;}
  
#ifndef NDEBUG
  void debug() const {std::cerr<<"Debug: "<<getArray()<<std::endl;}
#endif // NDEBUG
  
private:
  const dcomp index(const Idx& i, const Idx& j) const override {return getArray()(i)*conj(getArray()(j));} 
  
  double trace_v() const override {return norm();} 
  
};


template<int RANK1, int RANK2>
inline
const StateVector<RANK1+RANK2>
operator*(const StateVector<RANK1>& t1, const StateVector<RANK2>& t2)
{
  return StateVector<RANK1+RANK2>(t1,t2);
}


} // quantumdata

#endif // CPPQEDCORE_QUANTUMDATA_STATEVECTOR_H_INCLUDED