-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathlstdrv_readsmapct.pro
418 lines (299 loc) · 11.9 KB
/
lstdrv_readsmapct.pro
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
pro lstdrv_readSMAPCT,pswfits,pmwfits,plwfits,sra,sdec,prfsize,fcat,f_src=f_src,name=name,astromcorr=astromcorr,fcatmod=fcatmod,noseg=noseg,mralims=mralims,mdeclims=mdeclims,segthresh=segthresh
;
;----------------------------------------------------------------------
; parameter checking & defaults
;----------------------------------------------------------------------
; source parameter
;image file
;output file names
if not keyword_set(name) then name="testName"
; PRF parameters
if not keyword_set(segthresh) then segthresh=0.5
; OK read in fits image and header and list of srcs from fits file
; read in image and naxis
;blank=mrdfits(pswfits,0,bhdr250)
im250=mrdfits(pswfits,1,hdr250)*1e3
im350=mrdfits(pmwfits,1,hdr350)*1e3
im500=mrdfits(plwfits,1,hdr500)*1e3
naxis1_250=sxpar(hdr250,"naxis1")
naxis2_250=sxpar(hdr250,"naxis2")
naxis1_350=sxpar(hdr350,"naxis1")
naxis2_350=sxpar(hdr350,"naxis2")
naxis1_500=sxpar(hdr500,"naxis1")
naxis2_500=sxpar(hdr500,"naxis2")
pixsize250=3600.*sxpar(hdr250,"cd2_2")
pixsize350=3600.*sxpar(hdr350,"cd2_2")
pixsize500=3600.*sxpar(hdr500,"cd2_2")
if(keyword_set(astromcorr)) then begin
oldcrval1=sxpar(hdr250,'crval1')
fxaddpar,hdr250,'crval1',(oldcrval1+astromcorr[0]/3600.)
oldcrval2=sxpar(hdr250,'crval2')
fxaddpar,hdr250,'crval2',(oldcrval2+astromcorr[1]/3600.)
oldcrval1=sxpar(hdr350,'crval1')
fxaddpar,hdr350,'crval1',(oldcrval1+astromcorr[0]/3600.)
oldcrval2=sxpar(hdr350,'crval2')
fxaddpar,hdr350,'crval2',(oldcrval2+astromcorr[1]/3600.)
oldcrval1=sxpar(hdr500,'crval1')
fxaddpar,hdr500,'crval1',(oldcrval1+astromcorr[0]/3600.)
oldcrval2=sxpar(hdr500,'crval2')
fxaddpar,hdr500,'crval2',(oldcrval2+astromcorr[1]/3600.)
endif
make_2d,findgen(naxis1_250),findgen(naxis2_250),x_pix250,y_pix250
pixsize=[pixsize250,pixsize350,pixsize500]
nim250=mrdfits(pswfits,2,nhdr250)*1e3
nim350=mrdfits(pmwfits,2,nhdr350)*1e3
nim500=mrdfits(plwfits,2,nhdr500)*1e3
adxy,hdr250,sra,sdec,sx250,sy250
; filter out sources not in the map, just check 250
sgood=where((sx250 gt 0) and (sx250 lt naxis1_250) and (sy250 gt 0) and (sy250 lt naxis2_250) and finite(im250[sx250,sy250]),ngood)
if(sgood(0) eq -1) then stop
sx250=sx250[sgood]
sy250=sy250[sgood]
sra=sra[sgood]
sdec=sdec[sgood]
f_src=f_src[sgood]
n_src=ngood
adxy,hdr350,sra,sdec,sx350,sy350
adxy,hdr500,sra,sdec,sx500,sy500
bad=where(finite(im250) eq 0 or finite(nim250) eq 0 or nim250 eq 0)
if(bad[0] ne -1) then begin
im250[bad]=0.
nim250[bad]=1.
endif
bad=where(finite(im350) eq 0 or finite(nim350) eq 0 or nim250 eq 0)
if(bad[0] ne -1) then begin
im350[bad]=0.
nim350[bad]=1.
endif
bad=where(finite(im500) eq 0 or finite(nim500) eq 0 or nim250 eq 0)
if(bad[0] ne -1) then begin
im500[bad]=0.
nim500[bad]=1.
endif
; work out pseudo p-stat to weight models by
; calculate prob of source with f>fs in one beam
; work out input list area crudely by just assuming contiguous area
; from min to max ra & dec of sources
p_src=fltarr(n_src)
sarea=((max(sra)-min(sra))*cos(mean(sdec)*!pi/180.))*(max(sdec)-min(sdec))
for k=0L,n_src-1 do begin
tmp=where(f_src ge f_src[k],ntmp)
ns=ntmp/sarea
p_src[k]=!pi*(prfsize[0]/3600.)^2/4./alog(2.)*ns
endfor
;---------------
; load PRF from file
;---------------
pfwhm=prfsize/pixsize
sim250=im250;+3.2;+0.4
sim350=im350;+4.3;+0.7
sim500=im500;+5.7;+1.
segsrc=lonarr(n_src)+2
if not keyword_set(noseg) then lstdrv_segsrc,sx250,sy250,pfwhm[0],pfwhm[1],pfwhm[2],hdr250,im250,nim250,hdr350,im350,nim350,hdr500,im500,nim500,segsrc,bsn=segthresh
;stop
msegsize=max(histogram(segsrc))
if (msegsize gt 2000) then stop
;segsrc=fltarr(n_src)+2
fstruct={xid:0L,inra:0.,indec:0.,obs250:0.,chi250:0.,e250:0.,obs350:0.,e350:0.,chi350:0.,obs500:0.,e500:0.,chi500:0.,gid:0L,gsize:0}
fcat=replicate(fstruct,n_src)
fcat.inra=sra
fcat.indec=sdec
fcat.xid=long(sgood)
test250=1e10
tdiff250=1.
tdiff350=1.
tdiff500=1.
test250=1e4
test350=1e4
test500=1e4
;stop
;while(mean(abs([tdiff250,tdiff350,tdiff500])) gt 0.05) do begin
lstdrv_solvefluxes2,segsrc,sx250,sy250,sx350,sy350,sx500,sy500,pfwhm,sim250,sim350,sim500,nim250,nim350,nim500,naxis1_250,naxis2_250,naxis1_350,naxis2_350,naxis1_500,naxis2_500,p_src,fcat=fcat,reconmap=reconmap,f_src=f_src
oldtest250=test250
oldtest350=test350
oldtest500=test500
back250=sim250-reconmap.rmap250
back350=sim350-reconmap.rmap350
back500=sim500-reconmap.rmap500
;linfit,reconmap.rmap250,sim250,a,b
;kern=psf_gaussian(fwhm=15.,npix=[50,50],/norm);
;back250=convol(back250,kern)
;back350=convol(back350,kern)
;back500=convol(back500,kern)
;back250=smooth(back250,[21,21])
;back350=smooth(back350,[21,21])
;back500=smooth(back500,[21,21])
paxis=[11,11]
prf=psf_gaussian(fwhm=pfwhm[0],npix=paxis)
bmask250=fltarr(naxis1_250,naxis2_250)
make_2d,indgen(11)-5,indgen(11)-5,tx,ty
for k=0,n_src-1 do bmask250[sx250[k]+tx, sy250[k]+ty]=bmask250[sx250[k]+tx, sy250[k]+ty]+prf
;region250=fltarr(naxis1_250,naxis2_250)
;for k=0,n_src-1 do region250[sx250[k]+tx, sy250[k]+ty]=1.
region250=where(bmask250 ne 0.)
test250=linfit(bmask250[region250],(im250-reconmap.rmap250)[region250])
tdiff250=test250-oldtest250
prf=psf_gaussian(fwhm=pfwhm[1],npix=paxis)
bmask350=fltarr(naxis1_350,naxis2_350)
make_2d,indgen(11)-5,indgen(11)-5,tx,ty
for k=0,n_src-1 do bmask350[sx350[k]+tx, sy350[k]+ty]=bmask350[sx350[k]+tx, sy350[k]+ty]+prf
;testx=min(sx350)+findgen(max(sx350)-min(sx350)-15)+7
;testy=min(sy350)+findgen(max(sy350)-min(sy350)-15)+7
;make_2d,testx,testy,rx,ry
region350=where(bmask350 ne 0.)
test350=linfit(bmask350[region350],(im350-reconmap.rmap350)[region350])
tdiff350=test350-oldtest350
prf=psf_gaussian(fwhm=pfwhm[2],npix=paxis)
bmask500=fltarr(naxis1_500,naxis2_500)
make_2d,indgen(11)-5,indgen(11)-5,tx,ty
for k=0,n_src-1 do bmask500[sx500[k]+tx, sy500[k]+ty]=bmask500[sx500[k]+tx, sy500[k]+ty]+prf
;testx=min(sx500)+findgen(max(sx500)-min(sx500)-15)+7
;testy=min(sy500)+findgen(max(sy500)-min(sy500)-15)+7
;make_2d,testx,testy,rx,ry
region500=where(bmask500 ne 0.)
test500=linfit(bmask500[region500],(im500-reconmap.rmap500)[region500])
tdiff500=test500-oldtest500
sim250=im250-test250[0]
sim350=im350-test350[0]
sim500=im500-test500[0]
print,test250,tdiff250,test350,tdiff350,test500,tdiff500
print,mean(im250-sim250),mean(im350-sim350),mean(im500-sim500)
;stop;endwhile
mwrfits,blank,name+'_back.fits',bhdr250,/create
mwrfits,back250,name+'_back.fits',hdr250
mwrfits,back350,name+'_back.fits',hdr350
mwrfits,back500,name+'_back.fits',hdr500
; find missing sources
res250=(im250-reconmap.rmap250-test250[0])/nim250
res350=(im350-reconmap.rmap350-test350[0])/nim350
res500=(im500-reconmap.rmap500-test500[0])/nim500
bkg250=test250[0]
bkg350=test350[0]
bkg500=test500[0]
find,res250,nx250,ny250,nf250,sharp,round,3.,pfwhm[0],[0.,3.],[-3.,3.],/silent
find,res350,nx350,ny350,nf350,sharp,round,3.,pfwhm[1],[0.,3.],[-3.,3.],/silent
find,res500,nx500,ny500,nf500,sharp,round,3.,pfwhm[2],[0.,3.],[-3.,3.],/silent
; keep 250 list and filter out sources in 350 and 500
xyad,hdr250,nx250,ny250,nra250,ndec250
xyad,hdr350,nx350,ny350,nra350,ndec350
xyad,hdr500,nx500,ny500,nra500,ndec500
;f1=where(sqrt(((nra250-nra350)*cos(ndec250*!pi/180.))^2+(ndec250-ndec350)^2) lt 20.,comp=k1)
;f2=where(sqrt(((nra250-nra500)*cos(ndec250*!pi/180.))^2+(ndec250-ndec500)^2) lt 20.,comp=k1)
nra=nra250
ndec=ndec250
for i=0,n_elements(nx350)-1 do begin
sep=sqrt(((nra-nra350[i])*cos(ndec*!pi/180.))^2+(ndec-ndec350[i])^2)
if(min(sep) gt 20./3600.) then begin
nra=[nra, nra350[i]]
ndec=[ndec,ndec350[i]]
endif
endfor
for i=0,n_elements(nx500)-1 do begin
sep=sqrt(((nra-nra500[i])*cos(ndec*!pi/180.))^2+(ndec-ndec500[i])^2)
if(min(sep) gt 20./3600.) then begin
nra=[nra, nra500[i]]
ndec=[ndec,ndec500[i]]
endif
endfor
; check if any are within 0.5 FHWM of known sources, just check 250?
nm=-1
for i=0,n_elements(nra)-1 do begin
sep=sqrt(((nra[i]-sra)*cos(ndec[i]*!pi/180.))^2+(sdec-ndec[i])^2)
if(min(sep) gt 18./3600.) then nm=[nm,i]
endfor
nm=nm[1:n_elements(nm)-1]
nra=nra[nm]
ndec=ndec[nm]
; check within mralims and mdeclims
if(keyword_set(mralims)) then begin
gnew=where(nra gt mralims[0] and nra lt mralims[1] and ndec gt mdeclims[0] and ndec lt mdeclims[1])
nra=nra[gnew]
ndec=ndec[gnew]
endif
adxy,hdr250,nra,ndec,nx250,ny250
adxy,hdr350,nra,ndec,nx350,ny350
adxy,hdr500,nra,ndec,nx500,ny500
nnew=n_elements(nra)
nsources={nra:nra,ndec:ndec,nx250:nx250,ny250:ny250,nx350:nx350,ny350:ny350,nx500:nx500,ny500:ny500,obs250:fltarr(nnew), e250:fltarr(nnew), obs350:fltarr(nnew), e350:fltarr(nnew), obs500:fltarr(nnew), e500:fltarr(nnew)}
sx250=[sx250,nx250]
sy250=[sy250,ny250]
sx350=[sx350,nx350]
sy350=[sy350,ny350]
sx500=[sx500,nx500]
sy500=[sy500,ny500]
p_src=[p_src,fltarr(nnew)+1.]
f_src=[f_src,fltarr(nnew)+min(f_src)]
segsrc=lonarr(n_src+nnew)+2
if not keyword_set(noseg) then lstdrv_segsrc,sx250,sy250,pfwhm[0],pfwhm[1],pfwhm[2],hdr250,im250,nim250,hdr350,im350,nim350,hdr500,im500,nim500,segsrc,bsn=segthresh
;segsrc=intarr(n_src+nnew)+2.
fcatmod=replicate(fstruct,n_src+nnew)
fcatmod.inra=[sra, nra]
fcatmod.indec=[sdec,ndec]
nids=fltarr(nnew)-99.
fcatmod.xid=[fcat.xid,nids]
lstdrv_solvefluxes2,segsrc,sx250,sy250,sx350,sy350,sx500,sy500,pfwhm,sim250,sim350,sim500,nim250,nim350,nim500,naxis1_250,naxis2_250,naxis1_350,naxis2_350,naxis1_500,naxis2_500,p_src,fcat=fcatmod,reconmap=reconmap,f_src=f_src
; work out true error
;250
paxis=[11,11]
bmask250=fltarr(naxis1_250,naxis2_250)
make_2d,indgen(11)-5,indgen(11)-5,tx,ty
for k=0,n_src-1 do bmask250[sx250[k]+tx, sy250[k]+ty]=1.
region=where(bmask250 eq 1.)
tmap250=im250-reconmap.rmap250
prf250=psf_gaussian(fwhm=pfwhm[0],npix=paxis)
smap250=convol(tmap250,prf250)/total(prf250^2)
nm250=convol(1./nim250^2,prf250^2)
err250=stddev(smap250[region])^2-1./nm250[sx250,sy250]
good=where(err250 gt 0.)
fcatmod[good].e250=sqrt(fcatmod[good].e250^2+err250[good])
;350
bmask350=fltarr(naxis1_350,naxis2_350)
make_2d,indgen(11)-5,indgen(11)-5,tx,ty
for k=0,n_src-1 do bmask350[sx350[k]+tx, sy350[k]+ty]=1.
region=where(bmask350 eq 1.)
tmap350=im350-reconmap.rmap350
prf350=psf_gaussian(fwhm=pfwhm[1],npix=paxis)
smap350=convol(tmap350,prf350)/total(prf350^2)
nm350=convol(1./nim350^2,prf350^2)
err350=stddev(smap350[region])^2-1./nm350[sx350,sy350]
good=where(err350 gt 0)
fcatmod[good].e350=sqrt(fcatmod[good].e350^2+err350[good])
;500
bmask500=fltarr(naxis1_500,naxis2_500)
make_2d,indgen(11)-5,indgen(11)-6,tx,ty
for k=0,n_src-1 do bmask500[sx500[k]+tx, sy500[k]+ty]=1.
region=where(bmask500 eq 1.)
tmap500=im500-reconmap.rmap500
prf500=psf_gaussian(fwhm=pfwhm[2],npix=paxis)
smap500=convol(tmap500,prf500)/total(prf500^2)
nm500=convol(1./nim500^2,prf500^2)
err500=stddev(smap500[region])^2-1./nm500[sx500,sy500]
good=where(err500 gt 0)
fcatmod[good].e500=sqrt(fcatmod[good].e500^2+err500[good])
; make some residual maps
rmap250=im250-reconmap.rmap250
rmap350=im350-reconmap.rmap350
rmap500=im500-reconmap.rmap500
;shuffle sources around
testflx250=fcatmod[round(randomu(-6167,n_src)*n_src)].obs250
testflx350=fcatmod[round(randomu(-6167,n_src)*n_src)].obs350
testflx500=fcatmod[round(randomu(-6167,n_src)*n_src)].obs500
cp_src=p_src[round(randomu(-6167,n_src)*n_src)]
mkspiremap,naxis1_250,naxis2_250,sx250+2,sy250-2,testflx250,prf250,tmap250
mkspiremap,naxis1_350,naxis2_350,sx350+2,sy350-2,testflx350,prf350,tmap350
mkspiremap,naxis1_500,naxis2_500,sx500+1,sy500-1,testflx500,prf500,tmap500
tmap250=tmap250+rmap250
tmap350=tmap350+rmap350
tmap500=tmap500+rmap500
fcatcomp=replicate(fstruct,n_src+nnew)
fcatcomp.inra=[sra, nra]
fcatcomp.indec=[sdec,ndec]
segsrccomp=lonarr(n_src)+2
;if not keyword_set(noseg) then
lstdrv_segsrc,sx250+2,sy250-2,pfwhm[0],pfwhm[1],pfwhm[2],hdr250,tmap250,nim250,hdr350,tmap350,nim350,hdr500,tmap500,nim500,segsrccomp,bsn=segthresh
tmap250=tmap250-bkg250
tmap350=tmap350-bkg350
tmap500=tmap500-bkg500
lstdrv_solvefluxes2,segsrccomp,round(sx250+2),round(sy250-2),round(sx350+2),round(sy350-2),round(sx500+1),round(sy500-1),pfwhm,tmap250,tmap350,tmap500,nim250,nim350,nim500,naxis1_250,naxis2_250,naxis1_350,naxis2_350,naxis1_500,naxis2_500,cp_src,fcat=fcatcomp,reconmap=reconmap,f_src=f_src
stop
end