wxpythonとvtkでビデオプレーヤー作ってみる。その３。vtkImageViewerでjpeg表示。

目次

• その０。wxpythonのウィンドウ内部にvtkを表示するパネルを作ってみた
• その１。C++のJpeg表示サンプルをPythonに翻訳。
• その２。numpyアレイで用意した２Dテクスチャを四角ポリゴンに貼る。
• その３。vtkImageViewerでjpeg表示。
• その４。vtkImageViewerでnumpyアレイ表示。
• その５。完成。opencvとvtkImageViewerで読み込み・表示。

前置き

vtk使い始めなので、とりあえずなんか作ってみるという意味でビデオプレーヤーを作っていく企画の第３弾です。

vtkImageViewerを使ってみる

前回ちろっと触れましたが、vtkImageViewerというハイレベルなクラスが用意されているので、前回やったようにローレベルなポリゴンの設定とかはすっ飛ばすことが可能だった模様です。というわけで今回はvtkImageViewerをつかってやり直してみました。

画像サイズに合わせてウィンドウのサイズをリサイズしたかったのですが、vtkにはjpegReaderやtiffReadergが用意されています。

imageData = jpegReader.GetOutput()
w,h,z = imageData.GetDimensions()

としてからimageDataの下のメソッドでいろいろと画像の情報を取ってこれるようです。ｖｔｋは中身C++なので速いし、Pythonで実装されている部分が多いPILやpillowなんかの代わりとしても強力そうですね。

今回のコード

例によってメーリングリストにあった例を適当にいじってwxで使えるようにした程度です。

wx.Frame.__init__(self, None, title='TextureMapQuad', size=(w+40,h+100))

前回もそうしましたが、フレームを画像サイズより大きめにした上でpanelを置かないと、どうも後からリサイズしてもpanelが小さいままという問題があったためにハッキッシュなトリックをしています。すでに画像サイズにしたpanelをプロポーション１でEXPANDオプションつけてsizerに追加して、あとからFrameのFitを呼んでいるので、これでいいはずと思うんだけど。wxのsizerは時折わけわからない動作をするのであまり綺麗にデバックしないように心がけていたりする今日このごろ。追記）あほな単純ミスでした。panelをsizerに置いたつもりがvtkのrendererのrwiを置いてしまっているのでいけなかっただけでした。コードも修正。

import vtk
from vtk.wx.wxVTKRenderWindow import wxVTKRenderWindow
import wx


class Test(wx.Frame):

    def __init__ (self, fp):

        w,h,z,render = self.SetupRender(fp)

        wx.Frame.__init__(self, None, title='vtkImageViewer')
        canvas = wx.Panel(self, size=(w,h))
        rwi = wxVTKRenderWindow(canvas, -1)
        rwi.Enable(True)
        rwi.GetRenderWindow().AddRenderer(render)

        sizer = wx.BoxSizer(wx.VERTICAL)
        sizer.Add(canvas, 1, wx.EXPAND)
        self.SetSizerAndFit(sizer)
        self.Layout()
        self.Show()

    @staticmethod
    def SetupRender(fp):

        jpegReader = vtk.vtkJPEGReader()
        jpegReader.SetFileName( fp )
        jpegReader.Update()
        imageData = jpegReader.GetOutput()
        w,h,z = imageData.GetDimensions()

        imageviewer = vtk.vtkImageViewer()
        imageviewer.SetInputConnection( jpegReader.GetOutputPort() )
        imageviewer.SetColorWindow( 255 );
        imageviewer.SetColorLevel ( 128 );

        renderer = vtk.vtkRenderer()
        renderer.AddActor(imageviewer.GetActor2D())

        return w,h,z,renderer


if __name__ == '__main__':
    app = wx.App(0)
    frame = Test( "C:\Users\Public\Pictures\Sample Pictures\Penguins.jpg" )
    app.MainLoop()