def _update_curve(self, idx, f_axes, c, x, y, data, title):
if f_axes.num_child() == 0:
obj = FigPlot([0], [0])
f_axes.add_child('xslice_plot1', obj)
obj = FigPlot([0], [0])
f_axes.add_child('xslice_plot2', obj)
f_axes.realize()
j = 0
for name, child in f_axes.get_children():
if (isinstance(child, FigPlot)):
if j == idx:
child.setvar('x', data[0])
child.setvar('y', data[1])
child.set_suppress(False)
child.reset_artist()
a = child._artists[0]
a.set_color(c)
break
else:
j = j + 1
f_axes.adjust_axes_range()
v = f_axes.get_title(f_axes._artists[0])
v[0] = title
v = f_axes.set_title(v, f_axes._artists[0])
f_axes.set_bmp_update(False)
def generate_artist(self):
FigPlot.generate_artist(self)
container = self.get_container()
a = self._artists[0]
x = a.get_xdata()
y = a.get_ydata()
vm = self.getp('marked_point')
if vm is not None:
if self._mark_a is not None:
self._mark_a.remove()
self._mark_a = None
a3 = container.plot(x[vm].copy(),
y[vm].copy(),
marker='o',
color='k',
linestyle='None',
markerfacecolor='k',
markeredgecolor='k',
markersize=6,
scalex=False,
scaley=False)
self._mark_a = a3[0]
idx = np.where([not xx for xx in self.getp('enabled_point')])[0]
if self._enable_a is not None:
self._enable_a.remove()
self._enable_a = None
if idx.size == 0:
return
marker = a.get_marker()
fc = 'white'
ec = a.get_markeredgecolor()
ms = a.get_markersize()
a2 = container.plot(x[idx].copy(),
y[idx].copy(),
marker=marker,
color='k',
linestyle='None',
markerfacecolor=fc,
markeredgecolor=ec,
markersize=ms,
scalex=False,
scaley=False)
self._enable_a = a2[0]
def canvas_menu(self):
m = [
('Add Point', self.onAddPoint, None),
]
if self._figc_hit != -1:
if self.getp('enabled_point')[self._figc_hit]:
m = m + [
('Delete point', self.onDelPoint, None),
('Edit point', self.onEditPoint, None),
('Disable point', self.onDisablePoint, None),
]
else:
m = m + [
('Delete point', self.onDelPoint, None),
('Edit point', self.onEditPoint, None),
('Enable point', self.onEnablePoint, None),
]
vm = self.get_mark_data()
if vm[self._figc_hit]:
m = m + [
('Unmark point', self.onUnmarkPoint, None),
('Reset mark', self.onResetMark, None),
('Export mark', self.onExportMark, None),
]
else:
m = m + [
('Mark point', self.onMarkPoint, None),
('Reset mark', self.onResetMark, None),
('Export mark', self.onExportMark, None),
]
return m + FigPlot.canvas_menu(self)
def __new__(cls, *args,**kargs):
draggable, kargs = ProcessKeywords(kargs, 'draggable', True)
enabled_point, kargs = ProcessKeywords(kargs, 'enabled_point', 'all')
obj = FigPlot.__new__(cls, *args,**kargs)
if obj is not None:
obj.setvar('draggable', draggable)
if not obj.hasvar('enabled_point'): obj.setvar('enabled_point', enabled_point)
return obj
def del_artist(self, artist=None, delall=False):
if self._enable_a is not None:
self._enable_a.remove()
self._enable_a = None
if self._mark_a is not None:
self._mark_a.remove()
self._mark_a = None
return FigPlot.del_artist(self, artist=artist, delall=delall)
def __new__(cls, *args, **kargs):
draggable, kargs = ProcessKeywords(kargs, 'draggable', True)
enabled_point, kargs = ProcessKeywords(kargs, 'enabled_point', 'all')
obj = FigPlot.__new__(cls, *args, **kargs)
if obj is not None:
obj.setvar('draggable', draggable)
if not obj.hasvar('enabled_point'):
obj.setvar('enabled_point', enabled_point)
return obj
def del_artist(self, artist=None, delall=False):
if self._enable_a is not None:
self._enable_a.remove()
self._enable_a = None
if self._mark_a is not None:
self._mark_a.remove()
self._mark_a = None
return FigPlot.del_artist(self, artist=artist, delall=delall)
def generate_artist(self):
FigPlot.generate_artist(self)
container=self.get_container()
a = self._artists[0]
x=a.get_xdata()
y=a.get_ydata()
vm = self.getp('marked_point')
if vm is not None:
if self._mark_a is not None:
self._mark_a.remove()
self._mark_a = None
a3 = container.plot(x[vm].copy(), y[vm].copy(), marker='o',
color='k', linestyle='None',
markerfacecolor='k',
markeredgecolor='k',
markersize = 6,
scalex=False, scaley=False)
self._mark_a = a3[0]
idx = np.where([not xx for xx in self.getp('enabled_point')])[0]
if self._enable_a is not None:
self._enable_a.remove()
self._enable_a = None
if idx.size == 0:
return
marker = a.get_marker()
fc = 'white'
ec = a.get_markeredgecolor()
ms = a.get_markersize()
a2 = container.plot(x[idx].copy(), y[idx].copy(), marker=marker,
color='k', linestyle='None',
markerfacecolor=fc,
markeredgecolor=ec,
markersize = ms,
scalex=False, scaley=False)
self._enable_a = a2[0]
def picker_a(self, artist, evt):
hit = False
self._figc_hit = -1
self._figc_hit_pos = (evt.xdata, evt.ydata)
axes = artist.axes
if axes is None: return False, {}
x = artist.get_xdata()
y = artist.get_ydata()
for k, pd in enumerate(axes.transData.transform(np.transpose(np.vstack((x, y))))):
if (abs(evt.x - pd[0]) < 5 and
abs(evt.y - pd[1]) < 5):
hit = True
self._figc_hit = k
break
if hit:
return True, {'child_artist':artist}
return FigPlot.picker_a(self, artist, evt)
def picker_a(self, artist, evt):
hit = False
self._figc_hit = -1
self._figc_hit_pos = (evt.xdata, evt.ydata)
axes = artist.axes
if axes is None: return False, {}
x = artist.get_xdata()
y = artist.get_ydata()
for k, pd in enumerate(
axes.transData.transform(np.transpose(np.vstack((x, y))))):
if (abs(evt.x - pd[0]) < 5 and abs(evt.y - pd[1]) < 5):
hit = True
self._figc_hit = k
break
if hit:
return True, {'child_artist': artist}
return FigPlot.picker_a(self, artist, evt)
def canvas_menu(self):
m = [('Add Point', self.onAddPoint, None),]
if self._figc_hit != -1:
if self.getp('enabled_point')[self._figc_hit]:
m = m + [('Delete point', self.onDelPoint, None),
('Edit point', self.onEditPoint, None),
('Disable point', self.onDisablePoint, None),]
else:
m = m + [('Delete point', self.onDelPoint, None),
('Edit point', self.onEditPoint, None),
('Enable point', self.onEnablePoint, None),]
vm = self.get_mark_data()
if vm[self._figc_hit]:
m = m+ [ ('Unmark point', self.onUnmarkPoint, None),
('Reset mark', self.onResetMark, None),
('Export mark', self.onExportMark, None),]
else:
m = m + [('Mark point', self.onMarkPoint, None),
('Reset mark', self.onResetMark, None),
('Export mark', self.onExportMark, None),]
return m + FigPlot.canvas_menu(self)
def onCopyPath(self, event):
from ifigure.mto.fig_plot import FigPlot
from ifigure.mto.fig_fill import FigFill
if self._hit_path is not None:
canvas = event.GetEventObject()
v = self._hit_path.vertices
if self.getp('FillMode'):
obj = FigFill(v[:, 0], v[:, 1])
else:
obj = FigPlot(v[:, 0], v[:, 1], 'k')
ax = self.get_figaxes()
name = ax.get_next_name('Contour_path')
ax.add_child(name, obj)
obj.realize()
ax.set_bmp_update(False)
canvas.draw()
artists = [weakref.ref(obj._artists[0])]
h = [UndoRedoAddRemoveArtists(artists=artists, mode=0)]
window = canvas.GetTopLevelParent()
GlobalHistory().get_history(window).make_entry(
h, menu_name='add path plot')
def postprocess_data(
self,
ana,
viewer,
color_order=['red', 'blue', 'yellow'],
):
def set_auto_color_to_color(kargs, name, col):
if (not name in kargs) or (kargs[name] == 'auto'):
kargs[name] = col
return kargs
from ifigure.mto.fig_plot import FigPlot, TimeTrace, StepPlot
from ifigure.mto.fig_contour import FigContour
from ifigure.mto.fig_image import FigImage
from ifigure.mto.fig_surface import FigSurface
from ifigure.mto.fig_axline import FigAxline
from ifigure.mto.fig_axspan import FigAxspan
from ifigure.mto.fig_text import FigText
self._color_order = color_order
dprint2('process_data', self)
if ana.skipped:
if ana.do_title:
self.process_title(ana)
return True
if (self.isempty() and not self.get_figaxes().isempty()
and not self._suppress):
self.generate_artist()
# app = self.get_app()
# ax = self.get_container()
# fig_ax = ax.figobj
ishot = ana.ishot
self._analysis_flag[ishot] = False
col = color_order[ishot % len(color_order)]
for x in range(ishot - self.num_child() + 1):
plot_options = self.getvar('plot_options')
if self._plot_type == 'plot':
kargs = plot_options['plot'][1].copy()
kargs = set_auto_color_to_color(kargs, 'color', col)
kargs = set_auto_color_to_color(kargs, 'markerfacecolor', col)
kargs = set_auto_color_to_color(kargs, 'markeredgecolor', col)
obj = FigPlot([0], [0], *(plot_options['plot'][0]), **kargs)
elif self._plot_type == 'timetrace':
kargs = plot_options['timetrace'][1].copy()
kargs = set_auto_color_to_color(kargs, 'color', col)
kargs = set_auto_color_to_color(kargs, 'markerfacecolor', col)
kargs = set_auto_color_to_color(kargs, 'markeredgecolor', col)
obj = TimeTrace([0], [0], *(plot_options['timetrace'][0]),
**kargs)
elif self._plot_type == 'stepplot':
kargs = plot_options['stepplot'][1].copy()
kargs = set_auto_color_to_color(kargs, 'color', col)
kargs = set_auto_color_to_color(kargs, 'markerfacecolor', col)
kargs = set_auto_color_to_color(kargs, 'markeredgecolor', col)
obj = StepPlot([0], [0], *(plot_options['stepplot'][0]),
**kargs)
elif self._plot_type == 'contour':
obj = FigContour([0, 1], [0, 1],
np.arange(4).reshape(2, 2),
*(plot_options['contour'][0]))
elif self._plot_type == 'image':
obj = FigImage(np.zeros([2, 2]))
elif self._plot_type == 'surface':
obj = FigSurface(np.zeros([2, 2]))
elif self._plot_type == 'axspan':
obj = FigAxspan([0, 1])
elif self._plot_type == 'axline':
obj = FigAxline(np.zeros([1]))
elif self._plot_type == 'text':
obj = FigText(0, 0, '')
name = self._plot_type + str(self.num_child() + 1)
self.add_child(name, obj)
obj.set_container_idx(self._container_idx)
obj.set_suppress(True)
if self.get_script_local():
txt = self.make_script(ana.shot, short=True)
debug_mode = viewer.debug_mode
if txt is not None:
try:
vars = viewer.get_globals_for_figmds(ana.shot)
# print vars
for key in vars:
ana.result[key] = vars[key]
for key in viewer.startup_values:
ana.result[key] = viewer.startup_values[key]
ana.result['shot'] = ana.shot
if self.getvar('posvars') is not None:
a = {}
b = {}
exec self.getvar('posvars') in a, b
for key in b:
ana.result[key] = b[key]
filepath = os.path.join(self.owndir(), script_file_name)
from ifigure.widgets.debugger_core import get_breakpoint
code = compile(txt, filepath, 'exec')
if (len(get_breakpoint(filepath)) != 0 and debug_mode):
app = wx.GetApp().TopWindow
se = app.script_editor
import threading
if not se.CheckDebuggerStatus():
se.QueueSEDRequest(code, viewer.g, ana.result,
filepath)
# print threading.current_thread().name
# wx.Yield()
# time.sleep(3)
else:
se.RunSED(code, viewer.g, ana.result, filepath)
else:
exec code in viewer.g, ana.result
except:
dprint1('error occured when processing data by script')
print(
'error occured when processing data by following script'
)
print('#####')
print(txt)
print('#####')
print(traceback.format_exc())
self.change_suppress(True, self.get_child(ishot))
return False
if ana.do_title:
self.process_title(ana)
if self._plot_type == 'plot':
self._update_plot(ana, ishot) #, color_order)
elif self._plot_type == 'timetrace':
self._update_plot(ana, ishot) #, color_order)
elif self._plot_type == 'stepplot':
self._update_plot(ana, ishot) #, color_order)
elif self._plot_type == 'contour':
self._update_contour(ana, ishot)
elif self._plot_type == 'image':
self._update_image(ana, ishot)
elif self._plot_type == 'surface':
self._update_surface(ana, ishot)
elif self._plot_type == 'axspan':
self._update_axspan(ana, ishot)
elif self._plot_type == 'axline':
self._update_axline(ana, ishot)
elif self._plot_type == 'text':
self._update_text(ana, ishot)
obj = self.get_child(idx=ishot)
if obj.is_suppress():
return False
obj._data_extent = None
try:
if (self.get_figaxes() is not None):
self.get_figaxes().adjust_axes_range()
except:
dprint1(
"failed in adjusting axes at postprocess_data, continuing...")
pass
return True
def postprocess_data(self, ana, viewer,
color_order=['red','blue','yellow'],
):
def set_auto_color_to_color(kargs, name, col):
if (not name in kargs) or (kargs[name] == 'auto'):
kargs[name] = col
return kargs
from ifigure.mto.fig_plot import FigPlot, TimeTrace, StepPlot
from ifigure.mto.fig_contour import FigContour
from ifigure.mto.fig_image import FigImage
from ifigure.mto.fig_surface import FigSurface
from ifigure.mto.fig_axline import FigAxline
from ifigure.mto.fig_axspan import FigAxspan
from ifigure.mto.fig_text import FigText
self._color_order = color_order
dprint2('process_data', self)
if ana.skipped:
if ana.do_title:
self.process_title(ana)
return True
if (self.isempty() and
not self.get_figaxes().isempty() and
not self._suppress):
self.generate_artist()
# app = self.get_app()
# ax = self.get_container()
# fig_ax = ax.figobj
ishot = ana.ishot
self._analysis_flag[ishot] = False
col = color_order[ishot % len(color_order)]
for x in range(ishot - self.num_child()+1):
plot_options = self.getvar('plot_options')
if self._plot_type == 'plot':
kargs = plot_options['plot'][1].copy()
kargs = set_auto_color_to_color(kargs, 'color', col)
kargs = set_auto_color_to_color(kargs, 'markerfacecolor', col)
kargs = set_auto_color_to_color(kargs, 'markeredgecolor', col)
obj=FigPlot([0],[0],
*(plot_options['plot'][0]),
**kargs)
elif self._plot_type == 'timetrace':
kargs = plot_options['timetrace'][1].copy()
kargs = set_auto_color_to_color(kargs, 'color', col)
kargs = set_auto_color_to_color(kargs, 'markerfacecolor', col)
kargs = set_auto_color_to_color(kargs, 'markeredgecolor', col)
obj=TimeTrace([0],[0],
*(plot_options['timetrace'][0]),
**kargs)
elif self._plot_type == 'stepplot':
kargs = plot_options['stepplot'][1].copy()
kargs = set_auto_color_to_color(kargs, 'color', col)
kargs = set_auto_color_to_color(kargs, 'markerfacecolor', col)
kargs = set_auto_color_to_color(kargs, 'markeredgecolor', col)
obj=StepPlot([0],[0],
*(plot_options['stepplot'][0]),
**kargs)
elif self._plot_type == 'contour':
obj=FigContour([0,1], [0,1], np.arange(4).reshape(2,2),
*(plot_options['contour'][0]))
elif self._plot_type == 'image':
obj=FigImage(np.zeros([2,2]))
elif self._plot_type == 'surface':
obj=FigSurface(np.zeros([2,2]))
elif self._plot_type == 'axspan':
obj=FigAxspan([0,1])
elif self._plot_type == 'axline':
obj=FigAxline(np.zeros([1]))
elif self._plot_type == 'text':
obj=FigText(0, 0, '')
name = self._plot_type + str(self.num_child()+1)
self.add_child(name, obj)
obj.set_container_idx(self._container_idx)
obj.set_suppress(True)
if self.get_script_local():
txt = self.make_script(ana.shot, short = True)
debug_mode = viewer.debug_mode
if txt is not None:
try:
vars = viewer.get_globals_for_figmds(ana.shot)
# print vars
for key in vars:
ana.result[key] = vars[key]
for key in viewer.startup_values:
ana.result[key] = viewer.startup_values[key]
ana.result['shot'] = ana.shot
if self.getvar('posvars') is not None:
a = {}; b = {}
exec self.getvar('posvars') in a, b
for key in b:
ana.result[key] = b[key]
filepath = os.path.join(self.owndir(), script_file_name)
from ifigure.widgets.debugger_core import get_breakpoint
code = compile(txt, filepath, 'exec')
if (len(get_breakpoint(filepath)) != 0 and debug_mode):
app = wx.GetApp().TopWindow
se = app.script_editor
import threading
if not se.CheckDebuggerStatus():
se.QueueSEDRequest(code, viewer.g, ana.result,
filepath)
# print threading.current_thread().name
# wx.Yield()
# time.sleep(3)
else:
se.RunSED(code, viewer.g, ana.result, filepath)
else:
exec code in viewer.g, ana.result
except:
dprint1('error occured when processing data by script')
print('error occured when processing data by following script')
print('#####')
print(txt)
print('#####')
print(traceback.format_exc())
self.change_suppress(True, self.get_child(ishot))
return False
if ana.do_title:
self.process_title(ana)
if self._plot_type == 'plot':
self._update_plot(ana, ishot)#, color_order)
elif self._plot_type == 'timetrace':
self._update_plot(ana, ishot)#, color_order)
elif self._plot_type == 'stepplot':
self._update_plot(ana, ishot)#, color_order)
elif self._plot_type == 'contour':
self._update_contour(ana, ishot)
elif self._plot_type == 'image':
self._update_image(ana, ishot)
elif self._plot_type == 'surface':
self._update_surface(ana, ishot)
elif self._plot_type == 'axspan':
self._update_axspan(ana, ishot)
elif self._plot_type == 'axline':
self._update_axline(ana, ishot)
elif self._plot_type == 'text':
self._update_text(ana, ishot)
obj = self.get_child(idx=ishot)
if obj.is_suppress():
return False
obj._data_extent = None
try:
if (self.get_figaxes() is not None):
self.get_figaxes().adjust_axes_range()
except:
dprint1("failed in adjusting axes at postprocess_data, continuing...")
pass
return True
def onScope1D(self, evt):
from ifigure.mto.fig_plot import FigPlot
y = self._content().nc_eval(self._td)
obj = FigPlot(y)
self._on_scope(obj, False)
def args2var(self):
val = FigPlot.args2var(self)
if self.getvar('enabled_point') == 'all':
self.setp('enabled_point', [True]*(self.getp('x').size))
return val
def attr_in_file(self):
return ['enabled_point'] + FigPlot.attr_in_file()
def __init__(self, *args, **kargs):
FigPlot.__init__(self, *args, **kargs)
FigControl.__init__(self)
self._figc_hit = -1
self._enable_a = None
self._mark_a = None
def __init__(self, *args, **kargs):
FigPlot.__init__(self, *args, **kargs)
FigControl.__init__(self)
self._figc_hit = -1
self._enable_a = None
self._mark_a = None
def __init__(self, *args, **kargs):
kargs['style'] = (wx.FRAME_FLOAT_ON_PARENT | wx.MINIMIZE_BOX
| wx.MAXIMIZE_BOX | wx.RESIZE_BORDER
| wx.SYSTEM_MENU | wx.CAPTION | wx.CLOSE_BOX
| wx.CLIP_CHILDREN)
# print kargs
BookViewer.__init__(self, *args, **kargs)
f_page = self.get_page(0)
f_page.set_section(2)
f_axes = self.get_axes(0, 0)
for name, child in f_axes.get_children():
child.destroy()
obj = FigPlot([0], [0])
f_axes.add_child('xslice_plot1', obj)
obj.set_suppress(True)
obj = FigPlot([0], [0])
f_axes.add_child('xslice_plot2', obj)
obj.set_suppress(True)
f_axes.realize()
f_axes = self.get_axes(0, 1)
for name, child in f_axes.get_children():
child.destroy()
obj = FigPlot([0], [0])
f_axes.add_child('xslice_plot1', obj)
obj.set_suppress(True)
obj = FigPlot([0], [0])
f_axes.add_child('xslice_plot2', obj)
obj.set_suppress(True)
f_axes.realize()
def args2var(self):
val = FigPlot.args2var(self)
if self.getvar('enabled_point') == 'all':
self.setp('enabled_point', [True]*(self.getp('x').size))
return val
def attr_in_file(self):
return ['enabled_point'] + FigPlot.attr_in_file()