Skip to content

Commit bdcba10

Browse files
committed
Fix docs: Add visibility weights to example notebook
1 parent 12f67db commit bdcba10

1 file changed

Lines changed: 45 additions & 12 deletions

File tree

‎docs/source/notebooks/simulate_and_image.ipynb‎

Lines changed: 45 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -19,11 +19,14 @@
1919
"\n",
2020
"import astropy.units as u\n",
2121
"import astropy.constants as const\n",
22+
"import scipy.stats\n",
23+
"\n",
2224
"import fastimgproto.imager as imager\n",
2325
"import fastimgproto.visibility as visibility\n",
2426
"\n",
2527
"from astropy.coordinates import Angle, SkyCoord, AltAz, EarthLocation\n",
2628
"from astropy.time import Time\n",
29+
"\n",
2730
"from fastimgproto.gridder.conv_funcs import GaussianSinc\n",
2831
"from fastimgproto.skymodel.helpers import SkyRegion, SkySource\n",
2932
"from fastimgproto.sourcefind.image import SourceFindImage\n",
@@ -33,7 +36,9 @@
3336
{
3437
"cell_type": "code",
3538
"execution_count": null,
36-
"metadata": {},
39+
"metadata": {
40+
"collapsed": true
41+
},
3742
"outputs": [],
3843
"source": [
3944
"# from tqdm import tqdm_notebook as Tqdm\n",
@@ -43,7 +48,9 @@
4348
{
4449
"cell_type": "code",
4550
"execution_count": null,
46-
"metadata": {},
51+
"metadata": {
52+
"collapsed": true
53+
},
4754
"outputs": [],
4855
"source": [
4956
"import matplotlib.pyplot as plt\n",
@@ -70,7 +77,9 @@
7077
{
7178
"cell_type": "code",
7279
"execution_count": null,
73-
"metadata": {},
80+
"metadata": {
81+
"collapsed": true
82+
},
7483
"outputs": [],
7584
"source": [
7685
"pointing_centre = SkyCoord(0 * u.deg, -30 * u.deg)\n",
@@ -132,7 +141,8 @@
132141
" uvw = uvw_lambda)\n",
133142
"\n",
134143
"vis_noise_level = 0.1 * u.Jy\n",
135-
"data_vis = visibility.add_gaussian_noise(vis_noise_level, data_vis)"
144+
"data_vis = visibility.add_gaussian_noise(vis_noise_level, data_vis)\n",
145+
"vis_weights = np.minimum(1.0, scipy.stats.lognorm.rvs(s=0.25, size=len(data_vis)))"
136146
]
137147
},
138148
{
@@ -147,7 +157,9 @@
147157
{
148158
"cell_type": "code",
149159
"execution_count": null,
150-
"metadata": {},
160+
"metadata": {
161+
"collapsed": true
162+
},
151163
"outputs": [],
152164
"source": [
153165
"image_size=1024 * u.pixel\n",
@@ -163,7 +175,9 @@
163175
"kernel_support = 3\n",
164176
"kernel_func = GaussianSinc(trunc=kernel_support)\n",
165177
"with Tqdm() as progress_bar:\n",
166-
" image, beam = imager.image_visibilities(data_vis, uvw_lambda,\n",
178+
" image, beam = imager.image_visibilities(vis=data_vis, \n",
179+
" vis_weights=vis_weights,\n",
180+
" uvw_lambda=uvw_lambda,\n",
167181
" image_size=image_size,\n",
168182
" cell_size=cell_size,\n",
169183
" kernel_func=kernel_func,\n",
@@ -185,7 +199,9 @@
185199
{
186200
"cell_type": "code",
187201
"execution_count": null,
188-
"metadata": {},
202+
"metadata": {
203+
"collapsed": true
204+
},
189205
"outputs": [],
190206
"source": [
191207
"from fastimgproto.sourcefind.image import SourceFindImage"
@@ -194,7 +210,9 @@
194210
{
195211
"cell_type": "code",
196212
"execution_count": null,
197-
"metadata": {},
213+
"metadata": {
214+
"collapsed": true
215+
},
198216
"outputs": [],
199217
"source": [
200218
"detection_n_sigma=20\n",
@@ -217,7 +235,9 @@
217235
{
218236
"cell_type": "code",
219237
"execution_count": null,
220-
"metadata": {},
238+
"metadata": {
239+
"collapsed": true
240+
},
221241
"outputs": [],
222242
"source": [
223243
"src = sfimage.islands[0]"
@@ -241,11 +261,24 @@
241261
}
242262
],
243263
"metadata": {
264+
"kernelspec": {
265+
"display_name": "Python 3",
266+
"language": "python",
267+
"name": "python3"
268+
},
244269
"language_info": {
270+
"codemirror_mode": {
271+
"name": "ipython",
272+
"version": 3
273+
},
274+
"file_extension": ".py",
275+
"mimetype": "text/x-python",
245276
"name": "python",
246-
"pygments_lexer": "ipython3"
277+
"nbconvert_exporter": "python",
278+
"pygments_lexer": "ipython3",
279+
"version": "3.5.2"
247280
}
248281
},
249282
"nbformat": 4,
250-
"nbformat_minor": 0
251-
}
283+
"nbformat_minor": 1
284+
}

0 commit comments

Comments
 (0)