Skip to content

Commit ea778e3

Browse files
committed
small aesthetic improvements
1 parent f800d91 commit ea778e3

File tree

1 file changed

+13
-13
lines changed

1 file changed

+13
-13
lines changed

pocketfft_hdronly.h

+13-13
Original file line numberDiff line numberDiff line change
@@ -1456,7 +1456,7 @@ template<bool fwd, typename T> void pass_all(T c[], T0 fct) const
14561456
for (size_t i=0; i<length; ++i)
14571457
c[i] = ch[i]*fct;
14581458
else
1459-
std::copy (p1, p1+length, c);
1459+
std::copy_n (p1, length, c);
14601460
}
14611461
else
14621462
if (fct!=1.)
@@ -2204,36 +2204,36 @@ template<typename T> void radbg(size_t ido, size_t ip, size_t l1,
22042204
}
22052205
}
22062206

2207-
template<typename T> void copy_and_norm(T *c, T *p1, size_t n, T0 fct) const
2207+
template<typename T> void copy_and_norm(T *c, T *p1, T0 fct) const
22082208
{
22092209
if (p1!=c)
22102210
{
22112211
if (fct!=1.)
2212-
for (size_t i=0; i<n; ++i)
2212+
for (size_t i=0; i<length; ++i)
22132213
c[i] = fct*p1[i];
22142214
else
2215-
std::copy (p1, p1+length, c);
2215+
std::copy_n (p1, length, c);
22162216
}
22172217
else
22182218
if (fct!=1.)
2219-
for (size_t i=0; i<n; ++i)
2219+
for (size_t i=0; i<length; ++i)
22202220
c[i] *= fct;
22212221
}
22222222

22232223
public:
22242224
template<typename T> void exec(T c[], T0 fct, bool r2hc) const
22252225
{
22262226
if (length==1) { c[0]*=fct; return; }
2227-
size_t n=length, nf=fact.size();
2228-
arr<T> ch(n);
2227+
size_t nf=fact.size();
2228+
arr<T> ch(length);
22292229
T *p1=c, *p2=ch.data();
22302230

22312231
if (r2hc)
2232-
for(size_t k1=0, l1=n; k1<nf;++k1)
2232+
for(size_t k1=0, l1=length; k1<nf;++k1)
22332233
{
22342234
size_t k=nf-k1-1;
22352235
size_t ip=fact[k].fct;
2236-
size_t ido=n / l1;
2236+
size_t ido=length / l1;
22372237
l1 /= ip;
22382238
if(ip==4)
22392239
radf4(ido, l1, p1, p2, fact[k].tw);
@@ -2251,7 +2251,7 @@ template<typename T> void radbg(size_t ido, size_t ip, size_t l1,
22512251
for(size_t k=0, l1=1; k<nf; k++)
22522252
{
22532253
size_t ip = fact[k].fct,
2254-
ido= n/(ip*l1);
2254+
ido= length/(ip*l1);
22552255
if(ip==4)
22562256
radb4(ido, l1, p1, p2, fact[k].tw);
22572257
else if(ip==2)
@@ -2266,7 +2266,7 @@ template<typename T> void radbg(size_t ido, size_t ip, size_t l1,
22662266
l1*=ip;
22672267
}
22682268

2269-
copy_and_norm(c,p1,n,fct);
2269+
copy_and_norm(c,p1,fct);
22702270
}
22712271

22722272
private:
@@ -2437,12 +2437,12 @@ template<typename T0> class fftblue
24372437
tmp[m].Set(c[m], zero);
24382438
fft<true>(tmp.data(),fct);
24392439
c[0] = tmp[0].r;
2440-
std::copy (&tmp[1].r, &tmp[1].r+n-1, &c[1]);
2440+
std::copy_n (&tmp[1].r, n-1, &c[1]);
24412441
}
24422442
else
24432443
{
24442444
tmp[0].Set(c[0],c[0]*0);
2445-
std::copy (c+1, c+n, &tmp[1].r);
2445+
std::copy_n (c+1, n-1, &tmp[1].r);
24462446
if ((n&1)==0) tmp[n/2].i=T0(0)*c[0];
24472447
for (size_t m=1; 2*m<n; ++m)
24482448
tmp[n-m].Set(tmp[m].r, -tmp[m].i);

0 commit comments

Comments
 (0)