| 1 | /* ------------------------ Projet BAORadio --------------------
|
|---|
| 2 | Programme de calcul du spectre de puissance de bruit pour
|
|---|
| 3 | un interferometre (spectre moyenne sur 3D -> P_noise(k) )
|
|---|
| 4 |
|
|---|
| 5 | R. Ansari , C. Magneville - Juin 2010
|
|---|
| 6 |
|
|---|
| 7 | Usage: pknoise pknoise [-parname value] Diameter/Four2DRespTableFile OutPPFName
|
|---|
| 8 | -parname : -noise , -renmax, -scut, -ngen, -z, -bsize, -cszmpc, -nbin -prt
|
|---|
| 9 | --------------------------------------------------------------- */
|
|---|
| 10 |
|
|---|
| 11 | #include "machdefs.h"
|
|---|
| 12 | #include "sopnamsp.h"
|
|---|
| 13 | #include <iostream>
|
|---|
| 14 | #include <string>
|
|---|
| 15 | #include <math.h>
|
|---|
| 16 |
|
|---|
| 17 | #include <typeinfo>
|
|---|
| 18 |
|
|---|
| 19 | #include "mdish.h"
|
|---|
| 20 | #include "specpk.h"
|
|---|
| 21 | #include "interfconfigs.h"
|
|---|
| 22 | #include "radutil.h"
|
|---|
| 23 |
|
|---|
| 24 | #include "histinit.h"
|
|---|
| 25 | // #include "fiosinit.h"
|
|---|
| 26 | // #include "fitsioserver.h"
|
|---|
| 27 |
|
|---|
| 28 | #include "randr48.h"
|
|---|
| 29 |
|
|---|
| 30 | #include "timing.h"
|
|---|
| 31 | #include "ctimer.h"
|
|---|
| 32 | #include "slininterp.h"
|
|---|
| 33 |
|
|---|
| 34 | typedef DR48RandGen RandomGenerator ;
|
|---|
| 35 |
|
|---|
| 36 | // ---------------------------------------------------------------------
|
|---|
| 37 | // Test main
|
|---|
| 38 | // R. Ansari - Avril-Dec 2010
|
|---|
| 39 | // ---------------------------------------------------------------------
|
|---|
| 40 |
|
|---|
| 41 | void Usage()
|
|---|
| 42 | {
|
|---|
| 43 | cout << " Usage: pknoise [-parname value] Diameter/Four2DRespTableFile OutPPFName \n"
|
|---|
| 44 | << " -noise NoiseLevel (default=1.) \n"
|
|---|
| 45 | << " -renmax MaxValue (default : Do NOT renormalize 2D response value \n"
|
|---|
| 46 | << " -ngen NGen (default=0) number of noise fourier amp generations \n"
|
|---|
| 47 | << " NGen=0 -> Call ComputeNoisePk(), else generate Fourier Amplitudes (random) \n"
|
|---|
| 48 | << " NGen=-1 -> Call ComputeNoisePk(), dont change Four3DPk cell size \n"
|
|---|
| 49 | << " -bsize sx,sy,sz : 3D real space box size (default=512x512x256) \n"
|
|---|
| 50 | << " -cszmpc sx,sy,sz : 3D real space box cell size in Mpc (default=3x3x1.5) \n"
|
|---|
| 51 | << " -z redshift : redshift (default=1.0) \n"
|
|---|
| 52 | << " -pnf : Apply frequency dependent noise factor=ComovDa^2*(1+z)^2/Hz (default= NON) \n"
|
|---|
| 53 | << " -scut SCutValue (default= -100.) \n"
|
|---|
| 54 | << " if SCutValue<0. ==> SCut=MinNoisePower*(-SCutValue) \n"
|
|---|
| 55 | << " -prt PrtLev,PrtModulo (default=0,10) \n"
|
|---|
| 56 | << " -nbin PkNBin : nomber of histogram bins for Pnoise(k) (default= 256) \n"
|
|---|
| 57 | << " NOTE : this program needs redshift_da.txt andredshift_Hz.txt files" << endl;
|
|---|
| 58 | return;
|
|---|
| 59 | }
|
|---|
| 60 |
|
|---|
| 61 | //-------------------------------------------------------------------------
|
|---|
| 62 | // ------------------ MAIN PROGRAM ------------------------------
|
|---|
| 63 | //-------------------------------------------------------------------------
|
|---|
| 64 | int main(int narg, const char* arg[])
|
|---|
| 65 | {
|
|---|
| 66 | if ( (narg<3)||((narg>1)&&(strcmp(arg[1],"-h")==0)) ) {
|
|---|
| 67 | Usage();
|
|---|
| 68 | return 1;
|
|---|
| 69 | }
|
|---|
| 70 | cout << " ==== pknoise.cc : interferometer noise power spectrum computation ==== " << endl;
|
|---|
| 71 | // make sure SOPHYA modules are initialized
|
|---|
| 72 | SophyaInit();
|
|---|
| 73 | // FitsIOServerInit();
|
|---|
| 74 | InitTim();
|
|---|
| 75 | //--- decoding command line arguments
|
|---|
| 76 | string tits="pknoise";
|
|---|
| 77 | char cbuff[64];
|
|---|
| 78 | bool fgresptbl=false;
|
|---|
| 79 | double DIAMETRE=100.;
|
|---|
| 80 | string resptblname;
|
|---|
| 81 | double NoiseLevel=1.;
|
|---|
| 82 |
|
|---|
| 83 | bool fgrenorm=false;
|
|---|
| 84 | double rmax=1.;
|
|---|
| 85 | int NMAX = 0;
|
|---|
| 86 | double SCut=-100.;
|
|---|
| 87 | bool fgautoscut=true;
|
|---|
| 88 | double FacSCut=-SCut;
|
|---|
| 89 | double z_Redshift=1.0 ; // z=0.7 : 21 cm at z=0.7 -> 0.357 m
|
|---|
| 90 | double comov_dA_z=3330.; // Comoving radial distance = (1+z)dA
|
|---|
| 91 | double H_z=120.5; // Hubble_param(z)
|
|---|
| 92 | int box3dsz[3]={512,512,256};
|
|---|
| 93 | double cellsz[3]={3.,3.,3.};
|
|---|
| 94 | bool fgnoisefreq=false;
|
|---|
| 95 | int NBINPK=256;
|
|---|
| 96 | int prtlev=0;
|
|---|
| 97 | int prtmod=10;
|
|---|
| 98 |
|
|---|
| 99 | int ka=1;
|
|---|
| 100 | while (ka<(narg-1)) {
|
|---|
| 101 | if (strcmp(arg[ka],"-noise")==0) {
|
|---|
| 102 | NoiseLevel=atof(arg[ka+1]);
|
|---|
| 103 | ka+=2;
|
|---|
| 104 | }
|
|---|
| 105 | else if (strcmp(arg[ka],"-renmax")==0) {
|
|---|
| 106 | rmax=atof(arg[ka+1]); fgrenorm=true; ka+=2;
|
|---|
| 107 | }
|
|---|
| 108 | else if (strcmp(arg[ka],"-scut")==0) {
|
|---|
| 109 | SCut=atof(arg[ka+1]); ka+=2;
|
|---|
| 110 | if (SCut<0.) { FacSCut=-SCut; fgautoscut=true; }
|
|---|
| 111 | }
|
|---|
| 112 | else if (strcmp(arg[ka],"-ngen")==0) {
|
|---|
| 113 | NMAX=atoi(arg[ka+1]); ka+=2;
|
|---|
| 114 | }
|
|---|
| 115 | else if (strcmp(arg[ka],"-z")==0) {
|
|---|
| 116 | z_Redshift=atof(arg[ka+1]); ka+=2;
|
|---|
| 117 | }
|
|---|
| 118 | else if (strcmp(arg[ka],"-bsize")==0) {
|
|---|
| 119 | sscanf(arg[ka+1],"%d,%d,%d",box3dsz,box3dsz+1,box3dsz+2); ka+=2;
|
|---|
| 120 | }
|
|---|
| 121 | else if (strcmp(arg[ka],"-cszmpc")==0) {
|
|---|
| 122 | sscanf(arg[ka+1],"%lg,%lg,%lg",cellsz,cellsz+1,cellsz+2); ka+=2;
|
|---|
| 123 | }
|
|---|
| 124 | else if (strcmp(arg[ka],"-nbin")==0) {
|
|---|
| 125 | NBINPK=atoi(arg[ka+1]); ka+=2;
|
|---|
| 126 | }
|
|---|
| 127 | else if (strcmp(arg[ka],"-pnf")==0) {
|
|---|
| 128 | fgnoisefreq=true; ka++;
|
|---|
| 129 | }
|
|---|
| 130 | else if (strcmp(arg[ka],"-prt")==0) {
|
|---|
| 131 | sscanf(arg[ka+1],"%d,%d",&prtlev,&prtmod); ka+=2;
|
|---|
| 132 | }
|
|---|
| 133 | else break;
|
|---|
| 134 | }
|
|---|
| 135 |
|
|---|
| 136 | if ((ka+1)>=narg) {
|
|---|
| 137 | cout << " pknoise / Argument error " << endl;
|
|---|
| 138 | Usage();
|
|---|
| 139 | return 2;
|
|---|
| 140 | }
|
|---|
| 141 | if (isdigit(*arg[ka])) {
|
|---|
| 142 | fgresptbl=false;
|
|---|
| 143 | DIAMETRE=atof(arg[ka]);
|
|---|
| 144 | sprintf(cbuff,"pknoise_Dish(%g m)", DIAMETRE);
|
|---|
| 145 | }
|
|---|
| 146 | else {
|
|---|
| 147 | resptblname=arg[ka];
|
|---|
| 148 | fgresptbl=true;
|
|---|
| 149 | sprintf(cbuff,"pknoise_RespTblName=%s", arg[ka]);
|
|---|
| 150 | }
|
|---|
| 151 | tits=cbuff;
|
|---|
| 152 | string outfile = arg[ka+1];
|
|---|
| 153 | if (outfile==".") outfile = "pknoise.ppf";
|
|---|
| 154 | //-- end command line arguments
|
|---|
| 155 |
|
|---|
| 156 | int rc = 1;
|
|---|
| 157 | try { // exception handling try bloc at top level
|
|---|
| 158 | cout << " pknoise[0] : Executing, output file= " << outfile << endl;
|
|---|
| 159 | POutPersist po(outfile);
|
|---|
| 160 |
|
|---|
| 161 | cout << " pknoise[0.b] : Creating interpolating objects from redshift_da.txt and redshift_Hz.txt " << endl;
|
|---|
| 162 | SLinInterp1D redshift2da;
|
|---|
| 163 | string interpfilename="redshift_da.txt";
|
|---|
| 164 | redshift2da.ReadXYFromFile(interpfilename);
|
|---|
| 165 | SLinInterp1D redshift2hz;
|
|---|
| 166 | interpfilename="redshift_Hz.txt";
|
|---|
| 167 | redshift2hz.ReadXYFromFile(interpfilename);
|
|---|
| 168 | comov_dA_z=redshift2da(z_Redshift);
|
|---|
| 169 | H_z=redshift2hz(z_Redshift);
|
|---|
| 170 | cout << " Redshift= " << z_Redshift << " -> ComovdA= " << comov_dA_z
|
|---|
| 171 | << " Mpc H(z)= " << H_z << " km/s/Mpc " << endl;
|
|---|
| 172 |
|
|---|
| 173 | H21Conversions conv;
|
|---|
| 174 | conv.setRedshift(z_Redshift);
|
|---|
| 175 | double lambda = conv.getLambda();
|
|---|
| 176 |
|
|---|
| 177 | double Dol = DIAMETRE/lambda;
|
|---|
| 178 |
|
|---|
| 179 | Four2DResponse arep(2, DIAMETRE/lambda, DIAMETRE/lambda, lambda);
|
|---|
| 180 | Four2DResponse* arep_p=&arep;
|
|---|
| 181 | Four2DRespTable resptbl;
|
|---|
| 182 | if (fgresptbl) {
|
|---|
| 183 | cout << "pknoise[1]: initializing Four2DRespTable from file" << resptblname << endl;
|
|---|
| 184 | resptbl.readFromPPF(resptblname);
|
|---|
| 185 | cout << "pknoise[1.b] Four2DRespTable.LambdaRef=" << resptbl.getLambdaRef() << " setLambda("
|
|---|
| 186 | << lambda << ")" << endl;
|
|---|
| 187 | resptbl.setLambda(lambda);
|
|---|
| 188 | arep_p=&resptbl;
|
|---|
| 189 | if (fgrenorm) {
|
|---|
| 190 | cout << "pknoise[1.c] call to resptbl.renormalize(" << rmax << ")";
|
|---|
| 191 | double omax=resptbl.renormalize(rmax);
|
|---|
| 192 | cout << " ... Old Max=" << omax << endl;
|
|---|
| 193 | }
|
|---|
| 194 | }
|
|---|
| 195 | else cout << " pknoise[1]: Four2DResponse ( Diameter=" << DIAMETRE << " Lambda= " << lambda
|
|---|
| 196 | << " DoL=" << DIAMETRE/lambda << " ) " << endl;
|
|---|
| 197 | Histo2D h2drep = arep_p->GetResponse();
|
|---|
| 198 | Histo2D h2drepuv = arep_p->GetResponseUV();
|
|---|
| 199 | double repmax= h2drep.VMax();
|
|---|
| 200 | if (fgautoscut) {
|
|---|
| 201 | SCut = FacSCut/repmax;
|
|---|
| 202 | cout << " pknoise[1.b]: Four2DResponse.RepMax=" << repmax << " --> SCut=" << FacSCut << "/repmax="
|
|---|
| 203 | << SCut << endl;
|
|---|
| 204 | }
|
|---|
| 205 | else cout << " pknoise[1.b]: Four2DResponse.RepMax=" << repmax << " , SCut=" << SCut << endl;
|
|---|
| 206 |
|
|---|
| 207 | cout << " pknoise[2]: Instanciating object type Four3DPk " << endl;
|
|---|
| 208 | RandomGenerator rg;
|
|---|
| 209 |
|
|---|
| 210 | double dfreq=H_z*cellsz[2]/(1+z_Redshift)/lambda/1000.;
|
|---|
| 211 | double freq0=conv.getFrequency()-dfreq*box3dsz[2]/2;
|
|---|
| 212 |
|
|---|
| 213 | double dkxmpc=2.*DeuxPI;
|
|---|
| 214 | double dkympc=2.*DeuxPI;
|
|---|
| 215 | double dkzmpc=2.*DeuxPI;
|
|---|
| 216 | double angscale=1.;
|
|---|
| 217 | Vector angscales(box3dsz[2]);
|
|---|
| 218 | double pnoise0=comov_dA_z*comov_dA_z*(1.+z_Redshift)*(1.+z_Redshift)/H_z;
|
|---|
| 219 | Vector pnoisevec(box3dsz[2]);
|
|---|
| 220 | pnoisevec=1.;
|
|---|
| 221 | angscales=angscale;
|
|---|
| 222 | if (NMAX<0) { box3dsz[0]=256; box3dsz[1]=256; box3dsz[2]=128; }
|
|---|
| 223 | else {
|
|---|
| 224 | angscale=comov_dA_z;
|
|---|
| 225 | for(int kz=0; kz<box3dsz[2]; kz++) angscales(kz)=redshift2da(1420.4/(freq0+(double)kz*dfreq)-1.);
|
|---|
| 226 | dkxmpc = DeuxPI/(double)box3dsz[0]/cellsz[0];
|
|---|
| 227 | dkympc = DeuxPI/(double)box3dsz[1]/cellsz[1];
|
|---|
| 228 | dkzmpc = DeuxPI/(double)box3dsz[2]/cellsz[2];
|
|---|
| 229 | }
|
|---|
| 230 | cout << " pknoise[2.b]: ComovDist=" << comov_dA_z << " Mpc H(z)=" << H_z << " Mpc/km/s -> angscale=" << angscale << endl;
|
|---|
| 231 | cout << " pknoise[2.c]: Freq0=" << freq0 << " dFreq=" << dfreq << " freq(z=" << z_Redshift << ")="
|
|---|
| 232 | << conv.getFrequency() << " zmin=" << 1420.4/freq0-1. << " zmax=" << 1420.4/(freq0+box3dsz[2]*dfreq)-1. << endl;
|
|---|
| 233 | if(fgnoisefreq) {
|
|---|
| 234 | for(int kz=0; kz<box3dsz[2]; kz++) {
|
|---|
| 235 | double zreds=1420.4/(freq0+(double)kz*dfreq)-1.;
|
|---|
| 236 | double comda=redshift2da(zreds);
|
|---|
| 237 | pnoisevec(kz)=comda*comda*(1.+zreds)*(1.+zreds)/redshift2hz(zreds)/pnoise0;
|
|---|
| 238 | }
|
|---|
| 239 | cout << " pknoise[2.d]: PnoiseFactor:" << pnoisevec(0) << " ... " << pnoisevec(pnoisevec.Size()-1) << endl;
|
|---|
| 240 | }
|
|---|
| 241 | cout << " pknoise[2.e]: Four3DPk m3d(rg," << box3dsz[0]/2 << "," << box3dsz[1] << ","
|
|---|
| 242 | << box3dsz[2] << ")" << endl;
|
|---|
| 243 | Four3DPk m3d(rg,box3dsz[0]/2,box3dsz[1],box3dsz[2]);
|
|---|
| 244 | cout << " pknoise[2.f]: m3d.SetCellSize(" << dkxmpc << "," << dkympc << "," << dkzmpc
|
|---|
| 245 | << ") cell size (Mpc) : " << cellsz[0] << "x" << cellsz[1] << "x" << cellsz[2] << endl;
|
|---|
| 246 | m3d.SetCellSize(dkxmpc, dkympc, dkzmpc);
|
|---|
| 247 | m3d.SetPrtLevel(prtlev,prtmod);
|
|---|
| 248 |
|
|---|
| 249 | if (NMAX>0) {
|
|---|
| 250 | cout << " pknoise[3]: Computing Noise P(k) using PkNoiseCalculator ..." << endl;
|
|---|
| 251 | PkNoiseCalculator pkn(m3d, *(arep_p), SCut, NMAX, tits.c_str());
|
|---|
| 252 | pkn.SetFreqRange(freq0, dfreq);
|
|---|
| 253 | pkn.SetAngScaleConversion(angscales);
|
|---|
| 254 | pkn.SetPNoiseFactor(pnoisevec);
|
|---|
| 255 | pkn.SetPrtLevel(prtlev,prtmod);
|
|---|
| 256 | HProf hpn = pkn.Compute(NBINPK);
|
|---|
| 257 | }
|
|---|
| 258 | else {
|
|---|
| 259 | cout << " pknoise[3]: Computing Noise P(k) m3d.ComputeNoisePk(...) " << endl;
|
|---|
| 260 | HProf hpn = m3d.ComputeNoisePk(*(arep_p),angscale,SCut,NBINPK);
|
|---|
| 261 | }
|
|---|
| 262 |
|
|---|
| 263 | HProf hpnoise=m3d.GetPk();
|
|---|
| 264 | DataTable dtnoise;
|
|---|
| 265 | Histo fracmodok=m3d.FillPkDataTable(dtnoise);
|
|---|
| 266 | HProf h1dnoise=arep_p->GetProjNoiseLevel();
|
|---|
| 267 | HProf h1drep=arep_p->GetProjResponse();
|
|---|
| 268 | cout << " pknoise[3.b]: writing dtnoise,hpn,h2rep... with tags to " << outfile << endl;
|
|---|
| 269 | po << PPFNameTag("dtnoise") << dtnoise;
|
|---|
| 270 | po << PPFNameTag("hpnoise") << hpnoise;
|
|---|
| 271 | po << PPFNameTag("fracmodok") << fracmodok;
|
|---|
| 272 | po << PPFNameTag("h1dnoise") << h1dnoise;
|
|---|
| 273 | po << PPFNameTag("h1drep") << h1drep;
|
|---|
| 274 | po << PPFNameTag("h2drep") << h2drep;
|
|---|
| 275 | po << PPFNameTag("h2drepuv") << h2drepuv;
|
|---|
| 276 |
|
|---|
| 277 | rc = 0;
|
|---|
| 278 | } // End of try bloc
|
|---|
| 279 | catch (PThrowable & exc) { // catching SOPHYA exceptions
|
|---|
| 280 | cerr << " pknoise.cc: Catched Exception (PThrowable)" << (string)typeid(exc).name()
|
|---|
| 281 | << "\n...exc.Msg= " << exc.Msg() << endl;
|
|---|
| 282 | rc = 99;
|
|---|
| 283 | }
|
|---|
| 284 | catch (std::exception & e) { // catching standard C++ exceptions
|
|---|
| 285 | cerr << " pknoise.cc: Catched std::exception " << " - what()= " << e.what() << endl;
|
|---|
| 286 | rc = 98;
|
|---|
| 287 | }
|
|---|
| 288 | catch (...) { // catching other exceptions
|
|---|
| 289 | cerr << " pknoise.cc: some other exception (...) was caught ! " << endl;
|
|---|
| 290 | rc = 97;
|
|---|
| 291 | }
|
|---|
| 292 | PrtTim("End-pknoise");
|
|---|
| 293 | cout << " ==== End of pknoise.cc program Rc= " << rc << endl;
|
|---|
| 294 | return rc;
|
|---|
| 295 | }
|
|---|
| 296 |
|
|---|
| 297 |
|
|---|