pyqtgraphにWavファイルの波形を表示したい。

Question

Wavファイルはwave.open(ファイル名)で読み込みます。

pyqtgraphに表示するためにはどのような方法が考えられますでしょうか。

# -*- coding: utf-8 -*-
import sys
import numpy as np
from scipy import signal
from scipy.io.wavfile import write
from PySide2.QtWidgets import *
from PySide2.QtCore import *
import winsound as ws
import wave
import pyqtgraph as pg
#---------------------------------------------------
# Window
#---------------------------------------------------
class Gui(QDialog):
    def __init__(self, parent=None):
        super(Gui, self).__init__(parent)
        self.design()

    def design(self):
        # button
        self.button = QPushButton('Push')
        # Graph
        self.graph = pg.PlotWidget()
        self.graph.setMinimumSize(500, 250)
        self.graph.setMaximumSize(500, 250)
        # layout
        layout = QVBoxLayout(self)
        layout.addWidget(self.graph)
        layout.addWidget(self.button)
        # Connect
        self.button.clicked.connect(self.setWave)
    
    def setWave(self):
        pass

if __name__ == '__main__':
    app = QApplication(sys.argv)
    g = Gui()
    g.show()
    sys.exit(app.exec_())

Answer 1

以下のような感じで描画できます。

    def setWave(self):
        wf = wave.open('～.wav', 'r')
        buf = wf.readframes(wf.getnframes())
        data = np.frombuffer(buf, dtype="int16")
        self.graph.plot(data)

Comments

[OKA-G]

ありがとうございます。
すごく助かりました。
もう一つだけ質問なのですが、setWave()を実行するたびにグラフの中身をリセットする方法ってありますか？
何度実行しても常に一つだけ表示させたいです。

[can110]

分かりません。
とりあえずひととおりhttps://pyqtgraph.readthedocs.io/en/latest/index.htmlに目を通して目的のものがないか探してみてください。

[OKA-G]

分かりました。
ご丁寧にありがとうございます。

[OKA-G]

self.graph.clear()で出来ました！