유희평

Heepyung

# -*- coding: utf-8 -*-
# Form implementation generated from reading ui file 'Aditor_Ui_1.1.ui'
#
# Created by: PyQt5 UI code generator 5.15.1
#
# WARNING: Any manual changes made to this file will be lost when pyuic5 is
# run again. Do not edit this file unless you know what you are doing.
from PyQt5 import QtCore, QtGui, QtWidgets
from PyQt5.QtWidgets import QApplication, QMainWindow, QMainWindow, QFileDialog, QVBoxLayout, QInputDialog, QMessageBox, QAction
import sys
import cv2
from PyQt5.QtCore import QThread
import numpy as np
import pandas as pd
from time import sleep
from threading import Timer, Thread, Event
from matplotlib.figure import Figure
from matplotlib.backends.backend_qt5agg import FigureCanvasQTAgg as FigureCanvas
class perpetualTimer():#정해진 시간마다(주기) function수행
def __init__(self, t, hFunction):
self.t = t
self.hFunction = hFunction
self.thread = Timer(self.t, self.handle_function)
print("timer initate")
def handle_function(self):
self.hFunction()
self.thread = Timer(self.t, self.handle_function)
self.thread.start()
def start(self):
self.thread.start()
def cancel(self):
self.thread.cancel()
class workerThread(QThread):
updatedM = QtCore.pyqtSignal(int)
def __init__(self, mw):
self.mw = mw
#print("thread initialized")
QThread.__init__(self)
def __del__(self):
self.wait()
def run(self):
# print("run")
itr = 0
QApplication.processEvents()
while self.mw.isRun:
itr += 1
# print("waiting")
#print("frameID"+str(self.mw.frameID))
#print("endframe"+str(self.mw.endframe))
if self.mw.isthreadActive and self.mw.isbusy == False and self.mw.frameID != self.mw.cap.get(
cv2.CAP_PROP_POS_FRAMES):
# print(itr)\
#print("in 재생중")
if np.abs(self.mw.frameID - self.mw.cap.get(cv2.CAP_PROP_POS_FRAMES)) > 1:
self.mw.cap.set(cv2.CAP_PROP_POS_FRAMES, self.mw.frameID)
if self.mw.frameID >= self.mw.endframe-1: #끝에서 멈춰주는 기능
if not self.mw.timer is None:
print("end")
print(self.mw.frameID)
print(self.mw.horizontalSlider.value())
self.mw.timer.cancel()
self.mw.startButton.setEnabled(True)
self.mw.pauseButton.setEnabled(False)
self.mw.isthreadActive = False
self.mw.frameID = self.mw.endframe-1
if self.mw.timer is None:
self.mw.frameID += 1
self.mw.isbusy = True
sf = self.mw.scaleFactor
ret, image = self.mw.cap.read()
self.mw.limg = image
if sf <= 0:
self.mw.isbusy = False
continue
if ret == False:
self.mw.isthreadActive = False
self.mw.isbusy = False
continue
nchannel = image.shape[2]
limg2 = cv2.cvtColor(image, cv2.COLOR_RGB2BGR)
timg = cv2.resize(
limg2, (int(sf * limg2.shape[1]), int(sf * limg2.shape[0])))
limage = QtGui.QImage(timg.data, timg.shape[1], timg.shape[0], nchannel * timg.shape[1],
QtGui.QImage.Format_RGB888)
if self.mw.resizegoing == False:
self.mw.label.setPixmap(QtGui.QPixmap(limage))
# self.label.setPixmap(QtGui.QPixmap(limage))
if not self.mw.sliderbusy and not self.mw.sliderbusy2:
self.updatedM.emit(self.mw.frameID)
QApplication.processEvents()
self.mw.isbusy = False
else:
#print("in 멈춤!!!")
if self.mw.isthreadActive and self.mw.timer is None:
self.mw.frameID += 1
sleep(1.0 / 50)
class Ui_MainWindow(object):
def setupUi(self, MainWindow):
MainWindow.setObjectName("MainWindow")
MainWindow.resize(801, 773)
self.centralwidget = QtWidgets.QWidget(MainWindow)
self.centralwidget.setMinimumSize(QtCore.QSize(801, 732))
self.centralwidget.setObjectName("centralwidget")
self.horizontalLayoutWidget = QtWidgets.QWidget(self.centralwidget)
self.horizontalLayoutWidget.setGeometry(QtCore.QRect(10, 10, 781, 41))
self.horizontalLayoutWidget.setObjectName("horizontalLayoutWidget")
self.horizontalLayout_4b = QtWidgets.QHBoxLayout(
self.horizontalLayoutWidget)
self.horizontalLayout_4b.setContentsMargins(0, 0, 0, 0)
self.horizontalLayout_4b.setObjectName("horizontalLayout_4b")
self.openVideoButton = QtWidgets.QPushButton(
self.horizontalLayoutWidget)
self.openVideoButton.setToolTip("편집할 비디오를 선택하고 엽니다.")
self.openVideoButton.setObjectName("openVideoButton")
self.horizontalLayout_4b.addWidget(self.openVideoButton)
self.openDataButton = QtWidgets.QPushButton(self.horizontalLayoutWidget)
self.openDataButton.setToolTip(
"편집할 비디오와 관련된 데이터 파일을 선택하고 엽니다. csv 파일을 받습니다. 데이터 파일로 승률데이터(스포츠 경기 영상에 해당할 경우), 유튜브 댓글 데이터, 임의로 작성한 수치 데이터를 사용 가능합니다. \n*임의로 작성한 데이터는 시계열 데이터여야 합니다.\n*데이터가 필수적이지는 않습니다. 데이터를 넣지 않아도 두가지 기능(Check on Timeline, Export)을 이용할 수 있습니다.")
self.openDataButton.setObjectName("openDataButton")
self.horizontalLayout_4b.addWidget(self.openDataButton)
self.checkOnTimelineButton = QtWidgets.QPushButton(
self.horizontalLayoutWidget)
self.checkOnTimelineButton.setToolTip(
"입력된 영상의 추천 편집 지점을 Timeline에 표시해줍니다.")
self.checkOnTimelineButton.setObjectName("checkOnTimelineButton")
self.horizontalLayout_4b.addWidget(self.checkOnTimelineButton)
self.exportButton = QtWidgets.QPushButton(self.horizontalLayoutWidget)
self.exportButton.setToolTip(
"추천 편집 지점(Timeline에 체크된 부분)을 모아 하나의 영상으로 내보냅니다.")
self.exportButton.setObjectName("exportButton")
self.horizontalLayout_4b.addWidget(self.exportButton)
self.verticalLayoutWidget = QtWidgets.QWidget(self.centralwidget)
self.verticalLayoutWidget.setGeometry(QtCore.QRect(10, 60, 781, 411))
self.verticalLayoutWidget.setObjectName("verticalLayoutWidget")
self.verticalLayout_video = QtWidgets.QVBoxLayout(
self.verticalLayoutWidget)
self.verticalLayout_video.setContentsMargins(0, 0, 0, 0)
self.verticalLayout_video.setObjectName("verticalLayout_video")
self.label = QtWidgets.QLabel(self.verticalLayoutWidget)
self.label.setAlignment(QtCore.Qt.AlignCenter)
self.label.setObjectName("label")
self.verticalLayout_video.addWidget(self.label)
self.verticalLayoutWidget_2 = QtWidgets.QWidget(self.centralwidget)
self.verticalLayoutWidget_2.setGeometry(
QtCore.QRect(10, 530, 781, 131))
self.verticalLayoutWidget_2.setObjectName("verticalLayoutWidget_2")
self.verticalLayout_bottom = QtWidgets.QVBoxLayout(
self.verticalLayoutWidget_2)
self.verticalLayout_bottom.setContentsMargins(0, 0, 0, 0)
self.verticalLayout_bottom.setObjectName("verticalLayout_bottom")
self.tabWidget = QtWidgets.QTabWidget(self.verticalLayoutWidget_2)
self.tabWidget.setObjectName("tabWidget")
self.tab = QtWidgets.QWidget()
self.tab.setToolTip("추천 편집 지점을 나타내는 타임라인입니다.")
self.tab.setObjectName("tab")
self.label_2 = QtWidgets.QLabel(self.tab)
self.label_2.setGeometry(QtCore.QRect(0, 0, 771, 101))
self.label_2.setAlignment(QtCore.Qt.AlignCenter)
self.label_2.setObjectName("label_2")
self.tabWidget.addTab(self.tab, "")
self.tab_2 = QtWidgets.QWidget()
self.tab_2.setToolTip(
"데이터 파일과 영상에서 추출된 데이터를 바탕으로 하이라이트인 정도를 평가하는 그래프를 보여줍니다.")
self.tab_2.setObjectName("tab_2")
self.label_3 = QtWidgets.QLabel(self.tab_2)
self.label_3.setGeometry(QtCore.QRect(0, 0, 771, 101))
self.label_3.setAlignment(QtCore.Qt.AlignCenter)
self.label_3.setObjectName("label_3")
self.tabWidget.addTab(self.tab_2, "")
self.verticalLayout_bottom.addWidget(self.tabWidget)
self.progressBar = QtWidgets.QProgressBar(self.centralwidget)
self.progressBar.setGeometry(QtCore.QRect(10, 700, 781, 20))
self.progressBar.setProperty("value", 24)
self.progressBar.setToolTip(
"두가지 기능(Check on Timeline, Export)의 진행 상황을 나타냅니다.")
self.progressBar.setObjectName("progressBar")
self.horizontalLayoutWidget_2 = QtWidgets.QWidget(self.centralwidget)
self.horizontalLayoutWidget_2.setGeometry(
QtCore.QRect(10, 480, 781, 44))
self.horizontalLayoutWidget_2.setObjectName("horizontalLayoutWidget_2")
self.horizontalLayout_2b = QtWidgets.QHBoxLayout(
self.horizontalLayoutWidget_2)
self.horizontalLayout_2b.setContentsMargins(0, 0, 0, 0)
self.horizontalLayout_2b.setObjectName("horizontalLayout_2b")
self.startButton = QtWidgets.QPushButton(self.horizontalLayoutWidget_2)
self.startButton.setToolTip("동영상을 재생하는 버튼입니다.")
self.startButton.setObjectName("startButton")
self.horizontalLayout_2b.addWidget(self.startButton)
self.pauseButton = QtWidgets.QPushButton(self.horizontalLayoutWidget_2)
self.pauseButton.setToolTip("동영상을 일시정지하는 버튼입니다.")
self.pauseButton.setObjectName("pauseButton")
self.horizontalLayout_2b.addWidget(self.pauseButton)
self.horizontalSlider = QtWidgets.QSlider(self.centralwidget)
self.horizontalSlider.setGeometry(QtCore.QRect(18, 670, 762, 22))
self.horizontalSlider.setOrientation(QtCore.Qt.Horizontal)
self.horizontalSlider.setToolTip(
"동영상 재생 지점을 선택할 수 있습니다. Timeline에 체크된 부분을 확인할 때 쓰일 것입니다")
self.horizontalSlider.setObjectName("horizontalSlider")
MainWindow.setCentralWidget(self.centralwidget)
self.menubar = QtWidgets.QMenuBar(MainWindow)
self.menubar.setGeometry(QtCore.QRect(0, 0, 801, 21))
self.menubar.setObjectName("menubar")
self.menuFile = QtWidgets.QMenu(self.menubar)
self.menuFile.setToolTip(
"New Project: 현재 올려진 동영상과 타임라인을 지우고 초기화 시킵니다.\nSettings: 기타 설정을 변경할 수 있습니다. Check on Timeline의 민감도 설정을 할 수 있습니다. \nExit: 프로그램을 종료합니다.")
self.menuFile.setObjectName("menuFile")
self.menuTools = QtWidgets.QMenu(self.menubar)
self.menuTools.setToolTip(
"Add YouTube Link: 편집할 영상이 유튜브에 업로드 되어 있다면, 그 영상의 URL을 입력하여 댓글 데이터를 자동으로 가져옵니다.")
self.menuTools.setObjectName("menuTools")
self.menuHelp = QtWidgets.QMenu(self.menubar)
self.menuHelp.setToolTip("프로그램 사용 가이드와 제작정보")
self.menuHelp.setObjectName("menuHelp")
MainWindow.setMenuBar(self.menubar)
self.statusbar = QtWidgets.QStatusBar(MainWindow)
self.statusbar.setObjectName("statusbar")
MainWindow.setStatusBar(self.statusbar)
self.actionnew_project = QtWidgets.QAction(MainWindow)
self.actionnew_project.setObjectName("actionnew_project")
self.actionSetting = QtWidgets.QAction(MainWindow)
self.actionSetting.setObjectName("actionSetting")
self.actionExit = QtWidgets.QAction(MainWindow)
self.actionExit.setObjectName("actionExit")
self.actionHow_to_use = QtWidgets.QAction(MainWindow)
self.actionHow_to_use.setObjectName("actionHow_to_use")
self.actionAbout = QtWidgets.QAction(MainWindow)
self.actionAbout.setObjectName("actionAbout")
self.actionData = QtWidgets.QAction(MainWindow)
self.actionData.setObjectName("actionData")
self.menuFile.addAction(self.actionnew_project)
self.menuFile.addAction(self.actionSetting)
self.menuFile.addAction(self.actionExit)
self.menuTools.addAction(self.actionData)
self.menuHelp.addAction(self.actionHow_to_use)
self.menuHelp.addAction(self.actionAbout)
self.menubar.addAction(self.menuFile.menuAction())
self.menubar.addAction(self.menuTools.menuAction())
self.menubar.addAction(self.menuHelp.menuAction())
self.retranslateUi(MainWindow)
self.tabWidget.setCurrentIndex(1)
QtCore.QMetaObject.connectSlotsByName(MainWindow)
def retranslateUi(self, MainWindow):
_translate = QtCore.QCoreApplication.translate
MainWindow.setWindowTitle(_translate("MainWindow", "Auto Video Editor"))
self.openVideoButton.setText(_translate("MainWindow", "Open Video"))
self.openDataButton.setText(_translate("MainWindow", "Open Data File"))
self.checkOnTimelineButton.setText(
_translate("MainWindow", "Check on Timeline"))
self.exportButton.setText(_translate("MainWindow", "Export"))
self.label.setText(_translate("MainWindow", "video"))
self.label_2.setText(_translate(
"MainWindow", "Timeline: 추천 편집 지점을 나타내는 타임라인입니다."))
self.tabWidget.setTabText(self.tabWidget.indexOf(
self.tab), _translate("MainWindow", "Timeline"))
self.label_3.setText(_translate(
"MainWindow", "Data: 데이터 파일과 영상에서 추출된 데이터를 바탕으로 하이라이트인 정도를 평가하는 그래프를 보여줍니다."))
self.tabWidget.setTabText(self.tabWidget.indexOf(
self.tab_2), _translate("MainWindow", "Data"))
self.startButton.setText(_translate("MainWindow", "Start"))
self.pauseButton.setText(_translate("MainWindow", "Pause"))
self.menuFile.setTitle(_translate("MainWindow", "File"))
self.menuTools.setTitle(_translate("MainWindow", "Tools"))
self.menuHelp.setTitle(_translate("MainWindow", "Help"))
self.actionnew_project.setText(_translate("MainWindow", "New Project"))
self.actionSetting.setText(_translate("MainWindow", "Settings"))
self.actionExit.setText(_translate("MainWindow", "Exit"))
self.actionHow_to_use.setText(_translate("MainWindow", "How to use"))
self.actionAbout.setText(_translate("MainWindow", "About"))
self.actionData.setText(_translate("MainWindow", "XML action"))
class WindowClass(QMainWindow, Ui_MainWindow): # Ui_MainWindow를 받음
resized = QtCore.pyqtSignal()
def __init__(self):
super().__init__()
self.setupUi(self)
self.openVideoButton.clicked.connect(self.OVBtnClicked)
self.startButton.clicked.connect(self.startButtonClicked)
self.pauseButton.clicked.connect(self.pauseButtonclicked)
self.horizontalSlider.sliderPressed.connect(self.horizontalSliderPressed)
self.horizontalSlider.sliderReleased.connect(self.horizontalSliderReleased)
self.horizontalSlider.valueChanged.connect(self.slider_value_changed)
self.openDataButton.clicked.connect(self.ODBtnClicked)
self.checkOnTimelineButton.clicked.connect(self.CTLBtnCLicked)
self.exportButton.clicked.connect(self.EPBtnClicked)
action = QAction("&Add YouTube link", self)
self.menuTools.addAction(action)
action.triggered.connect(self.AYBtnClicked)
#버튼 설정
self.startButton.setEnabled(False)
self.pauseButton.setEnabled(False)
self.horizontalSlider.setEnabled(False)
self.openDataButton.setEnabled(False)
self.checkOnTimelineButton.setEnabled(False)
self.exportButton.setEnabled(False)
self.progressBar.setValue(0)
#프레임 정보
self.frameID = 0 #현재 프레임
self.stframe = 0 #시작 프레임
self.endframe = 0 #끝 프레임
self.fps = 0 #초당 프레임
#영상 크기 조절
self.scaleFactor = 0.5
# thread
self.wthread = workerThread(self)
self.wthread.updatedM.connect(self.horizontalSliderSet)
self.wthread.start()
#기타 멤버 변수들
self.limg = np.zeros((1, 1, 1))
self.isbusy = 0
self.frameHeight = 1
self.frameWidth = 1
self.isRun = True
self.resizegoing = False
self.sliderbusy = False
self.sliderbusy2 = False
self.linebusy = False
self.cap = None
self.timer = None
self.isvideo = False
self.videoFileName = ""
self.isthreadActive = False
#데이터
self.clip = 0
self.max_slice = 5
self.threshold = 0.4
self.inputData = pd.DataFrame()
self.audioData = pd.DataFrame()
self.commentsData = 0
self.highlightData = pd.DataFrame()
self.isdata = False
self.iscomments = False
self.videoID = ""
#그리기
layout1 = QVBoxLayout()
layout2 = QVBoxLayout()
self.figure1 = Figure()
self.figure2 = Figure()
self.label_2.canvas = FigureCanvas(self.figure1)
self.label_3.canvas = FigureCanvas(self.figure2)
layout1.addWidget(self.label_2.canvas)
layout2.addWidget(self.label_3.canvas)
self.label_2.setLayout(layout1)
self.label_3.setLayout(layout2)
self.ax1 = self.figure1.add_subplot(1, 1, 1)
self.ax2 = self.figure2.add_subplot(1, 1, 1)
self.ax1.get_yaxis().set_visible(False)
self.ax1.get_xaxis().set_visible(True)
self.figure1.subplots_adjust(left=0.001, right=0.999, top=1.0, bottom=0.1)
self.figure2.subplots_adjust(left=0.001, right=0.999, top=1.0, bottom=0.1)
def OVBtnClicked(self):
self.isthreadActive = False
fileName = QFileDialog.getOpenFileName(
None, caption="Select Video File", directory=QtCore.QDir.currentPath())
if len(fileName[0]) > 0:
self.videoFileName = fileName[0]
self.cap = cv2.VideoCapture(fileName[0]) # 비디오 캡쳐하는 부분
self.isvideo = True
else:
return
length = int(self.cap.get(cv2.CAP_PROP_FRAME_COUNT))
print("length: " + str(length))
self.fps = self.cap.get(cv2.CAP_PROP_FPS)
print("fps: " + str(self.fps))
self.stframe = 0
self.endframe = length
#self.stframe = int(self.data.values[1, 1] * self.fps)
#print("stframe: " + str(self.stframe))
#self.endframe = int(self.data.values[-1, 1] * self.fps)
#print("endframe: " + str(self.endframe))
#self.ui.horizontalSlider.setMaximum(self.endframe - self.stframe)
#self.cap.set(1, self.stframe)
#self.cap.set(1, 200)
ret, frame = self.cap.read()
self.drawmin = 1
self.frameID = self.stframe
self.limg = frame
self.frameHeight = frame.shape[0]
self.frameWidth = frame.shape[1]
nchannel = frame.shape[2]
limg2 = cv2.cvtColor(frame, cv2.COLOR_RGB2BGR)
timg = cv2.resize(limg2, (int(
self.scaleFactor * limg2.shape[1]), int(self.scaleFactor * limg2.shape[0])))
limage = QtGui.QImage(timg.data, timg.shape[1], timg.shape[0], nchannel * timg.shape[1],
QtGui.QImage.Format_RGB888)
self.label.setPixmap(QtGui.QPixmap(limage))
self.startButton.setEnabled(True)
self.horizontalSlider.setEnabled(True)
self.openDataButton.setEnabled(True)
self.checkOnTimelineButton.setEnabled(True)
self.exportButton.setEnabled(False)
self.horizontalSlider.setMaximum(self.endframe - self.stframe)
self.horizontalSlider.setValue(self.stframe)
self.isdata = False
self.ax1.clear()
self.ax2.clear()
def updateFrame(self):
self.frameID += 1
def startButtonClicked(self):
if self.isthreadActive:
return
self.startButton.setEnabled(False)
self.timer = perpetualTimer(1.0 / self.fps, self.updateFrame)
self.timer.start()
self.pauseButton.setEnabled(True)
self.isthreadActive = True
print("thread activatied")
def pauseButtonclicked(self):
if not self.isthreadActive:
return
self.startButton.setEnabled(True)
self.pauseButton.setEnabled(False)
if not self.timer is None:
self.timer.cancel()
self.isthreadActive = False
def videoSec(self):
return int(self.frameID / self.fps)
def horizontalSliderSet(self, cnt): #슬라이더 위치 업데이트
if cnt + 1 - self.stframe > self.horizontalSlider.maximum() or self.sliderbusy or self.resizegoing:
return
self.sliderbusy2 = True
self.horizontalSlider.setValue(cnt + 1 - self.stframe)
tsec = cnt / self.fps
tmin = int(tsec / 60)
ttsec = int(tsec - 60 * tmin)
ksec = tsec - 60 * tmin - ttsec
self.statusbar.showMessage(
"Time: " + str(tmin).zfill(2) + ":" + str(ttsec).zfill(2) + ":" + str(int(ksec * 100)))
self.sliderbusy2 = False
def horizontalSliderPressed(self):
self.sliderbusy = True
def slider_value_changed(self):
#print('slidervalue change')
self.horizontalSliderIncrease(0)
#if not self.isthreadActive:
# print("thread is not active")
# self.horizontalSliderIncrease(0)
def horizontalSliderIncrease(self, val):
if self.sliderbusy or self.resizegoing:
return
self.sliderbusy = True
# print(self.horizontalSlider.value())
# self.ui.horizontalSlider.setValue(self.ui.horizontalSlider.value()+val)
# print(self.frameID)
self.frameID = self.stframe + self.horizontalSlider.value() -1
# print(self.frameID)
# self.drawmin=1
if self.startButton.isEnabled():
self.cap.set(cv2.CAP_PROP_POS_FRAMES, self.frameID)
ret, frame = self.cap.read()
self.limg = frame
# self.on_zoomfit_clicked()
nchannel = frame.shape[2]
limg2 = cv2.cvtColor(frame, cv2.COLOR_RGB2BGR)
timg = cv2.resize(limg2, (int(self.scaleFactor * limg2.shape[1]), int(self.scaleFactor * limg2.shape[0])))
limage = QtGui.QImage(timg.data, timg.shape[1], timg.shape[0], nchannel * timg.shape[1],
QtGui.QImage.Format_RGB888)
self.label.setPixmap(QtGui.QPixmap(limage))
#self.lineSliderSet(self.frameID)
self.sliderbusy = False
def horizontalSliderReleased(self):
self.frameID = self.stframe + self.horizontalSlider.value()-1
print(self.frameID)
print(self.horizontalSlider.value())
self.drawmin = 1
if self.startButton.isEnabled():
self.cap.set(cv2.CAP_PROP_POS_FRAMES, self.frameID)
ret, frame = self.cap.read()
self.limg = frame
#self.on_zoomfit_clicked()
nchannel = frame.shape[2]
limg2 = cv2.cvtColor(frame, cv2.COLOR_RGB2BGR)
timg = cv2.resize(limg2, (int(self.scaleFactor * limg2.shape[1]), int(self.scaleFactor * limg2.shape[0])))
limage = QtGui.QImage(timg.data, timg.shape[1], timg.shape[0], nchannel * timg.shape[1],
QtGui.QImage.Format_RGB888)
self.label.setPixmap(QtGui.QPixmap(limage))
#self.lineSliderSet(self.frameID)
self.sliderbusy = False
def ODBtnClicked(self):
self.pauseButtonclicked()
fileName = QFileDialog.getOpenFileName(
None, caption="Select Data File(only csv)", directory=QtCore.QDir.currentPath())
if len(fileName[0]) > 0:
data = pd.read_csv(fileName[0])
self.inputData = data
self.isdata = True
#print(self.inputData)
return
def CTLBtnCLicked(self):
#데이터 가져오기
self.pauseButtonclicked()
choice = QMessageBox.question(self, '', '기본 설정값을 사용하겠습니까?', QMessageBox.Yes|QMessageBox.No)
if choice == QMessageBox.No:
input_max_slice, ok = QInputDialog.getInt(self, '단위 구간 설정', '하이라이트성을 평가할 구간의 길이를 설정하세요. (초)')
if ok:
if input_max_slice >= 0:
self.max_slice = input_max_slice
else:
return
input_threshold, ok = QInputDialog.getDouble(self, '임계값 설정', '하이라이트 포함 임계값을 설정하세요. (0~1 실수)')
if ok:
if 0 <= input_threshold <= 1:
self.threshold = input_threshold
else:
return
else:
self.max_slice = 5
self.threshold = 0.4
self.clip = Clip(self.videoFileName)
self.audioData = self.clip.getLoudSection(max_slice=self.max_slice, threshold=400)
self.drawData(max_slice=self.max_slice, threshold=self.threshold)
self.exportButton.setVisible(True)
self.exportButton.setEnabled(True)
return
#영상 내보내기
def EPBtnClicked(self):
#print(self.highlightData)
self.clip.concatVideo(self.highlightData, self.max_slice)
return
#타임라인 그리기
#def drawTimeline(self):
# return
#데이터 그리기
def decisionFunction(self, audioData, inputData = None):
data = pd.DataFrame()
#normalization(min-max scaling)
audioData["energy"] = (audioData["energy"] - audioData["energy"].min()) / (audioData["energy"].max() - audioData["energy"].min())
if inputData is None:
data = audioData
else:
self.inputData.iloc[:, 0] = (self.inputData.iloc[:, 0] - self.inputData.iloc[:, 0].min()) / (
self.inputData.iloc[:, 0].max() - self.inputData.iloc[:, 0].min())
self.inputData = self.inputData.reindex(self.inputData.index.repeat(int(60 / self.max_slice))).assign()
self.inputData = self.inputData.reset_index()
#print(self.inputData)
data = 0.9*audioData.iloc[:,0] + 0.1*self.inputData.iloc[:, 1]
data = data.to_frame()
data = data.iloc[:len(audioData), 0]
data = data.to_frame()
data["start"] = audioData["start"].tolist()
#result3 = pd.concat([df1, df2], axis=1)
#data.concat([data, df], axis=1)
return data
#timeline, data 그리기
def drawData(self, max_slice=10, threshold=0.4):
if self.isdata is False and self.isvideo is True:
data = self.decisionFunction(self.audioData) #그릴 데이터
elif self.isdata is True and self.isvideo is True:
data = self.decisionFunction(self.audioData, self.inputData) #그릴 데이터
else:
return
#ax2(Data)그리기
self.ax2.clear()
self.ax2.set_ylim([0, 1.1])
#print(data)
ndata = data.to_numpy()[:, 0]
print(ndata)
stretchedata = np.zeros(int(self.endframe/self.fps)+1) #영상의 실제 끝으로설정
#print(stretchedata[10])
#print("check")
for i in range(len(stretchedata)):
stretchedata[i] = ndata[int(i/max_slice)]
#print(stretchedata)
self.ax2.plot(stretchedata, '-c')
#self.ax2.hist(data.to_numpy()[:,0], 10, histtype='stepfilled')
self.ax2.set_xlim((self.stframe/self.fps), int(self.endframe/self.fps))
#print(data.to_numpy()[:,0])
#self.highlightData = data.to_numpy()[:,0]
self.ax2.plot([data.to_numpy()[0, 1], data.to_numpy()[0, 1]], [0, 1.1], 'k', linewidth=2.0)
#self.ax1.plot([self.data.values[1, 1], self.data.values[1, 1]], [0, 3.4], 'k', linewidth=2.0)
#print(data.to_numpy()[1, 1], data.to_numpy()[-1, 1])
self.ax2.xaxis.set_ticks(np.arange(self.stframe/self.fps, self.endframe/self.fps , max_slice))
self.ax2.xaxis.set_visible(True)
self.label_3.canvas.draw()
#ax1(Timeline)그리기
self.ax1.clear()
self.ax1.set_ylim([0.5, 1.5])
self.ax1.set_xlim((self.stframe / self.fps), int(self.endframe / self.fps))
timelinedata = np.zeros(int(self.endframe/self.fps)+1)
for i in range(len(timelinedata)):
if stretchedata[i] > threshold:
timelinedata[i] = 1
if self.iscomments is True:
os.environ['OAUTHLIB_INSECURE_TRANSPORT'] = '1'
service = get_authenticated_service()
comments = get_video_comments(
service, part='snippet', videoId=self.videoID, textFormat='plainText')
video_duration = get_video_duration(
service, part='contentDetails', id=self.videoID)
# print(video_duration)
# print(comments)
self.commentsData = make_video_timeline(comments, video_duration, self.max_slice)
print(str(len(timelinedata)) + "," + str(len(self.commentsData)))
if not (-1 <= len(timelinedata)/self.max_slice - len(self.commentsData) <= 1):
choice = QMessageBox.warning(self, 'warning', '유튜브 댓글 데이터가 올바르지 않습니다. 제외하고 진행하겠습니까?', QMessageBox.Yes | QMessageBox.No)
if choice == QMessageBox.Yes:
self.highlightData = timelinedata
self.iscomments = False
else:
sys.exit()
else: #commentsData반영
for i in range(len(timelinedata)):
#print(i)
if timelinedata[i] == 0:
#print(i/self.max_slice)
if self.commentsData[int(i/self.max_slice)] == 1:
timelinedata[i] = 1
print(self.commentsData)
self.highlightData = timelinedata
self.ax1.plot(timelinedata, '*y')
self.ax1.set_xlim((self.stframe / self.fps), int(self.endframe / self.fps))
self.ax1.xaxis.set_ticks(np.arange(self.stframe / self.fps, self.endframe / self.fps, max_slice))
self.ax1.plot([data.to_numpy()[0, 1], data.to_numpy()[0, 1]], [0.5, 1.5], 'k', linewidth=2.0)
self.label_2.canvas.draw()
#스케일링 방법 바꾸기
return
def AYBtnClicked(self):
video_id, ok = QInputDialog.getText(self, '영상 ID', '유튜브 영상 ID를 입력하세요. (/watch?v=~)')
if not ok:
return
self.videoID = video_id
self.iscomments = True
#check on timeline and export
import moviepy.editor as mp
import librosa
class Clip:
def __init__(self, fileName):
self.clip = mp.VideoFileClip(fileName)
self.fullFileName = fileName
self.fileName = fileName[:fileName.rfind('.')]
def getLoudSection(self, max_slice=10, threshold=180): #max_slice: 몇 초 단위로 나눌 것인지 결정
self.clip.audio.write_audiofile(self.fileName+'.wav')
x, sr = librosa.load(self.fileName + '.wav', sr=16000) #x: 오디오 파일을 나눈(1초에 sr개로 나눔) 샘플들의 진폭이 저장됨, 샘플의 개수는 sr*영상길이
#print(x)
#print(len(x))
window_length = max_slice * sr #한 구간에 해당하는 샘플의 개수
energy = np.array([sum(abs(x[i: i + window_length] ** 2)) #windwo_length구간 안에있는 모든 샘플들의 진폭의 제곱의 합->소리의 크기
for i in range(0, len(x), window_length)]) #len(x)는 sr*영상길이인데 이것을 window_length만큼 끊음 i = 0, winlength, winlength*2, winlength*3, ...
#energy data를 dataframe으로 만듦
df = pd.DataFrame(columns=['energy', 'start', 'end'])
row_index = 0
for i in range(len(energy)):
value = energy[i]
i = np.where(energy == value)[0]
df.loc[row_index, 'energy'] = value
df.loc[row_index, 'start'] = i[0] * max_slice
df.loc[row_index, 'end'] = (i[0] + 1) * max_slice
row_index = row_index + 1
#연속구간으로 나타내기
return df
def concatVideo(self, data, max_slice):
i = 0
clips = []
final1 = []
final2 = []
while i < len(data):
if data[i] == 1:
subclip = self.clip.subclip(i, i+max_slice)
clips.append(subclip)
i += max_slice
n= 5
result = [clips[i * n:(i + 1) * n] for i in range((len(clips) + n - 1) // n)]
print(len(result))
for j in range(len(result)):
final1.append(mp.concatenate_videoclips(result[j]))
final2 = mp.concatenate_videoclips(final1)
final2.write_videofile(self.fileName+'_highlights'+'.mp4')
import pickle
import csv
import os
import pandas as pd
import re
import math
import google.oauth2.credentials
from googleapiclient.discovery import build
from googleapiclient.errors import HttpError
from google_auth_oauthlib.flow import InstalledAppFlow
from google.auth.transport.requests import Request
# parameter 지정
CLIENT_SECRETS_FILE = "/Users/ASUS/Desktop/work/3-1/디자인적사고/project/client_secret.json"
CLIENT_SECRETS_FILE = "/Users/user/Documents/카카오톡 받은 파일/api/client_secret.json"
SCOPES = ['https://www.googleapis.com/auth/youtube.force-ssl']
API_SERVICE_NAME = 'youtube'
API_VERSION = 'v3'
# 인증 모듈
def get_authenticated_service():
credentials = None
if os.path.exists('token.pickle'):
with open('token.pickle', 'rb') as token:
credentials = pickle.load(token)
# Check if the credentials are invalid or do not exist
if not credentials or not credentials.valid:
# Check if the credentials have expired
if credentials and credentials.expired and credentials.refresh_token:
credentials.refresh(Request())
else:
flow = InstalledAppFlow.from_client_secrets_file(
CLIENT_SECRETS_FILE, SCOPES)
credentials = flow.run_console()
# Save the credentials for the next run
with open('token.pickle', 'wb') as token:
pickle.dump(credentials, token)
return build(API_SERVICE_NAME, API_VERSION, credentials=credentials)
# video comment 데이터프레임화
def get_video_comments(service, **kwargs):
results = service.commentThreads().list(**kwargs).execute()
youtube_pd = pd.DataFrame()
# filename = input('.csv 형태로 입력하세요')
while results:
for item in results['items']:
comment1 = item['snippet']['topLevelComment']['snippet']['textDisplay']
# comment2 = item['snippet']['topLevelComment']['snippet']['publishedAt']
# comment3 = item['snippet']['topLevelComment']['snippet']['authorDisplayName']
# comment2 = item['snippet']['topLevelComment']['snippet']['viewerRating']
comment2 = item['snippet']['topLevelComment']['snippet']['likeCount']
pd_data = {"comment": comment1, "likeCount": comment2}
youtube_pd = youtube_pd.append(pd_data, ignore_index=True)
# youtube_pd.to_csv(filename, index=False, encoding='utf-8-sig')
# Check if another page exists
if 'nextPageToken' in results:
kwargs['pageToken'] = results['nextPageToken']
results = service.commentThreads().list(**kwargs).execute()
else:
break
return youtube_pd
# 코멘트에서 타임라인만 추출
def make_video_timeline(commentFrame, duration, max_slice):
youtube_pd_sorted_by_likeCount = commentFrame.sort_values(by='likeCount')
# .str.split(" ") # .values.tolist()
#youtube_pd_sorted_by_likeCount = youtube_pd_sorted_by_likeCount.head(30) # 갯수 정해서 추출하기
comment = youtube_pd_sorted_by_likeCount['comment']
comment_split = comment.str.split(" ").values.tolist()
result = []
for i in range(len(comment_split)):
for j in range(len(comment_split[i])):
if ":" in comment_split[i][j]:
result.append(comment_split[i][j])
hi = ' '.join(result)
timeline = re.findall(r'(\d{1,2}):(\d{2})', hi)
timelist = []
for groups in timeline:
timelist.append(int(groups[0])*60+int(groups[1]))
timelist.sort()
# print(timelist)
timecheck = []
for i in range(math.ceil(duration/max_slice)):
timecheck.append(0)
for i in range(math.ceil(duration/max_slice)):
for j in timelist:
if i <= int(int(j)/max_slice) <= (i+10/max_slice):
timecheck[i] = 1
# print(timecheck)
return timecheck
# 비디오 총 길이 추출
def get_video_duration(service, **kwargs):
results = service.videos().list(**kwargs).execute()
comment_data = results['items']
comment_data = comment_data[0]['contentDetails']['duration']
match = re.match(r'PT(\d+H)?(\d+M)?(\d+S)?', comment_data).groups()
hours = _js_parseInt(match[0]) if match[0] else 0
minutes = _js_parseInt(match[1]) if match[1] else 0
seconds = _js_parseInt(match[2]) if match[2] else 0
return hours * 3600 + minutes * 60 + seconds
# 문자열에서 숫자만 추출
def _js_parseInt(string):
return int(''.join([x for x in string if x.isdigit()]))
if __name__ == "__main__":
app = QApplication(sys.argv)
myWindow = WindowClass()
myWindow.show()
sys.exit(app.exec_())