1 | from asap.asapplotter import asapplotter |
---|
2 | from asap.logging import asaplog, asaplog_post_dec |
---|
3 | |
---|
4 | from asap.parameters import rcParams |
---|
5 | from asap.selector import selector |
---|
6 | from asap.scantable import scantable |
---|
7 | import matplotlib.axes |
---|
8 | from matplotlib.font_manager import FontProperties |
---|
9 | from matplotlib.text import Text |
---|
10 | |
---|
11 | class flagplotter(asapplotter): |
---|
12 | """ |
---|
13 | The flag plotter |
---|
14 | Only row based panneling is allowed. |
---|
15 | |
---|
16 | Example: |
---|
17 | scan = asa p.scantable(filename='your_filename',average=False) |
---|
18 | guiflagger = asap.flagplotter(visible=True) |
---|
19 | guiflagger.plot(scan) |
---|
20 | ### flag/Unflag data graphically. |
---|
21 | guiflagger.save_data(name='flagged_file.asap',format='ASAP') |
---|
22 | |
---|
23 | NOTICE: |
---|
24 | The flagged data is not saved until you explicitly run scantable.save |
---|
25 | """ |
---|
26 | def __init__(self, visible=None, **kwargs): |
---|
27 | self._scan = None |
---|
28 | asapplotter.__init__(self,visible=visible, **kwargs) |
---|
29 | self._plotter._set_window_title('Flag Plotter') |
---|
30 | self._panelling = 'r' |
---|
31 | self.set_stacking('scan') |
---|
32 | self._ismodified = False |
---|
33 | |
---|
34 | def _new_custombar(self): |
---|
35 | backend = matplotlib.get_backend() |
---|
36 | # Flag plotter relys on supported GUI backends |
---|
37 | if not self._visible: |
---|
38 | asaplog.push("GUI backend is not available") |
---|
39 | asaplog.post("ERROR") |
---|
40 | elif backend == "TkAgg": |
---|
41 | from asap.customgui_tkagg import CustomFlagToolbarTkAgg |
---|
42 | return CustomFlagToolbarTkAgg(self) |
---|
43 | elif backend == "Qt4Agg": |
---|
44 | from asap.customgui_qt4agg import CustomFlagToolbarQT4Agg |
---|
45 | return CustomFlagToolbarQT4Agg(self) |
---|
46 | else: |
---|
47 | asaplog.push("Unsupported backend for interactive flagging. Use either TkAgg or PyQt4Agg") |
---|
48 | asaplog.post("ERROR") |
---|
49 | |
---|
50 | @asaplog_post_dec |
---|
51 | def _invalid_func(self, name): |
---|
52 | msg = "Invalid function 'flagplotter."+name+"'" |
---|
53 | #raise AttributeError(msg) |
---|
54 | asaplog.push(msg) |
---|
55 | asaplog.post('ERROR') |
---|
56 | |
---|
57 | def set_panelling(self,which='r'): |
---|
58 | """ This function is not available for the class flagplotter """ |
---|
59 | if which.lower().startswith('r'): |
---|
60 | return |
---|
61 | msg = "Pannel setting is fixed to row mode in 'flagplotter'" |
---|
62 | asaplog.push(msg) |
---|
63 | asaplog.post('ERROR') |
---|
64 | self._panelling = 'r' |
---|
65 | |
---|
66 | def plotazel(self,*args,**kwargs): |
---|
67 | """ This function is not available for the class flagplotter """ |
---|
68 | self._invalid_func(name='plotazel') |
---|
69 | |
---|
70 | def plotpointing(self,*args,**kwargs): |
---|
71 | """ This function is not available for the class flagplotter """ |
---|
72 | self._invalid_func(name='plotpointing') |
---|
73 | |
---|
74 | def plottp(self,*args,**kwargs): |
---|
75 | """ This function is not available for the class flagplotter """ |
---|
76 | self._invalid_func(name='plottp') |
---|
77 | |
---|
78 | def save_data(self, name=None, format=None, overwrite=False): |
---|
79 | """ |
---|
80 | Store the plotted scantable on disk. |
---|
81 | This function simply redirects call to scantable.save() |
---|
82 | |
---|
83 | Parameters: |
---|
84 | |
---|
85 | name: the name of the outputfile. For format "ASCII" |
---|
86 | this is the root file name (data in 'name'.txt |
---|
87 | and header in 'name'_header.txt) |
---|
88 | |
---|
89 | format: an optional file format. Default is ASAP. |
---|
90 | Allowed are: |
---|
91 | * 'ASAP' (save as ASAP [aips++] Table), |
---|
92 | * 'SDFITS' (save as SDFITS file) |
---|
93 | * 'ASCII' (saves as ascii text file) |
---|
94 | * 'MS2' (saves as an casacore MeasurementSet V2) |
---|
95 | * 'FITS' (save as image FITS - not readable by class) |
---|
96 | * 'CLASS' (save as FITS readable by CLASS) |
---|
97 | |
---|
98 | overwrite: If the file should be overwritten if it exists. |
---|
99 | The default False is to return with warning |
---|
100 | without writing the output. USE WITH CARE. |
---|
101 | """ |
---|
102 | # simply calls scantable.save |
---|
103 | self._data.save(name,format,overwrite) |
---|
104 | |
---|
105 | def set_data(self, scan, refresh=True): |
---|
106 | if self._is_new_scan(scan): |
---|
107 | self._ismodified = False |
---|
108 | asapplotter.set_data(self, scan, refresh) |
---|
109 | set_data.__doc__ = asapplotter.set_data.__doc__ |
---|
110 | |
---|
111 | @asaplog_post_dec |
---|
112 | def plot(self, scan=None): |
---|
113 | if self._is_new_scan(scan): |
---|
114 | self._ismodified = False |
---|
115 | asapplotter.plot(self,scan) |
---|
116 | plot.__doc__ = asapplotter.plot.__doc__ |
---|
117 | |
---|
118 | @asaplog_post_dec |
---|
119 | def _plot(self, scan): |
---|
120 | asapplotter._plot(self,scan) |
---|
121 | # rescale x-range of subplots 5% margins |
---|
122 | ganged = (self._plotter.axes._sharex != None) |
---|
123 | if ganged: |
---|
124 | np = 1 |
---|
125 | else: |
---|
126 | np = len(self._plotter.subplots) |
---|
127 | for ip in xrange(np): |
---|
128 | ax = self._plotter.subplots[ip]['axes'] |
---|
129 | lim0 = ax.get_xlim() |
---|
130 | offset = (lim0[1]-lim0[0])*0.05 |
---|
131 | ax.set_xlim(lim0[0]-offset,lim0[1]+offset) |
---|
132 | del ax, lim0, offset |
---|
133 | _plot.__doc__ = asapplotter._plot.__doc__ |
---|
134 | |
---|
135 | def _is_new_scan(self,scan): |
---|
136 | if isinstance(scan, scantable): |
---|
137 | if self._data is not None: |
---|
138 | if scan != self._data: |
---|
139 | return True |
---|
140 | else: |
---|
141 | return True |
---|
142 | return False |
---|