every_holiday/VRCT
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

👍️[Update] Model : Overlay() classをsizeに依存しない設計に修正

Browse Source
This commit is contained in:
misyaguziya
2024-11-10 17:55:06 +09:00
parent 109b7e62ba
commit 2df88bf698
5 changed files with 103 additions and 140 deletions
Download Patch File Download Diff File Expand all files Collapse all files

129
src-python/models/overlay/overlay.py
View File

@@ -63,7 +63,7 @@ def getRightHandBaseMatrix():
return arr
class Overlay:
def __init__(self, small_settings, large_settings):
def __init__(self, settings_dict):
self.system = None
self.overlay = None
self.handle = None
@@ -71,75 +71,66 @@ class Overlay:
self.loop = True
self.thread_overlay = None
self.settings = {"small": small_settings, "large": large_settings}
self.lastUpdate = {"small": time.monotonic(), "large": time.monotonic()}
self.fadeRatio = {"small": 1, "large": 1}
self.settings = {}
self.lastUpdate = {}
self.fadeRatio = {}
for key, value in settings_dict.items():
self.settings[key] = value
self.lastUpdate[key] = time.monotonic()
self.fadeRatio[key] = 1
def init(self):
try:
self.system = openvr.init(openvr.VRApplication_Background)
self.overlay = openvr.IVROverlay()
self.overlay_system = openvr.IVRSystem()
self.handle_small = self.overlay.createOverlay("VRCT_SMALL_LOG", "VRCT_SMALL_LOG")
self.handle_large = self.overlay.createOverlay("VRCT_LARGE_LOG", "VRCT_LARGE_LOG")
self.overlay.showOverlay(self.handle_small)
self.overlay.showOverlay(self.handle_large)
self.handle = {}
for i, size in enumerate(self.settings.keys()):
self.handle[size] = self.overlay.createOverlay(f"VRCT{i}", f"VRCT{i}")
self.overlay.showOverlay(self.handle[size])
self.initialized = True
self.updateImage(Image.new("RGBA", (1, 1), (0, 0, 0, 0)), "small")
self.updateColor([1, 1, 1], "small")
self.updateOpacity(self.settings["small"]["opacity"], "small")
self.updateUiScaling(self.settings["small"]["ui_scaling"], "small")
self.updatePosition(
self.settings["small"]["x_pos"],
self.settings["small"]["y_pos"],
self.settings["small"]["z_pos"],
self.settings["small"]["x_rotation"],
self.settings["small"]["y_rotation"],
self.settings["small"]["z_rotation"],
self.settings["small"]["tracker"]
)
self.updateImage(Image.new("RGBA", (1, 1), (0, 0, 0, 0)), "large")
self.updateColor([1, 1, 1], "large")
self.updateOpacity(self.settings["large"]["opacity"], "large")
self.updateUiScaling(self.settings["large"]["ui_scaling"], "large")
self.updatePosition(
self.settings["large"]["x_pos"],
self.settings["large"]["y_pos"],
self.settings["large"]["z_pos"],
self.settings["large"]["x_rotation"],
self.settings["large"]["y_rotation"],
self.settings["large"]["z_rotation"],
self.settings["large"]["tracker"]
)
for size in self.settings.keys():
self.updateImage(Image.new("RGBA", (1, 1), (0, 0, 0, 0)), size)
self.updateColor([1, 1, 1], size)
self.updateOpacity(self.settings[size]["opacity"], size, True)
self.updateUiScaling(self.settings[size]["ui_scaling"], size, True)
self.updatePosition(
self.settings[size]["x_pos"],
self.settings[size]["y_pos"],
self.settings[size]["z_pos"],
self.settings[size]["x_rotation"],
self.settings[size]["y_rotation"],
self.settings[size]["z_rotation"],
self.settings[size]["tracker"]
)
except Exception as e:
printLog("error:Could not initialise OpenVR", e)
def updateImage(self, img, size: str = "small"):
def updateImage(self, img, size):
if self.initialized is True:
width, height = img.size
img = img.tobytes()
img = (ctypes.c_char * len(img)).from_buffer_copy(img)
try:
self.overlay.setOverlayRaw(self.handle_small[size], img, width, height, 4)
self.overlay.setOverlayRaw(self.handle[size], img, width, height, 4)
except Exception as e:
printLog("error:Could not update image", e)
self.initialized = False
self.reStartOverlay()
while self.initialized is False:
time.sleep(0.1)
self.overlay.setOverlayRaw(self.handle_small[size], img, width, height, 4)
self.updateOpacity(self.settings[size]["opacity"], "small")
self.overlay.setOverlayRaw(self.handle[size], img, width, height, 4)
self.updateOpacity(self.settings[size]["opacity"], size, True)
self.lastUpdate[size] = time.monotonic()
def clearImage(self, size: str = "small"):
def clearImage(self, size):
if self.initialized is True:
self.updateImage(Image.new("RGBA", (1, 1), (0, 0, 0, 0)), size)
def updateColor(self, col, size: str = "small"):
def updateColor(self, col, size):
"""
col is a 3-tuple representing (r, g, b)
"""
@@ -147,7 +138,7 @@ class Overlay:
r, g, b = col
self.overlay.setOverlayColor(self.handle[size], r, g, b)
def updateOpacity(self, opacity, with_fade=False, size: str = "small"):
def updateOpacity(self, opacity, size, with_fade=True):
self.settings[size]["opacity"] = opacity
if self.initialized is True:
@@ -157,12 +148,12 @@ class Overlay:
else:
self.overlay.setOverlayAlpha(self.handle[size], self.settings[size]["opacity"])
def updateUiScaling(self, ui_scaling, size: str = "small"):
def updateUiScaling(self, ui_scaling, size):
self.settings[size]["ui_scaling"] = ui_scaling
if self.initialized is True:
self.overlay.setOverlayWidthInMeters(self.handle[size], self.settings[size]["ui_scaling"])
def updatePosition(self, x_pos, y_pos, z_pos, x_rotation, y_rotation, z_rotation, tracker: str="HMD", size: str = "small"):
def updatePosition(self, x_pos, y_pos, z_pos, x_rotation, y_rotation, z_rotation, tracker, size):
"""
x_pos, y_pos, z_pos are floats representing the position of overlay
x_rotation, y_rotation, z_rotation are floats representing the rotation of overlay
@@ -203,10 +194,10 @@ class Overlay:
transform
)
def updateDisplayDuration(self, display_duration, size: str = "small"):
def updateDisplayDuration(self, display_duration, size):
self.settings[size]["display_duration"] = display_duration
def updateFadeoutDuration(self, fadeout_duration, size: str = "small"):
def updateFadeoutDuration(self, fadeout_duration, size):
self.settings[size]["fadeout_duration"] = fadeout_duration
def checkActive(self):
@@ -221,7 +212,7 @@ class Overlay:
printLog("error:Could not check SteamVR running", e)
return False
def evaluateOpacityFade(self, lastUpdate, currentTime, size: str = "small"):
def evaluateOpacityFade(self, lastUpdate, currentTime, size):
if (currentTime - lastUpdate) > self.settings[size]["display_duration"]:
timeThroughInterval = currentTime - lastUpdate - self.settings[size]["display_duration"]
self.fadeRatio[size] = 1 - timeThroughInterval / self.settings[size]["fadeout_duration"]
@@ -229,19 +220,19 @@ class Overlay:
self.fadeRatio[size] = 0
self.overlay.setOverlayAlpha(self.handle[size], self.fadeRatio[size] * self.settings[size]["opacity"])
def update(self, size: str = "small"):
def update(self, size):
currTime = time.monotonic()
if self.settings[size]["fadeout_duration"] != 0:
self.evaluateOpacityFade(self.lastUpdate[size], currTime, size)
else:
self.updateOpacity(self.settings[size]["opacity"], size)
self.updateOpacity(self.settings[size]["opacity"], size, True)
def mainloop(self):
self.loop = True
while self.checkActive() is True and self.loop is True:
startTime = time.monotonic()
self.update("small")
self.update("large")
for size in self.settings.keys():
self.update(size)
sleepTime = (1 / 16) - (time.monotonic() - startTime)
if sleepTime > 0:
time.sleep(sleepTime)
@@ -261,9 +252,10 @@ class Overlay:
self.loop = False
self.thread_overlay.join()
self.thread_overlay = None
if isinstance(self.overlay, openvr.IVROverlay) and isinstance(self.handle["small"], int) and isinstance(self.handle["large"], int):
self.overlay.destroyOverlay(self.handle["small"])
self.overlay.destroyOverlay(self.handle["large"])
if isinstance(self.overlay, openvr.IVROverlay):
for size in self.settings.keys():
if isinstance(self.handle[size], int):
self.overlay.destroyOverlay(self.handle[size])
self.overlay = None
if isinstance(self.system, openvr.IVRSystem):
openvr.shutdown()
@@ -280,35 +272,6 @@ class Overlay:
return _proc_name in (p.name() for p in process_iter())
if __name__ == "__main__":
# from overlay_image import OverlayImage
# overlay_image = OverlayImage()
# overlay = Overlay(0, 0, 1, 1, 0, 1, 1)
# overlay.startOverlay()
# time.sleep(1)
# # Example usage
# img = overlay_image.createOverlayImageSmall("こんにちは、世界！さようなら", "Japanese", "Hello,World!Goodbye", "Japanese")
# overlay.updateImage(img)
# time.sleep(100000)
# for i in range(100):
# print(i)
# overlay = Overlay(0, 0, 1, 1, 1, 1, 1)
# overlay.startOverlay()
# time.sleep(1)
# # Example usage
# img = overlay_image.createOverlayImageSmall("こんにちは、世界！さようなら", "Japanese", "Hello,World!Goodbye", "Japanese", ui_type="sakura")
# overlay.updateImage(img)
# time.sleep(0.5)
# img = overlay_image.createOverlayImageSmall("こんにちは、世界！さようなら", "Japanese", "Hello,World!Goodbye", "Japanese")
# overlay.updateImage(img)
# time.sleep(0.5)
# overlay.shutdownOverlay()
x_pos = 0
y_pos = 0
z_pos = 0