class ExposureWidget(DebugObject):
""" Widget to show the current exposure """
def __init__(self, pipeline, parent):
""" Inits the widget """
DebugObject.__init__(self)
self._pipeline = pipeline
self._parent = parent
self._node = self._parent.attach_new_node("ExposureWidgetNode")
self._create_components()
def _create_components(self):
""" Internal method to init the widgets components """
self._node.hide()
# Create the texture where the gui component is rendered inside
self._storage_tex = Image.create_2d("ExposureDisplay", 140, 20, Texture.T_unsigned_byte, Texture.F_rgba8)
self._storage_tex.set_clear_color(Vec4(0.2, 0.6, 1.0, 1.0))
self._storage_tex.clear_image()
self._bg_frame = DirectFrame(parent=self._node, frameColor=(0.1, 0.1, 0.1, 1.0),
frameSize=(200, 0, -10, -85), pos=(0, 0, 0))
self._display_img = BetterOnscreenImage(
image=self._storage_tex, parent=self._node, w=140, h=20, x=20, y=50)
self._display_txt = BetterOnscreenText(
text="Current Exposure".upper(), parent=self._node, x=160, y=40, size=13,
color=Vec3(0.8), align="right")
# Create the shader which generates the visualization texture
self._cshader_node = ComputeNode("ExposureWidget")
self._cshader_node.add_dispatch(140 // 10, 20 // 4, 1)
self._cshader_np = self._node.attach_new_node(self._cshader_node)
# Defer the further loading
Globals.base.taskMgr.doMethodLater(1.0, self._late_init, "ExposureLateInit")
def _late_init(self, task):
""" Gets called after the pipeline initialized, this extracts the
exposure texture from the stage manager """
stage_mgr = self._pipeline.stage_mgr
if not stage_mgr.has_pipe("Exposure"):
self.debug("Disabling exposure widget, could not find the exposure data.")
self._node.remove_node()
return
self._node.show()
exposure_tex = stage_mgr.get_pipe("Exposure")
self._cshader = Shader.load_compute(Shader.SL_GLSL, "Shader/GUI/VisualizeExposure.compute.glsl")
self._cshader_np.set_shader(self._cshader)
self._cshader_np.set_shader_input("DestTex", self._storage_tex)
self._cshader_np.set_shader_input("ExposureTex", exposure_tex)
class FPSChart(RPObject):
""" Widget to show the FPS as a chart """
def __init__(self, pipeline, parent):
""" Inits the widget """
RPObject.__init__(self)
self._pipeline = pipeline
self._parent = parent
self._node = self._parent.attach_new_node("FPSChartNode")
self._create_components()
def _create_components(self):
""" Internal method to init the widgets components """
# Create the buffer which stores the last FPS values
self._storage_buffer = Image.create_buffer("FPSValues", 250, "R16")
self._storage_buffer.set_clear_color(Vec4(0))
self._storage_buffer.clear_image()
self._store_index = PTAInt.empty_array(1)
self._store_index[0] = 0
self._current_ftime = PTAFloat.empty_array(1)
self._current_ftime[0] = 16.0
self._chart_ms_max = PTAFloat.empty_array(1)
self._chart_ms_max[0] = 40
# Create the texture where the gui component is rendered inside
self._display_tex = Image.create_2d("FPSChartRender", 250, 120,
"RGBA8")
self._display_tex.set_clear_color(Vec4(0))
self._display_tex.clear_image()
self._display_img = Sprite(image=self._display_tex,
parent=self._node,
w=250,
h=120,
x=10,
y=10)
# Defer the further loading
Globals.base.taskMgr.doMethodLater(0.3, self._late_init,
"FPSChartInit")
def _late_init(self, task):
""" Gets called after the pipeline was initialized """
self._display_txt = Text(text="40 ms",
parent=self._node,
x=20,
y=25,
size=13,
color=Vec3(1),
may_change=True)
self._display_txt_bottom = Text(text="0 ms",
parent=self._node,
x=20,
y=120,
size=13,
color=Vec3(1),
may_change=True)
# Create the shader which generates the visualization texture
self._cshader_node = ComputeNode("FPSChartUpdateChart")
self._cshader_node.add_dispatch(250 // 10, 120 // 4, 1)
self._cshader_np = self._node.attach_new_node(self._cshader_node)
self._cshader = RPLoader.load_shader(
"/$$rp/shader/fps_chart.compute.glsl")
self._cshader_np.set_shader(self._cshader)
self._cshader_np.set_shader_inputs(DestTex=self._display_tex,
FPSValues=self._storage_buffer,
index=self._store_index,
maxMs=self._chart_ms_max)
self._update_shader_node = ComputeNode("FPSChartUpdateValues")
self._update_shader_node.add_dispatch(1, 1, 1)
self._update_shader_np = self._node.attach_new_node(
self._update_shader_node)
self._ushader = RPLoader.load_shader(
"/$$rp/shader/fps_chart_update.compute.glsl")
self._update_shader_np.set_shader(self._ushader)
self._update_shader_np.set_shader_inputs(
DestTex=self._storage_buffer,
index=self._store_index,
currentData=self._current_ftime)
Globals.base.addTask(self._update, "UpdateFPSChart", sort=-50)
return task.done
def _update(self, task):
""" Updates the widget """
self._store_index[0] = (self._store_index[0] + 1) % 250
self._current_ftime[0] = Globals.clock.get_dt() * 1000.0
avg_fps = Globals.clock.get_average_frame_rate()
if avg_fps > 122:
self._chart_ms_max[0] = 10.0
elif avg_fps > 62:
self._chart_ms_max[0] = 20.0
elif avg_fps > 32:
self._chart_ms_max[0] = 40.0
elif avg_fps > 15:
self._chart_ms_max[0] = 70.0
else:
self._chart_ms_max[0] = 200.0
self._display_txt.set_text(str(int(self._chart_ms_max[0])) + " ms")
return task.cont
class ExposureWidget(RPObject):
""" Widget to show the current exposure """
def __init__(self, pipeline, parent):
""" Inits the widget """
RPObject.__init__(self)
self._pipeline = pipeline
self._parent = parent
self._node = self._parent.attach_new_node("ExposureWidgetNode")
self._create_components()
def _create_components(self):
""" Internal method to init the widgets components """
self._node.hide()
# Create the texture where the gui component is rendered inside
self._storage_tex = Image.create_2d("ExposureDisplay", 140, 20,
"RGBA8")
self._storage_tex.set_clear_color(Vec4(0.2, 0.6, 1.0, 1.0))
self._storage_tex.clear_image()
self._bg_frame = DirectFrame(parent=self._node,
frameColor=(0.1, 0.1, 0.1, 1.0),
frameSize=(200, 0, -10, -85),
pos=(0, 0, 0))
self._display_img = Sprite(image=self._storage_tex,
parent=self._node,
w=140,
h=20,
x=20,
y=50)
self._display_txt = Text(text="Current Exposure".upper(),
parent=self._node,
x=160,
y=40,
size=13,
color=Vec3(0.8),
align="right")
# Create the shader which generates the visualization texture
self._cshader_node = ComputeNode("ExposureWidget")
self._cshader_node.add_dispatch(140 // 10, 20 // 4, 1)
self._cshader_np = self._node.attach_new_node(self._cshader_node)
# Defer the further loading
Globals.base.taskMgr.doMethodLater(1.0, self._late_init,
"ExposureLateInit")
def _late_init(self, task):
""" Gets called after the pipeline initialized, this extracts the
exposure texture from the stage manager """
stage_mgr = self._pipeline.stage_mgr
if "Exposure" not in stage_mgr.pipes:
self.debug(
"Disabling exposure widget, could not find the exposure data.")
self._node.remove_node()
return
self._node.show()
exposure_tex = stage_mgr.pipes["Exposure"]
self._cshader = RPLoader.load_shader(
"/$$rp/shader/visualize_exposure.compute.glsl")
self._cshader_np.set_shader(self._cshader)
self._cshader_np.set_shader_input("DestTex", self._storage_tex)
self._cshader_np.set_shader_input("ExposureTex", exposure_tex)
return task.done
class FPSChart(RPObject):
""" Widget to show the FPS as a chart """
def __init__(self, pipeline, parent):
""" Inits the widget """
RPObject.__init__(self)
self._pipeline = pipeline
self._parent = parent
self._node = self._parent.attach_new_node("FPSChartNode")
self._create_components()
def _create_components(self):
""" Internal method to init the widgets components """
# Create the buffer which stores the last FPS values
self._storage_buffer = Image.create_buffer("FPSValues", 250, "R16")
self._storage_buffer.set_clear_color(Vec4(0))
self._storage_buffer.clear_image()
self._store_index = PTAInt.empty_array(1)
self._store_index[0] = 0
self._current_ftime = PTAFloat.empty_array(1)
self._current_ftime[0] = 16.0
self._chart_ms_max = PTAFloat.empty_array(1)
self._chart_ms_max[0] = 40
# Create the texture where the gui component is rendered inside
self._display_tex = Image.create_2d("FPSChartRender", 250, 120, "RGBA8")
self._display_tex.set_clear_color(Vec4(0))
self._display_tex.clear_image()
self._display_img = Sprite(
image=self._display_tex, parent=self._node, w=250, h=120, x=10, y=10)
# Defer the further loading
Globals.base.taskMgr.doMethodLater(0.3, self._late_init, "FPSChartInit")
def _late_init(self, task):
""" Gets called after the pipeline was initialized """
self._display_txt = Text(
text="40 ms", parent=self._node, x=20, y=25,
size=13, color=Vec3(1), may_change=True)
self._display_txt_bottom = Text(
text="0 ms", parent=self._node, x=20, y=120,
size=13, color=Vec3(1), may_change=True)
# Create the shader which generates the visualization texture
self._cshader_node = ComputeNode("FPSChartUpdateChart")
self._cshader_node.add_dispatch(250 // 10, 120 // 4, 1)
self._cshader_np = self._node.attach_new_node(self._cshader_node)
self._cshader = RPLoader.load_shader("/$$rp/shader/fps_chart.compute.glsl")
self._cshader_np.set_shader(self._cshader)
self._cshader_np.set_shader_input("DestTex", self._display_tex)
self._cshader_np.set_shader_input("FPSValues", self._storage_buffer)
self._cshader_np.set_shader_input("index", self._store_index)
self._cshader_np.set_shader_input("maxMs", self._chart_ms_max)
self._update_shader_node = ComputeNode("FPSChartUpdateValues")
self._update_shader_node.add_dispatch(1, 1, 1)
self._update_shader_np = self._node.attach_new_node(self._update_shader_node)
self._ushader = RPLoader.load_shader("/$$rp/shader/fps_chart_update.compute.glsl")
self._update_shader_np.set_shader(self._ushader)
self._update_shader_np.set_shader_input("DestTex", self._storage_buffer)
self._update_shader_np.set_shader_input("index", self._store_index)
self._update_shader_np.set_shader_input("currentData", self._current_ftime)
Globals.base.addTask(self._update, "UpdateFPSChart", sort=-50)
return task.done
def _update(self, task):
""" Updates the widget """
self._store_index[0] = (self._store_index[0] + 1) % 250
self._current_ftime[0] = Globals.clock.get_dt() * 1000.0
avg_fps = Globals.clock.get_average_frame_rate()
if avg_fps > 122:
self._chart_ms_max[0] = 10.0
elif avg_fps > 62:
self._chart_ms_max[0] = 20.0
elif avg_fps > 32:
self._chart_ms_max[0] = 40.0
elif avg_fps > 15:
self._chart_ms_max[0] = 70.0
else:
self._chart_ms_max[0] = 200.0
self._display_txt.set_text(str(int(self._chart_ms_max[0])) + " ms")
return task.cont