assign_pkmemory.c

/*
1;95;0cint prep_c2c(){
  nx = n2;

  overdensity         = (fftw_complex*) fftw_malloc(n0*n1*n2*sizeof(*overdensity));
  smooth_overdensity  = (fftw_complex*) fftw_malloc(n0*n1*n2*sizeof(*overdensity)); // 1d: N input elements -> N/2 + 1 output elements.
  
  H_k                 = (fftw_complex*) fftw_malloc(n0*n1*nx*sizeof(*H_k));

  plan                = fftw_plan_dft_3d(n0, n1, n2, overdensity, H_k, FFTW_FORWARD, FFTW_ESTIMATE); // FFTW_ESTIMATE, FFTW_MEASURE, FFTW_PATIENT, //FFTW_EXHAUSTIVE.

  return 0;
}
*/
int prep_r2c(){
  nx = n2/2 + 1;

  overdensity         = (double*)       fftw_malloc(n0*n1*n2*sizeof(*overdensity));                  // here overdensity is a double, rather than complex.
  //  smooth_overdensity  = (double*)       fftw_malloc(n0*n1*n2*sizeof(*smooth_overdensity));
  
  H_k                 = (fftw_complex*) fftw_malloc(n0*n1*nx*sizeof(*H_k)); // returns half the array, along the fastest memory change direction: x.

  // FFTW_ESTIMATE, FFTW_MEASURE, FFTW_PATIENT, FFTW_EXHAUSTIVE.
  plan                = fftw_plan_dft_r2c_3d(n0, n1, n2, overdensity, H_k, FFTW_MEASURE); // r2c is always forward.
  
  return 0;
}

int prep_x2c(){
  //  fftw_import_wisdom_from_filename("/home/mjw/HOD_MockRun/wisdom/wisdom.dat");

  // prep_c2c();
  prep_r2c();

  //  fftw_export_wisdom_to_filename("/home/mjw/HOD_MockRun/wisdom/wisdom.dat");

  for(j=0; j<n0*n1*n2; j++)  overdensity[j] = 0.0;

  /*
  for(j=0; j<n0*n1*n2; j++){
    overdensity[j][0] = 0.0;
    overdensity[j][1] = 0.0;
  }
  */
  
  regress_mem(&flat);
  //  regress_mem(&half);
  //  regress_mem(&quart);
  
  walltime("Wall time after pk array malloc");
  
  return 0;
}

int regress_mem(regress* inst){
  inst->kind = calloc(n0*n1*nx, sizeof(int));
  inst->kLi  = calloc(n0*n1*nx, sizeof(double));
  inst->kM2  = calloc(n0*n1*nx, sizeof(double));

  return 0;
}