| 1 | from __future__ import absolute_import
|
|---|
| 2 |
|
|---|
| 3 | import collections
|
|---|
| 4 | import pp
|
|---|
| 5 |
|
|---|
| 6 | import beamsgenerator
|
|---|
| 7 | import cleanimage
|
|---|
| 8 | import dirtyimage
|
|---|
| 9 | import fts
|
|---|
| 10 | import loadparameters
|
|---|
| 11 | import observe
|
|---|
| 12 | import renderer
|
|---|
| 13 | import skygenerator
|
|---|
| 14 | import uvmapgenerator
|
|---|
| 15 | import uvspectra
|
|---|
| 16 |
|
|---|
| 17 |
|
|---|
| 18 | class Fiins(object):
|
|---|
| 19 | """FISICA simulator.
|
|---|
| 20 | """
|
|---|
| 21 | def __init__(self, sky_spreadsheet='SkySparams.xlsx', sky_sheet='1point'):
|
|---|
| 22 | self.sky_spreadsheet = sky_spreadsheet
|
|---|
| 23 | self.sky_sheet = sky_sheet
|
|---|
| 24 | self.result = collections.OrderedDict()
|
|---|
| 25 |
|
|---|
| 26 | # start parallel python (pp), find the number of CPUS available
|
|---|
| 27 | ppservers = ()
|
|---|
| 28 | self.job_server = pp.Server(ppservers=ppservers)
|
|---|
| 29 | print 'fiins starting pp with %s workers' % \
|
|---|
| 30 | self.job_server.get_ncpus()
|
|---|
| 31 |
|
|---|
| 32 | def simulate(self):
|
|---|
| 33 | # read parameters
|
|---|
| 34 | loadparams = loadparameters.LoadParameters(
|
|---|
| 35 | sky_spreadsheet=self.sky_spreadsheet,
|
|---|
| 36 | sky_sheet=self.sky_sheet)
|
|---|
| 37 | obs_specification = loadparams.run()
|
|---|
| 38 | self.result['loadparameters'] = obs_specification
|
|---|
| 39 |
|
|---|
| 40 | # generate information on the FTS
|
|---|
| 41 | ftsd = fts.FTS(parameters=obs_specification)
|
|---|
| 42 | self.result['fts'] = ftsd.run()
|
|---|
| 43 | print ftsd
|
|---|
| 44 |
|
|---|
| 45 | # generate UV map and observation framework
|
|---|
| 46 | uvmapgen = uvmapgenerator.UVMapGenerator(parameters=obs_specification,
|
|---|
| 47 | previous_results=self.result)
|
|---|
| 48 | self.result['uvmapgenerator'] = uvmapgen.run()
|
|---|
| 49 | print uvmapgen
|
|---|
| 50 |
|
|---|
| 51 | # generate primary beams
|
|---|
| 52 | beamsgen = beamsgenerator.BeamsGenerator(
|
|---|
| 53 | previous_results=self.result)
|
|---|
| 54 | self.result['beamsgenerator'] = beamsgen.run()
|
|---|
| 55 |
|
|---|
| 56 | # construct sky
|
|---|
| 57 | skygen = skygenerator.SkyGenerator(parameters=obs_specification,
|
|---|
| 58 | previous_results=self.result)
|
|---|
| 59 | self.result['skygenerator'] = skygen.run()
|
|---|
| 60 |
|
|---|
| 61 | # calculate interferograms
|
|---|
| 62 | obs = observe.Observe(parameters=obs_specification,
|
|---|
| 63 | previous_results=self.result, job_server=self.job_server)
|
|---|
| 64 | self.result['observe'] = obs.run()
|
|---|
| 65 |
|
|---|
| 66 | # recover spectra from interferograms
|
|---|
| 67 | uvspec = uvspectra.UVspectra(previous_results=self.result,
|
|---|
| 68 | job_server=self.job_server)
|
|---|
| 69 | self.result['uvspectra'] = uvspec.run()
|
|---|
| 70 |
|
|---|
| 71 | # construct dirty image
|
|---|
| 72 | dirty = dirtyimage.DirtyImage(previous_results=self.result,
|
|---|
| 73 | job_server=self.job_server)
|
|---|
| 74 | self.result['dirtyimage'] = dirty.run()
|
|---|
| 75 |
|
|---|
| 76 | # construct clean image
|
|---|
| 77 | clean = cleanimage.CleanImage(previous_results=self.result,
|
|---|
| 78 | job_server=self.job_server)
|
|---|
| 79 | self.result['cleanimage'] = clean.run()
|
|---|
| 80 |
|
|---|
| 81 | # construct html description of result
|
|---|
| 82 | self.render()
|
|---|
| 83 |
|
|---|
| 84 | def import_result(self):
|
|---|
| 85 | print 'does nothing'
|
|---|
| 86 |
|
|---|
| 87 | def render(self):
|
|---|
| 88 | htmlrenderer = renderer.Renderer(result=self.result)
|
|---|
| 89 | htmlrenderer.run()
|
|---|
| 90 |
|
|---|
| 91 | def __repr__(self):
|
|---|
| 92 | return 'FISICA simulator'
|
|---|
| 93 |
|
|---|
