mwxlib 0.93.5__py3-none-any.whl → 0.93.7__py3-none-any.whl

mwx/framework.py

@@ -1,7 +1,7 @@
 #! python3
 """mwxlib framework.
 """
-__version__ = "0.93.5"
+__version__ = "0.93.7"
 __author__ = "Kazuya O'moto <komoto@jeol.co.jp>"
 
 from functools import wraps, partial

mwx/graphman.py

@@ -33,6 +33,19 @@ from .matplot2g import GraphPlot
 from .matplot2lg import Histogram
 
 
+def split_paths(obj):
+    """Split obj path into dirname and basename.
+    The object can be module name:str, module, or class.
+    """
+    if hasattr(obj, '__file__'): #<class 'module'>
+        obj = obj.__file__
+    elif isinstance(obj, type):  #<class 'type'>
+        obj = inspect.getsourcefile(obj)
+    if obj.endswith(".py"):
+        obj, _ = os.path.splitext(obj)
+    return os.path.split(obj)
+
+
 class Thread(object):
     """Thread manager for graphman.Layer
     
@@ -1062,20 +1075,6 @@ class Frame(mwx.Frame):
         module.Plugin = _Plugin
         return _Plugin
     
-    @staticmethod
-    def _split_paths(root):
-        if hasattr(root, '__file__'): #<class 'module'>
-            name = root.__file__
-        elif isinstance(root, type):  #<class 'type'>
-            name = inspect.getsourcefile(root)
-        else:
-            name = root
-        dirname = os.path.dirname(name)
-        name = os.path.basename(name)
-        if name.endswith(".py"):
-            name, _ = os.path.splitext(name)
-        return dirname, name
-    
     def load_module(self, root):
         """Load module of plugin (internal use only).
         
@@ -1083,7 +1082,7 @@ class Frame(mwx.Frame):
             This is called automatically from load_plug,
             and should not be called directly from user.
         """
-        dirname_, name = self._split_paths(root)
+        dirname_, name = split_paths(root)
         
         ## Update the include-path to load the module correctly.
         if os.path.isdir(dirname_):
@@ -1147,10 +1146,10 @@ class Frame(mwx.Frame):
         props = dict(show=show, dock=dock, layer=layer, pos=pos, row=row, prop=prop,
                      floating_pos=floating_pos, floating_size=floating_size)
         
-        _dirname, name = self._split_paths(root)
+        _dirname, name = split_paths(root)
         
         plug = self.get_plug(name)
-        if plug and not force: # <plug:name> is already registered.
+        if plug and not force:
             self.update_pane(name, **props)
             try:
                 if session:

mwx/matplot2.py

@@ -297,8 +297,8 @@ class MatplotPanel(wx.Panel):
         self.cursor.visible = 1
     
     def draw(self, art=None):
-        """Draw the plot.
-        Called every time the drawing is updated.
+        """Draw plots.
+        Call each time the drawing should be updated.
         """
         if isinstance(art, matplotlib.artist.Artist):
             self.axes.draw_artist(art)
@@ -403,16 +403,16 @@ class MatplotPanel(wx.Panel):
     
     def copy_to_clipboard(self):
         """Copy canvas image to clipboard."""
-        self.message("Copy image to clipboard.")
+        ## b = self.selected.get_visible()
+        c = self.cursor.visible
         try:
-            b = self.selected.get_visible()
-            c = self.cursor.visible
-            self.selected.set_visible(0)
+            ## self.selected.set_visible(0)
             self.cursor.visible = 0
             self.canvas.draw()
             self.canvas.Copy_to_Clipboard()
+            self.message("Copy image to clipboard.")
         finally:
-            self.selected.set_visible(b)
+            ## self.selected.set_visible(b)
             self.cursor.visible = c
             self.canvas.draw()
     
@@ -604,10 +604,6 @@ class MatplotPanel(wx.Panel):
     
     ZOOM_RATIO = 10**0.2
     
-    def update_position(self):
-        self.toolbar.update()
-        self.toolbar.push_current()
-    
     def OnDraw(self, evt):
         """Called before canvas.draw."""
         pass
@@ -630,12 +626,14 @@ class MatplotPanel(wx.Panel):
     def OnHomePosition(self, evt):
         """Go back to home position."""
         self.toolbar.home()
-        self.update_position()
+        self.toolbar.update()
+        self.toolbar.push_current()
         self.draw()
     
     def OnEscapeSelection(self, evt):
         """Escape from selection."""
         del self.Selector
+        self.canvas.draw_idle()
     
     ## def OnShiftLimit(self, evt, r=0.1):
     ##     w = self.xlim[1] - self.xlim[0]
@@ -686,7 +684,6 @@ class MatplotPanel(wx.Panel):
         ## self.toolbar.set_cursor(1)
         self.set_wxcursor(wx.CURSOR_ARROW)
         self.toolbar.pan()
-        ## self.draw()
         self.handler.current_state = self.__prev  # --> previous state of PAN
         del self.__prev
     
@@ -701,7 +698,6 @@ class MatplotPanel(wx.Panel):
         ## self.toolbar.set_cursor(1)
         self.set_wxcursor(wx.CURSOR_ARROW)
         self.toolbar.zoom()
-        ## self.draw()
         self.handler.current_state = self.__prev  # --> previous state of ZOOM
         del self.__prev
     

mwx/matplot2g.py

@@ -863,7 +863,8 @@ class GraphPlot(MatplotPanel):
         """Reset display range (xylim's), update home position."""
         if self.frame:
             self.axes.axis(self.frame.get_extent()) # reset xlim and ylim
-            self.update_position()
+            self.toolbar.update()
+            self.toolbar.push_current()
             self.draw()
     
     def fit_to_canvas(self):
@@ -980,24 +981,24 @@ class GraphPlot(MatplotPanel):
     clipboard_data = None
     
     def write_buffer_to_clipboard(self):
-        """Copy - Write buffer data to clipboard."""
+        """Write buffer data to clipboard."""
         if not self.frame:
             self.message("No frame")
             return
         try:
-            self.message("Write buffer to clipboard.")
             name = self.frame.name
             data = self.frame.roi
             GraphPlot.clipboard_name = name
             GraphPlot.clipboard_data = data
             bins, vlim, img = imconvert(data, self.frame.vlim)
             Clipboard.imwrite(img)
+            self.message("Write buffer to clipboard.")
         except Exception as e:
-            self.message("- Failure in clipboard:", e)
             traceback.print_exc()
+            self.message("- Failed to write to clipboard:", e)
     
     def read_buffer_from_clipboard(self):
-        """Paste - Read buffer data from clipboard."""
+        """Read buffer data from clipboard."""
         try:
             name = GraphPlot.clipboard_name
             data = GraphPlot.clipboard_data
@@ -1010,8 +1011,8 @@ class GraphPlot(MatplotPanel):
                 self.message("Read image from clipboard.")
                 self.load(Clipboard.imread())
         except Exception as e:
-            self.message("- No data in clipboard:", e)
             traceback.print_exc()
+            self.message("- No data in clipboard:", e)
     
     def destroy_colorbar(self):
         if self.cbar:

mwx/matplot2lg.py

@@ -31,7 +31,7 @@ class LinePlot(MatplotPanel):
                'delete pressed' : (NORMAL, self.OnEscapeSelection),
                   'M-a pressed' : (NORMAL, self.OnHomePosition),
                   'C-a pressed' : (NORMAL, self.OnHomePosition),
-             'Lbutton dblclick' : (NORMAL, self.OnEscapeSelection),
+             'Lbutton dblclick' : (NORMAL, self.OnEscapeSelection, self.OnDragLock),
              '*Lbutton pressed' : (NORMAL, self.OnDragLock),
                  '*Ldrag begin' : (REGION, self.OnDragBegin),
             },
@@ -166,7 +166,7 @@ class LinePlot(MatplotPanel):
             if   x < l: x = l
             elif x > r: x = r
             self.region = (self.__lastpoint, x)
-        else:
+        elif self.region is not None:
             a, b = self.region
             d = x - self.__lastpoint
             if self.boundary is not None:
@@ -181,6 +181,8 @@ class LinePlot(MatplotPanel):
             else:
                 self.region = (a+d, b+d)
                 self.__lastpoint = x
+        else:
+            self.message("- No region.") #<FSM logic-error>
         self.draw()
     
     def OnDragEnd(self, evt):
@@ -282,7 +284,8 @@ class Histogram(LinePlot):
             self.xlim = x.min(), x.max()
             self.ylim = 0, y.max()
             self.region = None
-            self.update_position()
+            self.toolbar.update()
+            self.toolbar.push_current()
             self.draw()
     
     def hreplot(self, frame):
@@ -308,7 +311,8 @@ class Histogram(LinePlot):
             self.__plot.set_data([], [])
             self.region = None
         
-        self.update_position()
+        self.toolbar.update()
+        self.toolbar.push_current()
         self.draw()
     
     def writeln(self):
@@ -392,9 +396,9 @@ class LineProfile(LinePlot):
                  '*Ldrag begin' : (REGION, self.OnDragBegin),
             },
             REGION : {
-                 'S-Ldrag move' : (REGION+LINE, self.OnRegionLock),
+                 'S-Ldrag move' : (REGION+LINE, self.OnRegionLock, self.OnDragLineBegin),
                  'M-Ldrag move' : (REGION+MARK, self.OnMarkPeaks, self.OnMarkSelectionBegin),
-                  '*Ldrag move' : (REGION, self.OnDragMove),
+                  '*Ldrag move' : (REGION, self.OnDragMove, self.OnDragTrace),
                    '*Ldrag end' : (NORMAL, self.OnDragEnd),
             },
             LINE: {
@@ -431,7 +435,7 @@ class LineProfile(LinePlot):
         
         self.menu += [
             (mwx.ID_(510), "&Copy data", "Copy data to clipboard",
-                lambda v: self.copy_data_to_clipboard()),
+                lambda v: self.write_data_to_clipboard()),
             (),
             (mwx.ID_(511), "Logic length", "Set axis-unit in logic base", wx.ITEM_RADIO,
                 lambda v: self.set_logic(1),
@@ -516,6 +520,14 @@ class LineProfile(LinePlot):
         """Plotted (xdata, ydata) in single plot."""
         return self.__plot.get_data(orig=0)
     
+    def calc_average(self):
+        X, Y = self.plotdata
+        if self.region is not None:
+            a, b = self.region
+            Y = Y[(a <= X) & (X <= b)]
+        if Y.size:
+            return Y.mean()
+    
     def linplot(self, frame, fit=True, force=True):
         if not force:
             if frame is self.__frame:
@@ -572,7 +584,8 @@ class LineProfile(LinePlot):
                 self.xlim = ls[0], ls[-1]
                 self.ylim = ly[0], max(ly[1], max(zs))
             
-        self.update_position()
+        self.toolbar.update()
+        self.toolbar.push_current()
         self.draw()
     
     def writeln(self):
@@ -594,14 +607,14 @@ class LineProfile(LinePlot):
         else:
             self.modeline.write("")
     
-    def copy_data_to_clipboard(self):
-        """Copy plotdata to clipboard."""
-        self.message("Copy data to clipboard.")
+    def write_data_to_clipboard(self):
+        """Write plot data to clipboard."""
         X, Y = self.plotdata
         with io.StringIO() as o:
             for x, y in zip(X, Y):
                 o.write("{:g}\t{:g}\n".format(x, y))
             Clipboard.write(o.getvalue(), verbose=0)
+            self.message("Write data to clipboard.")
     
     ## --------------------------------
     ## Motion/Drag actions (override)
@@ -629,7 +642,7 @@ class LineProfile(LinePlot):
         if x.size:
             self.__fil.set_xy(list(chain([(x[0],0)], zip(x,y), [(x[-1],0)])))
         self.writeln()
-        
+    
     def OnLineWidth(self, evt):
         n = -2 if evt.key[-1] == '-' else 2
         self.set_linewidth(self.__linewidth + n)
@@ -648,7 +661,17 @@ class LineProfile(LinePlot):
     def OnDragLineBegin(self, evt):
         self.set_wxcursor(wx.CURSOR_SIZENS)
     
+    def OnDragTrace(self, evt):
+        """Show average value."""
+        y = self.calc_average()
+        if y is not None:
+            ## self.__hline.set_ydata([y])
+            ## self.__hline.set_visible(1)
+            ## self.canvas.draw_idle()
+            self.message(f"ya = {y:g}")
+    
     def OnRegionLock(self, evt):
+        """Show FWHM region."""
         x, y = self.plotdata
         if x.size:
             xc, yc = evt.xdata, evt.ydata
@@ -678,10 +701,11 @@ class LineProfile(LinePlot):
             
             self.__hline.set_ydata([yc])
             self.__hline.set_visible(1)
-            self.message("y = {:g}, xr = {}".format(yc, self.region))
             self.draw()
+            self.message(f"yc = {yc:g}")
     
     def OnMarkPeaks(self, evt):
+        """Set markers on peaks."""
         x, y = self.plotdata
         if x.size:
             lw = 5
@@ -699,7 +723,9 @@ class LineProfile(LinePlot):
                 self.Selector = x[peaks], y[peaks]
     
     def OnMarkErase(self, evt):
-        del self.Selector
+        """Erase markers on peaks."""
+        ## del self.Selector
+        self.OnEscapeSelection(evt)
     
     def OnMarkSelectionBegin(self, evt):
         org = self.p_event

mwx/py/__init__.py

@@ -0,0 +1,2 @@
+#! python3
+"""Plug-ins / Add-ins."""

mwx/py/ffmpeg_view.py

@@ -0,0 +1,257 @@
+#! python3
+"""FFmpeg wrapper.
+"""
+from subprocess import Popen, PIPE
+import numpy as np
+import os
+import wx
+import wx.media
+
+from mwx.graphman import Layer, Frame
+from mwx.controls import LParam, Icon, Button, TextCtrl
+
+
+def read_info(path):
+    command = ['ffprobe',
+               '-i', path,
+               '-loglevel', 'quiet',    # no verbose
+               '-print_format', 'json', # -format json
+               '-show_streams',         # -streams info
+               ]
+    with Popen(command, stdout=PIPE, stderr=PIPE) as fp:
+        ret, err = fp.communicate()
+        if not err:
+            return eval(ret)
+
+
+def capture_video(path, ss=0):
+    command = ['ffmpeg',
+               '-ss', f"{ss}",          # Note: placing -ss before -i will be faster,
+               '-i', path,              #       but maybe not accurate.
+               '-frames:v', '1',        # -frame one shot
+               '-f', 'rawvideo',        # -format raw
+               '-pix_fmt', 'rgb24',     # rgb24, gray, etc.
+               'pipe:'                  # pipe to stdout: '-'
+               ]
+    bufsize = 4096 # w * h * 3
+    buf = b"" # bytearray()
+    with Popen(command, stdout=PIPE) as fp:
+        while 1:
+            s = fp.stdout.read(bufsize)
+            buf += s
+            if len(s) < bufsize:
+                break
+    return np.frombuffer(buf, np.uint8)
+
+
+def export_video(path, crop, ss, to, filename):
+    command = ['ffmpeg',
+               '-i', path,
+               '-vf', f"{crop=}",
+               '-ss', f"{ss}",
+               '-to', f"{to}",
+               '-y', filename,
+               ]
+    print('>', ' '.join(command))
+    with Popen(command) as fp:
+        ret, err = fp.communicate()
+
+
+class MyFileDropLoader(wx.FileDropTarget):
+    def __init__(self, target):
+        wx.FileDropTarget.__init__(self)
+        self.target = target
+    
+    def OnDropFiles(self, x, y, filenames):
+        path = filenames[-1] # Only the last one will be loaded.
+        if len(filenames) > 1:
+            print("- Drop only one file please."
+                  "Loading {!r} ...".format(path))
+        self.target.load_media(path)
+        return True
+
+
+class Plugin(Layer):
+    """Media loader using FFMpeg (installation required).
+    """
+    menukey = "Plugins/Extensions/FFMpeg viewer"
+    ## menukey = "FFMpeg/"
+    dockable = False
+    
+    def Init(self):
+        self.mc = wx.media.MediaCtrl()
+        self.mc.Create(self, size=(300,300),
+                       style=wx.SIMPLE_BORDER,
+                       szBackend=wx.media.MEDIABACKEND_WMP10
+                       ## szBackend=wx.media.MEDIABACKEND_DIRECTSHOW
+        )
+        self.mc.ShowPlayerControls()
+        self.mc.Bind(wx.media.EVT_MEDIA_LOADED, self.OnMediaLoaded)
+        self.mc.Bind(wx.media.EVT_MEDIA_PAUSE, self.OnMediaPause)
+        
+        self.mc.SetDropTarget(MyFileDropLoader(self))
+        
+        self._path = None
+        
+        self.ss = LParam("ss:", # range/value will be set when loaded later.
+                        handler=self.set_offset,
+                        updater=self.get_offset,
+                        )
+        self.to = LParam("to:", # range/value will be set when loaded later.
+                        handler=self.set_offset,
+                        updater=self.get_offset,
+                        )
+        self.crop = TextCtrl(self, icon="cut", size=(130,-1),
+                        handler=self.set_crop,
+                        updater=self.get_crop,
+                        )
+        
+        self.snp = Button(self, handler=self.snapshot, tip='Snapshot', icon='clock')
+        self.exp = Button(self, handler=self.export, tip="Export", icon='save')
+        
+        self.rw = Button(self, handler=lambda v: self.seekdelta(-100), icon='|<-')
+        self.fw = Button(self, handler=lambda v: self.seekdelta(+100), icon='->|')
+        
+        self.layout((self.mc,), expand=2)
+        self.layout((self.ss, self.to, self.rw, self.fw,
+                     self.snp, self.crop, self.exp),
+                    expand=0, row=7, style='button', lw=12, cw=0, tw=64)
+        
+        self.menu[0:5] = [
+            (1, "&Load file", Icon('open'),
+                lambda v: self.load_media()),
+                
+            (2, "&Snapshot", Icon('clock'),
+                lambda v: self.snapshot(),
+                lambda v: v.Enable(self._path is not None)),
+            (),
+        ]
+        
+        self.parent.handler.bind("unknown_format", self.load_media)
+    
+    def Destroy(self):
+        try:
+            self.parent.handler.unbind("unknown_format", self.load_media)
+            self.mc.Stop()
+        finally:
+            return Layer.Destroy(self)
+    
+    def OnMediaLoaded(self, evt):
+        self.ss.range = (0, self.video_dur, 0.01)
+        self.to.range = (0, self.video_dur, 0.01)
+        self.Show()
+        evt.Skip()
+    
+    def OnMediaPause(self, evt):
+        evt.Skip()
+    
+    def load_media(self, path=None):
+        if path is None:
+            with wx.FileDialog(self, "Choose a media file",
+                style=wx.FD_OPEN|wx.FD_CHANGE_DIR|wx.FD_FILE_MUST_EXIST) as dlg:
+                if dlg.ShowModal() != wx.ID_OK:
+                    return None
+                path = dlg.Path
+        self.mc.Load(path) # -> True (always)
+        self.info = read_info(path)
+        if self.info:
+            v = next(x for x in self.info['streams'] if x['codec_type'] == 'video')
+            ## self.video_fps = eval(v['r_frame_rate']) # real base frame rate
+            self.video_fps = eval(v['avg_frame_rate'])  # averaged frame rate
+            self.video_dur = eval(v['duration'])        # duration [s]
+            w, h = v['width'], v['height']
+            if v['tags'].get('rotate') in ('90', '270'):
+                w, h = h, w  # transpose
+            self.video_size = w, h
+            self._path = path
+            self.message(f"Loaded {path!r} successfully.")
+            return True
+        else:
+            self.message(f"Failed to load file {path!r}.")
+            return False
+    
+    DELTA = 1000 # correction ▲理由は不明 (WMP10 backend only?)
+    
+    def set_offset(self, tc):
+        """Set offset value by referring to ss/to value."""
+        try:
+            self.mc.Seek(self.DELTA + int(tc.value * 1000))
+        except Exception:
+            pass
+    
+    def get_offset(self, tc):
+        """Get offset value and assigns it to ss/to value."""
+        try:
+            tc.value = round(self.mc.Tell()) / 1000
+        except Exception:
+            pass
+    
+    def set_crop(self):
+        """Set crop area (W:H:Left:Top) to roi."""
+        frame = self.graph.frame
+        if frame:
+            try:
+                w, h, xo, yo = eval(self.crop.Value.replace(':', ','))
+                xo -= 0.5  # Correction with half-pixel
+                yo -= 0.5  # to select left-top (not center) position
+                nx = xo, xo+w
+                ny = yo, yo+h
+                frame.region = frame.xyfrompixel(nx, ny)
+            except Exception:
+                self.message("Failed to evaluate crop text.")
+    
+    def get_crop(self):
+        """Get crop area (W:H:Left:Top) from roi."""
+        crop = ''
+        frame = self.graph.frame
+        if frame:
+            nx, ny = frame.xytopixel(*frame.region)
+            if nx.size:
+                xo, yo = nx[0], ny[1]
+                xp, yp = nx[1], ny[0]
+                crop = "{}:{}:{}:{}".format(xp-xo, yp-yo, xo, yo)
+        if self._path and not crop:
+            crop = "{}:{}:0:0".format(*self.video_size)
+        self.crop.Value = crop
+    
+    def seekdelta(self, offset):
+        """Seek relative position [ms]."""
+        if wx.GetKeyState(wx.WXK_SHIFT):
+            offset /= 10
+        try:
+            t = self.to.value + offset/1000
+        except Exception as e:
+            print(e)
+        else:
+            if self._path and 0 <= t < self.video_dur:
+                self.to.value = round(t, 3)
+            self.set_offset(self.to) # => seek
+    
+    def snapshot(self):
+        """Create a snapshot of the current frame.
+        Load the snapshot image into the graph window.
+        """
+        if not self._path:
+            return
+        t = self.mc.Tell()
+        w, h = self.video_size
+        buf = capture_video(self._path, t/1000).reshape((h,w,3))
+        name = "{}-ss{}".format(os.path.basename(self._path), int(t))
+        self.graph.load(buf, name)
+    
+    def export(self):
+        """Export the cropped / clipped data to a media file."""
+        if not self._path:
+            return
+        fout = "{}_clip".format(os.path.splitext(self._path)[0])
+        with wx.FileDialog(self, "Save as",
+                defaultFile=os.path.basename(fout),
+                wildcard="Media file (*.mp4)|*.mp4|"
+                         "Animiation (*.gif)|*.gif",
+                style=wx.FD_SAVE|wx.FD_OVERWRITE_PROMPT) as dlg:
+            if dlg.ShowModal() != wx.ID_OK:
+                return
+            fout = dlg.Path
+        export_video(self._path,
+                     self.crop.Value or "{}:{}:0:0".format(*self.video_size),
+                     self.ss.value, self.to.value, fout)

mwx/py/fft_view.py

@@ -0,0 +1,79 @@
+#! python3
+"""View of FFT/iFFT.
+"""
+import wx
+import numpy as np
+from numpy.fft import fft2,ifft2,fftshift,ifftshift
+## from scipy.fftpack import fft,ifft,fft2,ifft2 Memory Leak? <scipy 0.16.1>
+## import cv2
+
+from jgdk import Layer, Param
+import editor as edi
+
+
+class Plugin(Layer):
+    """FFT view.
+    
+    FFT src (graph.buffer) to dst (output.buffer).
+    Note:
+        Rectangular regions will result in distorted patterns.
+        長方形のリージョンは歪んだパターンになるので要注意
+    """
+    menukey = "Plugins/Extensions/&FFT view"
+    caption = "FFT view"
+    
+    def Init(self):
+        self.pchk = wx.CheckBox(self, label="logical unit")
+        self.pchk.Value = True
+        
+        self.ftor = Param("mask", (2,4,8,16,32,64)) # masking area factor of 1/2
+        
+        self.layout((self.pchk,), title="normal FFT")
+        self.layout((self.ftor,), title="inverse FFT", style='chkbox', tw=32)
+        
+        self.parent.define_key('C-f', self.newfft)
+        self.parent.define_key('C-S-f', self.newifft)
+    
+    def Destroy(self):
+        self.parent.define_key('C-f', None)
+        self.parent.define_key('C-S-f', None)
+        return Layer.Destroy(self)
+    
+    def newfft(self, evt):
+        """New FFT of graph to output."""
+        frame = self.graph.frame
+        if frame:
+            self.message("FFT execution...")
+            src = edi.fftcrop(frame.roi)
+            h, w = src.shape
+            
+            dst = fftshift(fft2(src))
+            
+            self.message("\b Loading image...")
+            u = 1 / w
+            if self.pchk.Value:
+                u /= frame.unit
+            self.output.load(dst, "*fft of {}*".format(frame.name),
+                                  localunit=u)
+            self.message("\b done")
+    
+    def newifft(self, evt):
+        """New inverse FFT of output to graph."""
+        frame = self.output.frame
+        if frame:
+            self.message("iFFT execution...")
+            src = frame.roi
+            h, w = src.shape
+            
+            if self.ftor.check:
+                y, x = np.ogrid[-h/2:h/2, -w/2:w/2]
+                mask = np.hypot(y,x) > w / self.ftor.value
+                src = src.copy() # apply mask to the copy
+                src[mask] = 0
+            
+            dst = ifft2(ifftshift(src))
+            
+            self.message("\b Loading image...")
+            self.graph.load(dst.real, "*ifft of {}*".format(frame.name),
+                                      localunit=1/w/frame.unit)
+            self.message("\b done")

mwx/py/frame_listview.py

@@ -0,0 +1,283 @@
+#! python3
+"""Property list of buffers.
+"""
+from pprint import pformat
+import wx
+from wx import aui
+from wx.lib.mixins.listctrl import ListCtrlAutoWidthMixin
+
+from mwx.framework import CtrlInterface, Menu, StatusBar
+from mwx.controls import Icon, Icon2, Clipboard
+from mwx.graphman import Layer
+
+
+class CheckList(wx.ListCtrl, ListCtrlAutoWidthMixin, CtrlInterface):
+    """CheckList of Graph buffers.
+    
+    Note:
+        list item order = buffer order.
+        (リストアイテムとバッファの並び順 0..n は常に一致します)
+    """
+    @property
+    def selected_items(self):
+        return filter(self.IsSelected, range(self.ItemCount))
+    
+    @property
+    def checked_items(self):
+        return filter(self.IsItemChecked, range(self.ItemCount))
+    
+    @property
+    def focused_item(self):
+        return self.FocusedItem
+    
+    @property
+    def all_items(self):
+        rows = range(self.ItemCount)
+        cols = range(self.ColumnCount)
+        ## return [[self.GetItemText(j, k) for k in cols] for j in rows]
+        for j in rows:
+            yield [self.GetItemText(j, k) for k in cols]
+    
+    def __init__(self, parent, target, **kwargs):
+        wx.ListCtrl.__init__(self, parent, size=(400,130),
+                             style=wx.LC_REPORT|wx.LC_HRULES, **kwargs)
+        ListCtrlAutoWidthMixin.__init__(self)
+        CtrlInterface.__init__(self)
+        
+        self.EnableCheckBoxes()
+        
+        self.parent = parent
+        self.Target = target
+        
+        self.__dir = True
+        
+        _alist = ( # assoc-list of column names
+            ("id", 42),
+            ("name", 160),
+            ("shape", 90),
+            ("dtype", 60),
+            ("Mb",   40),
+            ("unit", 60),
+            ("annotation", 240),
+        )
+        for k, (name, w) in enumerate(_alist):
+            self.InsertColumn(k, name, width=w)
+        
+        for j, frame in enumerate(self.Target.all_frames):
+            self.InsertItem(j, str(j))
+            self.UpdateInfo(frame) # update all --> 計算が入ると時間がかかる
+        
+        self.handler.update({
+            0 : {
+             'Lbutton dblclick' : (0, self.OnShowItems), # -> frame_shown
+                'enter pressed' : (0, self.OnShowItems), # -> frame_shown
+               'delete pressed' : (0, self.OnRemoveItems), # -> frame_removed/shown
+                   'f2 pressed' : (0, self.OnEditAnnotation),
+                  'C-a pressed' : (0, self.OnSelectAllItems),
+                  'C-o pressed' : (0, self.OnLoadItems),
+                  'C-s pressed' : (0, self.OnSaveItems),
+                 'M-up pressed' : (0, self.Target.OnPageUp),
+               'M-down pressed' : (0, self.Target.OnPageDown),
+            },
+        })
+        self.handler.clear(0)
+        
+        self.Bind(wx.EVT_LIST_COL_CLICK, self.OnSortItems)
+        self.Bind(wx.EVT_LIST_ITEM_SELECTED, self.OnItemSelected)
+        
+        self.context = { # bound to the target
+            None: {
+                  'frame_shown' : [ None, self.on_frame_shown ],
+                 'frame_hidden' : [ None, self.on_frame_hidden ],
+                 'frame_loaded' : [ None, self.on_frame_loaded ],
+                'frame_removed' : [ None, self.on_frames_removed ],
+               'frame_modified' : [ None, self.UpdateInfo ],
+                'frame_updated' : [ None, self.UpdateInfo ],
+            }
+        }
+        self.Target.handler.append(self.context)
+        
+        def copy(all=True):
+            frames = self.Target.all_frames
+            if frames:
+                frame = frames[self.focused_item]
+                if all:
+                    text = pformat(frame.attributes, sort_dicts=0)
+                else:
+                    text = "{}\n{}".format(frame.name, frame.annotation)
+                Clipboard.write(text)
+        
+        self.menu = [
+            (101, "&Edit annotation", "Edit annotation", Icon('pencil'),
+                self.OnEditAnnotation),
+            (),
+            (102, "Copy info", Icon('copy'),
+                lambda v: copy(0),
+                lambda v: v.Enable(len(list(self.selected_items)))),
+                
+            (103, "Copy ALL data", Icon2('copy', '+'),
+                lambda v: copy(),
+                lambda v: v.Enable(len(list(self.selected_items)))),
+        ]
+        self.Bind(wx.EVT_CONTEXT_MENU,
+                  lambda v: Menu.Popup(self, self.menu))
+    
+    def Destroy(self):
+        try:
+            self.Target.handler.remove(self.context)
+        finally:
+            return wx.ListCtrl.Destroy(self)
+    
+    def UpdateInfo(self, frame):
+        ls = ("{}".format(frame.index),
+              "{}".format(frame.name),
+              "{}".format(frame.buffer.shape),
+              "{}".format(frame.buffer.dtype),
+          "{:.1f}".format(frame.buffer.nbytes/1e6),
+          "{:g}{}".format(frame.unit, '*' if frame.localunit else ''),
+              "{}".format(frame.annotation),
+        )
+        j = frame.index
+        for k, v in enumerate(ls):
+            self.SetItem(j, k, v)
+        if frame.pathname:
+            self.CheckItem(j)
+    
+    def OnShowItems(self, evt):
+        self.Target.select(self.focused_item)
+    
+    def OnRemoveItems(self, evt):
+        del self.Target[self.selected_items]
+    
+    def OnSortItems(self, evt): #<wx._controls.ListEvent>
+        col = evt.Column
+        if col == 0: # reverse the first column
+            self.__dir = False
+        self.__dir = not self.__dir # toggle 0:ascend/1:descend
+        
+        frames = self.Target.all_frames
+        if frames:
+            def _eval(x):
+                try:
+                    return eval(x[col].replace('*', '')) # localunit* とか
+                except Exception:
+                    return x[col]
+            frame = self.Target.frame
+            items = sorted(self.all_items, reverse=self.__dir, key=_eval)
+            frames[:] = [frames[int(c[0])] for c in items] # sort by new Id of items
+            
+            lc = list(self.checked_items)
+            
+            for j, c in enumerate(items):
+                self.Select(j, False)
+                self.CheckItem(j, int(c[0]) in lc)
+                for k, v in enumerate(c[1:]): # update data except for id(0)
+                    self.SetItem(j, k+1, v)
+            self.Target.select(frame) # invokes [frame_shown] to select the item
+    
+    def OnSelectAllItems(self, evt):
+        for j in range(self.ItemCount):
+            self.Select(j)
+    
+    def OnLoadItems(self, evt):
+        self.parent.parent.import_index(view=self.Target)
+    
+    def OnSaveItems(self, evt):
+        frames = self.Target.all_frames
+        selected_frames = [frames[j] for j in self.selected_items]
+        if selected_frames:
+            self.parent.message("Exporting {} frames.".format(len(selected_frames)))
+            self.parent.parent.export_index(frames=selected_frames)
+        else:
+            self.parent.message("No frame selected.")
+    
+    def OnEditAnnotation(self, evt):
+        frames = self.Target.all_frames
+        if frames:
+            frame = frames[self.focused_item]
+            with wx.TextEntryDialog(self, frame.name,
+                    'Enter an annotation', frame.annotation) as dlg:
+                if dlg.ShowModal() == wx.ID_OK:
+                    frame.annotation = dlg.Value
+    
+    def OnItemSelected(self, evt):
+        frames = self.Target.all_frames
+        frame = frames[evt.Index]
+        self.parent.message(frame.pathname)
+        evt.Skip()
+    
+    ## --------------------------------
+    ## Actions of frame-handler
+    ## --------------------------------
+    
+    def on_frame_loaded(self, frame):
+        j = frame.index
+        self.InsertItem(j, str(j))
+        for k in range(j+1, self.ItemCount): # id(0) を更新する
+            self.SetItem(k, 0, str(k))
+        self.UpdateInfo(frame)
+    
+    def on_frame_shown(self, frame):
+        j = frame.index
+        self.SetItemFont(j, self.Font.Bold())
+        self.Select(j)
+        self.Focus(j)
+    
+    def on_frame_hidden(self, frame):
+        j = frame.index
+        self.SetItemFont(j, self.Font)
+        self.Select(j, False)
+    
+    def on_frames_removed(self, indices):
+        for j in reversed(indices):
+            self.DeleteItem(j)
+        for k in range(self.ItemCount): # id(0) を更新する
+            self.SetItem(k, 0, str(k))
+
+
+class Plugin(Layer):
+    """Property list of Grpah buffers.
+    """
+    menukey = "Plugins/Extensions/&Buffer listbox\tCtrl+b"
+    caption = "Property list"
+    dockable = False
+    
+    @property
+    def all_pages(self):
+        return [self.nb.GetPage(i) for i in range(self.nb.PageCount)]
+    
+    @property
+    def message(self):
+        return self.statusline
+    
+    def Init(self):
+        self.nb = aui.AuiNotebook(self, size=(400,150),
+            style = (aui.AUI_NB_DEFAULT_STYLE|aui.AUI_NB_RIGHT)
+                  &~(aui.AUI_NB_CLOSE_ON_ACTIVE_TAB|aui.AUI_NB_MIDDLE_CLICK_CLOSE)
+        )
+        self.attach(self.graph, "graph")
+        self.attach(self.output, "output")
+        
+        self.statusline = StatusBar(self)
+        self.layout((
+                self.nb,
+                (self.statusline, 0, wx.EXPAND),
+            ),
+            expand=2, border=0, vspacing=0,
+        )
+        
+        def on_focus_set(v):
+            self.parent.select_view(self.nb.CurrentPage.Target)
+            v.Skip()
+        
+        self.nb.Bind(wx.EVT_CHILD_FOCUS, on_focus_set)
+    
+    def attach(self, target, caption):
+        if target not in [lc.Target for lc in self.all_pages]:
+            lc = CheckList(self, target)
+            self.nb.AddPage(lc, caption)
+    
+    def detach(self, target):
+        for k, lc in enumerate(self.all_pages):
+            if target is lc.Target:
+                self.nb.DeletePage(k)

mwx/py/line_profile.py

@@ -0,0 +1,27 @@
+#! python3
+"""Line profile.
+"""
+from mwx.graphman import Layer
+from mwx.matplot2lg import LineProfile
+
+
+class Plugin(Layer):
+    """Line profile of the currently selected buffers.
+    """
+    menukey = "Plugins/Extensions/&Line profile\tCtrl+l"
+    caption = "Line profile"
+    dockable = False
+    
+    def Init(self):
+        self.plot = LineProfile(self, log=self.message, size=(300,200))
+        
+        self.layout((self.plot,), expand=2, border=0)
+        
+        @self.handler.bind('page_shown')
+        def activate(v):
+            self.plot.attach(*self.parent.graphic_windows)
+            self.plot.linplot(self.parent.selected_view.frame)
+        
+        @self.handler.bind('page_closed')
+        def deactivate(v):
+            self.plot.detach(*self.parent.graphic_windows)

mwx/utilus.py

@@ -570,7 +570,7 @@ class FSM(dict):
         self.update(contexts)
     
     def __missing__(self, key):
-        raise Exception("FSM:logic-error: undefined state {!r}".format(key))
+        raise Exception("FSM logic-error: undefined state {!r}".format(key))
     
     def __repr__(self):
         return "<{} object at 0x{:X}>".format(self.__class__.__name__, id(self))
@@ -662,7 +662,7 @@ class FSM(dict):
                 except BdbQuit:
                     pass
                 except Exception as e:
-                    self.dump("- FSM:exception: {!r}".format(e),
+                    self.dump("- FSM exception: {!r}".format(e),
                               "  event  : {}".format(event),
                               "  from   : {}".format(self.__prev_state),
                               "  to     : {}".format(self.__state),
@@ -799,7 +799,7 @@ class FSM(dict):
             warnings.warn(msg, stacklevel=3)
         
         if state not in self:
-            warn("- FSM:warning: [{!r}] context newly created.".format(state))
+            warn("- FSM warning: [{!r}] context newly created.".format(state))
             self[state] = SSM() # new context
         
         context = self[state]
@@ -808,14 +808,14 @@ class FSM(dict):
         
         if event in context:
             if state2 != context[event][0]:
-                warn("- FSM:warning: transaction may conflict.\n"
+                warn("- FSM warning: transaction may conflict.\n"
                      "  The state {2!r} and the original state is not the same."
                      "  {0!r} : {1!r} --> {2!r}".format(event, state, state2))
                 pass
                 context[event][0] = state2 # update transition
         else:
             ## if state2 not in self:
-            ##     warn("- FSM:warning: transaction may contradict\n"
+            ##     warn("- FSM warning: transaction may contradict\n"
             ##          "  The state {2!r} is not found in the contexts."
             ##          "  {0!r} : {1!r} --> {2!r}".format(event, state, state2))
             ##     pass
@@ -829,7 +829,7 @@ class FSM(dict):
             try:
                 transaction.append(action)
             except AttributeError:
-                warn("- FSM:warning: cannot append new transaction ({!r} : {!r})\n"
+                warn("- FSM warning: cannot append new transaction ({!r} : {!r})\n"
                      "  The transaction must be a list, not a tuple".format(state, event))
         return action
     
@@ -847,12 +847,12 @@ class FSM(dict):
             warnings.warn(msg, stacklevel=3)
         
         if state not in self:
-            warn("- FSM:warning: [{!r}] context does not exist.".format(state))
+            warn("- FSM warning: [{!r}] context does not exist.".format(state))
             return
         
         context = self[state]
         if event not in context:
-            warn("- FSM:warning: No such transaction ({!r} : {!r})".format(state, event))
+            warn("- FSM warning: No such transaction ({!r} : {!r})".format(state, event))
             return
         
         transaction = context[event]
@@ -866,7 +866,7 @@ class FSM(dict):
                 transaction.remove(action)
                 return True
             except AttributeError:
-                warn("- FSM:warning: removing action from context ({!r} : {!r})\n"
+                warn("- FSM warning: removing action from context ({!r} : {!r})\n"
                      "  The transaction must be a list, not a tuple".format(state, event))
         return False
 

{mwxlib-0.93.5.dist-info → mwxlib-0.93.7.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: mwxlib
-Version: 0.93.5
+Version: 0.93.7
 Summary: A wrapper of matplotlib and wxPython (phoenix)
 Home-page: https://github.com/komoto48g/mwxlib
 Author: Kazuya O'moto

mwxlib-0.93.7.dist-info/RECORD

@@ -0,0 +1,27 @@
+mwx/__init__.py,sha256=5B4YSOuijG1Uo5-FLtLHGB52Cp_F4vnN--4wGPBx7do,2398
+mwx/bookshelf.py,sha256=FrissUYdGXLABOzJmMaQU6GXvu6n_9DVW3d5wGwQzjM,6613
+mwx/controls.py,sha256=9C1sXBwQl-oZtCb3rXoT-doW48Sv1J359toNq9ScF9M,47173
+mwx/framework.py,sha256=HiXX52Go64R167S9GiKs1SEbkxx-sMps88VDI9mIw5w,73951
+mwx/graphman.py,sha256=SKYJFxi-TqWOALRczYz-M-2ZYEbqiZuAuKCfEIedctY,70156
+mwx/images.py,sha256=mrnUYH12I3XLVSZcEXlpVltX0XMxufbl2yRvDIQJZqc,49957
+mwx/matplot2.py,sha256=qaF_gvLoLn-TimLbRR59KUavNr1ZpZQdSMqjzJk47rk,32682
+mwx/matplot2g.py,sha256=Txcz3yusiFF0j1QnAS0dCGTbGrkx4VC6BZbWbjIRZnw,65585
+mwx/matplot2lg.py,sha256=nKXxy1XZRLTo0WijeIGd0WtQrlSpI1cxkGRwCktrJjY,27263
+mwx/mgplt.py,sha256=ITzxA97yDwr_35BUk5OqnyskSuKVDbpf2AQCKY1jHTI,5671
+mwx/nutshell.py,sha256=5HErOObnOprMc6gYnXV7wY-ALnlwNpmhV8f0EyipYA4,135816
+mwx/utilus.py,sha256=THDxQ-QqbHYVc8iX8qN9yxvfcf7Pvpm7sfTP9ipYvzs,37040
+mwx/wxmon.py,sha256=Qk86VbuuW2rR46pqEYLur13G_aloWz5SVv6sib30YY0,12717
+mwx/wxpdb.py,sha256=2z3ZD9Oo1H-ONBHlaprkB9hrTmAI7o03sqO46ppEFE4,19129
+mwx/wxwil.py,sha256=JK1du4i1RVMbDLqN8jLRDSu_JhKEp4mhHVMElzo4yoE,5578
+mwx/wxwit.py,sha256=2gFHi-8nwWRh26RdvZ_AUP--8PDjJB8TUU72y2fBUWM,7557
+mwx/py/__init__.py,sha256=NzBazvjN9KMFoylsHsLiNOHbcDrNflgSwEFlkT12Rnw,39
+mwx/py/ffmpeg_view.py,sha256=vUYNybIJsF1JGkDzjBgDyBQvDh8e1oKHlEMY5Fwc8L4,9399
+mwx/py/fft_view.py,sha256=1bc6e6muXoo2EHV-w1w4glB7ecPjc6aQXlbX-UplC6I,2642
+mwx/py/filling.py,sha256=KaHooM32hrGGgqw75Cbt8lAvACwC6RXadob9LGgNnEc,16806
+mwx/py/frame_listview.py,sha256=T-2xSv_D2bk9fJ64aiSEe1rJRTeqaIpLVAYEUXW5vz8,10110
+mwx/py/line_profile.py,sha256=WJB5z7F53yg4dII2R36IFZvtkSOGWT669b1HmAAXSnQ,816
+mwxlib-0.93.7.dist-info/LICENSE,sha256=PGtRKCaTkmUDlBQwpptJAxJtdqxIUtAmdBsaT9nUVkA,1091
+mwxlib-0.93.7.dist-info/METADATA,sha256=7uW6eS4CsshRUk0W1Qf7o4qqcpVZNJjbA-KKyWW-zQc,1925
+mwxlib-0.93.7.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
+mwxlib-0.93.7.dist-info/top_level.txt,sha256=SI1Mh118AstnUFGPNq5aMNKiAnVNmZk1S9Ij-OwAEpY,4
+mwxlib-0.93.7.dist-info/RECORD,,

mwxlib-0.93.5.dist-info/RECORD

@@ -1,23 +0,0 @@
-mwx/__init__.py,sha256=5B4YSOuijG1Uo5-FLtLHGB52Cp_F4vnN--4wGPBx7do,2398
-mwx/bookshelf.py,sha256=FrissUYdGXLABOzJmMaQU6GXvu6n_9DVW3d5wGwQzjM,6613
-mwx/controls.py,sha256=9C1sXBwQl-oZtCb3rXoT-doW48Sv1J359toNq9ScF9M,47173
-mwx/framework.py,sha256=SeFOWD9YzxgP5WlkYFeONWj9EoToH7IQ5LmyJbECqMw,73951
-mwx/graphman.py,sha256=0ppruAsbvUK6vUKnRc3-6gNW83V8XVp9Xr8mQrAaLTI,70267
-mwx/images.py,sha256=mrnUYH12I3XLVSZcEXlpVltX0XMxufbl2yRvDIQJZqc,49957
-mwx/matplot2.py,sha256=g902yipGCiO5SARTSafbE3c6-ZkQ-JbOkXSo7iltnDQ,32761
-mwx/matplot2g.py,sha256=d0g48JhBWvqqmHQKvuwEAQqa8ol9MX8UvEHU__hQiz4,65552
-mwx/matplot2lg.py,sha256=mCeT_OBwsq7OamW56WRg9YqyXe6WPJrjCfCVAzDvSlw,26343
-mwx/mgplt.py,sha256=ITzxA97yDwr_35BUk5OqnyskSuKVDbpf2AQCKY1jHTI,5671
-mwx/nutshell.py,sha256=5HErOObnOprMc6gYnXV7wY-ALnlwNpmhV8f0EyipYA4,135816
-mwx/utilus.py,sha256=IaB9RYPatprJXsulyrM2kEcVr-3ji0M88PUYBr9Nx6k,37040
-mwx/wxmon.py,sha256=Qk86VbuuW2rR46pqEYLur13G_aloWz5SVv6sib30YY0,12717
-mwx/wxpdb.py,sha256=2z3ZD9Oo1H-ONBHlaprkB9hrTmAI7o03sqO46ppEFE4,19129
-mwx/wxwil.py,sha256=JK1du4i1RVMbDLqN8jLRDSu_JhKEp4mhHVMElzo4yoE,5578
-mwx/wxwit.py,sha256=2gFHi-8nwWRh26RdvZ_AUP--8PDjJB8TUU72y2fBUWM,7557
-mwx/py/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-mwx/py/filling.py,sha256=KaHooM32hrGGgqw75Cbt8lAvACwC6RXadob9LGgNnEc,16806
-mwxlib-0.93.5.dist-info/LICENSE,sha256=PGtRKCaTkmUDlBQwpptJAxJtdqxIUtAmdBsaT9nUVkA,1091
-mwxlib-0.93.5.dist-info/METADATA,sha256=-z6_xKYj1N7D9gfLX6jJ6Yn1fPXwjnMvJO1utGox9Cc,1925
-mwxlib-0.93.5.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
-mwxlib-0.93.5.dist-info/top_level.txt,sha256=SI1Mh118AstnUFGPNq5aMNKiAnVNmZk1S9Ij-OwAEpY,4
-mwxlib-0.93.5.dist-info/RECORD,,