mirror of
https://github.com/paboyle/Grid.git
synced 2024-11-14 01:35:36 +00:00
74 lines
2.1 KiB
C++
74 lines
2.1 KiB
C++
#ifndef GRID_LATTICE_UNARY_H
|
|
#define GRID_LATTICE_UNARY_H
|
|
|
|
namespace Grid {
|
|
|
|
//////////////////////////////////////////////////////////////////////////////////////////////////////
|
|
// avoid copy back routines for mult, mac, sub, add
|
|
//////////////////////////////////////////////////////////////////////////////////////////////////////
|
|
template<class obj> Lattice<obj> sqrt(const Lattice<obj> &rhs){
|
|
Lattice<obj> ret(rhs._grid);
|
|
ret.checkerboard = rhs.checkerboard;
|
|
conformable(ret,rhs);
|
|
PARALLEL_FOR_LOOP
|
|
for(int ss=0;ss<rhs._grid->oSites();ss++){
|
|
ret._odata[ss]=sqrt(rhs._odata[ss]);
|
|
}
|
|
return ret;
|
|
}
|
|
template<class obj> Lattice<obj> rsqrt(const Lattice<obj> &rhs){
|
|
Lattice<obj> ret(rhs._grid);
|
|
ret.checkerboard = rhs.checkerboard;
|
|
conformable(ret,rhs);
|
|
PARALLEL_FOR_LOOP
|
|
for(int ss=0;ss<rhs._grid->oSites();ss++){
|
|
ret._odata[ss]=rsqrt(rhs._odata[ss]);
|
|
}
|
|
return ret;
|
|
}
|
|
template<class obj> Lattice<obj> sin(const Lattice<obj> &rhs){
|
|
Lattice<obj> ret(rhs._grid);
|
|
ret.checkerboard = rhs.checkerboard;
|
|
conformable(ret,rhs);
|
|
PARALLEL_FOR_LOOP
|
|
for(int ss=0;ss<rhs._grid->oSites();ss++){
|
|
ret._odata[ss]=sin(rhs._odata[ss]);
|
|
}
|
|
return ret;
|
|
}
|
|
template<class obj> Lattice<obj> cos(const Lattice<obj> &rhs){
|
|
Lattice<obj> ret(rhs._grid);
|
|
ret.checkerboard = rhs.checkerboard;
|
|
conformable(ret,rhs);
|
|
PARALLEL_FOR_LOOP
|
|
for(int ss=0;ss<rhs._grid->oSites();ss++){
|
|
ret._odata[ss]=cos(rhs._odata[ss]);
|
|
}
|
|
return ret;
|
|
}
|
|
|
|
|
|
template<class obj> Lattice<obj> pow(const Lattice<obj> &rhs,RealD y){
|
|
Lattice<obj> ret(rhs._grid);
|
|
ret.checkerboard = rhs.checkerboard;
|
|
conformable(ret,rhs);
|
|
PARALLEL_FOR_LOOP
|
|
for(int ss=0;ss<rhs._grid->oSites();ss++){
|
|
ret._odata[ss]=pow(rhs._odata[ss],y);
|
|
}
|
|
return ret;
|
|
}
|
|
template<class obj> Lattice<obj> mod(const Lattice<obj> &rhs,Integer y){
|
|
Lattice<obj> ret(rhs._grid);
|
|
ret.checkerboard = rhs.checkerboard;
|
|
conformable(ret,rhs);
|
|
PARALLEL_FOR_LOOP
|
|
for(int ss=0;ss<rhs._grid->oSites();ss++){
|
|
ret._odata[ss]=mod(rhs._odata[ss],y);
|
|
}
|
|
return ret;
|
|
}
|
|
|
|
}
|
|
#endif
|