mirror of https://github.com/aidygus/LinVAM.git
252 lines
9.2 KiB
Python
252 lines
9.2 KiB
Python
from PyQt5.QtCore import *
|
|
from PyQt5.QtGui import *
|
|
from PyQt5.QtWidgets import *
|
|
from ui_mainwnd import Ui_MainWidget
|
|
from profileeditwnd import ProfileEditWnd
|
|
import json
|
|
from profileexecutor import ProfileExecutor
|
|
import sys
|
|
import pickle
|
|
import time
|
|
import keyboard
|
|
import threading
|
|
|
|
class MainWnd(QWidget):
|
|
def __init__(self, p_parent = None):
|
|
super().__init__(p_parent)
|
|
self.ui = Ui_MainWidget()
|
|
self.ui.setupUi(self)
|
|
self.m_profileExecutor = ProfileExecutor()
|
|
|
|
self.ui.profileCbx.currentIndexChanged.connect(self.slotProfileChanged)
|
|
self.ui.addBut.clicked.connect(self.slotAddNewProfile)
|
|
self.ui.editBut.clicked.connect(self.slotEditProfile)
|
|
self.ui.removeBut.clicked.connect(self.slotRemoveProfile)
|
|
self.ui.listeningChk.stateChanged.connect(self.slotListeningEnabled)
|
|
self.ui.ok.clicked.connect(self.slotOK)
|
|
self.ui.cancel.clicked.connect(self.slotCancel)
|
|
|
|
if self.loadFromDatabase() > 0 :
|
|
# if self.loadTestProfiles() > 0:
|
|
w_jsonProfile = self.ui.profileCbx.itemData(0)
|
|
if w_jsonProfile != None:
|
|
self.m_activeProfile = json.loads(w_jsonProfile)
|
|
self.m_profileExecutor.setProfile(self.m_activeProfile)
|
|
self.m_profileExecutor.start()
|
|
|
|
def saveToDatabase(self):
|
|
w_profiles = []
|
|
w_profileCnt = self.ui.profileCbx.count()
|
|
for w_idx in range(w_profileCnt):
|
|
w_jsonProfile = self.ui.profileCbx.itemData(w_idx)
|
|
if w_jsonProfile == None:
|
|
continue
|
|
|
|
w_profile = json.loads(w_jsonProfile)
|
|
w_profiles.append(w_profile)
|
|
|
|
with open("profiles.dat", "wb") as f:
|
|
pickle.dump(w_profiles, f, pickle.HIGHEST_PROTOCOL)
|
|
|
|
def loadFromDatabase(self):
|
|
w_profiles = []
|
|
with open("profiles.dat", "rb") as f:
|
|
w_profiles = pickle.load(f)
|
|
|
|
for w_profile in w_profiles:
|
|
self.ui.profileCbx.addItem(w_profile['name'])
|
|
w_jsonProfile = json.dumps(w_profile)
|
|
self.ui.profileCbx.setItemData(self.ui.profileCbx.count() - 1, w_jsonProfile)
|
|
|
|
return len(w_profiles)
|
|
|
|
def loadTestProfiles(self):
|
|
|
|
w_carProfileDict = {
|
|
"name": "car game",
|
|
"commands": [
|
|
{'name': 'forward',
|
|
'actions': [
|
|
{'name': 'key action', 'key': 'up', 'type': 1}
|
|
],
|
|
'repeat': 1,
|
|
'async': False,
|
|
'threshold': 12
|
|
},
|
|
{'name': 'back',
|
|
'actions': [
|
|
{'name': 'key action', 'key': 'down', 'type': 1}
|
|
],
|
|
'repeat': 1,
|
|
'async': False,
|
|
'threshold': 3
|
|
},
|
|
{'name': 'left',
|
|
'actions': [{'name': 'key action', 'key': 'right', 'type': 0},
|
|
{'name': 'key action', 'key': 'left', 'type': 1},
|
|
],
|
|
'repeat': 1,
|
|
'async': False,
|
|
'threshold': 10
|
|
},
|
|
{'name': 'right',
|
|
'actions': [{'name': 'key action', 'key': 'left', 'type': 0},
|
|
{'name': 'key action', 'key': 'right', 'type': 1},
|
|
],
|
|
'repeat': 1,
|
|
'async': False,
|
|
'threshold': 3
|
|
},
|
|
{'name': 'stop',
|
|
'actions': [
|
|
{'name': 'key action', 'key': 'left', 'type': 0},
|
|
{'name': 'key action', 'key': 'right', 'type': 0},
|
|
{'name': 'key action', 'key': 'up', 'type': 0},
|
|
{'name': 'key action', 'key': 'down', 'type': 0}
|
|
],
|
|
'repeat': 1,
|
|
'async': False,
|
|
'threshold': 8
|
|
}
|
|
]
|
|
}
|
|
|
|
w_airplaneProfileDict = {
|
|
"name": "airplane game",
|
|
"commands": [
|
|
{'name': 'up',
|
|
'actions': [
|
|
{'name': 'command stop action', 'command name': 'down'},
|
|
{'name': 'mouse move action', 'x': 0, 'y': -5, 'absolute': False},
|
|
{'name': 'pause action', 'time': 0.01}
|
|
],
|
|
'repeat': -1,
|
|
'async': True,
|
|
'threshold': 3
|
|
},
|
|
{'name': 'left',
|
|
'actions': [
|
|
{'name': 'command stop action', 'command name': 'right'},
|
|
{'name': 'mouse move action', 'x': -5, 'y': 0, 'absolute': False},
|
|
{'name': 'pause action', 'time': 0.005}
|
|
],
|
|
'repeat': -1,
|
|
'async': True,
|
|
'threshold': 3
|
|
},
|
|
{'name': 'right',
|
|
'actions': [
|
|
{'name': 'command stop action', 'command name': 'left'},
|
|
{'name': 'mouse move action', 'x': 5, 'y': 0, 'absolute': False},
|
|
{'name': 'pause action', 'time': 0.005}
|
|
],
|
|
'repeat': -1,
|
|
'async': True,
|
|
'threshold': 3
|
|
},
|
|
{'name': 'down',
|
|
'actions': [
|
|
{'name': 'command stop action', 'command name': 'up'},
|
|
{'name': 'mouse move action', 'x': 0, 'y': 5, 'absolute': False},
|
|
{'name': 'pause action', 'time': 0.005}
|
|
],
|
|
'repeat': -1,
|
|
'async': True,
|
|
'threshold': 3
|
|
},
|
|
{'name': 'shoot',
|
|
'actions': [
|
|
{'name': 'mouse click action', 'button': 'left', 'type': 1},
|
|
{'name': 'pause action', 'time': 0.03}
|
|
],
|
|
'repeat': 1,
|
|
'async': False,
|
|
'threshold': 3
|
|
},
|
|
{'name': 'stop',
|
|
'actions': [
|
|
{'name': 'command stop action', 'command name': 'up'},
|
|
{'name': 'command stop action', 'command name': 'left'},
|
|
{'name': 'command stop action', 'command name': 'right'},
|
|
{'name': 'command stop action', 'command name': 'down'},
|
|
{'name': 'mouse click action', 'button': 'left', 'type': 0}
|
|
],
|
|
'repeat': 1,
|
|
'async': False,
|
|
'threshold': 3
|
|
}
|
|
]
|
|
}
|
|
w_profiles = []
|
|
w_profiles.append(w_carProfileDict)
|
|
w_profiles.append(w_airplaneProfileDict)
|
|
|
|
i = 0
|
|
for w_profile in w_profiles:
|
|
self.ui.profileCbx.addItem(w_profile['name'])
|
|
w_jsonProfile = json.dumps(w_profile)
|
|
self.ui.profileCbx.setItemData(i, w_jsonProfile)
|
|
i = i + 1
|
|
|
|
return i
|
|
|
|
def slotProfileChanged(self, p_idx):
|
|
w_jsonProfile = self.ui.profileCbx.itemData(p_idx)
|
|
if w_jsonProfile != None:
|
|
self.m_activeProfile = json.loads(w_jsonProfile)
|
|
self.m_profileExecutor.setProfile(self.m_activeProfile)
|
|
|
|
def slotAddNewProfile(self):
|
|
w_profileEditWnd = ProfileEditWnd(None, self)
|
|
if w_profileEditWnd.exec() == QDialog.Accepted:
|
|
w_profile = w_profileEditWnd.m_profile
|
|
self.m_profileExecutor.setProfile(w_profile)
|
|
self.ui.profileCbx.addItem(w_profile['name'])
|
|
w_jsonProfile = json.dumps(w_profile)
|
|
self.ui.profileCbx.setItemData(self.ui.profileCbx.count()-1, w_jsonProfile)
|
|
|
|
def slotEditProfile(self):
|
|
w_idx = self.ui.profileCbx.currentIndex()
|
|
w_jsonProfile = self.ui.profileCbx.itemData(w_idx)
|
|
w_profile = json.loads(w_jsonProfile)
|
|
w_profileEditWnd = ProfileEditWnd(w_profile, self)
|
|
if w_profileEditWnd.exec() == QDialog.Accepted:
|
|
w_profile = w_profileEditWnd.m_profile
|
|
self.m_profileExecutor.setProfile(w_profile)
|
|
self.ui.profileCbx.setItemText(w_idx, w_profile['name'])
|
|
w_jsonProfile = json.dumps(w_profile)
|
|
self.ui.profileCbx.setItemData(w_idx, w_jsonProfile)
|
|
|
|
def slotRemoveProfile(self):
|
|
w_curIdx = self.ui.profileCbx.currentIndex()
|
|
if w_curIdx >= 0:
|
|
self.ui.profileCbx.removeItem(w_curIdx)
|
|
|
|
w_curIdx = self.ui.profileCbx.currentIndex()
|
|
if w_curIdx >= 0:
|
|
w_jsonProfile = self.ui.profileCbx.itemData(w_curIdx)
|
|
w_profile = json.loads(w_jsonProfile)
|
|
self.m_profileExecutor.setProfile(w_profile)
|
|
|
|
def slotListeningEnabled(self, p_enabled):
|
|
if p_enabled:
|
|
self.m_profileExecutor.setEnableListening(True)
|
|
else:
|
|
self.m_profileExecutor.setEnableListening(False)
|
|
|
|
def slotOK(self):
|
|
self.saveToDatabase()
|
|
self.m_profileExecutor.stop()
|
|
self.close()
|
|
|
|
def slotCancel(self):
|
|
self.m_profileExecutor.stop()
|
|
self.close()
|
|
exit()
|
|
|
|
if __name__ == "__main__":
|
|
app = QApplication(sys.argv)
|
|
mainWnd = MainWnd()
|
|
mainWnd.show()
|
|
sys.exit(app.exec_())
|