Homec4science
Diffusion tamaas (archive/legacy_mpi) Browse

tamaas/src/surface_complex.hh
ab3353f71c30archive/legacy_mpi

Last Change
Raw File

surface_complex.hh
View Options

#ifndef SURFACE_COMPLEX_H
#define SURFACE_COMPLEX_H
/* -------------------------------------------------------------------------- */
template <typename T>
class SurfaceComplex : public Map2dSquare<std::complex<T> > {
public:
SurfaceComplex(UInt a, Real L):Map2dSquare<std::complex<T> >(a,L){}
SurfaceComplex(Surface<T> & s);
SurfaceComplex<T> & operator=(const Surface<T> & s);
SurfaceComplex<T> & operator=(const SurfaceComplex<T> & s);
//! make a surface real by taking norm as real part and imaginary 0
void makeItRealBySquare();
//! make it real by summing the imaginary and real parts
void makeItRealBySum();
//! make a surface real by taking norm as real part and imaginary 0
void makeItRealByAbs();
//! get real part
Surface<T> real() const;
//! get imaginary part
Surface<T> imag() const;
// set all the members of the map to a given value
using Map2d<std::complex<T> >::operator=;
virtual const SurfaceComplex<T> & getWrappedNumpy(){return *this;};
};
/* -------------------------------------------------------------------------- */
template <typename T>
SurfaceComplex<T> & SurfaceComplex<T>::operator=(const Surface<T> & s){
this->setGridSize(s.size());
UInt size = s.size()*s.size();
for (UInt i = 0; i < size; i++) {
this->at(i) = s(i);
}
return *this;
}
/* -------------------------------------------------------------------------- */
template <typename T>
SurfaceComplex<T> & SurfaceComplex<T>::operator=(const SurfaceComplex<T> & s){
this->setGridSize(s.size());
UInt size = s.size()*s.size();
for (UInt i = 0; i < size; i++) {
this->at(i) = s(i);
}
return *this;
}
/* -------------------------------------------------------------------------- */
template <typename T>
Surface<T> SurfaceComplex<T>::real()const{
Surface<T> res(this->size(),this->getL());
UInt n = this->size();
for (UInt i = 0 ; i < n ; ++i)
for (UInt j = 0 ; j < n ; ++j){
res(i,j) = this->at(i,j).real();
}
return res;
}
/* -------------------------------------------------------------------------- */
template <typename T>
Surface<T> SurfaceComplex<T>::imag()const{
Surface<T> res(this->size(),this->getL());
UInt n = this->size();
for (UInt i = 0 ; i < n ; ++i)
for (UInt j = 0 ; j < n ; ++j){
res(i,j) = this->at(i,j).imag();
}
return res;
}
/* -------------------------------------------------------------------------- */
template <typename T>
void SurfaceComplex<T>::makeItRealBySquare(){
UInt n = this->size();
for (UInt i = 0 ; i < n ; ++i)
for (UInt j = 0 ; j < n ; ++j){
Real norm = std::norm(this->at(i,j));
this->at(i,j) = norm;
}
}
/* -------------------------------------------------------------------------- */
template <typename T>
void SurfaceComplex<T>::makeItRealByAbs(){
UInt n = this->size();
for (UInt i = 0 ; i < n ; ++i)
for (UInt j = 0 ; j < n ; ++j){
Real abs = std::abs(this->at(i,j));
this->at(i,j) = abs;
}
}
/* -------------------------------------------------------------------------- */
template <typename T>
void SurfaceComplex<T>::makeItRealBySum(){
UInt n = this->size();
for (UInt i = 0 ; i < n*n ; ++i){
this->at(i) = this->at(i).real() + this->at(i).imag();
}
}
/* -------------------------------------------------------------------------- */
template <typename T>
SurfaceComplex<T>::SurfaceComplex(Surface<T> & surface):
Map2dSquare<std::complex<T> >(surface.size(),surface.getL()){
UInt n = this->size();
for (UInt i = 0; i < n*n; i++) {
this->at(i) = surface(i);
}
}
/* -------------------------------------------------------------------------- */
#endif /* SURFACE_COMPLEX_H */
c4science · Help