class PlotTracks(OutputRoutine):
"""A routine that plots the tracks and save to file, it is to be used with GetTracks"""
def __init__(self, output_dir, season='2016', array='AR3', spreads=True):
OutputRoutine.__init__(self, output_dir)
self._spreads = spreads # whether to plot spreads
self._pr = None
self._season = season
self._array = array
def initialize(self):
self._pr = PixelReader(season=self._season, array=self._array)
def execute(self):
tod_id = self.get_context().get_id()
tracks = self.get_context().get_store().get("tracks")
print '[INFO] n_tracks = %d' % len(tracks)
plt.figure(figsize=(10, 10))
self._pr.plot() # plot the array
for track in tracks:
if self._spreads:
plt.scatter(track[:, 0],
track[:, 1],
marker='o',
s=3000 * track[:, 2]**2,
alpha=0.1)
plt.plot(track[:, 0], track[:, 1])
else:
plt.scatter(track[:, 0], track[:, 1], alpha=0.7)
# save image
print '[INFO] Saving image ...'
plt.savefig(self._output_dir + "%d.png" % tod_id)
print '[INFO] Image saved: %s%d.png' % (self._output_dir, tod_id)
class PlotGlitches(Routine):
"""A routine that plots glitches """
def __init__(self, tag,input_key, tod_key,timeseries_key):
Routine.__init__(self)
self._tag = tag
self._input_key = input_key
self._tod_key = tod_key
self._timeseries_key = timeseries_key
self._pr = None
def initialize(self):
tod_data = self.get_store().get(self._tod_key) # retrieve tod_data
events = self.get_store().get(self._input_key) # retrieve tod_data
def execute(self):
print '[INFO] Loading Glitch Data ...'
tod_data = self.get_store().get(self._tod_key) # retrieve tod_data
array_name = self.get_array()
events = self.get_store().get(self._input_key)
peaks = [event['peak'] for event in events]
for i in range(len(peaks)):
print ('[INFO] Filtered peak: ', i,peaks[i])
self._pr = PixelReader(season= '2017', array=self.get_context().get_array())
plot = raw_input("Do you want to plot an event? Enter y/n: ")
if plot == "y":
tod_data = self.get_store().get(self._tod_key) # retrieve tod_data
events = self.get_store().get(self._input_key) # retrieve tod_data
peaks = [event['peak'] for event in events]
timeseries = self.get_store().get(self._timeseries_key)
"""
PLOTTING FUNCTION
Plot all pixels affected given an array of pixel ids
and a starting time and ending time
"""
def plotter(pixels,start_time,end_time):
for pid in pixels:
x = timeseries(pid,start_time,end_time)[0]
y1 = timeseries(pid,start_time,end_time)[1]
y2 = timeseries(pid,start_time,end_time)[2]
y3 = timeseries(pid,start_time,end_time)[3]
y4 = timeseries(pid,start_time,end_time)[4]
plt.title('Pixel affected from ' +str(start_time)+ '-' + str(end_time)+ ', Pixel ' + str(pid))
plt.xlabel('TOD track:' + str(self._tag))
plt.plot(x,y1,'.-',label='90 GHz')
plt.plot(x,y2,'.-',label='90 GHz')
plt.plot(x,y3,'.-',label='150 GHz')
plt.plot(x,y4,'.-',label='150 GHz')
plt.legend()
plt.show()
"""
SPECIFIC EVENT
To plot specific event, this interface will ask you to supply the event list, make sure you
manually convert the last string to a float or integer
"""
e = raw_input('Please copy the event index to plot 4 freq channels:')
event = events[int(e)]
stime = event['start']
etime = event['end']
pixels = event['pixels_affected']
plotter(pixels, stime, etime)
self._pr.plot(pixels)
plt.show()
y_n = ' '
while y_n != 'n':
y_n = raw_input ("Would you like to plot another event? Enter y/n...")
if y_n == 'y':
e= raw_input('Please copy the event index to plot 4 freq channels:')
event = events[int(e)]
stime = event['start']
etime = event['end']
pixels = event['pixels_affected']
print '[INFO] Plotting Glitch...'
plotter(pixels, stime, etime)
self._pr.plot(pixels)
plt.show()
else:
print 'No plot will be displayed!'
class PlotGlitches(Routine):
"""A routine that plots glitches """
def __init__(self, tag, cosig_key, tod_key, timeseries_key):
Routine.__init__(self)
self._tag = tag
self._cosig_key = cosig_key
self._tod_key = tod_key
self._timeseries_key = timeseries_key
self._pr = None
"""
def initialize(self):
tod_data = self.get_store().get(self._tod_key) # retrieve tod_data
cuts = self.get_store().get(self._cosig_key) # retrieve tod_data
"""
def execute(self, store):
print '[INFO] Loading Glitch Data ...'
tod_data = store.get(self._tod_key) # retrieve tod_data
cuts = store.get(self._cosig_key) # retrieve tod_data
array_name = self.get_array()
peaks = cuts['peaks']
#print('[INFO] All glitches, unfiltered...')
#print('[INFO] peaks: ', peaks)
#self._pr = PixelReader(season= '2017', array=self.get_context().get_array()) #for covered
self._pr = PixelReader() #for uncovered
#self._pr = PixelReader(season='2017',array = str(array_name))
#self._pr = PixelReader(season='2017', array=self.get_context().get_array())
plot = raw_input("Do you want to plot an event? Enter y/n: ")
if plot == "y":
tod_data = store.get(self._tod_key) # retrieve tod_data
cuts = store.get(self._cosig_key) # retrieve tod_data
peaks = cuts['peaks']
def cs_cuts():
cuts = store.get(self._cosig_key)
return cuts['coincident_signals']
timeseries = store.get(self._timeseries_key)
"""
PLOTTING FUNCTION
Plot all pixels affected given an array of pixel ids
and a starting time and ending time
"""
def plotter(pixels, start_time, end_time):
for pid in pixels:
x = timeseries(pid, start_time, end_time)[0]
y1 = timeseries(pid, start_time, end_time)[1]
y2 = timeseries(pid, start_time, end_time)[2]
y3 = timeseries(pid, start_time, end_time)[3]
y4 = timeseries(pid, start_time, end_time)[4]
plt.title('Pixel affected from ' + str(start_time) + '-' +
str(end_time) + ', Pixel ' + str(pid))
plt.xlabel('TOD track:' + str(self._tag))
plt.plot(x, y1, '.-', label='90 GHz')
plt.plot(x, y2, '.-', label='90 GHz')
plt.plot(x, y3, '.-', label='150 GHz')
plt.plot(x, y4, '.-', label='150 GHz')
plt.legend()
plt.show()
"""
SPECIFIC EVENT
To plot specific event, this interface will ask you to supply the event list, make sure you
manually convert the last string to a float or integer
"""
cs = cuts['coincident_signals']
e = raw_input(
'Please copy the event list to plot 4 freq channels:')
event = json.loads(e)
stime = event[0]
etime = event[1]
pixels = pixels_affected_in_event(cs, event)
plotter(pixels, stime, etime)
self._pr.plot(pixels)
plt.show()
y_n = ' '
while y_n != 'n':
y_n = raw_input(
"Would you like to plot another event? Enter y/n...")
if y_n == 'y':
e = raw_input(
'Please copy the event list to plot 4 freq channels:')
event = json.loads(e)
stime = event[0]
etime = event[1]
pixels = pixels_affected_in_event(cs, event)
print '[INFO] Plotting Glitch...'
plotter(pixels, stime, etime)
self._pr.plot(pixels)
plt.show()
else:
print 'No plot will be displayed!'
class PlotGlitches(Routine):
"""A routine that plot glitches"""
def __init__(self,tag, cosig_key, tod_key):
Routine.__init__(self)
self._tag = tag
self._cosig_key = cosig_key
self._tod_key = tod_key
self._pr = None
def initialize(self):
self._pr = PixelReader()
def execute(self):
print '[INFO] Loading Glitch Data ...'
tod_data = self.get_store().get(self._tod_key) # retrieve tod_data
cuts = self.get_store().get(self._cosig_key) # retrieve tod_data
# print('[INFO] pixels affected: ',pixels)
peaks = cuts['peaks']
#print('[INFO] peaks: ', peaks)
def cs_cuts():
cuts = self.get_store().get(self._cosig_key)
return cuts['coincident_signals']
def timeseries(pixel_id, s_time, e_time, buffer=10):
start_time = s_time - buffer
end_time = e_time + buffer
a1, a2 = self._pr.get_f1(pixel_id)
b1, b2 = self._pr.get_f2(pixel_id)
d1, d2 = tod_data.data[a1], tod_data.data[a2]
d3, d4 = tod_data.data[b1], tod_data.data[b2]
# try to remove the mean from start_time to end_time
d1 -= np.mean(d1[start_time:end_time])
d2 -= np.mean(d2[start_time:end_time])
d3 -= np.mean(d3[start_time:end_time])
d4 -= np.mean(d4[start_time:end_time])
time = tod_data.ctime - tod_data.ctime[0]
time = time[start_time:end_time]
d_1 = d1[start_time:end_time]
d_2 = d2[start_time:end_time]
d_3 = d3[start_time:end_time]
d_4 = d4[start_time:end_time]
"""
UNCOMMENT TO PLOT FOUR CORRESPONDING PIXELS WITH HI-LO FREQ
plt.plot(time,d_1, '.-', label=str(a1) + ' 90 GHz')
plt.plot(time, d_2, '.-', label=str(a2) + ' 90 GHz')
plt.plot(time, d_3, '.-', label=str(b1) + ' 150 GHz')
plt.plot(time, d_4, '.-', label=str(b2) + ' 150 GHz')
plt.legend(title='Detector UID')
plt.show()
"""
return time, d_1, d_2, d_3, d_4
"""
PLOTTING FUNCTION
Plot all pixels affected given an array of pixel ids
and a starting time and ending time
"""
def plotter(pixels,start_time,end_time):
for pid in pixels:
x = timeseries(pid,start_time,end_time)[0]
y1 = timeseries(pid,start_time,end_time)[1]
y2 = timeseries(pid,start_time,end_time)[2]
y3 = timeseries(pid,start_time,end_time)[3]
y4 = timeseries(pid,start_time,end_time)[4]
plt.title('Pixel affected from ' +str(start_time)+ '-' + str(end_time)+ ', Pixel ' + str(pid))
plt.xlabel('TOD track:' + str(self._tag)) # CHANGE TOD TRACK NAME
plt.plot(x,y1,'.-',label='90 GHz')
plt.plot(x,y2,'.-',label='90 GHz')
plt.plot(x,y3,'.-',label='150 GHz')
plt.plot(x,y4,'.-',label='150 GHz')
plt.legend()
plt.show()
"""
ALL EVENTS
From peaks, find cs, then use cs to find all pixels affected
then plot all pixels affected in all events in peak one by one
"""
cs = cuts['coincident_signals']
"""
for event in peaks:
all_pixels = pixels_affected_in_event(cs,event)
plotter(all_pixels, event[0], event[1])
"""
"""
SPECIFIC EVENT
To plot specific event, copy event from peaks below
"""
e = raw_input('Please copy the event list to plot 4 freq channels:')
event = json.loads(e)
stime = event[0]
etime = event[1]
pixels = pixels_affected_in_event(cs, event)
print 'Pixels Affected:', pixels
plotter(pixels, stime, etime)
self._pr.plot(pixels)
plt.show()
y_n = ' '
while y_n != 'n':
y_n = raw_input ("Would you like to plot another event? Enter y/n...")
if y_n == 'y':
e= raw_input('Please copy the event list to plot 4 freq channels:')
event = json.loads(e)
stime = event[0]
etime = event[1]
pixels = pixels_affected_in_event(cs, event)
print '[INFO] Plotting Glitch...'
plotter(pixels, stime, etime)
self._pr.plot(pixels)
plt.show()
else:
print 'No plot will be displayed!'
"""
