Aquila-Consortium/libsharp2
0
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Use macros to shorten alm2almtmp and almtmp2alm

Browse source
This commit is contained in:
Dag Sverre Seljebotn 2013-01-07 11:23:15 +01:00
parent 1ec2c44b85
commit 80b2f9abff
1 changed files with 30 additions and 32 deletions
Download patch file Download diff file Expand all files Collapse all files

62
libsharp/sharp.c
View file

@ -417,6 +417,15 @@ static void dealloc_almtmp (sharp_job *job)
static void alm2almtmp (sharp_job *job, int lmax, int mi) static void alm2almtmp (sharp_job *job, int lmax, int mi)
{ {
#define COPY_LOOP(source_t, expr_of_x) \
for (int l=job->ainfo->mval[mi]; l<=lmax; ++l) \
for (int i=0; i<job->ntrans*job->nalm; ++i) \
{ \
source_t x = ((source_t *)job->alm[i])[ofs+l*stride]; \
job->almtmp[job->ntrans*job->nalm*l+i] = expr_of_x; \
}
if (job->type!=SHARP_MAP2ALM) if (job->type!=SHARP_MAP2ALM)
{ {
ptrdiff_t ofs=job->ainfo->mvstart[mi]; ptrdiff_t ofs=job->ainfo->mvstart[mi];
@ -424,66 +433,55 @@ static void alm2almtmp (sharp_job *job, int lmax, int mi)
if (job->spin==0) if (job->spin==0)
{ {
if (job->flags&SHARP_DP) if (job->flags&SHARP_DP)
for (int l=job->ainfo->mval[mi]; l<=lmax; ++l) COPY_LOOP(dcmplx, x)
for (int i=0; i<job->ntrans*job->nalm; ++i)
job->almtmp[job->ntrans*job->nalm*l+i]
= ((dcmplx *)job->alm[i])[ofs+l*stride];
else else
for (int l=job->ainfo->mval[mi]; l<=lmax; ++l) COPY_LOOP(fcmplx, x)
for (int i=0; i<job->ntrans*job->nalm; ++i)
job->almtmp[job->ntrans*job->nalm*l+i]
= ((fcmplx *)job->alm[i])[ofs+l*stride];
} }
else else
{ {
if (job->flags&SHARP_DP) if (job->flags&SHARP_DP)
for (int l=job->ainfo->mval[mi]; l<=lmax; ++l) COPY_LOOP(dcmplx, x*job->norm_l[l])
for (int i=0; i<job->ntrans*job->nalm; ++i)
job->almtmp[job->ntrans*job->nalm*l+i]
= ((dcmplx *)job->alm[i])[ofs+l*stride]*job->norm_l[l];
else else
for (int l=job->ainfo->mval[mi]; l<=lmax; ++l) COPY_LOOP(fcmplx, x*job->norm_l[l])
for (int i=0; i<job->ntrans*job->nalm; ++i)
job->almtmp[job->ntrans*job->nalm*l+i]
= ((fcmplx *)job->alm[i])[ofs+l*stride]*job->norm_l[l];
} }
} }
else else
SET_ARRAY(job->almtmp,job->ntrans*job->nalm*job->ainfo->mval[mi], SET_ARRAY(job->almtmp,job->ntrans*job->nalm*job->ainfo->mval[mi],
job->ntrans*job->nalm*(lmax+1),0.); job->ntrans*job->nalm*(lmax+1),0.);
#undef COPY_LOOP
} }
static void almtmp2alm (sharp_job *job, int lmax, int mi) static void almtmp2alm (sharp_job *job, int lmax, int mi)
{ {
#define COPY_LOOP(target_t, expr_of_x) \
for (int l=job->ainfo->mval[mi]; l<=lmax; ++l) \
for (int i=0; i<job->ntrans*job->nalm; ++i) \
{ \
dcmplx x = job->almtmp[job->ntrans*job->nalm*l+i]; \
((target_t *)job->alm[i])[ofs+l*stride] += expr_of_x; \
}
if (job->type != SHARP_MAP2ALM) return; if (job->type != SHARP_MAP2ALM) return;
ptrdiff_t ofs=job->ainfo->mvstart[mi]; ptrdiff_t ofs=job->ainfo->mvstart[mi];
int stride=job->ainfo->stride; int stride=job->ainfo->stride;
if (job->spin==0) if (job->spin==0)
{ {
if (job->flags&SHARP_DP) if (job->flags&SHARP_DP)
for (int l=job->ainfo->mval[mi]; l<=lmax; ++l) COPY_LOOP(dcmplx, x)
for (int i=0;i<job->ntrans*job->nalm;++i)
((dcmplx *)job->alm[i])[ofs+l*stride] +=
job->almtmp[job->ntrans*job->nalm*l+i];
else else
for (int l=job->ainfo->mval[mi]; l<=lmax; ++l) COPY_LOOP(fcmplx, (fcmplx)x)
for (int i=0;i<job->ntrans*job->nalm;++i)
((fcmplx *)job->alm[i])[ofs+l*stride] +=
(fcmplx)(job->almtmp[job->ntrans*job->nalm*l+i]);
} }
else else
{ {
if (job->flags&SHARP_DP) if (job->flags&SHARP_DP)
for (int l=job->ainfo->mval[mi]; l<=lmax; ++l) COPY_LOOP(dcmplx, x * job->norm_l[l])
for (int i=0;i<job->ntrans*job->nalm;++i)
((dcmplx *)job->alm[i])[ofs+l*stride] +=
job->almtmp[job->ntrans*job->nalm*l+i]*job->norm_l[l];
else else
for (int l=job->ainfo->mval[mi]; l<=lmax; ++l) COPY_LOOP(fcmplx, (fcmplx)(x * job->norm_l[l]))
for (int i=0;i<job->ntrans*job->nalm;++i)
((fcmplx *)job->alm[i])[ofs+l*stride] +=
(fcmplx)(job->almtmp[job->ntrans*job->nalm*l+i]*job->norm_l[l]);
} }
#undef COPY_LOOP
} }
static void phase2map (sharp_job *job, int mmax, int llim, int ulim) static void phase2map (sharp_job *job, int mmax, int llim, int ulim)