👍️[Update] Model : energyの終了方法とreturnの値を修正

This commit is contained in:
misyaguziya
2024-09-11 13:28:35 +09:00
parent ccdd070fed
commit 0e226f7328
4 changed files with 41 additions and 22 deletions

View File

@@ -157,6 +157,24 @@ class Config:
if isinstance(value, bool): if isinstance(value, bool):
self._ENABLE_FOREGROUND = value self._ENABLE_FOREGROUND = value
@property
def ENABLE_CHECK_ENERGY_SEND(self):
return self._ENABLE_CHECK_ENERGY_SEND
@ENABLE_CHECK_ENERGY_SEND.setter
def ENABLE_CHECK_ENERGY_SEND(self, value):
if isinstance(value, bool):
self._ENABLE_CHECK_ENERGY_SEND = value
@property
def ENABLE_CHECK_ENERGY_RECEIVE(self):
return self._ENABLE_CHECK_ENERGY_RECEIVE
@ENABLE_CHECK_ENERGY_RECEIVE.setter
def ENABLE_CHECK_ENERGY_RECEIVE(self, value):
if isinstance(value, bool):
self._ENABLE_CHECK_ENERGY_RECEIVE = value
# @property # @property
# def SENT_MESSAGES_LOG(self): # def SENT_MESSAGES_LOG(self):
# return self._SENT_MESSAGES_LOG # return self._SENT_MESSAGES_LOG
@@ -959,6 +977,8 @@ class Config:
self._ENABLE_TRANSCRIPTION_SEND = False self._ENABLE_TRANSCRIPTION_SEND = False
self._ENABLE_TRANSCRIPTION_RECEIVE = False self._ENABLE_TRANSCRIPTION_RECEIVE = False
self._ENABLE_FOREGROUND = False self._ENABLE_FOREGROUND = False
self._ENABLE_CHECK_ENERGY_SEND = False
self._ENABLE_CHECK_ENERGY_RECEIVE = False
# self._SENT_MESSAGES_LOG = [] # self._SENT_MESSAGES_LOG = []
# self._CURRENT_SENT_MESSAGES_LOG_INDEX = 0 # self._CURRENT_SENT_MESSAGES_LOG_INDEX = 0

View File

@@ -534,7 +534,7 @@ class Model:
# self.mic_get_energy.stop() # self.mic_get_energy.stop()
# self.mic_get_energy = None # self.mic_get_energy = None
def startCheckMicEnergy(self, fnc, end_fnc, error_fnc=None): def startCheckMicEnergy(self, fnc, error_fnc=None):
mic_device_list = getInputDevices().get(config.CHOICE_MIC_HOST, [{"name": "NoDevice"}]) mic_device_list = getInputDevices().get(config.CHOICE_MIC_HOST, [{"name": "NoDevice"}])
choice_mic_device = [device for device in mic_device_list if device["name"] == config.CHOICE_MIC_DEVICE] choice_mic_device = [device for device in mic_device_list if device["name"] == config.CHOICE_MIC_DEVICE]
if len(choice_mic_device) == 0: if len(choice_mic_device) == 0:
@@ -557,15 +557,17 @@ class Model:
mic_device = choice_mic_device[0] mic_device = choice_mic_device[0]
self.mic_energy_recorder = SelectedMicEnergyRecorder(mic_device) self.mic_energy_recorder = SelectedMicEnergyRecorder(mic_device)
self.mic_energy_recorder.recordIntoQueue(mic_energy_queue) self.mic_energy_recorder.recordIntoQueue(mic_energy_queue)
self.mic_energy_plot_progressbar = threadFnc(sendMicEnergy, end_fnc=end_fnc) self.mic_energy_plot_progressbar = threadFnc(sendMicEnergy)
self.mic_energy_plot_progressbar.daemon = True self.mic_energy_plot_progressbar.daemon = True
self.mic_energy_plot_progressbar.start() self.mic_energy_plot_progressbar.start()
def stopCheckMicEnergy(self): def stopCheckMicEnergy(self):
if isinstance(self.mic_energy_plot_progressbar, threadFnc): if isinstance(self.mic_energy_plot_progressbar, threadFnc):
self.mic_energy_plot_progressbar.stop() self.mic_energy_plot_progressbar.stop()
self.mic_energy_plot_progressbar.join()
self.mic_energy_plot_progressbar = None self.mic_energy_plot_progressbar = None
if isinstance(self.mic_energy_recorder, SelectedMicEnergyRecorder): if isinstance(self.mic_energy_recorder, SelectedMicEnergyRecorder):
self.mic_energy_recorder.resume()
self.mic_energy_recorder.stop() self.mic_energy_recorder.stop()
self.mic_energy_recorder = None self.mic_energy_recorder = None
@@ -655,7 +657,7 @@ class Model:
# self.speaker_get_energy.stop() # self.speaker_get_energy.stop()
# self.speaker_get_energy = None # self.speaker_get_energy = None
def startCheckSpeakerEnergy(self, fnc, end_fnc, error_fnc=None): def startCheckSpeakerEnergy(self, fnc, error_fnc=None):
speaker_device_list = getOutputDevices() speaker_device_list = getOutputDevices()
choice_speaker_device = [device for device in speaker_device_list if device["name"] == config.CHOICE_SPEAKER_DEVICE] choice_speaker_device = [device for device in speaker_device_list if device["name"] == config.CHOICE_SPEAKER_DEVICE]
if len(choice_speaker_device) == 0: if len(choice_speaker_device) == 0:
@@ -678,15 +680,17 @@ class Model:
speaker_device = choice_speaker_device[0] speaker_device = choice_speaker_device[0]
self.speaker_energy_recorder = SelectedSpeakerEnergyRecorder(speaker_device) self.speaker_energy_recorder = SelectedSpeakerEnergyRecorder(speaker_device)
self.speaker_energy_recorder.recordIntoQueue(speaker_energy_queue) self.speaker_energy_recorder.recordIntoQueue(speaker_energy_queue)
self.speaker_energy_plot_progressbar = threadFnc(sendSpeakerEnergy, end_fnc=end_fnc) self.speaker_energy_plot_progressbar = threadFnc(sendSpeakerEnergy)
self.speaker_energy_plot_progressbar.daemon = True self.speaker_energy_plot_progressbar.daemon = True
self.speaker_energy_plot_progressbar.start() self.speaker_energy_plot_progressbar.start()
def stopCheckSpeakerEnergy(self): def stopCheckSpeakerEnergy(self):
if isinstance(self.speaker_energy_plot_progressbar, threadFnc): if isinstance(self.speaker_energy_plot_progressbar, threadFnc):
self.speaker_energy_plot_progressbar.stop() self.speaker_energy_plot_progressbar.stop()
self.speaker_energy_plot_progressbar.join()
self.speaker_energy_plot_progressbar = None self.speaker_energy_plot_progressbar = None
if isinstance(self.speaker_energy_recorder, SelectedSpeakerEnergyRecorder): if isinstance(self.speaker_energy_recorder, SelectedSpeakerEnergyRecorder):
self.speaker_energy_recorder.resume()
self.speaker_energy_recorder.stop() self.speaker_energy_recorder.stop()
self.speaker_energy_recorder = None self.speaker_energy_recorder = None

View File

@@ -454,30 +454,31 @@ def callbackDisableTranslation(*args, **kwargs) -> dict:
def callbackEnableTranscriptionSend(data, action, *args, **kwargs) -> dict: def callbackEnableTranscriptionSend(data, action, *args, **kwargs) -> dict:
printLog("Enable Transcription Send") printLog("Enable Transcription Send")
config.ENABLE_TRANSCRIPTION_SEND = True
startThreadingTranscriptionSendMessage(action) startThreadingTranscriptionSendMessage(action)
config.ENABLE_TRANSCRIPTION_SEND = True
return {"status":200, "result":config.ENABLE_TRANSCRIPTION_SEND} return {"status":200, "result":config.ENABLE_TRANSCRIPTION_SEND}
def callbackDisableTranscriptionSend(*args, **kwargs) -> dict: def callbackDisableTranscriptionSend(*args, **kwargs) -> dict:
printLog("Disable Transcription Send") printLog("Disable Transcription Send")
config.ENABLE_TRANSCRIPTION_SEND = False
stopThreadingTranscriptionSendMessage() stopThreadingTranscriptionSendMessage()
config.ENABLE_TRANSCRIPTION_SEND = False
return {"status":200, "result":config.ENABLE_TRANSCRIPTION_SEND} return {"status":200, "result":config.ENABLE_TRANSCRIPTION_SEND}
def callbackEnableTranscriptionReceive(data, action, *args, **kwargs) -> dict: def callbackEnableTranscriptionReceive(data, action, *args, **kwargs) -> dict:
printLog("Enable Transcription Receive") printLog("Enable Transcription Receive")
config.ENABLE_TRANSCRIPTION_RECEIVE = True
startThreadingTranscriptionReceiveMessage(action) startThreadingTranscriptionReceiveMessage(action)
if config.ENABLE_OVERLAY_SMALL_LOG is True: if config.ENABLE_OVERLAY_SMALL_LOG is True:
if model.overlay.initialized is False and model.overlay.checkSteamvrRunning() is True: if model.overlay.initialized is False and model.overlay.checkSteamvrRunning() is True:
model.startOverlay() model.startOverlay()
config.ENABLE_TRANSCRIPTION_RECEIVE = True
return {"status":200, "result":config.ENABLE_TRANSCRIPTION_RECEIVE} return {"status":200, "result":config.ENABLE_TRANSCRIPTION_RECEIVE}
def callbackDisableTranscriptionReceive(*args, **kwargs) -> dict: def callbackDisableTranscriptionReceive(*args, **kwargs) -> dict:
printLog("Disable Transcription Receive") printLog("Disable Transcription Receive")
config.ENABLE_TRANSCRIPTION_RECEIVE = False
stopThreadingTranscriptionReceiveMessage() stopThreadingTranscriptionReceiveMessage()
config.ENABLE_TRANSCRIPTION_RECEIVE = False
return {"status":200, "result":config.ENABLE_TRANSCRIPTION_RECEIVE} return {"status":200, "result":config.ENABLE_TRANSCRIPTION_RECEIVE}
def callbackEnableForeground(*args, **kwargs) -> dict: def callbackEnableForeground(*args, **kwargs) -> dict:
@@ -740,9 +741,6 @@ class ProgressBarMicEnergy:
def set(self, energy) -> None: def set(self, energy) -> None:
self.action("mic", {"status":200, "result":energy}) self.action("mic", {"status":200, "result":energy})
def stopped(self) -> None:
self.action("stopped", {"status":200})
def error(self) -> None: def error(self) -> None:
self.action("error_device", {"status":400,"result": {"message":"No mic device detected."}}) self.action("error_device", {"status":400,"result": {"message":"No mic device detected."}})
@@ -751,15 +749,16 @@ def callbackEnableCheckMicThreshold(data, action, *args, **kwargs) -> dict:
progressbar_mic_energy = ProgressBarMicEnergy(action) progressbar_mic_energy = ProgressBarMicEnergy(action)
model.startCheckMicEnergy( model.startCheckMicEnergy(
progressbar_mic_energy.set, progressbar_mic_energy.set,
progressbar_mic_energy.stopped,
progressbar_mic_energy.error progressbar_mic_energy.error
) )
return {"status":200} config.ENABLE_CHECK_ENERGY_SEND = True
return {"status":200, "result":config.ENABLE_CHECK_ENERGY_SEND}
def callbackDisableCheckMicThreshold(*args, **kwargs) -> dict: def callbackDisableCheckMicThreshold(*args, **kwargs) -> dict:
printLog("Disable Check Mic Threshold") printLog("Disable Check Mic Threshold")
model.stopCheckMicEnergy() model.stopCheckMicEnergy()
return {"status":200} config.ENABLE_CHECK_ENERGY_SEND = False
return {"status":200, "result":config.ENABLE_CHECK_ENERGY_SEND}
def callbackSetMicRecordTimeout(data, *args, **kwargs) -> dict: def callbackSetMicRecordTimeout(data, *args, **kwargs) -> dict:
printLog("Set Mic Record Timeout", data) printLog("Set Mic Record Timeout", data)
@@ -873,9 +872,6 @@ class ProgressBarSpeakerEnergy:
def set(self, energy) -> None: def set(self, energy) -> None:
self.action("speaker", {"status":200, "result":energy}) self.action("speaker", {"status":200, "result":energy})
def stopped(self) -> None:
self.action("stopped", {"status":200})
def error(self) -> None: def error(self) -> None:
self.action("error_device", {"status":400,"result": {"message":"No mic device detected."}}) self.action("error_device", {"status":400,"result": {"message":"No mic device detected."}})
@@ -884,15 +880,16 @@ def callbackEnableCheckSpeakerThreshold(data, action, *args, **kwargs) -> dict:
progressbar_speaker_energy = ProgressBarSpeakerEnergy(action) progressbar_speaker_energy = ProgressBarSpeakerEnergy(action)
model.startCheckSpeakerEnergy( model.startCheckSpeakerEnergy(
progressbar_speaker_energy.set, progressbar_speaker_energy.set,
progressbar_speaker_energy.stopped,
progressbar_speaker_energy.error progressbar_speaker_energy.error
) )
return {"status":200} config.ENABLE_CHECK_ENERGY_RECEIVE = True
return {"status":200, "result":config.ENABLE_CHECK_ENERGY_RECEIVE}
def callbackDisableCheckSpeakerThreshold(*args, **kwargs) -> dict: def callbackDisableCheckSpeakerThreshold(*args, **kwargs) -> dict:
printLog("Disable Check Speaker Threshold") printLog("Disable Check Speaker Threshold")
model.stopCheckSpeakerEnergy() model.stopCheckSpeakerEnergy()
return {"status":200} config.ENABLE_CHECK_ENERGY_RECEIVE = False
return {"status":200, "result":config.ENABLE_CHECK_ENERGY_RECEIVE}
def callbackSetSpeakerRecordTimeout(data, *args, **kwargs) -> dict: def callbackSetSpeakerRecordTimeout(data, *args, **kwargs) -> dict:
printLog("Set Speaker Record Timeout", data) printLog("Set Speaker Record Timeout", data)

View File

@@ -217,12 +217,10 @@ action_mapping = {
}, },
"/controller/callback_enable_check_mic_threshold": { "/controller/callback_enable_check_mic_threshold": {
"mic":"/action/check_mic_threshold_energy", "mic":"/action/check_mic_threshold_energy",
"stopped":"/action/check_mic_threshold_energy_stopped",
"error_device":"/action/error_device", "error_device":"/action/error_device",
}, },
"/controller/callback_enable_check_speaker_threshold": { "/controller/callback_enable_check_speaker_threshold": {
"speaker":"/action/check_speaker_threshold_energy", "speaker":"/action/check_speaker_threshold_energy",
"stopped":"/action/check_speaker_threshold_energy_stopped",
"error_device":"/action/error_device", "error_device":"/action/error_device",
}, },
"/controller/callback_messagebox_send": { "/controller/callback_messagebox_send": {