[[FrontPage/Python/PySide]] #! coding:utf-8 """ pyside-painter-skelton.py pysideでグラフィックを描画するためのスケルトンコード Created by 0160929 on 2015/11/19 7:28 """ __version__ = '0.3' import sys import os # PySide系モジュール from PySide.QtGui import * from PySide.QtCore import * # 演算系モジュール import numpy as np # logを保存 from datetime import datetime import csv def resetlog(file): if os.path.exists(file): os.remove(file) # ログファイルを削除 resetlog('log.cav') def debuglog(s=None, data=None): d = datetime.now().isoformat() with open('log.csv', 'a') as f: writer = csv.writer(f, lineterminator='\n') if not s is None: writer.writerow([d, s]) print d, s if not data is None: writer.writerow([d, data]) print d, data return # 描画用PySideクラス class GameWindow(QWidget): # -- 定数(画面サイズ) ----------------- SCREEN_HEIGHT = 500 SCREEN_WIDTH = 700 MARGIN_HEIGHT = 50 MARGIN_WIDTH = 50 STAGE_HEIGHT = SCREEN_HEIGHT - 2 * MARGIN_HEIGHT STAGE_WIDTH = SCREEN_WIDTH - 2 * MARGIN_WIDTH # -- 定数(タイマー ms) ------------------- INTERVAL_TIME = 1 # -- 定数(その他) --------------------- def __init__(self, parent=None): QWidget.__init__(self, parent) self.resize(self.SCREEN_WIDTH, self.SCREEN_HEIGHT) # -- 定数 ---------------------------- self.step = 0 # -- 画面バッファ -------------------- self.pixmap = QPixmap(self.size()) # -- 解析用オブジェクト -------------- # -- 操作用変数 ---------------------- # -- 初期画面の準備 ------------------ self.refreshPixmap() # 画面バッファの初期化 painter = QPainter(self.pixmap) self.drawGrid(painter) # グリッドの表示 self.update() # メインループの準備と開始 # ------------------------- if False: self.timer = QTimer() self.timer.timeout.connect(self.mainloop) self.timer.start(self.INTERVAL_TIME) # ************************************************************* # # メインループ # ************************************************************* # def mainloop(self): """ アニメーションのメインループ アルゴリズムの時間更新等はここで行う """ self.step += 1 # -- アルゴリズム処理----------- # -- 描画 ---------------------- painter = QPainter(self.pixmap) self.drawGrid(painter) self.drawGeoPoints(painter) # -- 画面更新 (AppのタイミングでpaintEventが呼ばれる) --- self.update() def paintEvent(self, *args, **kwargs): # -- おまじない --------------------------- painter = QStylePainter(self) # QPainterを生成 painter.setRenderHint(QPainter.Antialiasing, True) # アンチエイリアス painter.drawPixmap(0, 0, self.pixmap) # QPainterでバッファに準備したデータを描画 # 描画用QPenのデフォルト pen_default = QPen() pen_default.setColor(Qt.black) pen_default.setWidth(2) painter.setPen(pen_default) brush_default = QBrush(Qt.red, Qt.NoBrush) painter.setBrush(brush_default) # -- アクセス用定数 ----------------------- x, y = 0, 1 # -- 車両の位置情報を取得 ----------------- pos = self.locateXY([self.STAGE_WIDTH / 2., self.STAGE_HEIGHT / 2.]) # -- 車両位置のプロット ------------------- pen = QPen(Qt.red, 5) painter.setPen(pen) painter.drawPoint(pos[x], pos[y]) # -- 車両位置に 円を描画 ----------------------------- pen = QPen(Qt.black, 1, Qt.DotLine, Qt.RoundCap, Qt.RoundJoin) painter.setPen(pen) painter.setBrush(QBrush(QColor(0, 255, 0, 175))) circle_size = 50 painter.drawEllipse(pos[x] - circle_size / 2, pos[y] - circle_size / 2, circle_size, circle_size) # -- 車両位置に四角を描画 ----------------------------- pen = QPen(Qt.black, 1, Qt.SolidLine, Qt.RoundCap, Qt.RoundJoin) painter.setPen(pen) painter.setBrush(QBrush(QColor(0, 255, 0, 255), Qt.NoBrush)) painter.drawRect(pos[x] - circle_size / 2, pos[y] - circle_size / 2, circle_size, circle_size) # qpoints = [QPoint(*lv) for lv in v] # polygon = QPolygon(qpoints) # painter.drawPolygon(polygon) # -- 進行方向(速度ベクトル)を描画 --------- pen = QPen(Qt.gray, 1, Qt.SolidLine, Qt.RoundCap, Qt.RoundJoin) painter.setPen(pen) painter.drawLine(pos[x], pos[y], (pos[x] + 50), (pos[y] + 50)) # -- 棒グラフを表示 ------------------------- r = QRect(0, 0, 20, 30) r.setHeight(200) p0 = self.locateXY([20, 20], 'qp') r.moveBottomLeft(p0) painter.setBrush(QBrush(Qt.green, Qt.Dense4Pattern)) painter.drawRect(r) # ************************************************************* # # 描画系補助関数 # ************************************************************* # def drawDebugLog(self, painter): pass def drawGrid(self, painter): """ マップを表示する関数 """ # -- 定数 --------------------------------------- Nx = 10 Ny = 10 Lx = self.STAGE_WIDTH Ly = self.STAGE_HEIGHT # -- 線の色と種類の選定 -------------------------- painter.setPen(QPen(Qt.black, 0.5)) painter.setBrush(QBrush(QColor(200, 200, 200, 150), Qt.Dense1Pattern)) # -- 横線を描画 ---------------------------------- for xi in range(Nx + 1): p1 = self.locateXY([xi * Lx / Nx, 0], 'qp') p2 = self.locateXY([xi * Lx / Nx, Ly], 'qp') painter.drawLine(p1, p2) # -- 縦線を描画 ---------------------------------- for yi in range(Ny + 1): p1 = self.locateXY([0, yi * Ly / Ny], 'qp') p2 = self.locateXY([Lx, yi * Ly / Ny], 'qp') painter.drawLine(p1, p2) def locateXY(self, pos, type='np'): """ オブジェクトのローカル座標を、スクリーン上の座標へ変換 """ _pos = np.asarray(pos) local_x, local_y = _pos.copy() srn_x = local_x + self.MARGIN_WIDTH srn_y = (self.SCREEN_HEIGHT - local_y - self.MARGIN_HEIGHT) if type is 'np': return np.asarray([srn_x, srn_y]) elif type is 'qp': return QPoint(srn_x, srn_y) elif type is 'qpF': return QPointF(srn_x, srn_y) elif type is 'list': return [srn_x, srn_y] return 'ERROR:: check type option' def locateVec(self, vec, type='np'): """ オブジェクトのベクトルを、スクリーン上の座標空間へ変換 (※yを反転させるだけ) """ srn_vec = [vec[0], -1 * vec[1]] if type is 'np': return np.asarray(srn_vec) elif type is 'qp': return QPoint(*srn_vec) elif type is 'qpF': return QPointF(*srn_vec) elif type is 'list': return srn_vec return 'ERROR:: check type option' # ************************************************************* # # その他Qt関連補助関数 # ************************************************************* # def refreshPixmap(self): """ 画面バッファの初期化関数 """ # 画面バッファの初期化 self.pixmap = QPixmap(self.size()) # 画面を塗りつぶし (おまじない) self.pixmap.fill(self, 0, 0) self.pixmap.fill(Qt.white) # ぺインターの生成 (おまじない) painter = QPainter(self.pixmap) # ぺインターによる初期化 (おまじない) painter.initFrom(self) pass def sizeHint(self): return QSize(self.SCREEN_WIDTH, self.SCREEN_HEIGHT) def keyPressEvent(self, event): e = event.key() if e == Qt.Key_Up: pass elif e == Qt.Key_Down: pass elif e == Qt.Key_Left: pass elif e == Qt.Key_Right: pass elif e == Qt.Key_Plus: pass elif e == Qt.Key_Minus: pass elif e == Qt.Key_Q: self.close() else: pass print 'Presskey', e self.update() # ******************************************************* # main関数 # ******************************************************* def mainGUI(): app = QApplication(sys.argv) win = GameWindow() win.show() sys.exit(app.exec_()) if __name__ == "__main__": mainGUI() #ls()