def draw_img_for_viewing(self):
if(not options.inspectOnly):
print "Press 'p' to save PNG."
global colmax
global colmin
P.close('all')
fig = P.figure(num=None, figsize=(13.5, 5), dpi=100, facecolor='w', edgecolor='k')
cid1 = fig.canvas.mpl_connect('key_press_event', self.on_keypress_for_viewing)
cid2 = fig.canvas.mpl_connect('button_press_event', self.on_click)
canvas = fig.add_subplot(121)
canvas.set_title(self.filename)
self.axes = P.imshow(self.inarr, origin='lower', vmax = colmax, vmin = colmin)
#self.axes = P.imshow(self.inarr, origin='lower', vmax = 2000, vmin = 0)
self.colbar = P.colorbar(self.axes, pad=0.01)
self.orglims = self.axes.get_clim()
canvas = fig.add_subplot(122)
canvas.set_title("angular average")
maxAngAvg = (self.inangavg).max()
for i,j in eDD.iceHInvAngQ.iteritems():
self.HIceQ[i] = eDD.get_pix_from_invAngsQ_and_detectorDist(runtag,j,self.detectorDistance, wavelengthInAngs=self.wavelength)
numQLabels = len(self.HIceQ.keys())+1
labelPosition = maxAngAvg/numQLabels
for i,j in self.HIceQ.iteritems():
P.axvline(j,0,colmax,color='r')
P.text(j,labelPosition,str(i), rotation="45")
labelPosition += maxAngAvg/numQLabels
P.plot(self.inangavg)
P.show()
def draw_img_for_retagging(self):
if(not options.inspectOnly):
print "Press 1-"+ str(numTypes)+ " to save png (overwrites old PNGs); Press 0 to undo (deletes png if wrongly saved)."
global colmax
global colmin
for i,j in eDD.iceHInvAngQ.iteritems():
self.HIceQ[i] = eDD.get_pix_from_invAngsQ_and_detectorDist(runtag,j,self.detectorDistance, wavelengthInAngs=self.wavelength)
fig = P.figure(num=None, figsize=(13.5, 6), dpi=100, facecolor='w', edgecolor='k')
cid1 = fig.canvas.mpl_connect('key_press_event', self.on_keypress_for_retagging)
cid2 = fig.canvas.mpl_connect('button_press_event', self.on_click)
canvas = fig.add_axes([0.05,0.05,0.5,0.9])
canvas.set_title(self.filename)
self.axes = P.imshow(self.inarr, origin='lower', vmax = colmax, vmin = colmin)
self.colbar = P.colorbar(self.axes, pad=0.01)
self.orglims = self.axes.get_clim()
(rTemp,cTemp) = self.inarr.shape
#Approximate, center
for i,j in self.HIceQ.iteritems():
circ = P.Circle((rTemp/2, cTemp/2), radius=j)
circ.set_fill(False)
circ.set_edgecolor('k')
canvas.add_patch(circ)
canvas = fig.add_axes([0.6,0.05,0.35,0.4])
canvas.set_title("angular average")
maxAngAvg = (self.inangavg).max()
numQLabels = len(self.HIceQ.keys())+1
labelPosition = maxAngAvg/numQLabels
for i,j in self.HIceQ.iteritems():
P.axvline(j,0,colmax,color='r')
P.text(j,labelPosition,str(i), rotation="45")
labelPosition += maxAngAvg/numQLabels
P.plot(self.inangavg)
canvas = fig.add_axes([0.6,0.55,0.35,0.4], xlabel="pixel value", ylabel="count")
canvas.set_title("histogram (omiting values<1) ")
nonzeroarr = N.compress(self.origarr.flat>1, self.origarr.flat)
canvas.hist(nonzeroarr, bins=N.arange(0,2000,10),log=True)
P.show()
def draw_img_for_tagging(self):
if(not options.inspectOnly and not options.auto):
print "Press 1-"+ str(numTypes)+ " to save png (overwrites old PNGs); Press 0 to undo (deletes png if wrongly saved)."
global colmax
global colmin
global storeFlag
fig = P.figure(num=None, figsize=(13.5, 5), dpi=100, facecolor='w', edgecolor='k')
cid1 = fig.canvas.mpl_connect('key_press_event', self.on_keypress_for_tagging)
cid2 = fig.canvas.mpl_connect('button_press_event', self.on_click)
canvas = fig.add_subplot(121)
canvas.set_title(self.filename)
#self.axes = P.imshow(self.inarr, origin='lower', vmax = colmax, vmin = colmin)
self.axes = P.imshow(self.inarr, origin='lower', vmax = options.maxIntens, vmin = 0)
self.colbar = P.colorbar(self.axes, pad=0.01)
self.orglims = self.axes.get_clim()
canvas = fig.add_subplot(122)
canvas.set_title("angular average")
maxAngAvg = (self.inangavg).max()
for i,j in eDD.iceHInvAngQ.iteritems():
self.HIceQ[i] = eDD.get_pix_from_invAngsQ_and_detectorDist(runtag,j,self.detectorDistance, wavelengthInAngs=self.wavelength)
numQLabels = len(self.HIceQ.keys())+1
labelPosition = maxAngAvg/numQLabels
for i,j in self.HIceQ.iteritems():
P.axvline(j,0,colmax,color='r')
P.text(j,labelPosition,str(i), rotation="45")
labelPosition += maxAngAvg/numQLabels
P.plot(self.inangavg)
#Check for max intensity of angular average below threshold
if (not options.inspectOnly and (self.inangavg[options.lowerBoundType2:options.upperBoundType2]).max() < options.thresholdType2):
print "%s automatically characterized as type2." % (self.filename)
storeFlag = 2
if(not os.path.exists(write_anomaly_dir_types[storeFlag])):
os.mkdir(write_anomaly_dir_types[storeFlag])
self.axes.set_clim(0, 2000)
pngtag = write_anomaly_dir_types[storeFlag] + "%s.png" % (self.filename)
P.savefig(pngtag)
print "%s saved." % (pngtag)
self.tag = storeFlag
P.close()
elif (options.auto):
P.close()
else:
P.show()
def draw_img_for_thresholding(self):
global colmax
global colmin
global storeFlag
for i,j in eDD.iceHInvAngQ.iteritems():
self.HIceQ[i] = eDD.get_pix_from_invAngsQ_and_detectorDist(runtag,j,self.detectorDistance, wavelengthInAngs=self.wavelength)
fig = P.figure(num=None, figsize=(13.5, 6), dpi=100, facecolor='w', edgecolor='k')
cid1 = fig.canvas.mpl_connect('key_press_event', self.on_keypress_for_viewing)
cid2 = fig.canvas.mpl_connect('button_press_event', self.on_click)
canvas = fig.add_subplot(121)
canvas.set_title(self.filename)
self.axes = P.imshow(self.inarr, origin='lower', vmax = colmax, vmin = colmin)
self.colbar = P.colorbar(self.axes, pad=0.01)
self.orglims = self.axes.get_clim()
(rTemp,cTemp) = self.inarr.shape
#Approximate, center
for i,j in self.HIceQ.iteritems():
circ = P.Circle((rTemp/2, cTemp/2), radius=j)
circ.set_fill(False)
circ.set_edgecolor('k')
canvas.add_patch(circ)
canvas = fig.add_subplot(122)
canvas.set_title("Angular Average")
maxAngAvg = (self.inangavg).max()
numQLabels = len(self.HIceQ.keys())+1
labelPosition = maxAngAvg/numQLabels
for i,j in self.HIceQ.iteritems():
P.axvline(j,0,colmax,color='r')
P.text(j,labelPosition,str(i), rotation="45")
labelPosition += maxAngAvg/numQLabels
P.plot(self.inangavg)
pngtag = foundTypes[storeFlag] + '/' + "below%dADUs/%s.png" % (options.threshold, self.filename)
P.savefig(pngtag)
if (options.verbose):
print "%s saved." % (pngtag)
P.close()
