void get_sources(ParamCoLoRe *par)
{
//////
// Uses the gaussian matter density field to obtain a
// poisson sampling of point sources (returns an integer array
// with the number of sources in each cell).
int ii,nthr;
lint *np_tot_thr;
#ifdef _HAVE_OMP
nthr=omp_get_max_threads();
#else //_HAVE_OMP
nthr=1;
#endif //_HAVE_OMP
np_tot_thr=my_calloc(nthr,sizeof(lint));
print_info("*** Getting point sources\n");
if(NodeThis==0) timer(0);
print_info("Poisson-sampling\n");
#ifdef _HAVE_OMP
#pragma omp parallel default(none) \
shared(par,np_tot_thr,IThread0)
#endif //_HAVE_OMP
{
lint iz;
#ifdef _HAVE_OMP
int ithr=omp_get_thread_num();
#else //_HAVE_OMP
int ithr=0;
#endif //_HAVE_OMP
double dx=par->l_box/par->n_grid;
double cell_vol=dx*dx*dx;
int ngx=2*(par->n_grid/2+1);
unsigned int seed_thr=par->seed_rng+IThread0+ithr;
gsl_rng *rng_thr=init_rng(seed_thr);
#ifdef _HAVE_OMP
#pragma omp for schedule(static)
#endif //_HAVE_OMP
for(iz=0;iz<par->nz_here;iz++) {
int iy;
lint indexz=iz*((lint)(ngx*par->n_grid));
double z0=(iz+par->iz0_here+0.5)*dx-par->pos_obs[2];
for(iy=0;iy<par->n_grid;iy++) {
int ix;
lint indexy=iy*ngx;
double y0=(iy+0.5)*dx-par->pos_obs[1];
for(ix=0;ix<par->n_grid;ix++) {
int npp=0;
double dz_rsd=0;
lint index=ix+indexy+indexz;
double x0=(ix+0.5)*dx-par->pos_obs[1];
double r=sqrt(x0*x0+y0*y0+z0*z0);
double redshift=z_of_r(par,r);
double ndens=ndens_of_z(par,redshift);
if(ndens>0) {
double bias=bias_of_z(par,redshift);
double gfb=dgrowth_of_r(par,r)*bias;
double lambda=ndens*cell_vol*
exp(gfb*(par->grid_dens[index]-0.5*gfb*par->sigma2_gauss));
npp=rng_poisson(lambda,rng_thr);
dz_rsd=par->grid_rvel[index]*vgrowth_of_r(par,r);
}
par->grid_rvel[index]=dz_rsd;
par->nsources[index]=npp;
np_tot_thr[ithr]+=npp;
}
}
}//end omp for
end_rng(rng_thr);
}//end omp parallel
if(NodeThis==0) timer(2);
par->nsources_this=0;
for(ii=0;ii<nthr;ii++)
par->nsources_this+=np_tot_thr[ii];
par->nsources_total=0;
#ifdef _HAVE_MPI
MPI_Allreduce(&(par->nsources_this),&(par->nsources_total),1,LINT_MPI,MPI_SUM,MPI_COMM_WORLD);
#else //_HAVE_MPI
par->nsources_total=par->nsources_this;
#endif //_HAVE_MPI
print_info(" There will be %ld particles in total\n",(long)(par->nsources_total));
//#ifdef _DEBUG
// printf("Node %d has %ld particles\n",NodeThis,(long)(par->nsources_this));
//#endif //_DEBUG
for(ii=nthr-1;ii>0;ii--) {
int jj;
lint nh=0;
for(jj=0;jj<ii;jj++)
nh+=np_tot_thr[jj];
np_tot_thr[ii]=nh;
}
np_tot_thr[0]=0;
//np_tot_thr now contains the id of the first particle in the thread
#ifdef _SPREC
fftwf_free(par->grid_dens_f);
#else //_SPREC
fftw_free(par->grid_dens_f);
#endif //_SPREC
par->grid_dens_f=NULL;
par->gals=my_malloc(par->nsources_this*sizeof(Gal));
if(NodeThis==0) timer(0);
print_info("Assigning coordinates\n");
#ifdef _HAVE_OMP
#pragma omp parallel default(none) \
shared(par,IThread0,np_tot_thr)
#endif //_HAVE_OMP
{
lint iz;
#ifdef _HAVE_OMP
int ithr=omp_get_thread_num();
#else //_HAVE_OMP
int ithr=0;
#endif //_HAVE_OMP
double dx=par->l_box/par->n_grid;
int ngx=2*(par->n_grid/2+1);
unsigned int seed_thr=par->seed_rng+IThread0+ithr;
gsl_rng *rng_thr=init_rng(seed_thr);
#ifdef _HAVE_OMP
#pragma omp for schedule(static)
#endif //_HAVE_OMP
for(iz=0;iz<par->nz_here;iz++) {
int iy;
lint indexz=iz*((lint)(ngx*par->n_grid));
double z0=(iz+par->iz0_here)*dx-par->pos_obs[2];
for(iy=0;iy<par->n_grid;iy++) {
int ix;
lint indexy=iy*ngx;
double y0=iy*dx-par->pos_obs[1];
for(ix=0;ix<par->n_grid;ix++) {
double x0=ix*dx-par->pos_obs[1];
lint index=ix+indexy+indexz;
int npp=par->nsources[index];
if(npp>0) {
int ip;
double dz_rsd=par->grid_rvel[index];
for(ip=0;ip<npp;ip++) {
double cth,phi,r;
lint pid=np_tot_thr[ithr];
double x=x0+dx*rng_01(rng_thr);
double y=y0+dx*rng_01(rng_thr);
double z=z0+dx*rng_01(rng_thr);
cart2sph(x,y,z,&r,&cth,&phi);
par->gals[pid].ra=RTOD*phi;
par->gals[pid].dec=90-RTOD*acos(cth);
par->gals[pid].z0=z_of_r(par,r);
par->gals[pid].dz_rsd=dz_rsd;
np_tot_thr[ithr]++;
}
}
}
}
}//end omp for
end_rng(rng_thr);
}//end omp parallel
if(NodeThis==0) timer(2);
free(np_tot_thr);
print_info("\n");
}
quint32
APG::ConstraintSolver::playlist_size() const
{
return rng_poisson( (double)m_suggestedPlaylistSize );
}
