Function ws_handler

demo_render/interactive_viewer/server.py:592–776

(request)

        return web.json_response(state.animator.to_list())

    async def ws_handler(request):
        ws = web.WebSocketResponse()
        await ws.prepare(request)
        print("[ws] Client connected")

        # Send initial frame
        try:
            frame_data = state.render()
            if frame_data:
                await ws.send_bytes(frame_data)
                print(f"[ws] Sent initial frame ({len(frame_data)} bytes)")
        except Exception as e:
            print(f"[ws] Error sending initial frame: {e}")

        need_render = False

        async for msg in ws:
            if msg.type == web.WSMsgType.TEXT:
                data = json.loads(msg.data)
                t = data.get("type")

                if t == "orbit":
                    state.camera.orbit(data["dx"], data["dy"])
                    need_render = True
                elif t == "pan":
                    state.camera.pan(data["dx"], data["dy"])
                    need_render = True
                elif t == "zoom":
                    state.camera.zoom(data["delta"])
                    need_render = True
                elif t == "move":
                    speed = move_speed_base
                    if data.get("shift"):
                        speed *= 3.0
                    d = data["dir"]
                    if d == "forward":
                        state.camera.move_forward(speed)
                    elif d == "backward":
                        state.camera.move_forward(-speed)
                    elif d == "right":
                        state.camera.move_right(speed)
                    elif d == "left":
                        state.camera.move_right(-speed)
                    elif d == "up":
                        state.camera.move_up(speed)
                    elif d == "down":
                        state.camera.move_up(-speed)
                    need_render = True
                elif t == "set_frame":
                    state.frame_idx = int(data["value"])
                    state._dirty_geometry = True
                    need_render = True
                elif t == "set_display_mode":
                    state.display_mode = data["value"]
                    state._dirty_geometry = True
                    need_render = True
                elif t == "set_conf_threshold":
                    state.conf_threshold = float(data["value"])