| 1 | import os
|
|---|
| 2 | from asap.parameters import rcParams
|
|---|
| 3 | from asap import scantable
|
|---|
| 4 |
|
|---|
| 5 | def guiflag(sdfile=None,outfile=None,outform='ASAP', overwrite=False):
|
|---|
| 6 | """
|
|---|
| 7 | GUI flagging method
|
|---|
| 8 | Parameters
|
|---|
| 9 | sdfile: input filename
|
|---|
| 10 | outfile: output filename
|
|---|
| 11 | outform: output file format, should be either 'ASAP', 'MS2',
|
|---|
| 12 | 'ASCII', or 'STDIFTS'. Default: 'ASAP'
|
|---|
| 13 | overwrite: whether or not overwrite output file if exists.
|
|---|
| 14 | Default False
|
|---|
| 15 | """
|
|---|
| 16 | if not rcParams['plotter.gui']:
|
|---|
| 17 | print "ERROR - GUI is not available"
|
|---|
| 18 | return
|
|---|
| 19 | from asap.flagplotter import flagplotter
|
|---|
| 20 | if not (sdfile and outfile):
|
|---|
| 21 | print "ERROR - sdfile and outfile is necessary"
|
|---|
| 22 | return
|
|---|
| 23 | if not os.path.exists(sdfile):
|
|---|
| 24 | print "ERROR - invalid input filename '"+sdfile+"'"
|
|---|
| 25 | return
|
|---|
| 26 | if not overwrite and os.path.exists(outfile):
|
|---|
| 27 | print "ERROR - output file '"+outfile+"' already exists"
|
|---|
| 28 | return
|
|---|
| 29 | scan = scantable(filename=sdfile,average=False)
|
|---|
| 30 | guiflagger = flagplotter(visible=True)
|
|---|
| 31 | guiflagger._plotter.legend(loc=0)
|
|---|
| 32 | guiflagger.plot(scan)
|
|---|
| 33 | finish=raw_input("Press enter to finish interactive flagging:")
|
|---|
| 34 | guiflagger._plotter.unmap()
|
|---|
| 35 | guiflagger._plotter = None
|
|---|
| 36 | del guiflagger
|
|---|
| 37 | scan.save(outfile,outform.upper(),overwrite)
|
|---|
| 38 | if outform!='ASCII':
|
|---|
| 39 | print "Wrote output "+outform+" file "+outfile
|
|---|
| 40 |
|
|---|
