IfcOpenShell/src/ifcviewer/ViewportWindow.cpp

/********************************************************************************
 *                                                                              *
 * This file is part of IfcOpenShell.                                           *
 *                                                                              *
 * IfcOpenShell is free software: you can redistribute it and/or modify         *
 * it under the terms of the Lesser GNU General Public License as published by  *
 * the Free Software Foundation, either version 3.0 of the License, or          *
 * (at your option) any later version.                                          *
 *                                                                              *
 * IfcOpenShell is distributed in the hope that it will be useful,              *
 * but WITHOUT ANY WARRANTY; without even the implied warranty of               *
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the                 *
 * Lesser GNU General Public License for more details.                          *
 *                                                                              *
 * You should have received a copy of the Lesser GNU General Public License     *
 * along with this program. If not, see <http://www.gnu.org/licenses/>.         *
 *                                                                              *
 ********************************************************************************/

#include "ViewportWindow.h"

#include "AppSettings.h"

#include <QMouseEvent>
#include <QKeyEvent>
#include <QWheelEvent>
#include <QSurfaceFormat>
#include <QCoreApplication>
#include <QCursor>
#include <QTimer>
#include <QtMath>
#include <QtOpenGL/QOpenGLVersionFunctionsFactory>

#include <cstring>
#include <cstdlib>
#include <cmath>
#include <algorithm>
#include <limits>

static const size_t INITIAL_VBO_SIZE  = 64 * 1024 * 1024;   // 64 MB
static const size_t INITIAL_EBO_SIZE  = 32 * 1024 * 1024;   // 32 MB
static const size_t INITIAL_SSBO_SIZE = 4  * 1024 * 1024;   // 4 MB (~52k instances)
static const size_t MAX_BUFFER_SIZE   = 4ull * 1024 * 1024 * 1024;  // 4 GB

static_assert(sizeof(DrawElementsIndirectCommand) == 20, "indirect cmd must be 20 bytes");

// -----------------------------------------------------------------------------
// Shaders
// -----------------------------------------------------------------------------
//
// Vertex layout (GL side, 12 bytes — quantized; see InstancedGeometry.h):
//   location 0: vec3 a_position_q   (u16x3 normalized, per-mesh AABB basis)
//   location 1: vec2 a_normal_oct   (i8x2 normalized, octahedral)
//   location 2: vec4 a_color        (u8x4 normalized)
//
// Per-instance record in SSBO std430 (80 bytes):
//   mat4 transform
//   uint object_id
//   uint color_override_rgba8     -- 0 => use baked a_color
//   uint mesh_id                  -- index into per-model MeshGpu[]
//   uint _pad1
//
// The draw calls pass `u_instance_offset = mesh.first_instance`; the shader
// reads `instances[u_instance_offset + gl_InstanceID]`.

static const char* MAIN_VERTEX_SHADER = R"(
#version 450 core
#extension GL_ARB_shader_draw_parameters : require
// Quantized vertex inputs — see InstancedGeometry.h for layout.
layout(location = 0) in vec3 a_position_q;  // u16x3 normalized -> [0,1]
layout(location = 1) in vec2 a_normal_oct;  // i8x2 normalized -> [-1,1]
layout(location = 2) in vec4 a_color;

struct InstanceRecord {
    mat4 transform;
    uint object_id;
    uint color_override;
    uint mesh_id;
    uint _pad1;
};
layout(std430, binding = 0) readonly buffer Instances {
    InstanceRecord instances[];
};
layout(std430, binding = 1) readonly buffer VisibleIndices {
    uint visible[];
};
struct MeshQuant { vec4 aabb_min; vec4 aabb_max; };
layout(std430, binding = 2) readonly buffer Meshes {
    MeshQuant meshes[];
};
// Per-object_id selection flag (bit 0 = in selection).  Sized so that
// any object_id present in the scene is a valid index — see SelectionState.
layout(std430, binding = 3) readonly buffer SelectionFlags {
    uint sel_flags[];
};

uniform mat4 u_view_projection;
uniform uint u_active_id;       // last single-clicked id (0 = none)

out vec3 v_normal;
out vec4 v_color;
out vec3 v_world_pos;
flat out uint v_object_id;
flat out uint v_selected;
flat out uint v_active;

// Meyer et al. octahedral normal decode.  Input is in [-1,1]^2.
vec3 octDecode(vec2 e) {
    vec3 n = vec3(e.xy, 1.0 - abs(e.x) - abs(e.y));
    if (n.z < 0.0) n.xy = (1.0 - abs(n.yx)) * vec2(n.x >= 0.0 ? 1.0 : -1.0,
                                                    n.y >= 0.0 ? 1.0 : -1.0);
    return normalize(n);
}

void main() {
    uint slot = uint(gl_BaseInstanceARB) + uint(gl_InstanceID);
    uint iid = visible[slot];
    InstanceRecord inst = instances[iid];
    MeshQuant mq = meshes[inst.mesh_id];

    // Dequantize local position against this mesh's AABB.
    vec3 pos_local = mix(mq.aabb_min.xyz, mq.aabb_max.xyz, a_position_q);

    vec4 world = inst.transform * vec4(pos_local, 1.0);
    v_world_pos = world.xyz;
    gl_Position = u_view_projection * world;

    // Rotate the normal by the upper-3x3 of the transform.  BIM placements
    // are overwhelmingly rigid rotations (+ optional uniform scale +
    // optional reflection), so we skip the full inverse-transpose but do
    // need to flip the normal when the transform contains a reflection,
    // otherwise mirrored instances shade as if inside-out.  The same
    // determinant sign is what GL_CULL_FACE uses to decide winding, so
    // keeping them in agreement means backface culling is safe to enable.
    vec3 n_local = octDecode(a_normal_oct);
    mat3 rot = mat3(inst.transform);
    vec3 n = rot * n_local;
    if (determinant(rot) < 0.0) n = -n;
    v_normal = normalize(n);

    vec4 baked = a_color;
    if (inst.color_override != 0u) {
        float r = float((inst.color_override      ) & 0xFFu) / 255.0;
        float g = float((inst.color_override >>  8) & 0xFFu) / 255.0;
        float b = float((inst.color_override >> 16) & 0xFFu) / 255.0;
        float a = float((inst.color_override >> 24) & 0xFFu) / 255.0;
        if (a > 0.0) baked = vec4(r, g, b, a);
    }
    v_color = baked;

    v_object_id = inst.object_id;
    // Index by object_id directly; SelectionState guarantees the buffer
    // is sized to cover every id allocated to the scene.
    v_selected  = ((sel_flags[v_object_id] & 1u) != 0u) ? 1u : 0u;
    v_active    = (v_object_id == u_active_id) ? 1u : 0u;
}
)";

static const char* MAIN_FRAGMENT_SHADER = R"(
#version 450 core
in vec3 v_normal;
in vec4 v_color;
in vec3 v_world_pos;
flat in uint v_object_id;
flat in uint v_selected;
flat in uint v_active;

uniform vec3 u_light_dir;       // primary key direction (world-space)
uniform vec3 u_fill_dir;        // secondary fill direction
uniform vec3 u_sky_color;       // hemisphere top tint
uniform vec3 u_ground_color;    // hemisphere bottom tint

// Section planes — clip the half-space dot(n,p)+d > 0.  AND-combined.
const int MAX_CLIP_PLANES = 8;
uniform int  u_clip_count;
uniform vec4 u_clip_planes[MAX_CLIP_PLANES];

out vec4 frag_color;

void main() {
    for (int i = 0; i < u_clip_count; ++i) {
        if (dot(u_clip_planes[i].xyz, v_world_pos) + u_clip_planes[i].w > 0.0) {
            discard;
        }
    }
    // v_normal already has the reflection flip applied in the vertex
    // shader.  When backface culling is off, open shells let us see the
    // "wrong" side of a face — flip based on gl_FrontFacing so both
    // sides light correctly.  When culling is on this branch is always
    // true and has no effect.
    vec3 n = normalize(v_normal);
    if (!gl_FrontFacing) n = -n;

    // Hemisphere ambient: faces pointing up read sky, down read ground.
    // World-up is +Z so n.z drives the mix.  Floors/ceilings/walls get
    // visibly distinct ambient even when shadowed.
    float hemi_t = 0.5 + 0.5 * n.z;
    vec3  ambient = mix(u_ground_color, u_sky_color, hemi_t);

    // Key + fill direct light.  Fill is a softer secondary so backs of
    // objects don't go pitch black — keeps shape readable from any angle.
    float key  = max(dot(n, u_light_dir), 0.0);
    float fill = max(dot(n, u_fill_dir),  0.0) * 0.35;

    vec3 color = v_color.rgb * (ambient + (key + fill) * 0.7);

    // Cavity shading: where adjacent fragments have a sharp normal change
    // (concave creases, edges where two faces meet), darken slightly.
    // length(fwidth(n)) spikes at those boundaries; the clamp caps how
    // dark a single edge can get so the effect is a hint, not a heavy
    // outline.
    float cavity = clamp(length(fwidth(n)) * 1.5, 0.0, 0.35);
    color *= (1.0 - cavity);

    // Selected (member of the multi-set) gets a blue tint; the active
    // (last single-clicked) gets a stronger mix so the user can always
    // tell which one drives the properties panel.
    if (v_selected == 1u) color = mix(color, vec3(0.2, 0.6, 1.0), 0.45);
    if (v_active   == 1u) color = mix(color, vec3(0.4, 0.8, 1.0), 0.40);
    frag_color = vec4(color, v_color.a);
}
)";

static const char* PICK_VERTEX_SHADER = R"(
#version 450 core
#extension GL_ARB_shader_draw_parameters : require
layout(location = 0) in vec3 a_position_q;
layout(location = 1) in vec2 a_normal_oct;

struct InstanceRecord {
    mat4 transform;
    uint object_id;
    uint color_override;
    uint mesh_id;
    uint _pad1;
};
layout(std430, binding = 0) readonly buffer Instances {
    InstanceRecord instances[];
};
layout(std430, binding = 1) readonly buffer VisibleIndices {
    uint visible[];
};
struct MeshQuant { vec4 aabb_min; vec4 aabb_max; };
layout(std430, binding = 2) readonly buffer Meshes {
    MeshQuant meshes[];
};

uniform mat4 u_view_projection;

flat out uint v_object_id;
out vec3 v_world_pos;
out vec3 v_world_normal;

vec3 octDecode(vec2 e) {
    vec3 n = vec3(e.xy, 1.0 - abs(e.x) - abs(e.y));
    if (n.z < 0.0) n.xy = (1.0 - abs(n.yx)) * vec2(n.x >= 0.0 ? 1.0 : -1.0,
                                                    n.y >= 0.0 ? 1.0 : -1.0);
    return normalize(n);
}

void main() {
    uint slot = uint(gl_BaseInstanceARB) + uint(gl_InstanceID);
    uint iid = visible[slot];
    InstanceRecord inst = instances[iid];
    MeshQuant mq = meshes[inst.mesh_id];
    vec3 pos_local = mix(mq.aabb_min.xyz, mq.aabb_max.xyz, a_position_q);
    vec4 world = inst.transform * vec4(pos_local, 1.0);
    v_world_pos = world.xyz;
    gl_Position = u_view_projection * world;

    vec3 n_local = octDecode(a_normal_oct);
    mat3 rot = mat3(inst.transform);
    vec3 n = rot * n_local;
    if (determinant(rot) < 0.0) n = -n;
    v_world_normal = normalize(n);

    v_object_id = inst.object_id;
}
)";

static const char* PICK_FRAGMENT_SHADER = R"(
#version 450 core
flat in uint v_object_id;
in vec3 v_world_pos;
in vec3 v_world_normal;

const int MAX_CLIP_PLANES = 8;
uniform int  u_clip_count;
uniform vec4 u_clip_planes[MAX_CLIP_PLANES];

layout(location = 0) out uint frag_id;
layout(location = 1) out vec3 frag_pos;
layout(location = 2) out vec3 frag_normal;

void main() {
    for (int i = 0; i < u_clip_count; ++i) {
        if (dot(u_clip_planes[i].xyz, v_world_pos) + u_clip_planes[i].w > 0.0) {
            discard;
        }
    }
    frag_id     = v_object_id;
    frag_pos    = v_world_pos;
    frag_normal = normalize(v_world_normal);
}
)";

static const char* AXIS_VERTEX_SHADER = R"(
#version 450 core
layout(location = 0) in vec3 a_position;
layout(location = 1) in vec3 a_color;
uniform mat4 u_mvp;
out vec3 v_color;
void main() {
    gl_Position = u_mvp * vec4(a_position, 1.0);
    v_color = a_color;
}
)";

static const char* AXIS_FRAGMENT_SHADER = R"(
#version 450 core
in vec3 v_color;
out vec4 frag_color;
void main() { frag_color = vec4(v_color, 1.0); }
)";

// Pivot indicator: a small 3D axis cross drawn at u_pivot in world space.
// a_local is the unit endpoint of one of the six arm vertices (±X,±Y,±Z),
// scaled by u_arm_world so the cross keeps a roughly constant on-screen size
// across zoom levels.  Drawing in world space (not screen-aligned) is the
// whole point: it tumbles visibly when the camera orbits, which makes it
// unambiguous that the marker is anchored to a 3D point and not glued to
// the screen.  Per-vertex color gives RGB axes matching the corner gizmo.
static const char* PIVOT_VERTEX_SHADER = R"(
#version 450 core
layout(location = 0) in vec3 a_local;
layout(location = 1) in vec3 a_color;
uniform mat4  u_mvp;
uniform vec3  u_pivot;
uniform float u_arm_world;
out vec3 v_color;
void main() {
    gl_Position = u_mvp * vec4(u_pivot + a_local * u_arm_world, 1.0);
    v_color = a_color;
}
)";

static const char* PIVOT_FRAGMENT_SHADER = R"(
#version 450 core
in vec3 v_color;
uniform float u_alpha;
out vec4 frag_color;
void main() { frag_color = vec4(v_color, u_alpha); }
)";

// Section plane gizmo: a unit-extent quad + arrow expressed in plane-local
// space, transformed to world via the per-plane basis (u, v, n).  The arrow
// shaft sits along +z (== +n in plane-local terms), so dragging the arrow
// always slides the plane along its normal regardless of camera orientation.
static const char* PLANE_VERTEX_SHADER = R"(
#version 450 core
layout(location = 0) in vec3 a_local;     // (u, v, n) in plane-local space
layout(location = 1) in vec3 a_color;
uniform mat4  u_vp;
uniform vec3  u_origin;
uniform vec3  u_axis_u;
uniform vec3  u_axis_v;
uniform vec3  u_axis_n;
uniform float u_size;
uniform vec4  u_tint;
out vec4 v_color;
void main() {
    vec3 world = u_origin
               + a_local.x * u_axis_u * u_size
               + a_local.y * u_axis_v * u_size
               + a_local.z * u_axis_n * u_size;
    gl_Position = u_vp * vec4(world, 1.0);
    v_color = vec4(a_color, 1.0) * u_tint;
}
)";

static const char* PLANE_FRAGMENT_SHADER = R"(
#version 450 core
in vec4 v_color;
out vec4 frag_color;
void main() { frag_color = v_color; }
)";

// Edge pass — fullscreen triangle generated from gl_VertexID, samples
// resolved depth at four cardinal neighbours, computes a laplacian, and
// outputs a per-pixel darkening factor that gets multiplied into the
// existing colour via GL_DST_COLOR * GL_ZERO blending.
static const char* EDGE_VERTEX_SHADER = R"(
#version 450 core
out vec2 v_uv;
void main() {
    vec2 pos = vec2((gl_VertexID & 1) << 2, (gl_VertexID & 2) << 1) - 1.0;
    v_uv = pos * 0.5 + 0.5;
    gl_Position = vec4(pos, 0.0, 1.0);
}
)";

static const char* EDGE_FRAGMENT_SHADER = R"(
#version 450 core
in vec2 v_uv;
uniform sampler2D u_depth;
uniform vec2  u_texel;       // 1.0 / depth-texture size
uniform float u_near;
uniform float u_far;
uniform float u_is_ortho;    // 0 or 1
uniform float u_scale;       // edge-darkening multiplier
uniform float u_threshold;   // laplacian floor (relative to depth)
out vec4 frag_color;

float linearize(float z) {
    if (u_is_ortho > 0.5) {
        // Ortho: depth buffer is already a linear remap of view-z.
        return mix(u_near, u_far, z);
    }
    // Perspective: standard NDC -> view-z reverse projection.
    float ndc = z * 2.0 - 1.0;
    return (2.0 * u_near * u_far) / (u_far + u_near - ndc * (u_far - u_near));
}

void main() {
    float c = linearize(texture(u_depth, v_uv).r);
    float n = linearize(texture(u_depth, v_uv + vec2( 0.0,  u_texel.y)).r);
    float s = linearize(texture(u_depth, v_uv + vec2( 0.0, -u_texel.y)).r);
    float e = linearize(texture(u_depth, v_uv + vec2( u_texel.x, 0.0)).r);
    float w = linearize(texture(u_depth, v_uv + vec2(-u_texel.x, 0.0)).r);

    // Laplacian magnitude: ~0 on smooth surfaces, large at depth jumps.
    // Threshold scales with depth so distant edges still register.
    float lap   = abs(4.0 * c - n - s - e - w);
    float t     = u_threshold * c;
    float edge  = clamp((lap - t) * u_scale, 0.0, 0.6);

    // Multiplicative blend (GL_DST_COLOR, GL_ZERO): out = dst * rgb.
    frag_color = vec4(vec3(1.0 - edge), 1.0);
}
)";

static GLuint compileShader(QOpenGLFunctions_4_5_Core* gl, GLenum type, const char* source) {
    GLuint shader = gl->glCreateShader(type);
    gl->glShaderSource(shader, 1, &source, nullptr);
    gl->glCompileShader(shader);
    GLint ok = 0;
    gl->glGetShaderiv(shader, GL_COMPILE_STATUS, &ok);
    if (!ok) {
        char log[2048];
        gl->glGetShaderInfoLog(shader, sizeof(log), nullptr, log);
        qWarning("Shader compile error: %s", log);
    }
    return shader;
}

static const char* HIZ_DOWNSAMPLE_VS = R"(
#version 450 core
void main() {
    vec2 pos = vec2((gl_VertexID & 1) * 4.0 - 1.0,
                    (gl_VertexID & 2) * 2.0 - 1.0);
    gl_Position = vec4(pos, 0.0, 1.0);
}
)";

static const char* HIZ_DOWNSAMPLE_FS = R"(
#version 450 core
uniform sampler2D u_depth;
uniform vec2 u_inv_dest_size;
void main() {
    vec2 uv = gl_FragCoord.xy * u_inv_dest_size;
    gl_FragDepth = texture(u_depth, uv).r;
}
)";


static GLuint linkProgram(QOpenGLFunctions_4_5_Core* gl, GLuint vert, GLuint frag) {
    GLuint prog = gl->glCreateProgram();
    gl->glAttachShader(prog, vert);
    gl->glAttachShader(prog, frag);
    gl->glLinkProgram(prog);
    GLint ok = 0;
    gl->glGetProgramiv(prog, GL_LINK_STATUS, &ok);
    if (!ok) {
        char log[2048];
        gl->glGetProgramInfoLog(prog, sizeof(log), nullptr, log);
        qWarning("Program link error: %s", log);
    }
    gl->glDeleteShader(vert);
    gl->glDeleteShader(frag);
    return prog;
}

// -----------------------------------------------------------------------------

// Meyer et al. octahedral normal encode.  Input unit vector -> [-1,1]^2.
static void octEncode(const float n[3], float out[2]) {
    float ax = std::fabs(n[0]), ay = std::fabs(n[1]), az = std::fabs(n[2]);
    float denom = ax + ay + az;
    if (denom < 1e-12f) { out[0] = 0.0f; out[1] = 0.0f; return; }
    float px = n[0] / denom;
    float py = n[1] / denom;
    if (n[2] < 0.0f) {
        float sx = px >= 0.0f ? 1.0f : -1.0f;
        float sy = py >= 0.0f ? 1.0f : -1.0f;
        float nx = (1.0f - std::fabs(py)) * sx;
        float ny = (1.0f - std::fabs(px)) * sy;
        px = nx; py = ny;
    }
    out[0] = px;
    out[1] = py;
}

// Quantize a streamer-format vertex (pos3 + normal3 + color-as-float) into
// the 12 B VBO record, given the mesh's tight local AABB.  `extent_recip`
// is 1/(max-min) per axis, or 0 for degenerate axes (quantum becomes 0).
static void quantizeVertex(const float src[7],
                           const float aabb_min[3],
                           const float extent_recip[3],
                           uint8_t dst[INSTANCED_VERTEX_STRIDE_BYTES]) {
    // Position -> u16 normalized.
    uint16_t* p = reinterpret_cast<uint16_t*>(dst + INSTANCED_VERTEX_POS_OFFSET);
    for (int a = 0; a < 3; ++a) {
        float t = (src[a] - aabb_min[a]) * extent_recip[a];
        if (t < 0.0f) t = 0.0f; else if (t > 1.0f) t = 1.0f;
        p[a] = static_cast<uint16_t>(t * 65535.0f + 0.5f);
    }
    // Normal -> oct i8x2.  int8 gives ~1.4° worst-case error — fine for BIM.
    float oct[2];
    octEncode(src + 3, oct);
    int8_t* n = reinterpret_cast<int8_t*>(dst + INSTANCED_VERTEX_NORMAL_OFFSET);
    for (int a = 0; a < 2; ++a) {
        float v = oct[a];
        if (v < -1.0f) v = -1.0f; else if (v > 1.0f) v = 1.0f;
        n[a] = static_cast<int8_t>(std::lrintf(v * 127.0f));
    }
    // Color passes through — streamer packs 4 bytes into the 7th float slot.
    std::memcpy(dst + INSTANCED_VERTEX_COLOR_OFFSET, src + 6, 4);
}

// Determinant of the upper-left 3x3 of a column-major mat4 stored as 16 floats.
// Sign tells us whether the transform contains a reflection, which is what
// decides which glFrontFace winding to draw the instance with.
static bool transformIsReflected(const float t[16]) {
    const float det =
        t[0] * (t[5] * t[10] - t[9] * t[6])
      - t[4] * (t[1] * t[10] - t[9] * t[2])
      + t[8] * (t[1] * t[6]  - t[5] * t[2]);
    return det < 0.0f;
}

static bool aabbInFrustum(const float aabb_min[3], const float aabb_max[3],
                          const float planes[6][4]) {
    for (int p = 0; p < 6; ++p) {
        float px = planes[p][0] >= 0.0f ? aabb_max[0] : aabb_min[0];
        float py = planes[p][1] >= 0.0f ? aabb_max[1] : aabb_min[1];
        float pz = planes[p][2] >= 0.0f ? aabb_max[2] : aabb_min[2];
        float dist = planes[p][0] * px + planes[p][1] * py + planes[p][2] * pz + planes[p][3];
        if (dist < 0.0f) return false;
    }
    return true;
}

static void extractFrustumPlanes(const QMatrix4x4& vp, float planes[6][4]) {
    for (int i = 0; i < 4; ++i) {
        planes[0][i] = vp(3, i) + vp(0, i);
        planes[1][i] = vp(3, i) - vp(0, i);
        planes[2][i] = vp(3, i) + vp(1, i);
        planes[3][i] = vp(3, i) - vp(1, i);
        planes[4][i] = vp(3, i) + vp(2, i);
        planes[5][i] = vp(3, i) - vp(2, i);
    }
    for (int p = 0; p < 6; ++p) {
        float len = std::sqrt(planes[p][0]*planes[p][0] +
                              planes[p][1]*planes[p][1] +
                              planes[p][2]*planes[p][2]);
        if (len > 0.0f) {
            float inv = 1.0f / len;
            planes[p][0] *= inv; planes[p][1] *= inv;
            planes[p][2] *= inv; planes[p][3] *= inv;
        }
    }
}

// Compute world AABB by transforming the 8 corners of `local_min..local_max`
// through the column-major 4x4 `M` and bounding the result.  Same maths as
// GeometryStreamer's worldAabbFromLocal; duplicated here so the viewport
// can recompute world AABBs independently when stage matrices change.
static void worldAabbFromLocalVp(const float local_min[3],
                                 const float local_max[3],
                                 const float M[16],
                                 float out_min[3], float out_max[3]) {
    out_min[0] = out_min[1] = out_min[2] =  std::numeric_limits<float>::max();
    out_max[0] = out_max[1] = out_max[2] = -std::numeric_limits<float>::max();
    for (int c = 0; c < 8; ++c) {
        float x = (c & 1) ? local_max[0] : local_min[0];
        float y = (c & 2) ? local_max[1] : local_min[1];
        float z = (c & 4) ? local_max[2] : local_min[2];
        float wx = M[0]*x + M[4]*y + M[8]*z  + M[12];
        float wy = M[1]*x + M[5]*y + M[9]*z  + M[13];
        float wz = M[2]*x + M[6]*y + M[10]*z + M[14];
        if (wx < out_min[0]) out_min[0] = wx; if (wx > out_max[0]) out_max[0] = wx;
        if (wy < out_min[1]) out_min[1] = wy; if (wy > out_max[1]) out_max[1] = wy;
        if (wz < out_min[2]) out_min[2] = wz; if (wz > out_max[2]) out_max[2] = wz;
    }
}

// Build bvh_items (one per instance, 1:1 ordering) and a per-model BVH.
// Items with instances.size() < BVH_MIN_OBJECTS leave bvh empty — the
// render path falls back to drawing every instance.
static void buildBvhForModel(ModelGpuData& m, uint32_t model_id) {
    m.bvh_items.clear();
    m.bvh_items.reserve(m.instances.size());
    for (const auto& inst : m.instances) {
        BvhItem it;
        std::memcpy(it.aabb_min, inst.world_aabb_min, sizeof(it.aabb_min));
        std::memcpy(it.aabb_max, inst.world_aabb_max, sizeof(it.aabb_max));
        it.model_id = inst.model_id;
        m.bvh_items.push_back(it);
    }
    if (m.bvh_items.size() >= BVH_MIN_OBJECTS) {
        m.bvh = buildModelBvhOne(m.bvh_items, model_id);
    } else {
        m.bvh = ModelBvh{};
    }
}

ViewportWindow::ViewportWindow(QWindow* parent)
    : QWindow(parent)
{
    setSurfaceType(QWindow::OpenGLSurface);

    QSurfaceFormat fmt;
    fmt.setVersion(4, 5);
    fmt.setProfile(QSurfaceFormat::CoreProfile);
    fmt.setDepthBufferSize(24);
    fmt.setSwapBehavior(QSurfaceFormat::DoubleBuffer);
    fmt.setSamples(4);
    setFormat(fmt);

    // Redraw is driven by QEvent::UpdateRequest.  We post one via
    // requestUpdate() from every function that mutates visible state
    // (mouse/wheel, model lifecycle, selection, resize).  When nothing
    // changes — the common case for a static BIM model — we don't burn
    // CPU/GPU redrawing the same frame.  Qt coalesces multiple
    // requestUpdate() calls inside a single vblank.
}

ViewportWindow::~ViewportWindow() {
    if (context_) {
        context_->makeCurrent(this);
        if (gl_) {
            for (auto& [mid, m] : models_gpu_) {
                if (m.vao)  gl_->glDeleteVertexArrays(1, &m.vao);
                if (m.vbo)  gl_->glDeleteBuffers(1, &m.vbo);
                if (m.ebo)  gl_->glDeleteBuffers(1, &m.ebo);
                if (m.ssbo) gl_->glDeleteBuffers(1, &m.ssbo);
                if (m.mesh_info_ssbo) gl_->glDeleteBuffers(1, &m.mesh_info_ssbo);
                if (m.visible_ssbo) gl_->glDeleteBuffers(1, &m.visible_ssbo);
                if (m.indirect_buffer) gl_->glDeleteBuffers(1, &m.indirect_buffer);
            }
            if (axis_vao_)      gl_->glDeleteVertexArrays(1, &axis_vao_);
            if (axis_vbo_)      gl_->glDeleteBuffers(1, &axis_vbo_);
            if (pivot_vao_)     gl_->glDeleteVertexArrays(1, &pivot_vao_);
            if (pivot_vbo_)     gl_->glDeleteBuffers(1, &pivot_vbo_);
            if (plane_vao_)     gl_->glDeleteVertexArrays(1, &plane_vao_);
            if (plane_vbo_)     gl_->glDeleteBuffers(1, &plane_vbo_);
            if (edge_vao_)      gl_->glDeleteVertexArrays(1, &edge_vao_);
            if (edge_depth_fbo_) gl_->glDeleteFramebuffers(1, &edge_depth_fbo_);
            if (edge_depth_tex_) gl_->glDeleteTextures(1, &edge_depth_tex_);
            if (main_program_)  gl_->glDeleteProgram(main_program_);
            if (pick_program_)  gl_->glDeleteProgram(pick_program_);
            if (axis_program_)  gl_->glDeleteProgram(axis_program_);
            if (pivot_program_) gl_->glDeleteProgram(pivot_program_);
            if (plane_program_) gl_->glDeleteProgram(plane_program_);
            if (edge_program_)  gl_->glDeleteProgram(edge_program_);
            if (pick_fbo_)      gl_->glDeleteFramebuffers(1, &pick_fbo_);
            if (pick_color_tex_)  gl_->glDeleteTextures(1, &pick_color_tex_);
            if (pick_pos_tex_)    gl_->glDeleteTextures(1, &pick_pos_tex_);
            if (pick_normal_tex_) gl_->glDeleteTextures(1, &pick_normal_tex_);
            if (pick_depth_rbo_) gl_->glDeleteRenderbuffers(1, &pick_depth_rbo_);
            if (hiz_fbo_)         gl_->glDeleteFramebuffers(1, &hiz_fbo_);
            if (hiz_depth_tex_)   gl_->glDeleteTextures(1, &hiz_depth_tex_);
            if (hiz_resolve_fbo_) gl_->glDeleteFramebuffers(1, &hiz_resolve_fbo_);
            if (hiz_resolve_depth_tex_) gl_->glDeleteTextures(1, &hiz_resolve_depth_tex_);
            if (hiz_downsample_program_) gl_->glDeleteProgram(hiz_downsample_program_);
            if (hiz_downsample_vao_) gl_->glDeleteVertexArrays(1, &hiz_downsample_vao_);
            overlay_renderer_.release();
            selection_.releaseGl();
        }
        context_->doneCurrent();
    }
}

void ViewportWindow::initGL() {
    if (gl_initialized_) return;

    context_ = new QOpenGLContext(this);
    context_->setFormat(requestedFormat());
    if (!context_->create()) { qFatal("Failed to create OpenGL context"); return; }
    context_->makeCurrent(this);

    gl_ = QOpenGLVersionFunctionsFactory::get<QOpenGLFunctions_4_5_Core>(context_);
    if (!gl_) { qWarning("OpenGL 4.5 not available"); return; }

    buildShaders();
    buildAxisGizmo();
    buildPivotIndicator();
    buildSectionPlaneGizmo();
    overlay_renderer_.initialize(gl_);
    selection_.initializeGl(gl_);
    // Drive viewport repaints + tree/properties sync from selection state.
    connect(&selection_, &SelectionState::changed, this,
            [this](uint32_t active_id) {
                requestUpdate();
                emit objectPicked(active_id);
            });

    gl_->glEnable(GL_DEPTH_TEST);
    gl_->glEnable(GL_MULTISAMPLE);
    gl_->glClearColor(0.18f, 0.20f, 0.22f, 1.0f);
    gl_->glCullFace(GL_BACK);
    if (AppSettings::instance().backfaceCulling()) gl_->glEnable(GL_CULL_FACE);
    else                                            gl_->glDisable(GL_CULL_FACE);

    // Hot-toggle cull state when the setting changes.  Queued so we touch GL
    // state only when render() is about to run.
    connect(&AppSettings::instance(), &AppSettings::backfaceCullingChanged,
            this, [this](bool on) {
                if (!gl_initialized_ || !gl_) return;
                context_->makeCurrent(this);
                if (on) gl_->glEnable(GL_CULL_FACE);
                else    gl_->glDisable(GL_CULL_FACE);
                requestUpdate();
            });

    // Cull thresholds and HiZ live-toggle: cached cull results assume the
    // active values, so any change must invalidate them and request a
    // re-render so the new threshold actually takes effect.
    auto invalidate_cull = [this]() {
        if (!gl_initialized_) return;
        have_cached_cull_ = false;
        requestUpdate();
    };
    connect(&AppSettings::instance(), &AppSettings::minPixelRadiusChanged,
            this, [invalidate_cull](double){ invalidate_cull(); });
    connect(&AppSettings::instance(), &AppSettings::motionMinPixelRadiusChanged,
            this, [invalidate_cull](double){ invalidate_cull(); });
    connect(&AppSettings::instance(), &AppSettings::lod1PixelThresholdChanged,
            this, [invalidate_cull](double){ invalidate_cull(); });
    connect(&AppSettings::instance(), &AppSettings::hizEnabledChanged,
            this, [this, invalidate_cull](bool){
                hiz_vp_valid_ = false;  // pyramid no longer trusted
                invalidate_cull();
            });

    gl_initialized_ = true;
    flushPendingOperations();
    requestUpdate();

    emit initialized();
}

void ViewportWindow::enqueuePendingOperation(PendingOperation op) {
    pending_ops_.push_back(std::move(op));
}

void ViewportWindow::flushPendingOperations() {
    while (!pending_ops_.empty()) {
        PendingOperation op = std::move(pending_ops_.front());
        pending_ops_.pop_front();

        switch (op.type) {
        case PendingOpType::UploadMeshChunk:
            uploadMeshChunk(op.mesh_chunk);
            break;
        case PendingOpType::UploadInstanceChunk:
            uploadInstanceChunk(op.instance_chunk);
            break;
        case PendingOpType::FinalizeModel:
            finalizeModel(op.model_id);
            break;
        case PendingOpType::ApplyCachedModel:
            applyCachedModel(op.model_id, std::move(op.sidecar_data));
            break;
        case PendingOpType::ApplyLodExtension:
            applyLodExtension(op.model_id, op.sidecar_data);
            break;
        case PendingOpType::ResetScene:
            resetScene();
            break;
        case PendingOpType::HideModel:
            hideModel(op.model_id);
            break;
        case PendingOpType::ShowModel:
            showModel(op.model_id);
            break;
        case PendingOpType::RemoveModel:
            removeModel(op.model_id);
            break;
        }
    }
}

void ViewportWindow::setupVaoLayout(GLuint vao, GLuint vbo, GLuint ebo) {
    gl_->glVertexArrayVertexBuffer(vao, 0, vbo, 0, INSTANCED_VERTEX_STRIDE_BYTES);
    gl_->glVertexArrayElementBuffer(vao, ebo);

    // position (3 x u16 normalized @ 0)
    gl_->glEnableVertexArrayAttrib(vao, 0);
    gl_->glVertexArrayAttribFormat(vao, 0, 3, GL_UNSIGNED_SHORT, GL_TRUE,
                                   INSTANCED_VERTEX_POS_OFFSET);
    gl_->glVertexArrayAttribBinding(vao, 0, 0);

    // normal oct-encoded (2 x i8 normalized @ 6)
    gl_->glEnableVertexArrayAttrib(vao, 1);
    gl_->glVertexArrayAttribFormat(vao, 1, 2, GL_BYTE, GL_TRUE,
                                   INSTANCED_VERTEX_NORMAL_OFFSET);
    gl_->glVertexArrayAttribBinding(vao, 1, 0);

    // color (4 x u8 normalized @ 8)
    gl_->glEnableVertexArrayAttrib(vao, 2);
    gl_->glVertexArrayAttribFormat(vao, 2, 4, GL_UNSIGNED_BYTE, GL_TRUE,
                                   INSTANCED_VERTEX_COLOR_OFFSET);
    gl_->glVertexArrayAttribBinding(vao, 2, 0);
}

void ViewportWindow::buildShaders() {
    {
        GLuint vs = compileShader(gl_, GL_VERTEX_SHADER, MAIN_VERTEX_SHADER);
        GLuint fs = compileShader(gl_, GL_FRAGMENT_SHADER, MAIN_FRAGMENT_SHADER);
        main_program_ = linkProgram(gl_, vs, fs);
    }
    {
        GLuint vs = compileShader(gl_, GL_VERTEX_SHADER, PICK_VERTEX_SHADER);
        GLuint fs = compileShader(gl_, GL_FRAGMENT_SHADER, PICK_FRAGMENT_SHADER);
        pick_program_ = linkProgram(gl_, vs, fs);
    }
    {
        GLuint vs = compileShader(gl_, GL_VERTEX_SHADER, AXIS_VERTEX_SHADER);
        GLuint fs = compileShader(gl_, GL_FRAGMENT_SHADER, AXIS_FRAGMENT_SHADER);
        axis_program_ = linkProgram(gl_, vs, fs);
    }
    {
        GLuint vs = compileShader(gl_, GL_VERTEX_SHADER, PIVOT_VERTEX_SHADER);
        GLuint fs = compileShader(gl_, GL_FRAGMENT_SHADER, PIVOT_FRAGMENT_SHADER);
        pivot_program_ = linkProgram(gl_, vs, fs);
    }
    {
        GLuint vs = compileShader(gl_, GL_VERTEX_SHADER, PLANE_VERTEX_SHADER);
        GLuint fs = compileShader(gl_, GL_FRAGMENT_SHADER, PLANE_FRAGMENT_SHADER);
        plane_program_ = linkProgram(gl_, vs, fs);
    }
    {
        GLuint vs = compileShader(gl_, GL_VERTEX_SHADER, EDGE_VERTEX_SHADER);
        GLuint fs = compileShader(gl_, GL_FRAGMENT_SHADER, EDGE_FRAGMENT_SHADER);
        edge_program_ = linkProgram(gl_, vs, fs);
        gl_->glCreateVertexArrays(1, &edge_vao_);
    }
    {
        GLuint vs = compileShader(gl_, GL_VERTEX_SHADER, HIZ_DOWNSAMPLE_VS);
        GLuint fs = compileShader(gl_, GL_FRAGMENT_SHADER, HIZ_DOWNSAMPLE_FS);
        hiz_downsample_program_ = linkProgram(gl_, vs, fs);
        gl_->glCreateVertexArrays(1, &hiz_downsample_vao_);
    }
}

void ViewportWindow::buildAxisGizmo() {
    static const float axis_data[] = {
        0,0,0,  1.0f,0.25f,0.25f,
        1,0,0,  1.0f,0.25f,0.25f,
        0,0,0,  0.30f,0.95f,0.30f,
        0,1,0,  0.30f,0.95f,0.30f,
        0,0,0,  0.30f,0.55f,1.0f,
        0,0,1,  0.30f,0.55f,1.0f,
    };
    gl_->glCreateVertexArrays(1, &axis_vao_);
    gl_->glCreateBuffers(1, &axis_vbo_);
    gl_->glNamedBufferStorage(axis_vbo_, sizeof(axis_data), axis_data, 0);
    gl_->glVertexArrayVertexBuffer(axis_vao_, 0, axis_vbo_, 0, 6 * sizeof(float));
    gl_->glEnableVertexArrayAttrib(axis_vao_, 0);
    gl_->glVertexArrayAttribFormat(axis_vao_, 0, 3, GL_FLOAT, GL_FALSE, 0);
    gl_->glVertexArrayAttribBinding(axis_vao_, 0, 0);
    gl_->glEnableVertexArrayAttrib(axis_vao_, 1);
    gl_->glVertexArrayAttribFormat(axis_vao_, 1, 3, GL_FLOAT, GL_FALSE, 3 * sizeof(float));
    gl_->glVertexArrayAttribBinding(axis_vao_, 1, 0);
}

void ViewportWindow::buildPivotIndicator() {
    // 6 verts = 3 line segments along world ±X, ±Y, ±Z, RGB-coded.
    static const float verts[] = {
        // local x,y,z      r,g,b
        -1, 0, 0,           1.0f, 0.30f, 0.30f,
         1, 0, 0,           1.0f, 0.30f, 0.30f,
         0,-1, 0,           0.35f, 0.95f, 0.35f,
         0, 1, 0,           0.35f, 0.95f, 0.35f,
         0, 0,-1,           0.35f, 0.55f, 1.0f,
         0, 0, 1,           0.35f, 0.55f, 1.0f,
    };
    pivot_rim_count_ = 6;  // total vertex count drawn as GL_LINES

    gl_->glCreateVertexArrays(1, &pivot_vao_);
    gl_->glCreateBuffers(1, &pivot_vbo_);
    gl_->glNamedBufferStorage(pivot_vbo_, sizeof(verts), verts, 0);
    gl_->glVertexArrayVertexBuffer(pivot_vao_, 0, pivot_vbo_, 0, 6 * sizeof(float));
    gl_->glEnableVertexArrayAttrib(pivot_vao_, 0);
    gl_->glVertexArrayAttribFormat(pivot_vao_, 0, 3, GL_FLOAT, GL_FALSE, 0);
    gl_->glVertexArrayAttribBinding(pivot_vao_, 0, 0);
    gl_->glEnableVertexArrayAttrib(pivot_vao_, 1);
    gl_->glVertexArrayAttribFormat(pivot_vao_, 1, 3, GL_FLOAT, GL_FALSE, 3 * sizeof(float));
    gl_->glVertexArrayAttribBinding(pivot_vao_, 1, 0);
}

bool ViewportWindow::growModelVbo(ModelGpuData& m, size_t needed_total) {
    size_t new_capacity = m.vbo_capacity;
    while (new_capacity < needed_total) new_capacity *= 2;
    if (new_capacity > MAX_BUFFER_SIZE) {
        qWarning("VBO grow request (%zu MB) exceeds cap", new_capacity / (1024*1024));
        return false;
    }
    GLuint new_vbo = 0;
    gl_->glCreateBuffers(1, &new_vbo);
    gl_->glNamedBufferStorage(new_vbo, new_capacity, nullptr, GL_DYNAMIC_STORAGE_BIT);
    if (m.vbo_used > 0) {
        gl_->glCopyNamedBufferSubData(m.vbo, new_vbo, 0, 0, m.vbo_used);
    }
    gl_->glDeleteBuffers(1, &m.vbo);
    m.vbo = new_vbo;
    m.vbo_capacity = new_capacity;
    gl_->glVertexArrayVertexBuffer(m.vao, 0, m.vbo, 0, INSTANCED_VERTEX_STRIDE_BYTES);
    qInfo("Model VBO grew to %zu MB", m.vbo_capacity / (1024*1024));
    return true;
}

bool ViewportWindow::growModelSsbo(ModelGpuData& m, size_t needed_total) {
    size_t new_capacity = m.ssbo_capacity ? m.ssbo_capacity : INITIAL_SSBO_SIZE;
    while (new_capacity < needed_total) new_capacity *= 2;
    if (new_capacity > MAX_BUFFER_SIZE) {
        qWarning("Instance SSBO grow request (%zu MB) exceeds cap", new_capacity / (1024*1024));
        return false;
    }
    GLuint new_ssbo = 0;
    gl_->glCreateBuffers(1, &new_ssbo);
    gl_->glNamedBufferStorage(new_ssbo, new_capacity, nullptr, GL_DYNAMIC_STORAGE_BIT);
    const size_t used = m.ssbo_instance_count * sizeof(InstanceGpu);
    if (m.ssbo && used > 0) {
        gl_->glCopyNamedBufferSubData(m.ssbo, new_ssbo, 0, 0, used);
    }
    if (m.ssbo) gl_->glDeleteBuffers(1, &m.ssbo);
    m.ssbo = new_ssbo;
    m.ssbo_capacity = new_capacity;
    qInfo("Model instance SSBO grew to %zu MB", m.ssbo_capacity / (1024*1024));
    return true;
}

bool ViewportWindow::growModelEbo(ModelGpuData& m, size_t needed_total) {
    size_t new_capacity = m.ebo_capacity;
    while (new_capacity < needed_total) new_capacity *= 2;
    if (new_capacity > MAX_BUFFER_SIZE) {
        qWarning("EBO grow request (%zu MB) exceeds cap", new_capacity / (1024*1024));
        return false;
    }
    GLuint new_ebo = 0;
    gl_->glCreateBuffers(1, &new_ebo);
    gl_->glNamedBufferStorage(new_ebo, new_capacity, nullptr, GL_DYNAMIC_STORAGE_BIT);
    if (m.ebo_used > 0) {
        gl_->glCopyNamedBufferSubData(m.ebo, new_ebo, 0, 0, m.ebo_used);
    }
    gl_->glDeleteBuffers(1, &m.ebo);
    m.ebo = new_ebo;
    m.ebo_capacity = new_capacity;
    gl_->glVertexArrayElementBuffer(m.vao, m.ebo);
    qInfo("Model EBO grew to %zu MB", m.ebo_capacity / (1024*1024));
    return true;
}

ModelGpuData& ViewportWindow::getOrCreateModel(uint32_t model_id) {
    auto it = models_gpu_.find(model_id);
    if (it != models_gpu_.end()) return it->second;

    ModelGpuData m;
    gl_->glCreateVertexArrays(1, &m.vao);
    gl_->glCreateBuffers(1, &m.vbo);
    gl_->glCreateBuffers(1, &m.ebo);

    m.vbo_capacity = INITIAL_VBO_SIZE;
    m.ebo_capacity = INITIAL_EBO_SIZE;
    gl_->glNamedBufferStorage(m.vbo, m.vbo_capacity, nullptr, GL_DYNAMIC_STORAGE_BIT);
    gl_->glNamedBufferStorage(m.ebo, m.ebo_capacity, nullptr, GL_DYNAMIC_STORAGE_BIT);
    setupVaoLayout(m.vao, m.vbo, m.ebo);

    // Pre-allocate instance SSBO so we can append during streaming.
    gl_->glCreateBuffers(1, &m.ssbo);
    m.ssbo_capacity = INITIAL_SSBO_SIZE;
    gl_->glNamedBufferStorage(m.ssbo, m.ssbo_capacity, nullptr, GL_DYNAMIC_STORAGE_BIT);

    return models_gpu_.emplace(model_id, std::move(m)).first->second;
}

void ViewportWindow::uploadMeshChunk(const MeshChunk& chunk) {
    if (!gl_initialized_) {
        PendingOperation op;
        op.type = PendingOpType::UploadMeshChunk;
        op.mesh_chunk = chunk;
        enqueuePendingOperation(std::move(op));
        return;
    }
    if (chunk.vertices.empty() || chunk.indices.empty()) return;
    context_->makeCurrent(this);

    ModelGpuData& m = getOrCreateModel(chunk.model_id);

    // Streamer format: 7 floats/vertex (pos3 + normal3 + color-as-float).
    const size_t src_stride_floats = 7;
    const size_t n_verts = chunk.vertices.size() / src_stride_floats;

    // Recompute a tight local AABB from the actual vertex positions — the
    // chunk-provided AABB can be slightly loose, which wastes quantization
    // precision.  Also derives the dequant basis we'll ship to the GPU.
    float bmin[3] = {  std::numeric_limits<float>::infinity(),
                       std::numeric_limits<float>::infinity(),
                       std::numeric_limits<float>::infinity() };
    float bmax[3] = { -std::numeric_limits<float>::infinity(),
                      -std::numeric_limits<float>::infinity(),
                      -std::numeric_limits<float>::infinity() };
    for (size_t i = 0; i < n_verts; ++i) {
        const float* v = chunk.vertices.data() + i * src_stride_floats;
        for (int a = 0; a < 3; ++a) {
            if (v[a] < bmin[a]) bmin[a] = v[a];
            if (v[a] > bmax[a]) bmax[a] = v[a];
        }
    }
    // Degenerate / zero-extent axis: collapse to a single quantum.  The
    // dequant shader will output bmin[a] for every vertex, which is correct.
    float extent_recip[3];
    for (int a = 0; a < 3; ++a) {
        float ext = bmax[a] - bmin[a];
        extent_recip[a] = ext > 0.0f ? 1.0f / ext : 0.0f;
    }

    // Quantize into a scratch buffer sized to the destination layout.
    std::vector<uint8_t> quant(n_verts * INSTANCED_VERTEX_STRIDE_BYTES);
    for (size_t i = 0; i < n_verts; ++i) {
        quantizeVertex(chunk.vertices.data() + i * src_stride_floats,
                       bmin, extent_recip,
                       quant.data() + i * INSTANCED_VERTEX_STRIDE_BYTES);
    }

    const size_t vb_size = quant.size();
    const size_t ib_size = chunk.indices.size() * sizeof(uint32_t);

    if (m.vbo_used + vb_size > m.vbo_capacity) {
        if (!growModelVbo(m, m.vbo_used + vb_size)) return;
    }
    if (m.ebo_used + ib_size > m.ebo_capacity) {
        if (!growModelEbo(m, m.ebo_used + ib_size)) return;
    }

    MeshInfo info;
    info.vbo_byte_offset = static_cast<uint32_t>(m.vbo_used);
    info.vertex_count    = static_cast<uint32_t>(n_verts);
    info.ebo_byte_offset = static_cast<uint32_t>(m.ebo_used);
    info.index_count     = static_cast<uint32_t>(chunk.indices.size());
    for (int a = 0; a < 3; ++a) {
        info.local_aabb_min[a] = bmin[a];
        info.local_aabb_max[a] = bmax[a];
    }
    info.first_instance = 0;
    info.instance_count = 0;

    gl_->glNamedBufferSubData(m.vbo, m.vbo_used, vb_size, quant.data());
    gl_->glNamedBufferSubData(m.ebo, m.ebo_used, ib_size, chunk.indices.data());
    m.vbo_used += vb_size;
    m.ebo_used += ib_size;
    m.vertex_count += info.vertex_count;

    if (m.meshes.size() <= chunk.local_mesh_id) m.meshes.resize(chunk.local_mesh_id + 1);
    m.meshes[chunk.local_mesh_id] = info;

    // Write the matching dequant basis into the MeshGpu SSBO.  Grow on
    // demand; geometrically doubling keeps this amortized O(1) over streaming.
    MeshGpu mg{};
    for (int a = 0; a < 3; ++a) {
        mg.aabb_min[a] = bmin[a];
        mg.aabb_max[a] = bmax[a];
    }
    mg.aabb_min[3] = 0.0f;
    mg.aabb_max[3] = 0.0f;

    const size_t mg_offset = chunk.local_mesh_id * sizeof(MeshGpu);
    if (mg_offset + sizeof(MeshGpu) > m.mesh_info_capacity) {
        size_t new_cap = m.mesh_info_capacity ? m.mesh_info_capacity : 32 * sizeof(MeshGpu);
        while (new_cap < mg_offset + sizeof(MeshGpu)) new_cap *= 2;
        GLuint new_ssbo = 0;
        gl_->glCreateBuffers(1, &new_ssbo);
        gl_->glNamedBufferStorage(new_ssbo, new_cap, nullptr, GL_DYNAMIC_STORAGE_BIT);
        if (m.mesh_info_ssbo && m.mesh_info_capacity > 0) {
            gl_->glCopyNamedBufferSubData(m.mesh_info_ssbo, new_ssbo, 0, 0,
                                          m.mesh_info_capacity);
            gl_->glDeleteBuffers(1, &m.mesh_info_ssbo);
        }
        m.mesh_info_ssbo = new_ssbo;
        m.mesh_info_capacity = new_cap;
    }
    gl_->glNamedBufferSubData(m.mesh_info_ssbo, mg_offset, sizeof(MeshGpu), &mg);
}

void ViewportWindow::uploadInstanceChunk(const InstanceChunk& chunk) {
    if (!gl_initialized_) {
        PendingOperation op;
        op.type = PendingOpType::UploadInstanceChunk;
        op.instance_chunk = chunk;
        enqueuePendingOperation(std::move(op));
        return;
    }
    context_->makeCurrent(this);

    ModelGpuData& m = getOrCreateModel(chunk.model_id);

    selection_.noteObjectId(chunk.object_id);

    InstanceCpu inst;
    inst.mesh_id  = chunk.local_mesh_id;
    inst.object_id = chunk.object_id;
    inst.color_override_rgba8 = chunk.color_override_rgba8;
    inst.model_id = chunk.model_id;
    // Streamer's `chunk.transform` is the placement_transformation (the
    // iterator's per-shape transform with vertex-rebasing offset folded in).
    std::memcpy(inst.placement_transformation, chunk.transform,
                sizeof(inst.placement_transformation));
    // Compose against the model's current stage matrices to fill in
    // inst.transform + inst.world_aabb_*.  When all stages are identity
    // (the default until a setter is called), this reduces to
    // transform == placement_transformation and the world AABB matches
    // the streamer's pre-computed chunk.world_aabb_* exactly.
    composeInstanceFromPlacement(inst, m);
    m.instances.push_back(inst);
    m.instance_reflected.push_back(transformIsReflected(inst.transform) ? 1 : 0);

    // Mirror into bvh_items so the hot cull path (which reads AABBs out of
    // bvh_items even when no BVH has been built yet) stays correct during
    // streaming.  finalizeModel rebuilds the real BVH over these items.
    BvhItem bi;
    std::memcpy(bi.aabb_min, inst.world_aabb_min, sizeof(bi.aabb_min));
    std::memcpy(bi.aabb_max, inst.world_aabb_max, sizeof(bi.aabb_max));
    bi.model_id = inst.model_id;
    m.bvh_items.push_back(bi);

    // Append the GPU record to the instance SSBO so the model is drawable
    // immediately, without waiting for finalizeModel.  The visible-list
    // architecture means SSBO order is irrelevant to correctness.
    InstanceGpu gpu;
    std::memcpy(gpu.transform, inst.transform, sizeof(gpu.transform));
    gpu.object_id = inst.object_id;
    gpu.color_override_rgba8 = inst.color_override_rgba8;
    gpu.mesh_id = inst.mesh_id;
    gpu._pad1 = 0;

    const size_t offset = m.ssbo_instance_count * sizeof(InstanceGpu);
    if (offset + sizeof(InstanceGpu) > m.ssbo_capacity) {
        if (!growModelSsbo(m, offset + sizeof(InstanceGpu))) return;
    }
    gl_->glNamedBufferSubData(m.ssbo, offset, sizeof(InstanceGpu), &gpu);
    m.ssbo_instance_count++;

    if (chunk.local_mesh_id < m.meshes.size()) {
        m.total_triangles += m.meshes[chunk.local_mesh_id].index_count / 3;
    }
    have_cached_cull_ = false;
    requestUpdate();
}

void ViewportWindow::finalizeModel(uint32_t model_id) {
    if (!gl_initialized_) {
        PendingOperation op;
        op.type = PendingOpType::FinalizeModel;
        op.model_id = model_id;
        enqueuePendingOperation(std::move(op));
        return;
    }
    context_->makeCurrent(this);

    auto it = models_gpu_.find(model_id);
    if (it == models_gpu_.end()) return;
    ModelGpuData& m = it->second;

    // Instance SSBO has been populated incrementally during streaming, so
    // we don't re-upload here.  What finalize still does:
    //   (1) compute per-mesh instance counts — used by stats and the sidecar
    //       round-trip (first_instance is unused by the visible-list renderer),
    //   (2) build the per-model BVH over instance world AABBs.
    for (auto& mesh : m.meshes) { mesh.first_instance = 0; mesh.instance_count = 0; }
    for (const auto& inst : m.instances) {
        if (inst.mesh_id < m.meshes.size()) ++m.meshes[inst.mesh_id].instance_count;
    }

    buildBvhForModel(m, model_id);

    m.finalized = true;
    have_cached_cull_ = false;
    requestUpdate();

    const size_t ssbo_bytes = m.ssbo_instance_count * sizeof(InstanceGpu);
    qDebug("Model %u finalized: %zu verts, %zu meshes, %zu instances, %.1f MB vram "
           "(vbo %.1f + ebo %.1f + ssbo-used %.1f / %.1f cap)",
           model_id, size_t(m.vertex_count), m.meshes.size(), m.instances.size(),
           (m.vbo_capacity + m.ebo_capacity + m.ssbo_capacity) / (1024.0*1024.0),
           m.vbo_capacity / (1024.0*1024.0),
           m.ebo_capacity / (1024.0*1024.0),
           ssbo_bytes / (1024.0*1024.0),
           m.ssbo_capacity / (1024.0*1024.0));
}

bool ViewportWindow::snapshotModel(uint32_t model_id, SidecarData& out) const {
    auto it = models_gpu_.find(model_id);
    if (!gl_ || it == models_gpu_.end()) return false;
    const auto& m = it->second;
    if (!m.finalized) return false;

    // GPU readback of the packed VBO/EBO ranges actually in use.  VBO is
    // raw bytes at the quantized layout.
    if (m.vbo_used > 0) {
        out.vertices.resize(m.vbo_used);
        gl_->glGetNamedBufferSubData(m.vbo, 0, m.vbo_used, out.vertices.data());
    }
    if (m.ebo_used > 0) {
        out.indices.resize(m.ebo_used / sizeof(uint32_t));
        gl_->glGetNamedBufferSubData(m.ebo, 0, m.ebo_used, out.indices.data());
    }

    out.meshes    = m.meshes;
    out.instances = m.instances;
    return true;
}

void ViewportWindow::applyCachedModel(uint32_t model_id, SidecarData data) {
    if (!gl_initialized_) {
        PendingOperation op;
        op.type = PendingOpType::ApplyCachedModel;
        op.model_id = model_id;
        op.sidecar_data = std::move(data);
        enqueuePendingOperation(std::move(op));
        return;
    }
    context_->makeCurrent(this);

    // Drop any existing state for this model_id.
    auto existing = models_gpu_.find(model_id);
    if (existing != models_gpu_.end()) {
        if (existing->second.vao)  gl_->glDeleteVertexArrays(1, &existing->second.vao);
        if (existing->second.vbo)  gl_->glDeleteBuffers(1, &existing->second.vbo);
        if (existing->second.ebo)  gl_->glDeleteBuffers(1, &existing->second.ebo);
        if (existing->second.ssbo) gl_->glDeleteBuffers(1, &existing->second.ssbo);
        if (existing->second.mesh_info_ssbo) gl_->glDeleteBuffers(1, &existing->second.mesh_info_ssbo);
        if (existing->second.visible_ssbo) gl_->glDeleteBuffers(1, &existing->second.visible_ssbo);
        if (existing->second.indirect_buffer) gl_->glDeleteBuffers(1, &existing->second.indirect_buffer);
        models_gpu_.erase(existing);
    }

    ModelGpuData m;
    gl_->glCreateVertexArrays(1, &m.vao);
    gl_->glCreateBuffers(1, &m.vbo);
    gl_->glCreateBuffers(1, &m.ebo);

    const size_t vb_bytes = data.vertices.size();
    const size_t ib_bytes = data.indices.size()  * sizeof(uint32_t);
    m.vbo_capacity = std::max<size_t>(vb_bytes, 1);
    m.ebo_capacity = std::max<size_t>(ib_bytes, 1);
    gl_->glNamedBufferStorage(m.vbo, m.vbo_capacity,
                              vb_bytes ? data.vertices.data() : nullptr,
                              GL_DYNAMIC_STORAGE_BIT);
    gl_->glNamedBufferStorage(m.ebo, m.ebo_capacity,
                              ib_bytes ? data.indices.data() : nullptr,
                              GL_DYNAMIC_STORAGE_BIT);
    setupVaoLayout(m.vao, m.vbo, m.ebo);

    m.vbo_used = vb_bytes;
    m.ebo_used = ib_bytes;
    m.vertex_count = static_cast<uint32_t>(vb_bytes / INSTANCED_VERTEX_STRIDE_BYTES);
    m.meshes = std::move(data.meshes);
    m.instances = std::move(data.instances);

    // Sidecar path bypasses uploadInstanceChunk, so register every
    // instance's object_id with the selection state up front — otherwise
    // the per-object_id flags SSBO would be too small for these ids and
    // their selection bit reads would silently fall outside the buffer.
    for (const auto& inst : m.instances) {
        selection_.noteObjectId(inst.object_id);
    }

    uint32_t total_tri = 0;
    for (const auto& mesh : m.meshes) {
        total_tri += (mesh.index_count / 3) * mesh.instance_count;
    }
    m.total_triangles = total_tri;

    // Recompose every instance against the model's current stage matrices.
    // The cached InstanceCpu::transform / world_aabb_* may have been frozen
    // against a different .ifcfed's stages — placement_transformation is the
    // authoritative input, so we always rebuild transform + world_aabb here.
    for (auto& inst : m.instances) {
        composeInstanceFromPlacement(inst, m);
    }

    // Build and upload the instance SSBO.
    std::vector<InstanceGpu> gpu(m.instances.size());
    for (size_t i = 0; i < m.instances.size(); ++i) {
        const InstanceCpu& src = m.instances[i];
        InstanceGpu& dst = gpu[i];
        std::memcpy(dst.transform, src.transform, sizeof(dst.transform));
        dst.object_id = src.object_id;
        dst.color_override_rgba8 = src.color_override_rgba8;
        dst.mesh_id = src.mesh_id;
        dst._pad1 = 0;
    }
    gl_->glCreateBuffers(1, &m.ssbo);
    const size_t ssbo_bytes = gpu.size() * sizeof(InstanceGpu);
    if (ssbo_bytes > 0) {
        gl_->glNamedBufferStorage(m.ssbo, ssbo_bytes, gpu.data(), 0);
    }
    m.ssbo_instance_count = static_cast<uint32_t>(gpu.size());

    // Build and upload the per-mesh quantization SSBO from cached meshes.
    {
        std::vector<MeshGpu> mesh_gpu(m.meshes.size());
        for (size_t i = 0; i < m.meshes.size(); ++i) {
            for (int a = 0; a < 3; ++a) {
                mesh_gpu[i].aabb_min[a] = m.meshes[i].local_aabb_min[a];
                mesh_gpu[i].aabb_max[a] = m.meshes[i].local_aabb_max[a];
            }
            mesh_gpu[i].aabb_min[3] = 0.0f;
            mesh_gpu[i].aabb_max[3] = 0.0f;
        }
        const size_t mg_bytes = mesh_gpu.size() * sizeof(MeshGpu);
        gl_->glCreateBuffers(1, &m.mesh_info_ssbo);
        if (mg_bytes > 0) {
            gl_->glNamedBufferStorage(m.mesh_info_ssbo, mg_bytes,
                                      mesh_gpu.data(), GL_DYNAMIC_STORAGE_BIT);
            m.mesh_info_capacity = mg_bytes;
        } else {
            gl_->glNamedBufferStorage(m.mesh_info_ssbo, sizeof(MeshGpu),
                                      nullptr, GL_DYNAMIC_STORAGE_BIT);
            m.mesh_info_capacity = sizeof(MeshGpu);
        }
    }

    // Recompute the reflection flag from each instance's transform — the
    // sidecar only caches InstanceCpu, not the parallel reflection flags.
    m.instance_reflected.resize(m.instances.size());
    for (size_t i = 0; i < m.instances.size(); ++i) {
        m.instance_reflected[i] = transformIsReflected(m.instances[i].transform) ? 1 : 0;
    }

    buildBvhForModel(m, model_id);

    m.finalized = true;
    models_gpu_.emplace(model_id, std::move(m));
    have_cached_cull_ = false;
    requestUpdate();

    qDebug("Sidecar apply: model %u  %zu verts, %zu meshes, %zu instances  "
           "%.1f MB vram (vbo %.1f + ebo %.1f + ssbo %.1f)",
           model_id, vb_bytes / INSTANCED_VERTEX_STRIDE_BYTES,
           models_gpu_[model_id].meshes.size(),
           models_gpu_[model_id].instances.size(),
           (vb_bytes + ib_bytes + ssbo_bytes) / (1024.0*1024.0),
           vb_bytes / (1024.0*1024.0),
           ib_bytes / (1024.0*1024.0),
           ssbo_bytes / (1024.0*1024.0));
}

void ViewportWindow::applyLodExtension(uint32_t model_id, const SidecarData& sd) {
    if (!gl_initialized_) {
        PendingOperation op;
        op.type = PendingOpType::ApplyLodExtension;
        op.model_id = model_id;
        op.sidecar_data = sd;
        enqueuePendingOperation(std::move(op));
        return;
    }
    auto it = models_gpu_.find(model_id);
    if (it == models_gpu_.end() || !it->second.finalized) return;
    ModelGpuData& m = it->second;

    const size_t total_ib_bytes = sd.indices.size() * sizeof(uint32_t);
    if (total_ib_bytes <= m.ebo_used) {
        // buildLods didn't add anything; just refresh the meshes vector in
        // case lod1_* fields were touched.
        m.meshes = sd.meshes;
        have_cached_cull_ = false;
        requestUpdate();
        return;
    }

    context_->makeCurrent(this);
    if (total_ib_bytes > m.ebo_capacity) {
        if (!growModelEbo(m, total_ib_bytes)) return;
    }
    const size_t append_bytes = total_ib_bytes - m.ebo_used;
    const uint32_t* appended_src =
        sd.indices.data() + (m.ebo_used / sizeof(uint32_t));
    gl_->glNamedBufferSubData(m.ebo, m.ebo_used, append_bytes, appended_src);
    m.ebo_used = total_ib_bytes;

    // Replace mesh metadata so cullAndUploadVisible sees the new lod1_ fields.
    m.meshes = sd.meshes;
    have_cached_cull_ = false;
    requestUpdate();
}

void ViewportWindow::resetScene() {
    if (!gl_initialized_) {
        PendingOperation op;
        op.type = PendingOpType::ResetScene;
        pending_ops_.clear();
        enqueuePendingOperation(std::move(op));
        return;
    }
    context_->makeCurrent(this);
    for (auto& [mid, m] : models_gpu_) {
        if (m.vao)  gl_->glDeleteVertexArrays(1, &m.vao);
        if (m.vbo)  gl_->glDeleteBuffers(1, &m.vbo);
        if (m.ebo)  gl_->glDeleteBuffers(1, &m.ebo);
        if (m.ssbo) gl_->glDeleteBuffers(1, &m.ssbo);
        if (m.mesh_info_ssbo) gl_->glDeleteBuffers(1, &m.mesh_info_ssbo);
        if (m.visible_ssbo) gl_->glDeleteBuffers(1, &m.visible_ssbo);
        if (m.indirect_buffer) gl_->glDeleteBuffers(1, &m.indirect_buffer);
    }
    models_gpu_.clear();
    selection_.reset();
    have_cached_cull_ = false;
    requestUpdate();
}

void ViewportWindow::hideModel(uint32_t model_id) {
    if (!gl_initialized_) {
        PendingOperation op;
        op.type = PendingOpType::HideModel;
        op.model_id = model_id;
        enqueuePendingOperation(std::move(op));
        return;
    }
    auto it = models_gpu_.find(model_id);
    if (it != models_gpu_.end()) {
        it->second.hidden = true;
        have_cached_cull_ = false;
        requestUpdate();
    }
}

void ViewportWindow::showModel(uint32_t model_id) {
    if (!gl_initialized_) {
        PendingOperation op;
        op.type = PendingOpType::ShowModel;
        op.model_id = model_id;
        enqueuePendingOperation(std::move(op));
        return;
    }
    auto it = models_gpu_.find(model_id);
    if (it != models_gpu_.end()) {
        it->second.hidden = false;
        have_cached_cull_ = false;
        requestUpdate();
    }
}

void ViewportWindow::removeModel(uint32_t model_id) {
    if (!gl_initialized_) {
        PendingOperation op;
        op.type = PendingOpType::RemoveModel;
        op.model_id = model_id;
        enqueuePendingOperation(std::move(op));
        return;
    }
    context_->makeCurrent(this);
    auto it = models_gpu_.find(model_id);
    if (it != models_gpu_.end()) {
        if (it->second.vao)  gl_->glDeleteVertexArrays(1, &it->second.vao);
        if (it->second.vbo)  gl_->glDeleteBuffers(1, &it->second.vbo);
        if (it->second.ebo)  gl_->glDeleteBuffers(1, &it->second.ebo);
        if (it->second.ssbo) gl_->glDeleteBuffers(1, &it->second.ssbo);
        if (it->second.mesh_info_ssbo) gl_->glDeleteBuffers(1, &it->second.mesh_info_ssbo);
        if (it->second.visible_ssbo) gl_->glDeleteBuffers(1, &it->second.visible_ssbo);
        if (it->second.indirect_buffer) gl_->glDeleteBuffers(1, &it->second.indirect_buffer);
        models_gpu_.erase(it);
        have_cached_cull_ = false;
        requestUpdate();
    }
}

void ViewportWindow::setSelectedObjectId(uint32_t id) {
    // Convenience pass-through.  SelectionState fires `changed` which
    // re-emits objectPicked + requestUpdate via the lambda hooked up in
    // initGL, so we don't need to do those manually here.
    selection_.setSelectedObjectId(id);
}

void ViewportWindow::setCamera(float tx, float ty, float tz,
                                float dist, float yaw, float pitch) {
    camera_target_ = QVector3D(tx, ty, tz);
    camera_distance_ = dist;
    camera_yaw_ = yaw;
    camera_pitch_ = pitch;
    have_cached_cull_ = false;
    requestUpdate();
}

bool ViewportWindow::computeObjectAabb(uint32_t object_id,
                                       QVector3D& mn, QVector3D& mx) const {
    if (object_id == 0) return false;
    bool found = false;
    QVector3D lo( std::numeric_limits<float>::max(),
                  std::numeric_limits<float>::max(),
                  std::numeric_limits<float>::max());
    QVector3D hi(-std::numeric_limits<float>::max(),
                 -std::numeric_limits<float>::max(),
                 -std::numeric_limits<float>::max());
    for (const auto& [mid, m] : models_gpu_) {
        if (!m.finalized || m.hidden) continue;
        for (const InstanceCpu& inst : m.instances) {
            if (inst.object_id != object_id) continue;
            for (int a = 0; a < 3; ++a) {
                if (inst.world_aabb_min[a] < lo[a]) lo[a] = inst.world_aabb_min[a];
                if (inst.world_aabb_max[a] > hi[a]) hi[a] = inst.world_aabb_max[a];
            }
            found = true;
        }
    }
    if (found) { mn = lo; mx = hi; }
    return found;
}

bool ViewportWindow::computeSceneAabb(QVector3D& mn, QVector3D& mx) const {
    bool found = false;
    QVector3D lo( std::numeric_limits<float>::max(),
                  std::numeric_limits<float>::max(),
                  std::numeric_limits<float>::max());
    QVector3D hi(-std::numeric_limits<float>::max(),
                 -std::numeric_limits<float>::max(),
                 -std::numeric_limits<float>::max());
    for (const auto& [mid, m] : models_gpu_) {
        if (!m.finalized || m.hidden) continue;
        if (!m.bvh.nodes.empty()) {
            const BvhNode& root = m.bvh.nodes[0];
            for (int a = 0; a < 3; ++a) {
                if (root.aabb_min[a] < lo[a]) lo[a] = root.aabb_min[a];
                if (root.aabb_max[a] > hi[a]) hi[a] = root.aabb_max[a];
            }
            found = true;
        } else {
            for (const InstanceCpu& inst : m.instances) {
                for (int a = 0; a < 3; ++a) {
                    if (inst.world_aabb_min[a] < lo[a]) lo[a] = inst.world_aabb_min[a];
                    if (inst.world_aabb_max[a] > hi[a]) hi[a] = inst.world_aabb_max[a];
                }
                found = true;
            }
        }
    }
    if (found) { mn = lo; mx = hi; }
    return found;
}

void ViewportWindow::frameAabb(const QVector3D& mn, const QVector3D& mx,
                               float padding) {
    const QVector3D centroid = (mn + mx) * 0.5f;
    const float radius = ((mx - mn).length() * 0.5f);

    // Empty / point AABB: keep the existing distance so we just recenter.
    float new_distance = camera_distance_;
    if (radius > 1e-4f) {
        const float fovy_rad = qDegreesToRadians(camera_fov_y_deg_);
        const float tan_half = tanf(fovy_rad * 0.5f);
        // tan_half == 0 is impossible at fov 45°, but guard anyway.
        if (tan_half > 1e-6f) {
            const int   h = qMax(height(), 1);
            const float aspect = float(qMax(width(), 1)) / float(h);
            // Use the tighter axis: portrait windows need a larger pull-back.
            const float min_aspect = aspect < 1.0f ? aspect : 1.0f;
            new_distance = (radius / (tan_half * min_aspect)) * padding;
        }
    }

    camera_target_ = centroid;
    camera_distance_ = qMax(0.1f, new_distance);
    have_cached_cull_ = false;
    requestUpdate();
}

void ViewportWindow::focusOnSelectedObject() {
    if (camera_mode_ == CameraMode::Fps) return;
    // F frames the *active* object so it's predictable across multi-select
    // states.  Framing the union of every selected member would be more
    // expansive but ambiguous in a large set.
    const uint32_t target = selection_.activeObjectId();
    QVector3D mn, mx;
    if (!computeObjectAabb(target, mn, mx)) {
        qDebug("Focus: no object selected or no AABB available");
        return;
    }
    frameAabb(mn, mx, 1.30f);  // a bit of headroom around small objects
}

void ViewportWindow::viewAll() {
    if (camera_mode_ == CameraMode::Fps) return;
    QVector3D mn, mx;
    if (!computeSceneAabb(mn, mx)) {
        qDebug("View All: scene is empty");
        return;
    }
    frameAabb(mn, mx, 1.10f);
}

void ViewportWindow::setBenchmarkFrames(int n) {
    benchmark_total_ = n;
    benchmark_count_ = 0;
    benchmark_warmup_ = 5;
    benchmark_yaw_start_ = camera_yaw_;
    benchmark_frame_times_.clear();
    benchmark_frame_times_.reserve(n);
    requestUpdate();
}

QString ViewportWindow::cameraString() const {
    return QString("%1,%2,%3,%4,%5,%6")
        .arg(camera_target_.x(), 0, 'f', 4)
        .arg(camera_target_.y(), 0, 'f', 4)
        .arg(camera_target_.z(), 0, 'f', 4)
        .arg(camera_distance_, 0, 'f', 4)
        .arg(camera_yaw_, 0, 'f', 2)
        .arg(camera_pitch_, 0, 'f', 2);
}

ViewportWindow::CameraState ViewportWindow::cameraState() const {
    return { camera_target_, camera_distance_, camera_yaw_, camera_pitch_ };
}

void ViewportWindow::keyPressEvent(QKeyEvent* event) {
    const int key = event->key();

    // Shift+F toggles FPS/fly mode.  Checked before auto-repeat filtering so
    // a held Shift+F doesn't thrash between modes.
    if (key == Qt::Key_F
        && (event->modifiers() & Qt::ShiftModifier)
        && !event->isAutoRepeat()) {
        if (camera_mode_ == CameraMode::Fps) exitFpsMode();
        else                                 enterFpsMode();
        return;
    }

    if (camera_mode_ == CameraMode::Fps) {
        if (key == Qt::Key_Escape && !event->isAutoRepeat()) {
            exitFpsMode();
            return;
        }
        switch (key) {
        case Qt::Key_W: case Qt::Key_A: case Qt::Key_S: case Qt::Key_D:
        case Qt::Key_Q: case Qt::Key_E: case Qt::Key_Shift:
            if (!event->isAutoRepeat()) {
                const bool was_empty = fps_keys_held_.isEmpty();
                fps_keys_held_.insert(key);
                // Kick the render loop; subsequent frames self-schedule while
                // any key stays held.  Reset the dt baseline so the first
                // frame doesn't integrate idle time.
                if (was_empty) {
                    fps_last_tick_.restart();
                    requestUpdate();
                }
            }
            // Swallow auto-repeats too so they don't leak to shortcuts.
            return;
        default: break;
        }
    }

    if (key == Qt::Key_C && !(event->modifiers() & Qt::ControlModifier)) {
        qDebug("--camera %s", qPrintable(cameraString()));
        return;
    }

    // Plain F (no modifiers): focus camera on the currently selected object.
    // Shift+F is FPS-mode toggle and was handled above.
    if (key == Qt::Key_F
        && event->modifiers() == Qt::NoModifier
        && !event->isAutoRepeat()) {
        focusOnSelectedObject();
        return;
    }
    // Home: frame the entire scene.
    if (key == Qt::Key_Home && !event->isAutoRepeat()) {
        viewAll();
        return;
    }
    // P toggles orthographic / perspective projection.
    if (key == Qt::Key_P
        && event->modifiers() == Qt::NoModifier
        && !event->isAutoRepeat()) {
        toggleProjection();
        return;
    }
    // Standard axis-aligned views: X/Y/Z look toward the target from the
    // positive axis (eye on +X/+Y/+Z), Shift+X/Y/Z from the negative side.
    // Yaw is the orbit angle in the world XY plane (0° = +X, 90° = +Y);
    // pitch is the elevation (0° = horizon, +90° = looking down).  Top/
    // bottom intentionally use pitch = ±90° — the up-vector switch in
    // updateCamera() keeps lookAt well-conditioned there and yields
    // world-Y as screen-up (architectural "north").
    if ((key == Qt::Key_X || key == Qt::Key_Y || key == Qt::Key_Z)
        && (event->modifiers() == Qt::NoModifier
            || event->modifiers() == Qt::ShiftModifier)
        && !event->isAutoRepeat()) {
        const bool neg = (event->modifiers() & Qt::ShiftModifier);
        switch (key) {
        case Qt::Key_X: setStandardView(neg ? 180.0f : 0.0f,   0.0f);  break;
        case Qt::Key_Y: setStandardView(neg ? 270.0f : 90.0f,  0.0f);  break;
        case Qt::Key_Z: setStandardView(camera_yaw_, neg ? -90.0f : 90.0f); break;
        }
        return;
    }
    // K toggles the section tool.
    if (key == Qt::Key_K
        && event->modifiers() == Qt::NoModifier
        && !event->isAutoRepeat()) {
        toggleSectionTool();
        return;
    }
    // Shift+K clears every section plane (and the selection).
    if (key == Qt::Key_K
        && event->modifiers() == Qt::ShiftModifier
        && !event->isAutoRepeat()) {
        clearSectionPlanes();
        section_plane_selected_ = -1;
        section_drag_active_ = false;
        section_drag_index_  = -1;
        return;
    }
    // While the section tool is active: Esc exits the tool, Delete removes
    // the selected plane.
    if (section_tool_active_) {
        if (key == Qt::Key_Escape && !event->isAutoRepeat()) {
            toggleSectionTool();
            return;
        }
        if ((key == Qt::Key_Delete || key == Qt::Key_Backspace)
            && !event->isAutoRepeat()
            && section_plane_selected_ >= 0) {
            removeSectionPlane(section_plane_selected_);
            section_plane_selected_ = -1;
            return;
        }
    }
    // Esc exits any active measurement tool.  Backspace/Delete in length
    // mode removes the last point — emitted as a signal because the
    // library doesn't track per-tool semantic state.
    if (tool_mode_ != ToolMode::None && !event->isAutoRepeat()) {
        if (key == Qt::Key_Escape) {
            setToolMode(ToolMode::None);
            return;
        }
        if (tool_mode_ == ToolMode::Length
            && (key == Qt::Key_Backspace || key == Qt::Key_Delete)) {
            emit toolBackspacePressed();
            return;
        }
    }
    QWindow::keyPressEvent(event);
}

void ViewportWindow::keyReleaseEvent(QKeyEvent* event) {
    if (camera_mode_ == CameraMode::Fps && !event->isAutoRepeat()) {
        fps_keys_held_.remove(event->key());
    }
    QWindow::keyReleaseEvent(event);
}

void ViewportWindow::enterFpsMode() {
    if (camera_mode_ == CameraMode::Fps) return;
    camera_mode_ = CameraMode::Fps;
    fps_keys_held_.clear();
    setCursor(Qt::BlankCursor);
    fps_ignore_next_mouse_move_ = true;
    recenterFpsCursor();
    fps_last_tick_.start();
}

void ViewportWindow::exitFpsMode() {
    if (camera_mode_ == CameraMode::Orbit) return;
    camera_mode_ = CameraMode::Orbit;
    fps_keys_held_.clear();
    unsetCursor();
}

void ViewportWindow::recenterFpsCursor() {
    const QPoint center(width() / 2, height() / 2);
    QCursor::setPos(mapToGlobal(center));
    last_mouse_pos_ = center;
}

void ViewportWindow::fpsIntegrate() {
    if (camera_mode_ != CameraMode::Fps || fps_keys_held_.isEmpty()) return;

    qint64 ns = fps_last_tick_.nsecsElapsed();
    fps_last_tick_.restart();
    float dt = static_cast<float>(ns) * 1e-9f;
    if (dt > 0.1f) dt = 0.1f;  // clamp after stalls

    // View direction is target - eye = -offset.  offset components match
    // updateCamera() so forward stays consistent when mode flips.
    const float yaw_rad   = qDegreesToRadians(camera_yaw_);
    const float pitch_rad = qDegreesToRadians(camera_pitch_);
    const QVector3D offset(cosf(pitch_rad) * cosf(yaw_rad),
                           cosf(pitch_rad) * sinf(yaw_rad),
                           sinf(pitch_rad));
    const QVector3D world_up(0, 0, 1);
    const QVector3D forward = -offset;
    QVector3D right = QVector3D::crossProduct(forward, world_up);
    if (right.lengthSquared() > 1e-8f) right.normalize();

    QVector3D move(0, 0, 0);
    if (fps_keys_held_.contains(Qt::Key_W)) move += forward;
    if (fps_keys_held_.contains(Qt::Key_S)) move -= forward;
    if (fps_keys_held_.contains(Qt::Key_D)) move += right;
    if (fps_keys_held_.contains(Qt::Key_A)) move -= right;
    if (fps_keys_held_.contains(Qt::Key_E)) move += world_up;
    if (fps_keys_held_.contains(Qt::Key_Q)) move -= world_up;
    if (move.isNull()) return;
    move.normalize();

    const float speed_mul = fps_keys_held_.contains(Qt::Key_Shift) ? 5.0f : 1.0f;
    camera_target_ += move * (fps_move_speed_ * speed_mul * dt);
    have_cached_cull_ = false;
}

// --- HiZ occlusion culling (Phase 3C) -----------------------------------

// Baseline HiZ resolution.  256x128 is enough to cull big occluders
// (walls, slabs) reliably; finer detail doesn't help much because we're
// sampling the pyramid at the mip level where the AABB's rect is ~2
// texels anyway.  Readback cost is ~128 KB/frame ≈ negligible.
// AppSettings::hizResolution() drives the width; height tracks aspect.
// Captured once at first call — changing the setting requires a viewport
// reinitialization to take effect (rebuilding the pyramid mid-session
// would tangle with the framebuffer state machine for little gain).
static int hizBaseWidth() {
    static const int w = std::max(64, AppSettings::instance().hizResolution());
    return w;
}

static bool hizEnabled() {
    return AppSettings::instance().hizEnabled();
}

void ViewportWindow::buildHizPyramid() {
    if (!gl_initialized_) return;

    const int win_w = width()  * devicePixelRatio();
    const int win_h = height() * devicePixelRatio();
    if (win_w <= 0 || win_h <= 0) return;

    const int base_w = hizBaseWidth();
    const int base_h = std::max(1, (base_w * win_h) / win_w);

    // Resolve target (full window size, single sample, D24S8 to match Qt's
    // default FBO which uses depth+stencil even when only depth is requested).
    if (win_w != hiz_resolve_w_ || win_h != hiz_resolve_h_) {
        if (hiz_resolve_fbo_)        gl_->glDeleteFramebuffers(1, &hiz_resolve_fbo_);
        if (hiz_resolve_depth_tex_)  gl_->glDeleteTextures(1, &hiz_resolve_depth_tex_);
        gl_->glCreateTextures(GL_TEXTURE_2D, 1, &hiz_resolve_depth_tex_);
        gl_->glTextureStorage2D(hiz_resolve_depth_tex_, 1,
                                GL_DEPTH24_STENCIL8, win_w, win_h);
        gl_->glCreateFramebuffers(1, &hiz_resolve_fbo_);
        gl_->glNamedFramebufferTexture(hiz_resolve_fbo_, GL_DEPTH_STENCIL_ATTACHMENT,
                                       hiz_resolve_depth_tex_, 0);
        hiz_resolve_w_ = win_w;
        hiz_resolve_h_ = win_h;
    }

    if (base_w != hiz_base_w_ || base_h != hiz_base_h_) {
        if (hiz_fbo_)       gl_->glDeleteFramebuffers(1, &hiz_fbo_);
        if (hiz_depth_tex_) gl_->glDeleteTextures(1, &hiz_depth_tex_);
        gl_->glCreateTextures(GL_TEXTURE_2D, 1, &hiz_depth_tex_);
        gl_->glTextureStorage2D(hiz_depth_tex_, 1, GL_DEPTH_COMPONENT24,
                                base_w, base_h);
        gl_->glCreateFramebuffers(1, &hiz_fbo_);
        gl_->glNamedFramebufferTexture(hiz_fbo_, GL_DEPTH_ATTACHMENT,
                                       hiz_depth_tex_, 0);
        gl_->glNamedFramebufferDrawBuffer(hiz_fbo_, GL_NONE);
        gl_->glNamedFramebufferReadBuffer(hiz_fbo_, GL_NONE);
        {
            GLenum s = gl_->glCheckNamedFramebufferStatus(hiz_fbo_, GL_FRAMEBUFFER);
            if (s != GL_FRAMEBUFFER_COMPLETE)
                qWarning("HiZ FBO incomplete: 0x%04x", s);
        }

        hiz_base_w_ = base_w;
        hiz_base_h_ = base_h;
        hiz_depth_readback_.assign(base_w * base_h, 1.0f);

        // Build the mip-offset table.  Level 0 = base_w x base_h.
        hiz_mip_offset_.clear();
        hiz_mip_w_.clear();
        hiz_mip_h_.clear();
        uint32_t off = 0;
        int mw = base_w, mh = base_h;
        while (mw >= 1 && mh >= 1) {
            hiz_mip_offset_.push_back(off);
            hiz_mip_w_.push_back(static_cast<uint32_t>(mw));
            hiz_mip_h_.push_back(static_cast<uint32_t>(mh));
            off += static_cast<uint32_t>(mw) * static_cast<uint32_t>(mh);
            if (mw == 1 && mh == 1) break;
            mw = std::max(1, mw / 2);
            mh = std::max(1, mh / 2);
        }
        hiz_pyramid_.assign(off, 1.0f);
    }

    // Drain stale GL errors before HiZ pipeline.
    while (gl_->glGetError() != GL_NO_ERROR) {}

    // Step 1: MSAA default-fb → full-size SS resolve (same-size blit).
    gl_->glBindFramebuffer(GL_READ_FRAMEBUFFER, 0);
    gl_->glBindFramebuffer(GL_DRAW_FRAMEBUFFER, hiz_resolve_fbo_);
    gl_->glBlitFramebuffer(0, 0, win_w, win_h,
                           0, 0, win_w, win_h,
                           GL_DEPTH_BUFFER_BIT | GL_STENCIL_BUFFER_BIT, GL_NEAREST);

    // Step 2: downsample resolved depth to HiZ base via fullscreen-triangle.
    // glBlitFramebuffer with depth + scaling produces GL_INVALID_VALUE on
    // some drivers, so we sample the resolve texture and write gl_FragDepth.
    gl_->glBindFramebuffer(GL_FRAMEBUFFER, hiz_fbo_);
    gl_->glViewport(0, 0, hiz_base_w_, hiz_base_h_);
    gl_->glEnable(GL_DEPTH_TEST);
    gl_->glDepthFunc(GL_ALWAYS);
    gl_->glDepthMask(GL_TRUE);
    gl_->glColorMask(GL_FALSE, GL_FALSE, GL_FALSE, GL_FALSE);
    gl_->glUseProgram(hiz_downsample_program_);
    gl_->glTextureParameteri(hiz_resolve_depth_tex_, GL_TEXTURE_MIN_FILTER, GL_NEAREST);
    gl_->glTextureParameteri(hiz_resolve_depth_tex_, GL_TEXTURE_MAG_FILTER, GL_NEAREST);
    gl_->glTextureParameteri(hiz_resolve_depth_tex_, GL_TEXTURE_COMPARE_MODE, GL_NONE);
    gl_->glBindTextureUnit(0, hiz_resolve_depth_tex_);
    gl_->glUniform1i(gl_->glGetUniformLocation(hiz_downsample_program_, "u_depth"), 0);
    gl_->glUniform2f(gl_->glGetUniformLocation(hiz_downsample_program_, "u_inv_dest_size"),
                     1.0f / static_cast<float>(hiz_base_w_),
                     1.0f / static_cast<float>(hiz_base_h_));
    gl_->glBindVertexArray(hiz_downsample_vao_);
    gl_->glDrawArrays(GL_TRIANGLES, 0, 3);
    gl_->glBindVertexArray(0);
    gl_->glColorMask(GL_TRUE, GL_TRUE, GL_TRUE, GL_TRUE);
    gl_->glDepthFunc(GL_LESS);
    gl_->glBindFramebuffer(GL_FRAMEBUFFER, 0);
    gl_->glViewport(0, 0, win_w, win_h);

    // Synchronous readback into level 0 of the pyramid.  At 256x128 this
    // is ~128 KB and the driver copy is fast enough not to matter in
    // practice; PBO-ring async was tried and made orbiting flicker worse
    // (2-frame-stale depth vs 1-frame).
    gl_->glGetTextureImage(hiz_depth_tex_, 0, GL_DEPTH_COMPONENT, GL_FLOAT,
                           static_cast<GLsizei>(hiz_depth_readback_.size() * sizeof(float)),
                           hiz_depth_readback_.data());

    {
        static int diag = 5;
        static int skip = 60;
        if (skip > 0) { --skip; }
        else if (diag > 0) {
            --diag;
            float mn = 1.0f, mx = 0.0f;
            int zeros = 0, ones = 0;
            for (size_t i = 0; i < hiz_depth_readback_.size(); ++i) {
                float v = hiz_depth_readback_[i];
                if (v < mn) mn = v;
                if (v > mx) mx = v;
                if (v == 0.0f) ++zeros;
                if (v == 1.0f) ++ones;
            }
            int geom = (int)hiz_depth_readback_.size() - zeros - ones;
            qWarning("HiZ readback %dx%d: min=%.6f max=%.6f zeros=%d ones=%d geom=%d total=%d",
                     hiz_base_w_, hiz_base_h_, mn, mx, zeros, ones, geom,
                     (int)hiz_depth_readback_.size());
        }
    }

    // Copy level 0 into the pyramid, then max-reduce subsequent levels.
    std::memcpy(hiz_pyramid_.data() + hiz_mip_offset_[0],
                hiz_depth_readback_.data(),
                hiz_depth_readback_.size() * sizeof(float));
    for (size_t lvl = 1; lvl < hiz_mip_offset_.size(); ++lvl) {
        const uint32_t pw = hiz_mip_w_[lvl - 1];
        const uint32_t ph = hiz_mip_h_[lvl - 1];
        const uint32_t cw = hiz_mip_w_[lvl];
        const uint32_t ch = hiz_mip_h_[lvl];
        const float* parent = hiz_pyramid_.data() + hiz_mip_offset_[lvl - 1];
        float* child  = hiz_pyramid_.data() + hiz_mip_offset_[lvl];
        for (uint32_t y = 0; y < ch; ++y) {
            const uint32_t py0 = std::min(2 * y,     ph - 1);
            const uint32_t py1 = std::min(2 * y + 1, ph - 1);
            for (uint32_t x = 0; x < cw; ++x) {
                const uint32_t px0 = std::min(2 * x,     pw - 1);
                const uint32_t px1 = std::min(2 * x + 1, pw - 1);
                const float a = parent[py0 * pw + px0];
                const float b = parent[py0 * pw + px1];
                const float c = parent[py1 * pw + px0];
                const float d = parent[py1 * pw + px1];
                child[y * cw + x] = std::max(std::max(a, b), std::max(c, d));
            }
        }
    }

    hiz_vp_ = proj_matrix_ * view_matrix_;
    hiz_vp_valid_ = true;
}

bool ViewportWindow::aabbOccludedByHiz(const float mn[3], const float mx[3]) const {
    if (!hiz_vp_valid_ || hiz_pyramid_.empty()) return false;

    // Project all 8 corners through the HiZ frame's VP (stored last frame).
    // Track NDC min/max over x, y, z.  If any corner has w <= 0, the AABB
    // straddles the near plane and we skip (behaves like "not occluded").
    float sx_min =  std::numeric_limits<float>::infinity();
    float sx_max = -std::numeric_limits<float>::infinity();
    float sy_min =  std::numeric_limits<float>::infinity();
    float sy_max = -std::numeric_limits<float>::infinity();
    float sz_min =  std::numeric_limits<float>::infinity();
    const float* vp = hiz_vp_.constData();  // column-major
    for (int c = 0; c < 8; ++c) {
        const float x = (c & 1) ? mx[0] : mn[0];
        const float y = (c & 2) ? mx[1] : mn[1];
        const float z = (c & 4) ? mx[2] : mn[2];
        const float cx = vp[0]*x + vp[4]*y + vp[8]*z  + vp[12];
        const float cy = vp[1]*x + vp[5]*y + vp[9]*z  + vp[13];
        const float cz = vp[2]*x + vp[6]*y + vp[10]*z + vp[14];
        const float cw = vp[3]*x + vp[7]*y + vp[11]*z + vp[15];
        if (cw <= 1e-4f) return false;  // near-plane straddle
        const float inv = 1.0f / cw;
        const float nx = cx * inv;
        const float ny = cy * inv;
        const float nz = cz * inv;
        if (nx < sx_min) sx_min = nx;  if (nx > sx_max) sx_max = nx;
        if (ny < sy_min) sy_min = ny;  if (ny > sy_max) sy_max = ny;
        if (nz < sz_min) sz_min = nz;
    }

    if (sx_max < -1.0f || sx_min > 1.0f ||
        sy_max < -1.0f || sy_min > 1.0f) return false;
    if (sz_min < -1.0f) return false;

    sx_min = std::max(sx_min, -1.0f);
    sx_max = std::min(sx_max,  1.0f);
    sy_min = std::max(sy_min, -1.0f);
    sy_max = std::min(sy_max,  1.0f);

    const float u_min = 0.5f * (sx_min + 1.0f);
    const float u_max = 0.5f * (sx_max + 1.0f);
    const float v_min = 0.5f * (sy_min + 1.0f);
    const float v_max = 0.5f * (sy_max + 1.0f);
    const float aabb_near_depth = 0.5f * (sz_min + 1.0f);

    // Sample at a fine mip and reject ONLY if every texel agrees the AABB
    // is behind it.  One non-occluding texel → visible, early-out.  Cap
    // iteration to avoid slow queries on large projected rects.
    const int mip = std::min(1, (int)hiz_mip_offset_.size() - 1);
    const uint32_t mw = hiz_mip_w_[mip];
    const uint32_t mh = hiz_mip_h_[mip];
    int x0 = static_cast<int>(std::floor(u_min * mw));
    int x1 = static_cast<int>(std::ceil (u_max * mw));
    int y0 = static_cast<int>(std::floor(v_min * mh));
    int y1 = static_cast<int>(std::ceil (v_max * mh));
    if (x0 < 0) x0 = 0;
    if (y0 < 0) y0 = 0;
    if (x1 > (int)mw) x1 = mw;
    if (y1 > (int)mh) y1 = mh;
    if (x1 <= x0 || y1 <= y0) return false;

    static constexpr int MAX_HIZ_SAMPLES = 64;
    if ((x1 - x0) * (y1 - y0) > MAX_HIZ_SAMPLES) return false;

    const float* level = hiz_pyramid_.data() + hiz_mip_offset_[mip];
    for (int y = y0; y < y1; ++y) {
        const float* row = level + static_cast<size_t>(y) * mw;
        for (int x = x0; x < x1; ++x) {
            if (aabb_near_depth <= row[x]) return false;
        }
    }
    return true;
}

uint32_t ViewportWindow::pickObjectAt(int x, int y) {
    if (!gl_initialized_) return 0;
    context_->makeCurrent(this);

    int w = width() * devicePixelRatio();
    int h = height() * devicePixelRatio();
    if (pick_width_ != w || pick_height_ != h) {
        if (pick_fbo_) gl_->glDeleteFramebuffers(1, &pick_fbo_);
        if (pick_color_tex_)  gl_->glDeleteTextures(1, &pick_color_tex_);
        if (pick_pos_tex_)    gl_->glDeleteTextures(1, &pick_pos_tex_);
        if (pick_normal_tex_) gl_->glDeleteTextures(1, &pick_normal_tex_);
        if (pick_depth_rbo_)  gl_->glDeleteRenderbuffers(1, &pick_depth_rbo_);
        gl_->glCreateFramebuffers(1, &pick_fbo_);
        gl_->glCreateTextures(GL_TEXTURE_2D, 1, &pick_color_tex_);
        gl_->glTextureStorage2D(pick_color_tex_, 1, GL_R32UI, w, h);
        gl_->glNamedFramebufferTexture(pick_fbo_, GL_COLOR_ATTACHMENT0, pick_color_tex_, 0);
        gl_->glCreateTextures(GL_TEXTURE_2D, 1, &pick_pos_tex_);
        gl_->glTextureStorage2D(pick_pos_tex_, 1, GL_RGB32F, w, h);
        gl_->glNamedFramebufferTexture(pick_fbo_, GL_COLOR_ATTACHMENT1, pick_pos_tex_, 0);
        gl_->glCreateTextures(GL_TEXTURE_2D, 1, &pick_normal_tex_);
        gl_->glTextureStorage2D(pick_normal_tex_, 1, GL_RGB16F, w, h);
        gl_->glNamedFramebufferTexture(pick_fbo_, GL_COLOR_ATTACHMENT2, pick_normal_tex_, 0);
        gl_->glCreateRenderbuffers(1, &pick_depth_rbo_);
        gl_->glNamedRenderbufferStorage(pick_depth_rbo_, GL_DEPTH_COMPONENT24, w, h);
        gl_->glNamedFramebufferRenderbuffer(pick_fbo_, GL_DEPTH_ATTACHMENT, GL_RENDERBUFFER, pick_depth_rbo_);
        static const GLenum draw_bufs[3] = {
            GL_COLOR_ATTACHMENT0, GL_COLOR_ATTACHMENT1, GL_COLOR_ATTACHMENT2
        };
        gl_->glNamedFramebufferDrawBuffers(pick_fbo_, 3, draw_bufs);
        pick_width_ = w;
        pick_height_ = h;
    }

    renderPickPass();

    // The pick pass overwrote each model's visible_ssbo / indirect_buffer with
    // pick-specific cull params (no contribution cull, no HiZ).  Invalidate
    // the cached cull so the next render() rebuilds them with main-render
    // params; otherwise the viewport draws with stale pick-pass buffers and
    // shading looks wrong until the camera moves.
    have_cached_cull_ = false;

    int px = x * devicePixelRatio();
    int py = (height() - y) * devicePixelRatio();
    uint32_t pixel = 0;
    gl_->glGetTextureSubImage(pick_color_tex_, 0, px, py, 0, 1, 1, 1,
                              GL_RED_INTEGER, GL_UNSIGNED_INT, sizeof(pixel), &pixel);
    return pixel;
}

std::unordered_set<uint32_t> ViewportWindow::picksInRect(const QRect& rect_logical) {
    std::unordered_set<uint32_t> out;
    if (!gl_initialized_) return out;
    if (rect_logical.isEmpty() || rect_logical.width() < 1 || rect_logical.height() < 1) {
        return out;
    }
    context_->makeCurrent(this);

    const int w = width()  * devicePixelRatio();
    const int h = height() * devicePixelRatio();
    if (w <= 0 || h <= 0) return out;

    // Reuse the same allocation path as pickObjectAt — call it once at
    // (0,0) just to ensure the framebuffer matches the current size and
    // the pick pass has been rendered for this state.
    pickObjectAt(rect_logical.left(), rect_logical.top());

    // Convert the logical-coords rect to physical pick-tex coords with
    // GL's bottom-left origin.  Clamp to surface to avoid OOB reads.
    const qreal dpr = devicePixelRatio();
    QRect r = rect_logical
        .intersected(QRect(0, 0, width(), height()));
    if (r.isEmpty()) return out;
    const int rx = int(r.left() * dpr);
    const int ry_top = int((height() - (r.top() + r.height())) * dpr);
    const int rw = std::max(1, int(r.width()  * dpr));
    const int rh = std::max(1, int(r.height() * dpr));
    if (rx < 0 || ry_top < 0 || rx + rw > w || ry_top + rh > h) {
        // Final defensive clamp — devicePixelRatio + integer rounding can
        // push the rect a pixel off the texture; readback would silently
        // return zeros for those rows.
        const int cx  = std::max(0, std::min(rx, w - 1));
        const int cy  = std::max(0, std::min(ry_top, h - 1));
        const int cw2 = std::max(1, std::min(rw, w - cx));
        const int ch2 = std::max(1, std::min(rh, h - cy));
        std::vector<uint32_t> pixels(size_t(cw2) * size_t(ch2), 0);
        gl_->glGetTextureSubImage(pick_color_tex_, 0, cx, cy, 0, cw2, ch2, 1,
                                  GL_RED_INTEGER, GL_UNSIGNED_INT,
                                  GLsizei(pixels.size() * sizeof(uint32_t)),
                                  pixels.data());
        for (uint32_t id : pixels) if (id != 0) out.insert(id);
        return out;
    }

    std::vector<uint32_t> pixels(size_t(rw) * size_t(rh), 0);
    gl_->glGetTextureSubImage(pick_color_tex_, 0, rx, ry_top, 0, rw, rh, 1,
                              GL_RED_INTEGER, GL_UNSIGNED_INT,
                              GLsizei(pixels.size() * sizeof(uint32_t)),
                              pixels.data());
    for (uint32_t id : pixels) if (id != 0) out.insert(id);
    return out;
}

bool ViewportWindow::pickSurfaceAt(int x, int y,
                                   uint32_t& object_id_out,
                                   QVector3D& world_pos_out,
                                   QVector3D& world_normal_out) {
    const uint32_t id = pickObjectAt(x, y);
    if (id == 0) return false;

    const int px = x * devicePixelRatio();
    const int py = (height() - y) * devicePixelRatio();
    float pos[3]    = {0, 0, 0};
    float normal[3] = {0, 0, 0};
    gl_->glGetTextureSubImage(pick_pos_tex_, 0, px, py, 0, 1, 1, 1,
                              GL_RGB, GL_FLOAT, sizeof(pos), pos);
    gl_->glGetTextureSubImage(pick_normal_tex_, 0, px, py, 0, 1, 1, 1,
                              GL_RGB, GL_FLOAT, sizeof(normal), normal);

    QVector3D n(normal[0], normal[1], normal[2]);
    if (n.lengthSquared() < 1e-8f) {
        // Pick succeeded for object_id but normal attachment was empty —
        // unexpected (shader writes it on every covered fragment), so bail
        // rather than hand the caller a degenerate plane.
        return false;
    }
    object_id_out    = id;
    world_pos_out    = QVector3D(pos[0], pos[1], pos[2]);
    world_normal_out = n.normalized();
    return true;
}

void ViewportWindow::uploadClipPlaneUniforms(GLuint program) {
    const GLint u_count = gl_->glGetUniformLocation(program, "u_clip_count");
    if (u_count < 0) return;  // program does not declare clipping uniforms
    const int n = qMin(int(section_planes_.size()), MaxSectionPlanes);
    gl_->glUniform1i(u_count, n);
    if (n == 0) return;
    float packed[MaxSectionPlanes * 4] = {};
    for (int i = 0; i < n; ++i) {
        packed[i * 4 + 0] = section_planes_[i].n.x();
        packed[i * 4 + 1] = section_planes_[i].n.y();
        packed[i * 4 + 2] = section_planes_[i].n.z();
        packed[i * 4 + 3] = section_planes_[i].d;
    }
    const GLint u_planes = gl_->glGetUniformLocation(program, "u_clip_planes");
    if (u_planes >= 0) {
        gl_->glUniform4fv(u_planes, n, packed);
    }
}

bool ViewportWindow::addSectionPlaneAtSurface(const QVector3D& point,
                                              const QVector3D& normal) {
    if (int(section_planes_.size()) >= MaxSectionPlanes) {
        qWarning("Section plane cap (%d) reached", MaxSectionPlanes);
        return false;
    }
    QVector3D n = normal;
    if (n.lengthSquared() < 1e-8f) return false;
    n.normalize();
    // Auto-flip so the plane clips the camera-facing half — first click
    // immediately cuts away what's between the user and the clicked surface.
    const QVector3D eye_dir = (camera_eye_ - point);
    if (QVector3D::dotProduct(n, eye_dir) < 0.0f) n = -n;

    SectionPlane p;
    p.n = n;
    p.origin = point;
    p.d = -QVector3D::dotProduct(n, point);
    section_planes_.push_back(p);
    have_cached_cull_ = false;
    requestUpdate();
    return true;
}

void ViewportWindow::toggleSectionTool() {
    section_tool_active_ = !section_tool_active_;
    if (!section_tool_active_) {
        section_drag_active_ = false;
        section_drag_index_  = -1;
    }
    requestUpdate();
}

void ViewportWindow::toggleProjection() {
    projection_ortho_ = !projection_ortho_;
    have_cached_cull_ = false;  // proj_matrix_ changes -> frustum planes change
    requestUpdate();
}

void ViewportWindow::setStandardView(float yaw_deg, float pitch_deg) {
    // Bypasses the orbit-MMB pitch clamp so top/bottom can land exactly on
    // ±90°.  updateCamera() picks the up vector based on |pitch|, so the
    // resulting view is well-conditioned at the poles too.
    camera_yaw_   = yaw_deg;
    camera_pitch_ = pitch_deg;
    have_cached_cull_ = false;
    requestUpdate();
}

void ViewportWindow::removeSectionPlane(int index) {
    if (index < 0 || index >= int(section_planes_.size())) return;
    section_planes_.erase(section_planes_.begin() + index);
    have_cached_cull_ = false;
    requestUpdate();
}

void ViewportWindow::clearSectionPlanes() {
    if (section_planes_.empty()) return;
    section_planes_.clear();
    have_cached_cull_ = false;
    requestUpdate();
}

void ViewportWindow::cullAndUploadVisible(ModelGpuData& m, const float planes[6][4],
                                          float focal_px, float min_pixel_radius) {
    cullModelCpu(m, planes, focal_px, min_pixel_radius);
    uploadCullResults(m);
}

void ViewportWindow::cullModelCpu(ModelGpuData& m, const float planes[6][4],
                                  float focal_px, float min_pixel_radius) {
    // Per-mesh scratch, split by winding × LOD.  Winding split lets the draw
    // pass toggle glFrontFace once between two MDI calls so GL_CULL_FACE does
    // the right thing for both.  LOD split means instances that want the
    // decimated mesh go into a different bucket that emits against
    // mesh.lod1_ebo_byte_offset / lod1_index_count.
    QElapsedTimer phase_timer;
    phase_timer.start();

    auto resize_if = [&](std::vector<std::vector<uint32_t>>& v) {
        if (v.size() < m.meshes.size()) v.resize(m.meshes.size());
    };
    resize_if(m.vis_fwd_lod0);
    resize_if(m.vis_fwd_lod1);
    resize_if(m.vis_rev_lod0);
    resize_if(m.vis_rev_lod1);
    for (size_t i = 0; i < m.meshes.size(); ++i) {
        m.vis_fwd_lod0[i].clear();
        m.vis_fwd_lod1[i].clear();
        m.vis_rev_lod0[i].clear();
        m.vis_rev_lod1[i].clear();
    }
    cull_clear_ns_ += phase_timer.nsecsElapsed();
    phase_timer.restart();

    // LOD1 switches in when projected sphere radius (in pixels) drops below
    // this threshold.  AppSettings drives it (default 30).  Set to 0 in
    // settings to disable LOD1 entirely (always draw LOD0).
    const float lod1_px_threshold =
        static_cast<float>(AppSettings::instance().lod1PixelThreshold());

    // Bounding-sphere contribution test: approximate an AABB by its enclosing
    // sphere (centre = midpoint, radius = half-diagonal).  Project radius to
    // pixels as r_px = focal_px * r / distance (perspective).  Reject if
    // smaller than the threshold.  Returns true when the node/instance
    // should be kept.
    //
    // If the camera is inside the AABB the sphere-radius test would reject
    // by distance going to zero / negative — we handle that by skipping the
    // test whenever the camera lies within an inflated AABB.  Cheap and
    // conservative: never drops things you're standing next to.
    const float cx = camera_eye_.x();
    const float cy = camera_eye_.y();
    const float cz = camera_eye_.z();
    // In ortho the projected pixel size of a bounding sphere doesn't depend
    // on per-instance distance — the ortho box scales the entire scene by a
    // constant.  Substitute that constant (camera_distance_, which is
    // exactly the half-height/tan(fovy/2) used to size the box) for the
    // per-instance dist so the same `r_px = focal_px * r / dist` formula
    // and threshold survive both modes.  Snapshot now so the worker threads
    // see a consistent value.
    const bool  ortho_mode    = projection_ortho_;
    const float ortho_dist    = camera_distance_;

    auto contributionPasses = [&](const float mn[3], const float mx[3]) -> bool {
        if (min_pixel_radius <= 0.0f) return true;
        // Camera inside AABB? Always keep — only relevant in perspective
        // where dist→0 would make r_px blow up; harmless in ortho too.
        if (cx >= mn[0] && cx <= mx[0] &&
            cy >= mn[1] && cy <= mx[1] &&
            cz >= mn[2] && cz <= mx[2]) {
            return true;
        }
        float ex = 0.5f * (mx[0] - mn[0]);
        float ey = 0.5f * (mx[1] - mn[1]);
        float ez = 0.5f * (mx[2] - mn[2]);
        float radius = std::sqrt(ex*ex + ey*ey + ez*ez);
        float dist;
        if (ortho_mode) {
            dist = ortho_dist;
        } else {
            float dx = 0.5f * (mx[0] + mn[0]) - cx;
            float dy = 0.5f * (mx[1] + mn[1]) - cy;
            float dz = 0.5f * (mx[2] + mn[2]) - cz;
            dist = std::sqrt(dx*dx + dy*dy + dz*dz);
        }
        // r_px = focal_px * radius / dist; compare r_px >= min_pixel_radius,
        // rearranged to avoid the divide.
        return focal_px * radius >= min_pixel_radius * dist;
    };

    // Returns projected sphere radius in pixels (or +inf when camera is
    // inside the AABB).  Shares the geometry with contributionPasses; this
    // version returns the value so we can also use it for LOD selection.
    auto pixelRadius = [&](const float mn[3], const float mx[3]) -> float {
        if (cx >= mn[0] && cx <= mx[0] &&
            cy >= mn[1] && cy <= mx[1] &&
            cz >= mn[2] && cz <= mx[2]) {
            return std::numeric_limits<float>::infinity();
        }
        float ex = 0.5f * (mx[0] - mn[0]);
        float ey = 0.5f * (mx[1] - mn[1]);
        float ez = 0.5f * (mx[2] - mn[2]);
        float radius = std::sqrt(ex*ex + ey*ey + ez*ez);
        float dist;
        if (ortho_mode) {
            dist = ortho_dist;
        } else {
            float dx = 0.5f * (mx[0] + mn[0]) - cx;
            float dy = 0.5f * (mx[1] + mn[1]) - cy;
            float dz = 0.5f * (mx[2] + mn[2]) - cz;
            dist = std::sqrt(dx*dx + dy*dy + dz*dz);
        }
        return dist > 0.0f ? focal_px * radius / dist
                           : std::numeric_limits<float>::infinity();
    };

    // HiZ occlusion is skipped entirely when the pick pass runs
    // (min_pixel_radius == 0 on that path), when the user disables it via
    // env var, or before the first pyramid has been built.
    //
    // Crucially, HiZ is also skipped when the stored VP (hiz_vp_, captured at
    // the end of the previous frame) differs from this frame's VP — i.e.
    // whenever the camera has moved.  The stored depth buffer encodes what
    // was visible from hiz_vp_'s viewpoint; projecting a current-frame AABB
    // through that VP answers "was this occluded LAST frame?", which is only
    // a correct proxy for "is this occluded NOW?" when the camera is static.
    // Orbiting past a wall would otherwise leave objects persistently culled
    // because prior frames' depth buffers only ever contained the wall (the
    // objects behind it were themselves HiZ-culled, never drawn, so never in
    // the buffer — a self-reinforcing feedback loop).  On static views HiZ
    // kicks in after a single frame of lag.
    const QMatrix4x4 current_vp = proj_matrix_ * view_matrix_;
    // Default on: motion-frame HiZ stays trusted across small VP changes.
    // Env var =0 reverts to the strict gate (previous frame's pyramid is
    // discarded the moment the view-projection drifts) — useful when
    // chasing the missing-geometry class of HiZ correctness bug.
    static const bool hiz_force_motion = []{
        const char* e = std::getenv("IFC_HIZ_MOTION");
        return !(e && e[0] == '0');
    }();
    const bool hiz_vp_matches = hiz_vp_valid_
        && (hiz_force_motion || hiz_vp_ == current_vp);
    const bool hiz_on = hizEnabled() && min_pixel_radius > 0.0f && hiz_vp_matches;

    // Hot path: read the AABB from the compact bvh_items array (28 B stride)
    // rather than the wide InstanceCpu (104 B stride).  Most instances fail
    // frustum or contribution, so we want to avoid touching the wider struct
    // until a survivor needs its mesh_id.  This alone turns the cull from
    // cache-miss-per-instance into stream-friendly linear reads.
    auto test_and_push = [&](uint32_t inst_idx) {
        const BvhItem& item = m.bvh_items[inst_idx];
        if (!aabbInFrustum(item.aabb_min, item.aabb_max, planes)) return;
        if (!contributionPasses(item.aabb_min, item.aabb_max)) return;
        if (hiz_on && aabbOccludedByHiz(item.aabb_min, item.aabb_max)) {
            hiz_reject_count_.fetch_add(1, std::memory_order_relaxed);
            return;
        }
        // Survivor — now pay the wide-struct fetch for mesh_id.
        const InstanceCpu& inst = m.instances[inst_idx];
        if (inst.mesh_id >= m.meshes.size()) return;
        const MeshInfo& mesh = m.meshes[inst.mesh_id];
        const bool want_lod1 = mesh.lod1_index_count > 0 &&
            lod1_px_threshold > 0.0f &&
            pixelRadius(item.aabb_min, item.aabb_max) < lod1_px_threshold;
        const bool reflected = inst_idx < m.instance_reflected.size()
            && m.instance_reflected[inst_idx] != 0;
        auto& bucket =
            reflected ? (want_lod1 ? m.vis_rev_lod1
                                   : m.vis_rev_lod0)
                      : (want_lod1 ? m.vis_fwd_lod1
                                   : m.vis_fwd_lod0);
        bucket[inst.mesh_id].push_back(inst_idx);
    };

    if (!m.bvh.nodes.empty()) {
        uint32_t stack[64];
        int sp = 0;
        stack[sp++] = 0;
        while (sp > 0) {
            uint32_t ni = stack[--sp];
            const BvhNode& n = m.bvh.nodes[ni];
            if (!aabbInFrustum(n.aabb_min, n.aabb_max, planes)) continue;
            // Contribution cull the whole subtree: if the node's enclosing
            // sphere is below threshold, every child is too.
            if (!contributionPasses(n.aabb_min, n.aabb_max)) continue;
            // HiZ cull the whole subtree: if the node AABB is fully
            // occluded, every leaf is too.  The conservative test (AABB
            // near-depth vs max pyramid depth) never rejects a visible
            // parent wrongly even when some children could have peeked
            // through.
            if (hiz_on && aabbOccludedByHiz(n.aabb_min, n.aabb_max)) continue;
            if (n.count > 0) {
                for (uint32_t k = 0; k < n.count; ++k) {
                    uint32_t item_idx = m.bvh.item_indices[n.right_or_first + k];
                    test_and_push(item_idx);
                }
            } else {
                // Left child = ni + 1, right child = n.right_or_first.
                // Push right first so left is popped next (DFS order).
                if (sp + 2 <= 64) {
                    stack[sp++] = n.right_or_first;
                    stack[sp++] = ni + 1;
                }
            }
        }
    } else {
        for (uint32_t i = 0; i < m.instances.size(); ++i) test_and_push(i);
    }
    cull_traverse_ns_ += phase_timer.nsecsElapsed();
    phase_timer.restart();

    // Flatten fwd-slice first (LOD0 then LOD1), then rev-slice (ditto), into
    // visible_flat_.  Commands for the fwd slice fill [0, indirect_forward_count),
    // rev fills [indirect_forward_count, end).  LOD0/LOD1 within a winding
    // slice are contiguous — winding is what requires glFrontFace to flip
    // between MDI calls, LOD is not.
    m.visible_flat.clear();
    m.indirect_scratch.clear();

    auto emit_slice = [&](std::vector<std::vector<uint32_t>>& by_mesh, int lod) {
        for (size_t mi = 0; mi < m.meshes.size(); ++mi) {
            const auto& mesh = m.meshes[mi];
            const uint32_t vis_count = static_cast<uint32_t>(by_mesh[mi].size());
            const uint32_t idx_count =
                (lod == 1) ? mesh.lod1_index_count : mesh.index_count;
            const uint32_t ebo_off =
                (lod == 1) ? mesh.lod1_ebo_byte_offset : mesh.ebo_byte_offset;
            if (vis_count == 0 || idx_count == 0) continue;

            DrawElementsIndirectCommand cmd;
            cmd.count         = idx_count;
            cmd.instanceCount = vis_count;
            cmd.firstIndex    = ebo_off / sizeof(uint32_t);
            cmd.baseVertex    = mesh.vbo_byte_offset / INSTANCED_VERTEX_STRIDE_BYTES;
            cmd.baseInstance  = static_cast<uint32_t>(m.visible_flat.size());
            m.indirect_scratch.push_back(cmd);

            m.visible_flat.insert(m.visible_flat.end(),
                                  by_mesh[mi].begin(), by_mesh[mi].end());
        }
    };

    emit_slice(m.vis_fwd_lod0, 0);
    emit_slice(m.vis_fwd_lod1, 1);
    m.indirect_forward_count = static_cast<uint32_t>(m.indirect_scratch.size());
    emit_slice(m.vis_rev_lod0, 0);
    emit_slice(m.vis_rev_lod1, 1);
    m.indirect_command_count = static_cast<uint32_t>(m.indirect_scratch.size());

    // Per-model stats snapshot — summed into the frame counters regardless
    // of whether this frame ran a full cull or reused the cached one.
    uint32_t model_vis_obj = 0, model_vis_tri = 0;
    for (const auto& cmd : m.indirect_scratch) {
        model_vis_tri += (cmd.count / 3) * cmd.instanceCount;
        model_vis_obj += cmd.instanceCount;
    }
    m.cached_visible_objects   = model_vis_obj;
    m.cached_visible_triangles = model_vis_tri;

    cull_emit_ns_ += phase_timer.nsecsElapsed();
}

void ViewportWindow::uploadCullResults(ModelGpuData& m) {
    QElapsedTimer phase_timer;
    phase_timer.start();

    // Upload visible list (keep binding alive even when empty).
    size_t vis_bytes = std::max<size_t>(m.visible_flat.size() * sizeof(uint32_t),
                                        sizeof(uint32_t));
    if (m.visible_ssbo == 0 || m.visible_ssbo_capacity < vis_bytes) {
        if (m.visible_ssbo) gl_->glDeleteBuffers(1, &m.visible_ssbo);
        size_t new_cap = m.visible_ssbo_capacity ? m.visible_ssbo_capacity : 4096;
        while (new_cap < vis_bytes) new_cap *= 2;
        gl_->glCreateBuffers(1, &m.visible_ssbo);
        gl_->glNamedBufferStorage(m.visible_ssbo, new_cap, nullptr, GL_DYNAMIC_STORAGE_BIT);
        m.visible_ssbo_capacity = new_cap;
    }
    if (!m.visible_flat.empty()) {
        gl_->glNamedBufferSubData(m.visible_ssbo, 0,
            m.visible_flat.size() * sizeof(uint32_t), m.visible_flat.data());
    }

    // Upload indirect command buffer.
    size_t ind_bytes = m.indirect_scratch.size() * sizeof(DrawElementsIndirectCommand);
    if (ind_bytes == 0) {
        cull_upload_ns_ += phase_timer.nsecsElapsed();
        return;
    }
    if (m.indirect_buffer == 0 || m.indirect_capacity < ind_bytes) {
        if (m.indirect_buffer) gl_->glDeleteBuffers(1, &m.indirect_buffer);
        size_t new_cap = m.indirect_capacity ? m.indirect_capacity : 4096;
        while (new_cap < ind_bytes) new_cap *= 2;
        gl_->glCreateBuffers(1, &m.indirect_buffer);
        gl_->glNamedBufferStorage(m.indirect_buffer, new_cap, nullptr, GL_DYNAMIC_STORAGE_BIT);
        m.indirect_capacity = new_cap;
    }
    gl_->glNamedBufferSubData(m.indirect_buffer, 0, ind_bytes, m.indirect_scratch.data());
    cull_upload_ns_ += phase_timer.nsecsElapsed();
}

void ViewportWindow::updateCamera() {
    float yaw_rad = qDegreesToRadians(camera_yaw_);
    float pitch_rad = qDegreesToRadians(camera_pitch_);
    QVector3D eye;
    eye.setX(camera_target_.x() + camera_distance_ * cosf(pitch_rad) * cosf(yaw_rad));
    eye.setY(camera_target_.y() + camera_distance_ * cosf(pitch_rad) * sinf(yaw_rad));
    eye.setZ(camera_target_.z() + camera_distance_ * sinf(pitch_rad));
    camera_eye_ = eye;
    view_matrix_.setToIdentity();
    // Default up is world +Z.  Within ~1° of straight-up/down, switch to
    // world +Y so lookAt's side vector doesn't degenerate (forward × up
    // → 0).  Y-as-north is the architectural top-view convention.
    const QVector3D up = (std::abs(camera_pitch_) >= 89.0f)
                       ? QVector3D(0, 1, 0)
                       : QVector3D(0, 0, 1);
    view_matrix_.lookAt(eye, camera_target_, up);
    proj_matrix_.setToIdentity();
    float aspect = width() > 0 ? float(width()) / float(height()) : 1.0f;
    if (projection_ortho_) {
        // Size the ortho box so it shows the same world rectangle the
        // perspective camera would see at the pivot's distance — toggling
        // at any zoom keeps the framing roughly identical.
        const float half_h = camera_distance_ * tanf(qDegreesToRadians(camera_fov_y_deg_ * 0.5f));
        const float half_w = half_h * aspect;
        // Near/far span ±10× distance; matches the perspective far so
        // typical scenes always fit between the planes.
        const float depth = camera_distance_ * 10.0f;
        proj_matrix_.ortho(-half_w, half_w, -half_h, half_h, -depth, depth);
    } else {
        proj_matrix_.perspective(camera_fov_y_deg_, aspect, 0.1f, camera_distance_ * 10.0f);
    }
}

void ViewportWindow::render() {
    if (!gl_initialized_ || !isExposed()) return;

    QElapsedTimer frame_cost_clock;
    frame_cost_clock.start();

    // Advance FPS-mode camera by wall-clock dt since the last frame, then let
    // the view matrix derive from the new target.  Driving this from render()
    // rather than a QTimer means a long frame costs exactly one missed step
    // (caught up on the next frame) instead of a backlog that prints as a
    // stall.
    fpsIntegrate();

    context_->makeCurrent(this);
    updateCamera();

    int w = width()  * devicePixelRatio();
    int h = height() * devicePixelRatio();
    gl_->glViewport(0, 0, w, h);
    gl_->glClear(GL_COLOR_BUFFER_BIT | GL_DEPTH_BUFFER_BIT);

    QMatrix4x4 vp = proj_matrix_ * view_matrix_;
    float planes[6][4];
    extractFrustumPlanes(vp, planes);

    // Pixels-per-radian vertical focal length.  Combined with per-instance
    // world-space radius this gives screen-space pixel size for contribution
    // culling below.
    const float focal_px = 0.5f * static_cast<float>(h) /
        std::tan(qDegreesToRadians(0.5f * camera_fov_y_deg_));
    const float base_min_pixel_radius =
        static_cast<float>(AppSettings::instance().minPixelRadius());
    // 0 = disabled (motion uses the same threshold as still frames).
    const float motion_min_pixel_radius =
        static_cast<float>(AppSettings::instance().motionMinPixelRadius());

    gl_->glUseProgram(main_program_);
    GLint u_vp        = gl_->glGetUniformLocation(main_program_, "u_view_projection");
    GLint u_light     = gl_->glGetUniformLocation(main_program_, "u_light_dir");
    GLint u_fill      = gl_->glGetUniformLocation(main_program_, "u_fill_dir");
    GLint u_sky       = gl_->glGetUniformLocation(main_program_, "u_sky_color");
    GLint u_ground    = gl_->glGetUniformLocation(main_program_, "u_ground_color");
    GLint u_active    = gl_->glGetUniformLocation(main_program_, "u_active_id");
    gl_->glUniformMatrix4fv(u_vp, 1, GL_FALSE, vp.constData());
    // Key light: high noon-ish from off-camera; fill ~120° away so back-of-
    // object surfaces still get some direct contribution.  Sky/ground tints
    // are a neutral cool-on-warm pairing — readable on white walls and grey
    // slabs without colouring shaded faces noticeably.
    // Both directions are roughly unit-length (~0.99) — matches how
    // u_light_dir was already supplied and keeps the brightness math sane.
    gl_->glUniform3f(u_light,  0.3f,  0.5f, 0.8f);
    gl_->glUniform3f(u_fill,  -0.3f, -0.5f, 0.8f);
    gl_->glUniform3f(u_sky,    0.55f, 0.60f, 0.70f);
    gl_->glUniform3f(u_ground, 0.35f, 0.32f, 0.28f);
    gl_->glUniform1ui(u_active, selection_.activeObjectId());
    selection_.bindForRender(/*binding_index=*/3);
    uploadClipPlaneUniforms(main_program_);

    visible_triangles_ = 0;
    visible_objects_ = 0;
    gl_draw_calls_ = 0;
    indirect_sub_draws_ = 0;
    // Only reset hiz_reject_count_ on frames where we actually re-cull;
    // otherwise we'd wipe the previous cull's number and print 0 every
    // still frame.  See the cull_this_frame branch below.

    // Decide whether this frame's view+scene is identical to the last
    // successful cull.  If so the per-model indirect buffers / visible
    // SSBOs are still valid — we just re-issue the draws from them and
    // skip the expensive cull traversal entirely.
    const bool camera_unchanged = have_cached_cull_
        && last_cull_view_ == view_matrix_
        && last_cull_proj_ == proj_matrix_;
    const bool camera_moving = !camera_unchanged;
    const bool use_motion_threshold = camera_moving
        && motion_min_pixel_radius > base_min_pixel_radius;
    // Force a re-cull on the first still frame after motion so we
    // restore the base contribution threshold and clear stale HiZ.
    const bool needs_settle_recull = !camera_moving
        && last_cull_was_motion_;
    const bool cull_this_frame = camera_moving || needs_settle_recull;
    // Invalidate HiZ on the settle frame: the pyramid was built from the
    // motion frame's sparse depth (aggressive threshold hid objects whose
    // depth would normally populate the pyramid), causing false occlusion.
    if (needs_settle_recull)
        hiz_vp_valid_ = false;
    // Contribution culling works in both modes: cullModelCpu substitutes
    // camera_distance_ for the per-instance dist when projection_ortho_ is
    // set, which matches the ortho box's constant pixels-per-world.
    const float min_pixel_radius = use_motion_threshold
        ? motion_min_pixel_radius : base_min_pixel_radius;
    if (cull_this_frame) {
        hiz_reject_count_.store(0, std::memory_order_relaxed);
        last_cull_was_motion_ = camera_moving;
    } else {
        ++cull_skipped_frames_;
    }

    // Start each frame with CCW-is-front; the two-pass draw below flips
    // back and forth.  Harmless when culling is off.
    gl_->glFrontFace(GL_CCW);

    static const bool mt_cull_enabled = []{
        const char* e = std::getenv("IFC_CULL_THREADS");
        return !(e && e[0] == '0');
    }();

    QElapsedTimer cull_wall_timer;
    if (cull_this_frame) {
        cull_wall_timer.start();
        std::vector<ModelGpuData*> cull_targets;
        cull_targets.reserve(models_gpu_.size());
        for (auto& [mid, m] : models_gpu_) {
            if (m.hidden || !m.ssbo || m.ssbo_instance_count == 0) continue;
            cull_targets.push_back(&m);
        }

        if (mt_cull_enabled && cull_targets.size() > 1) {
            std::vector<std::future<void>> futs;
            futs.reserve(cull_targets.size());
            for (ModelGpuData* mp : cull_targets) {
                const float mpr = min_pixel_radius;
                futs.emplace_back(std::async(std::launch::async,
                    [this, mp, &planes, focal_px, mpr]() {
                        cullModelCpu(*mp, planes, focal_px, mpr);
                    }));
            }
            for (auto& f : futs) f.get();
        } else {
            for (ModelGpuData* mp : cull_targets) {
                cullModelCpu(*mp, planes, focal_px, min_pixel_radius);
            }
        }

        cull_wall_ns_ += cull_wall_timer.nsecsElapsed();
    }

    for (auto& [model_id, m] : models_gpu_) {
        if (m.hidden || !m.ssbo || m.ssbo_instance_count == 0) continue;

        if (cull_this_frame) {
            uploadCullResults(m);
        }
        if (m.indirect_command_count == 0) continue;

        gl_->glBindVertexArray(m.vao);
        gl_->glBindBufferBase(GL_SHADER_STORAGE_BUFFER, 0, m.ssbo);
        gl_->glBindBufferBase(GL_SHADER_STORAGE_BUFFER, 1, m.visible_ssbo);
        gl_->glBindBufferBase(GL_SHADER_STORAGE_BUFFER, 2, m.mesh_info_ssbo);
        gl_->glBindBuffer(GL_DRAW_INDIRECT_BUFFER, m.indirect_buffer);

        uint32_t fwd = m.indirect_forward_count;
        uint32_t rev = m.indirect_command_count - fwd;
        // Perf diagnostics (confirmed 2026-04 on GTX 1650 @ 128M tris:
        // draw-bound, not upload-bound — see README Phase 3):
        //   IFC_SKIP_MDI=1         skip the actual MDI draws (keeps cull +
        //                          upload + binds).  FPS jump == draw-bound.
        //   IFC_MAX_SUBDRAWS=N     truncate drawcount to N per MDI.  Lets
        //                          you distinguish per-subdraw command-
        //                          processor overhead from raw tri work.
        static const bool skip_mdi = []{
            const char* e = std::getenv("IFC_SKIP_MDI");
            return e && e[0] == '1';
        }();
        static const uint32_t max_subdraws = []{
            const char* e = std::getenv("IFC_MAX_SUBDRAWS");
            return (e && *e) ? static_cast<uint32_t>(std::atoi(e))
                             : std::numeric_limits<uint32_t>::max();
        }();
        if (max_subdraws < m.indirect_command_count) {
            // Keep the fwd/rev ratio so the workload mix is preserved.
            const uint32_t total = m.indirect_command_count;
            fwd = static_cast<uint32_t>((uint64_t)fwd * max_subdraws / total);
            rev = max_subdraws - fwd;
        }
        // Forward pass: non-reflected instances, standard CCW winding.
        if (fwd > 0 && !skip_mdi) {
            gl_->glFrontFace(GL_CCW);
            gl_->glMultiDrawElementsIndirect(
                GL_TRIANGLES, GL_UNSIGNED_INT, nullptr,
                static_cast<GLsizei>(fwd), 0);
            ++gl_draw_calls_;
        }
        // Reverse pass: reflected instances — their world-space winding is
        // flipped, so telling GL the front is CW keeps cull-back working.
        if (rev > 0 && !skip_mdi) {
            gl_->glFrontFace(GL_CW);
            gl_->glMultiDrawElementsIndirect(
                GL_TRIANGLES, GL_UNSIGNED_INT,
                reinterpret_cast<const void*>(m.indirect_forward_count * sizeof(DrawElementsIndirectCommand)),
                static_cast<GLsizei>(rev), 0);
            ++gl_draw_calls_;
            gl_->glFrontFace(GL_CCW);
        }

        visible_triangles_  += m.cached_visible_triangles;
        visible_objects_    += m.cached_visible_objects;
        indirect_sub_draws_ += m.indirect_command_count;
    }
    if (cull_this_frame) {
        last_cull_view_     = view_matrix_;
        last_cull_proj_     = proj_matrix_;
        have_cached_cull_   = true;
    }
    if (needs_settle_recull)
        qDebug("[motion-cull] settle result: obj=%u sub_draws=%u hiz_rej=%u",
               visible_objects_, indirect_sub_draws_,
               hiz_reject_count_.load());
    gl_->glBindBuffer(GL_DRAW_INDIRECT_BUFFER, 0);

    renderEdgePass();
    renderPivotIndicator();
    renderSectionPlanes();
    // Overlay handles highlight tris + HUD text; renderAxisGizmo() must
    // come after because it shrinks glViewport to the corner badge and
    // does not restore.
    {
        const qreal dpr = devicePixelRatio();
        // Push the live box-select rect (or empty when not dragging) so
        // the overlay draws/clears it as part of the per-frame pass.
        QRect sel_rect;
        if (box_select_active_) {
            sel_rect = QRect(box_select_start_pos_,
                             box_select_current_pos_).normalized();
        }
        overlay_renderer_.setSelectionRect(sel_rect);
        overlay_renderer_.render(vp.constData(),
                                 int(width()  * dpr),
                                 int(height() * dpr),
                                 dpr);
    }
    renderAxisGizmo();

    // Build HiZ from this frame's resolved depth for next frame's cull.
    // Synchronous glReadPixels inside — cost ~0.5 ms at 256x128 on a
    // mid-range dGPU.  Skippable via IFC_NO_HIZ=1.  Also skipped on
    // still frames: if we didn't re-cull, the depth buffer is
    // bit-identical to the one we already turned into a pyramid.
    if (hizEnabled() && cull_this_frame) {
        buildHizPyramid();
    }

    context_->swapBuffers(this);

    // Ensure one more frame runs after the last motion frame so the
    // settle recull can detect the camera has stopped and restore the
    // base contribution threshold.
    if (last_cull_was_motion_)
        requestUpdate();

    // Measure frame *cost* (time spent inside render()) rather than the
    // wall-clock gap between frames.  With event-driven rendering, idle gaps
    // between requestUpdate() calls would otherwise pollute the FPS window.
    // Reported fps = "if I rendered continuously, this is the rate I'd hit",
    // which is what profiling actually wants.
    const float frame_cost_s = frame_cost_clock.nsecsElapsed() * 1e-9f;

    // FPS-mode continuous redraw: keep asking for frames while any movement
    // key is held.  When all keys release, the loop drops out and the
    // viewport goes idle until the next input event.  Log hitches if
    // IFC_FPS_HITCH_MS is set so stalls can be attributed.
    if (camera_mode_ == CameraMode::Fps && !fps_keys_held_.isEmpty()) {
        static const float hitch_ms = []{
            const char* e = std::getenv("IFC_FPS_HITCH_MS");
            return (e && *e) ? static_cast<float>(std::atof(e)) : 0.0f;
        }();
        if (hitch_ms > 0.0f && frame_cost_s * 1000.0f > hitch_ms) {
            qDebug("[fps-hitch] frame %.1f ms  vis=%u  sub_draws=%u",
                   frame_cost_s * 1000.0f, visible_objects_, indirect_sub_draws_);
        }
        requestUpdate();
    }

    if (benchmark_total_ > 0) {
        camera_yaw_ += benchmark_yaw_speed_;
        have_cached_cull_ = false;

        if (benchmark_warmup_ > 0) {
            --benchmark_warmup_;
        } else {
            benchmark_frame_times_.push_back(frame_cost_s * 1000.0f);
            ++benchmark_count_;
        }
        if (benchmark_count_ >= benchmark_total_) {
            std::sort(benchmark_frame_times_.begin(), benchmark_frame_times_.end());
            float sum = 0.0f;
            for (float t : benchmark_frame_times_) sum += t;
            float avg = sum / benchmark_frame_times_.size();
            float median = benchmark_frame_times_[benchmark_frame_times_.size() / 2];
            float p1  = benchmark_frame_times_[(size_t)(benchmark_frame_times_.size() * 0.01f)];
            float p99 = benchmark_frame_times_[(size_t)(benchmark_frame_times_.size() * 0.99f)];
            float total_arc = benchmark_yaw_speed_ * (benchmark_total_ + 5);
            qDebug("\n=== BENCHMARK (%d frames, orbit %.0f° at %.1f°/frame) ===",
                   benchmark_total_, total_arc, benchmark_yaw_speed_);
            qDebug("  avg: %.2f ms (%.1f fps)", avg, 1000.0f / avg);
            qDebug("  median: %.2f ms (%.1f fps)", median, 1000.0f / median);
            qDebug("  p1: %.2f ms  p99: %.2f ms", p1, p99);
            qDebug("  last frame: obj %u  tri %u  sub_draws %u  hiz_rej %u",
                   visible_objects_, visible_triangles_,
                   indirect_sub_draws_,
                   hiz_reject_count_.load());
            qDebug("=== END BENCHMARK ===\n");
            QCoreApplication::quit();
            return;
        }
        requestUpdate();
    }

    accumulated_time_ += frame_cost_s;
    frame_count_++;
    if (accumulated_time_ >= 1.0f) {
        last_fps_ = static_cast<float>(frame_count_) / accumulated_time_;
        const uint32_t frames_in_window = static_cast<uint32_t>(frame_count_);
        frame_count_ = 0;
        accumulated_time_ = 0.0f;

        uint32_t total_obj = 0, total_tri = 0, total_meshes = 0;
        size_t total_vbo = 0, total_ebo = 0, total_ssbo = 0;
        size_t num_models = 0, num_hidden = 0;
        for (const auto& [mid, mm] : models_gpu_) {
            num_models++;
            if (mm.hidden) { num_hidden++; continue; }
            total_obj += static_cast<uint32_t>(mm.instances.size());
            total_tri += mm.total_triangles;
            total_meshes += static_cast<uint32_t>(mm.meshes.size());
            total_vbo += mm.vbo_capacity;
            total_ebo += mm.ebo_capacity;
            total_ssbo += mm.ssbo_instance_count * sizeof(InstanceGpu);
        }

        FrameStats stats;
        stats.fps = last_fps_;
        stats.frame_time_ms = 1000.0f / last_fps_;
        stats.total_objects = total_obj;
        stats.visible_objects = visible_objects_;
        stats.total_triangles = total_tri;
        stats.visible_triangles = visible_triangles_;
        stats.unique_meshes = total_meshes;
        stats.gl_draw_calls = gl_draw_calls_;
        stats.indirect_sub_draws = indirect_sub_draws_;
        emit frameStatsUpdated(stats);

        const double inv_frames = frames_in_window > 0
            ? 1.0 / static_cast<double>(frames_in_window) : 0.0;
        const double clr_ms = cull_clear_ns_.load()    * 1e-6 * inv_frames;
        const double trv_ms = cull_traverse_ns_.load() * 1e-6 * inv_frames;
        const double emt_ms = cull_emit_ns_.load()     * 1e-6 * inv_frames;
        const double upl_ms = cull_upload_ns_.load()   * 1e-6 * inv_frames;
        const double wall_ms = cull_wall_ns_           * 1e-6 * inv_frames;
        cull_clear_ns_.store(0);
        cull_traverse_ns_.store(0);
        cull_emit_ns_.store(0);
        cull_upload_ns_.store(0);
        cull_wall_ns_ = 0;
        const uint32_t skipped = cull_skipped_frames_;
        cull_skipped_frames_ = 0;

        qDebug("[frame] %.1f fps  %.2f ms  obj %u/%u  tri %u/%u  "
               "meshes %u  gl_draws %u  sub_draws %u  hiz_rej %u  "
               "cull[wall %.2f | work: clr %.2f trv %.2f emt %.2f upl %.2f]ms  skipped %u/%u  "
               "vram %.1f MB (vbo %.1f + ebo %.1f + ssbo %.1f)  models %zu (%zu hidden)",
               last_fps_, 1000.0f / last_fps_,
               visible_objects_, total_obj,
               visible_triangles_, total_tri,
               total_meshes, gl_draw_calls_, indirect_sub_draws_,
               hiz_reject_count_.load(),
               wall_ms, clr_ms, trv_ms, emt_ms, upl_ms,
               skipped, frames_in_window,
               (total_vbo + total_ebo + total_ssbo) / (1024.0*1024.0),
               total_vbo / (1024.0*1024.0),
               total_ebo / (1024.0*1024.0),
               total_ssbo / (1024.0*1024.0),
               num_models, num_hidden);

        // One-shot sub_draw composition diagnostic.
        static const bool subdraw_diag = std::getenv("IFC_SUBDRAW_DIAG") != nullptr;
        if (subdraw_diag) {
            uint32_t total_subdraws = 0;
            uint32_t hist[8] = {};
            uint32_t instances_in_bucket[8] = {};
            uint32_t tris_in_bucket[8] = {};

            struct ModelStats {
                uint32_t model_id;
                uint32_t subdraws;
                uint32_t single_instance;
                uint32_t total_meshes;
                uint32_t total_instances;
            };
            std::vector<ModelStats> per_model;

            auto bucket_idx = [](uint32_t ic) -> int {
                if (ic <= 1) return 0;
                if (ic <= 2) return 1;
                if (ic <= 4) return 2;
                if (ic <= 8) return 3;
                if (ic <= 16) return 4;
                if (ic <= 64) return 5;
                if (ic <= 256) return 6;
                return 7;
            };

            // --- Mesh-level consolidation analysis ---
            // Per mesh_id, count visible instances across all 4 buckets.
            // Also count how many buckets each mesh_id appears in.
            uint32_t unique_visible_meshes = 0;
            uint32_t meshes_truly_single = 0;   // 1 instance total, 1 bucket
            uint32_t meshes_split_by_state = 0; // >1 bucket but each has 1 instance
            uint32_t subdraws_if_merged_buckets = 0; // sub_draws if winding+LOD ignored
            uint32_t mesh_vis_hist[8] = {};     // histogram of per-mesh visible instance counts

            for (const auto& [mid, mm] : models_gpu_) {
                if (mm.hidden) continue;
                ModelStats ms{mid, mm.indirect_command_count, 0,
                              static_cast<uint32_t>(mm.meshes.size()),
                              static_cast<uint32_t>(mm.instances.size())};
                for (const auto& cmd : mm.indirect_scratch) {
                    int b = bucket_idx(cmd.instanceCount);
                    hist[b]++;
                    instances_in_bucket[b] += cmd.instanceCount;
                    tris_in_bucket[b] += (cmd.count / 3) * cmd.instanceCount;
                    if (cmd.instanceCount == 1) ms.single_instance++;
                    total_subdraws++;
                }
                per_model.push_back(ms);

                const size_t nm = mm.meshes.size();
                for (size_t mi = 0; mi < nm; ++mi) {
                    uint32_t total_vis = 0;
                    uint32_t buckets_present = 0;
                    auto count_bucket = [&](const std::vector<std::vector<uint32_t>>& v) {
                        if (mi < v.size() && !v[mi].empty()) {
                            total_vis += static_cast<uint32_t>(v[mi].size());
                            buckets_present++;
                        }
                    };
                    count_bucket(mm.vis_fwd_lod0);
                    count_bucket(mm.vis_fwd_lod1);
                    count_bucket(mm.vis_rev_lod0);
                    count_bucket(mm.vis_rev_lod1);
                    if (total_vis == 0) continue;

                    unique_visible_meshes++;
                    mesh_vis_hist[bucket_idx(total_vis)]++;
                    if (total_vis > 0) subdraws_if_merged_buckets++;

                    if (total_vis == 1 && buckets_present == 1)
                        meshes_truly_single++;
                    else if (buckets_present > 1) {
                        bool all_single = true;
                        auto check = [&](const std::vector<std::vector<uint32_t>>& v) {
                            if (mi < v.size() && v[mi].size() > 1) all_single = false;
                        };
                        check(mm.vis_fwd_lod0); check(mm.vis_fwd_lod1);
                        check(mm.vis_rev_lod0); check(mm.vis_rev_lod1);
                        if (all_single) meshes_split_by_state++;
                    }
                }
            }

            qDebug("\n=== SUB_DRAW COMPOSITION (this frame) ===");
            qDebug("Total sub_draws: %u", total_subdraws);
            const char* labels[] = {"    1", "    2", "  3-4", "  5-8",
                                    " 9-16", "17-64", "65-256", " 257+"};
            qDebug("instanceCount histogram:");
            qDebug("  range     | sub_draws | instances  | triangles");
            for (int i = 0; i < 8; ++i) {
                if (hist[i] == 0) continue;
                qDebug("  %s   | %7u   | %9u  | %10u",
                       labels[i], hist[i], instances_in_bucket[i], tris_in_bucket[i]);
            }

            uint32_t single = hist[0], small = hist[0] + hist[1] + hist[2];
            qDebug("Single-instance sub_draws: %u (%.1f%%)",
                   single, total_subdraws ? 100.0 * single / total_subdraws : 0.0);
            qDebug("Small (<=4) sub_draws:     %u (%.1f%%)",
                   small, total_subdraws ? 100.0 * small / total_subdraws : 0.0);

            qDebug("\n--- MESH-LEVEL CONSOLIDATION ---");
            qDebug("Unique visible mesh IDs: %u", unique_visible_meshes);
            qDebug("Visible instance count per mesh_id:");
            qDebug("  range     | mesh_ids");
            for (int i = 0; i < 8; ++i) {
                if (mesh_vis_hist[i] == 0) continue;
                qDebug("  %s   | %7u", labels[i], mesh_vis_hist[i]);
            }

            qDebug("\nAmong single-instance sub_draws (%u):", single);
            qDebug("  Truly unique (1 inst, 1 bucket):     %u", meshes_truly_single);
            qDebug("  Split by state (>1 bucket, each =1): %u  (saves %u sub_draws if merged)",
                   meshes_split_by_state, meshes_split_by_state);

            qDebug("\nEstimated sub_draws by grouping strategy:");
            qDebug("  Current (mesh_id x winding x LOD):   %u", total_subdraws);
            qDebug("  Merged buckets (mesh_id only):        %u  (%.0f%% reduction)",
                   subdraws_if_merged_buckets,
                   total_subdraws ? 100.0 * (1.0 - (double)subdraws_if_merged_buckets / total_subdraws) : 0.0);

            std::sort(per_model.begin(), per_model.end(),
                      [](const ModelStats& a, const ModelStats& b) {
                          return a.subdraws > b.subdraws;
                      });
            qDebug("\nTop 15 models by sub_draw count:");
            qDebug("  model_id | sub_draws | single_inst | meshes   | instances");
            for (size_t i = 0; i < std::min<size_t>(15, per_model.size()); ++i) {
                const auto& ms = per_model[i];
                qDebug("  %7u  | %7u   | %7u     | %7u  | %7u",
                       ms.model_id, ms.subdraws, ms.single_instance,
                       ms.total_meshes, ms.total_instances);
            }
            qDebug("=== END SUB_DRAW COMPOSITION ===\n");
        }
    }
}

void ViewportWindow::renderPickPass() {
    gl_->glBindFramebuffer(GL_FRAMEBUFFER, pick_fbo_);
    gl_->glViewport(0, 0, pick_width_, pick_height_);
    const GLuint  zero_id  = 0;
    const float   zero3[4] = {0, 0, 0, 0};
    gl_->glClearBufferuiv(GL_COLOR, 0, &zero_id);
    gl_->glClearBufferfv (GL_COLOR, 1, zero3);
    gl_->glClearBufferfv (GL_COLOR, 2, zero3);
    gl_->glClear(GL_DEPTH_BUFFER_BIT);

    QMatrix4x4 vp = proj_matrix_ * view_matrix_;
    float planes[6][4];
    extractFrustumPlanes(vp, planes);

    gl_->glUseProgram(pick_program_);
    GLint u_vp       = gl_->glGetUniformLocation(pick_program_, "u_view_projection");
    gl_->glUniformMatrix4fv(u_vp, 1, GL_FALSE, vp.constData());
    uploadClipPlaneUniforms(pick_program_);

    gl_->glFrontFace(GL_CCW);

    for (auto& [model_id, m] : models_gpu_) {
        if (m.hidden || !m.ssbo || m.ssbo_instance_count == 0) continue;

        // Pick pass: contribution-cull disabled (0.0 threshold) so every
        // frustum-visible object is clickable, even sub-pixel ones.
        cullAndUploadVisible(m, planes, 1.0f, 0.0f);
        if (m.indirect_command_count == 0) continue;

        gl_->glBindVertexArray(m.vao);
        gl_->glBindBufferBase(GL_SHADER_STORAGE_BUFFER, 0, m.ssbo);
        gl_->glBindBufferBase(GL_SHADER_STORAGE_BUFFER, 1, m.visible_ssbo);
        gl_->glBindBufferBase(GL_SHADER_STORAGE_BUFFER, 2, m.mesh_info_ssbo);
        gl_->glBindBuffer(GL_DRAW_INDIRECT_BUFFER, m.indirect_buffer);

        const uint32_t fwd = m.indirect_forward_count;
        const uint32_t rev = m.indirect_command_count - fwd;
        if (fwd > 0) {
            gl_->glFrontFace(GL_CCW);
            gl_->glMultiDrawElementsIndirect(
                GL_TRIANGLES, GL_UNSIGNED_INT, nullptr,
                static_cast<GLsizei>(fwd), 0);
        }
        if (rev > 0) {
            gl_->glFrontFace(GL_CW);
            gl_->glMultiDrawElementsIndirect(
                GL_TRIANGLES, GL_UNSIGNED_INT,
                reinterpret_cast<const void*>(fwd * sizeof(DrawElementsIndirectCommand)),
                static_cast<GLsizei>(rev), 0);
            gl_->glFrontFace(GL_CCW);
        }
    }
    gl_->glBindBuffer(GL_DRAW_INDIRECT_BUFFER, 0);
    gl_->glBindFramebuffer(GL_FRAMEBUFFER, 0);
}

void ViewportWindow::buildSectionPlaneGizmo() {
    // Plane-local geometry, all GL_LINES.  Quad is drawn in plane-local
    // (u, v); arrow shaft + head extend along +n.  Layout per vertex:
    //   x,y,z (plane-local)   r,g,b (color, modulated by u_tint)
    static const float verts[] = {
        // --- quad outline (4 segments = 8 verts, white) ---
        -1, -1, 0,   1,1,1,   1, -1, 0,   1,1,1,
         1, -1, 0,   1,1,1,   1,  1, 0,   1,1,1,
         1,  1, 0,   1,1,1,  -1,  1, 0,   1,1,1,
        -1,  1, 0,   1,1,1,  -1, -1, 0,   1,1,1,
        // --- arrow shaft along +n (yellow) ---
         0,  0, 0,   1.0f, 0.85f, 0.2f,
         0,  0, 1,   1.0f, 0.85f, 0.2f,
        // --- arrow head (4 diagonals from tip back to a ring at z=0.7) ---
         0,  0, 1,   1.0f, 0.85f, 0.2f,
        -0.18f, 0, 0.78f,  1.0f, 0.85f, 0.2f,
         0,  0, 1,   1.0f, 0.85f, 0.2f,
         0.18f, 0, 0.78f,  1.0f, 0.85f, 0.2f,
         0,  0, 1,   1.0f, 0.85f, 0.2f,
         0, -0.18f, 0.78f, 1.0f, 0.85f, 0.2f,
         0,  0, 1,   1.0f, 0.85f, 0.2f,
         0,  0.18f, 0.78f, 1.0f, 0.85f, 0.2f,
    };
    plane_quad_offset_  = 0;
    plane_quad_count_   = 8;
    plane_arrow_offset_ = 8;
    plane_arrow_count_  = 10;  // 2 shaft + 8 head diagonals

    gl_->glCreateVertexArrays(1, &plane_vao_);
    gl_->glCreateBuffers(1, &plane_vbo_);
    gl_->glNamedBufferStorage(plane_vbo_, sizeof(verts), verts, 0);
    gl_->glVertexArrayVertexBuffer(plane_vao_, 0, plane_vbo_, 0, 6 * sizeof(float));
    gl_->glEnableVertexArrayAttrib(plane_vao_, 0);
    gl_->glVertexArrayAttribFormat(plane_vao_, 0, 3, GL_FLOAT, GL_FALSE, 0);
    gl_->glVertexArrayAttribBinding(plane_vao_, 0, 0);
    gl_->glEnableVertexArrayAttrib(plane_vao_, 1);
    gl_->glVertexArrayAttribFormat(plane_vao_, 1, 3, GL_FLOAT, GL_FALSE, 3 * sizeof(float));
    gl_->glVertexArrayAttribBinding(plane_vao_, 1, 0);
}

// Pick any unit vector orthogonal to n.  Avoids the degenerate case where n
// is parallel to the seed by choosing the seed with the smallest |n_i|.
static QVector3D anyOrthogonal(const QVector3D& n) {
    const float ax = std::abs(n.x()), ay = std::abs(n.y()), az = std::abs(n.z());
    QVector3D seed = (ax < ay && ax < az) ? QVector3D(1, 0, 0)
                   : (ay < az)            ? QVector3D(0, 1, 0)
                                          : QVector3D(0, 0, 1);
    QVector3D u = QVector3D::crossProduct(n, seed);
    if (u.lengthSquared() < 1e-12f) u = QVector3D(1, 0, 0);
    return u.normalized();
}

void ViewportWindow::renderSectionPlanes() {
    if (section_planes_.empty() || !plane_program_ || !plane_vao_) return;

    // Plane half-extent in metres — quad is drawn at (±1, ±1) in plane-local
    // space, so size = 1.0 produces the 2x2m footprint we want.
    constexpr float kHalfSize = 1.0f;

    const QMatrix4x4 vp = proj_matrix_ * view_matrix_;

    gl_->glUseProgram(plane_program_);
    gl_->glUniformMatrix4fv(gl_->glGetUniformLocation(plane_program_, "u_vp"),
                            1, GL_FALSE, vp.constData());
    gl_->glUniform1f(gl_->glGetUniformLocation(plane_program_, "u_size"), kHalfSize);

    gl_->glEnable(GL_BLEND);
    gl_->glBlendFunc(GL_SRC_ALPHA, GL_ONE_MINUS_SRC_ALPHA);
    gl_->glBindVertexArray(plane_vao_);

    const GLint loc_origin = gl_->glGetUniformLocation(plane_program_, "u_origin");
    const GLint loc_u      = gl_->glGetUniformLocation(plane_program_, "u_axis_u");
    const GLint loc_v      = gl_->glGetUniformLocation(plane_program_, "u_axis_v");
    const GLint loc_n      = gl_->glGetUniformLocation(plane_program_, "u_axis_n");
    const GLint loc_tint   = gl_->glGetUniformLocation(plane_program_, "u_tint");

    for (int i = 0; i < int(section_planes_.size()); ++i) {
        const SectionPlane& p = section_planes_[i];
        const QVector3D u = anyOrthogonal(p.n);
        const QVector3D v = QVector3D::crossProduct(p.n, u).normalized();

        gl_->glUniform3f(loc_origin, p.origin.x(), p.origin.y(), p.origin.z());
        gl_->glUniform3f(loc_u,      u.x(), u.y(), u.z());
        gl_->glUniform3f(loc_v,      v.x(), v.y(), v.z());
        gl_->glUniform3f(loc_n,      p.n.x(), p.n.y(), p.n.z());

        const bool selected = (i == section_plane_selected_);
        // Selected plane: cyan-tinted, full alpha.  Unselected: warm tint,
        // dimmer.  Multiplied onto the per-vertex color.
        if (selected) gl_->glUniform4f(loc_tint, 0.55f, 0.95f, 1.0f, 1.0f);
        else          gl_->glUniform4f(loc_tint, 1.0f,  0.85f, 0.4f, 0.75f);

        gl_->glLineWidth(selected ? 2.5f : 1.5f);
        gl_->glDrawArrays(GL_LINES, plane_quad_offset_,  plane_quad_count_);
        gl_->glDrawArrays(GL_LINES, plane_arrow_offset_, plane_arrow_count_);
    }

    gl_->glBindVertexArray(0);
    gl_->glDisable(GL_BLEND);
}

int ViewportWindow::hitTestSectionGizmo(int x, int y) const {
    if (section_planes_.empty()) return -1;
    const int w = width();
    const int h = height();
    if (w <= 0 || h <= 0) return -1;
    const QMatrix4x4 vp = proj_matrix_ * view_matrix_;
    const float grab_px = 12.0f;
    int best = -1;
    float best_d2 = grab_px * grab_px;

    auto project = [&](const QVector3D& world, QVector2D& out) -> bool {
        QVector4D clip = vp * QVector4D(world, 1.0f);
        if (clip.w() <= 0.0f) return false;  // behind camera
        const float invw = 1.0f / clip.w();
        // Qt mouse coords have y down from the top — match that here.
        out = QVector2D(
            (clip.x() * invw * 0.5f + 0.5f) * float(w),
            (1.0f - (clip.y() * invw * 0.5f + 0.5f)) * float(h));
        return true;
    };

    for (int i = 0; i < int(section_planes_.size()); ++i) {
        const SectionPlane& p = section_planes_[i];
        QVector2D s_origin, s_tip;
        if (!project(p.origin, s_origin))               continue;
        if (!project(p.origin + p.n * 1.0f, s_tip))     continue;

        // Distance from (x,y) to the line segment (s_origin, s_tip).
        const QVector2D q{float(x), float(y)};
        const QVector2D ab = s_tip - s_origin;
        const float ab_len2 = ab.lengthSquared();
        if (ab_len2 < 1e-3f) continue;  // degenerate (axis edge-on)
        float t = QVector2D::dotProduct(q - s_origin, ab) / ab_len2;
        t = qBound(0.0f, t, 1.0f);
        const QVector2D proj = s_origin + ab * t;
        const float d2 = (q - proj).lengthSquared();
        if (d2 < best_d2) { best_d2 = d2; best = i; }
    }
    return best;
}

void ViewportWindow::updateSectionDrag(int x, int y) {
    if (!section_drag_active_) return;
    if (section_drag_index_ < 0
        || section_drag_index_ >= int(section_planes_.size())) return;
    SectionPlane& p = section_planes_[section_drag_index_];

    const int w = width();
    const int h = height();
    if (w <= 0 || h <= 0) return;
    const QMatrix4x4 vp = proj_matrix_ * view_matrix_;

    auto project = [&](const QVector3D& world, QVector2D& out) -> bool {
        QVector4D clip = vp * QVector4D(world, 1.0f);
        if (clip.w() <= 0.0f) return false;
        const float invw = 1.0f / clip.w();
        out = QVector2D(
            (clip.x() * invw * 0.5f + 0.5f) * float(w),
            (1.0f - (clip.y() * invw * 0.5f + 0.5f)) * float(h));
        return true;
    };

    QVector2D s_origin, s_n;
    if (!project(section_drag_start_origin_, s_origin)) return;
    if (!project(section_drag_start_origin_ + p.n, s_n)) return;
    const QVector2D screen_axis = s_n - s_origin;
    const float screen_axis_len2 = screen_axis.lengthSquared();
    if (screen_axis_len2 < 1e-3f) return;  // axis is edge-on; drag would be ill-conditioned

    // Project the cursor delta onto the screen-space axis to get the world-
    // space slide along n.  delta_pixels / |screen_axis_pixels_per_meter|.
    const QVector2D delta_px(float(x - section_drag_start_mouse_.x()),
                             float(y - section_drag_start_mouse_.y()));
    const float delta_along_axis_px = QVector2D::dotProduct(delta_px, screen_axis);
    const float meters = delta_along_axis_px / screen_axis_len2;

    p.origin = section_drag_start_origin_ + p.n * meters;
    p.d      = -QVector3D::dotProduct(p.n, p.origin);
    have_cached_cull_ = false;
    requestUpdate();
}

void ViewportWindow::renderEdgePass() {
    if (!edge_program_) return;
    const int w = width()  * devicePixelRatio();
    const int h = height() * devicePixelRatio();
    if (w <= 0 || h <= 0) return;

    // Lazy resize.  D24S8 to match Qt's default FBO format (depth+stencil
    // even though we only sample depth) so the blit doesn't fail.
    if (edge_w_ != w || edge_h_ != h) {
        if (edge_depth_fbo_) gl_->glDeleteFramebuffers(1, &edge_depth_fbo_);
        if (edge_depth_tex_) gl_->glDeleteTextures(1, &edge_depth_tex_);
        gl_->glCreateTextures(GL_TEXTURE_2D, 1, &edge_depth_tex_);
        gl_->glTextureStorage2D(edge_depth_tex_, 1, GL_DEPTH24_STENCIL8, w, h);
        gl_->glCreateFramebuffers(1, &edge_depth_fbo_);
        gl_->glNamedFramebufferTexture(edge_depth_fbo_, GL_DEPTH_STENCIL_ATTACHMENT,
                                       edge_depth_tex_, 0);
        edge_w_ = w; edge_h_ = h;
    }

    // Step 1: resolve MSAA depth from default FB → single-sample texture.
    gl_->glBindFramebuffer(GL_READ_FRAMEBUFFER, 0);
    gl_->glBindFramebuffer(GL_DRAW_FRAMEBUFFER, edge_depth_fbo_);
    gl_->glBlitFramebuffer(0, 0, w, h, 0, 0, w, h,
                           GL_DEPTH_BUFFER_BIT | GL_STENCIL_BUFFER_BIT,
                           GL_NEAREST);

    // Step 2: fullscreen darkening pass into default FB colour.
    gl_->glBindFramebuffer(GL_FRAMEBUFFER, 0);
    gl_->glViewport(0, 0, w, h);
    gl_->glDisable(GL_DEPTH_TEST);
    gl_->glDepthMask(GL_FALSE);
    gl_->glEnable(GL_BLEND);
    gl_->glBlendFunc(GL_DST_COLOR, GL_ZERO);

    gl_->glUseProgram(edge_program_);
    gl_->glTextureParameteri(edge_depth_tex_, GL_TEXTURE_MIN_FILTER, GL_NEAREST);
    gl_->glTextureParameteri(edge_depth_tex_, GL_TEXTURE_MAG_FILTER, GL_NEAREST);
    gl_->glTextureParameteri(edge_depth_tex_, GL_TEXTURE_COMPARE_MODE, GL_NONE);
    gl_->glBindTextureUnit(0, edge_depth_tex_);

    // Match the (near, far) used in updateCamera().  For ortho the near
    // plane is at -depth_extent, far at +depth_extent.
    const float depth_extent = camera_distance_ * 10.0f;
    const float near_z = projection_ortho_ ? -depth_extent : 0.1f;
    const float far_z  = depth_extent;

    gl_->glUniform1i(gl_->glGetUniformLocation(edge_program_, "u_depth"), 0);
    gl_->glUniform2f(gl_->glGetUniformLocation(edge_program_, "u_texel"),
                     1.0f / float(w), 1.0f / float(h));
    gl_->glUniform1f(gl_->glGetUniformLocation(edge_program_, "u_near"),     near_z);
    gl_->glUniform1f(gl_->glGetUniformLocation(edge_program_, "u_far"),      far_z);
    gl_->glUniform1f(gl_->glGetUniformLocation(edge_program_, "u_is_ortho"),
                     projection_ortho_ ? 1.0f : 0.0f);
    gl_->glUniform1f(gl_->glGetUniformLocation(edge_program_, "u_scale"),     6.0f);
    gl_->glUniform1f(gl_->glGetUniformLocation(edge_program_, "u_threshold"), 0.004f);

    gl_->glBindVertexArray(edge_vao_);
    gl_->glDrawArrays(GL_TRIANGLES, 0, 3);
    gl_->glBindVertexArray(0);

    gl_->glDisable(GL_BLEND);
    gl_->glDepthMask(GL_TRUE);
    gl_->glEnable(GL_DEPTH_TEST);
}

void ViewportWindow::renderPivotIndicator() {
    if (!pivot_indicator_visible_ || !pivot_program_ || !pivot_vao_) return;

    const int h = height() * devicePixelRatio();
    if (h <= 0) return;

    // Pick a world-space arm length that projects to ~30 pixels.  In
    // perspective, world-per-pixel grows with distance (factored from
    // fovy and viewport height); in ortho it's set by the box height
    // (camera_distance * tan(fovy/2)) and is independent of distance —
    // both reduce to the same formula here because the ortho box is
    // sized to match perspective at the pivot's distance.
    const float fovy_rad = qDegreesToRadians(camera_fov_y_deg_);
    const float world_per_pixel = camera_distance_ * tanf(fovy_rad * 0.5f) * 2.0f / float(h);
    const float arm_pixels = 30.0f * float(devicePixelRatio());
    const float arm_world  = arm_pixels * world_per_pixel;

    const QMatrix4x4 mvp = proj_matrix_ * view_matrix_;

    gl_->glUseProgram(pivot_program_);
    gl_->glUniformMatrix4fv(gl_->glGetUniformLocation(pivot_program_, "u_mvp"),
                            1, GL_FALSE, mvp.constData());
    gl_->glUniform3f(gl_->glGetUniformLocation(pivot_program_, "u_pivot"),
                     camera_target_.x(), camera_target_.y(), camera_target_.z());
    gl_->glUniform1f(gl_->glGetUniformLocation(pivot_program_, "u_arm_world"),
                     arm_world);

    gl_->glEnable(GL_BLEND);
    gl_->glBlendFunc(GL_SRC_ALPHA, GL_ONE_MINUS_SRC_ALPHA);
    gl_->glLineWidth(2.0f);
    gl_->glBindVertexArray(pivot_vao_);

    // Pass 1: occluded portions, depth test reversed, dim — gives an X-ray
    // hint that the pivot lives behind geometry.
    gl_->glDepthFunc(GL_GREATER);
    gl_->glUniform1f(gl_->glGetUniformLocation(pivot_program_, "u_alpha"), 0.30f);
    gl_->glDrawArrays(GL_LINES, 0, pivot_rim_count_);

    // Pass 2: visible portions, normal depth, full alpha.
    gl_->glDepthFunc(GL_LEQUAL);
    gl_->glUniform1f(gl_->glGetUniformLocation(pivot_program_, "u_alpha"), 1.0f);
    gl_->glDrawArrays(GL_LINES, 0, pivot_rim_count_);

    gl_->glBindVertexArray(0);
    gl_->glDisable(GL_BLEND);
    gl_->glDepthFunc(GL_LESS);
}

void ViewportWindow::setPivotIndicatorVisible(bool visible, int hide_after_ms) {
    if (!pivot_indicator_hide_timer_) {
        pivot_indicator_hide_timer_ = new QTimer(this);
        pivot_indicator_hide_timer_->setSingleShot(true);
        connect(pivot_indicator_hide_timer_, &QTimer::timeout, this, [this]() {
            pivot_indicator_visible_ = false;
            requestUpdate();
        });
    }
    pivot_indicator_visible_ = visible;
    if (visible && hide_after_ms > 0) {
        pivot_indicator_hide_timer_->start(hide_after_ms);
    } else {
        pivot_indicator_hide_timer_->stop();
    }
}

void ViewportWindow::renderAxisGizmo() {
    if (!axis_program_ || !axis_vao_) return;
    const int dpr = devicePixelRatio();
    const int gizmo_size = 110 * dpr;
    const int margin = 10 * dpr;
    gl_->glViewport(margin, margin, gizmo_size, gizmo_size);
    gl_->glDisable(GL_DEPTH_TEST);

    float yaw_rad = qDegreesToRadians(camera_yaw_);
    float pitch_rad = qDegreesToRadians(camera_pitch_);
    QVector3D eye_dir(cosf(pitch_rad) * cosf(yaw_rad),
                      cosf(pitch_rad) * sinf(yaw_rad),
                      sinf(pitch_rad));
    QMatrix4x4 gv; gv.lookAt(eye_dir * 3.0f, QVector3D(0,0,0), QVector3D(0,0,1));
    QMatrix4x4 gp; gp.ortho(-1.4f, 1.4f, -1.4f, 1.4f, 0.1f, 10.0f);
    QMatrix4x4 mvp = gp * gv;

    gl_->glUseProgram(axis_program_);
    gl_->glUniformMatrix4fv(gl_->glGetUniformLocation(axis_program_, "u_mvp"), 1, GL_FALSE, mvp.constData());
    gl_->glLineWidth(2.5f);
    gl_->glBindVertexArray(axis_vao_);
    gl_->glDrawArrays(GL_LINES, 0, 6);
    gl_->glEnable(GL_DEPTH_TEST);
}

void ViewportWindow::exposeEvent(QExposeEvent*) {
    if (isExposed()) {
        if (!gl_initialized_) initGL();
        else                  requestUpdate();
    }
}
void ViewportWindow::resizeEvent(QResizeEvent*) {
    if (gl_initialized_) requestUpdate();
}
bool ViewportWindow::event(QEvent* e) {
    switch (e->type()) {
    case QEvent::UpdateRequest:
        if (isExposed() && gl_initialized_) render();
        return true;
    case QEvent::MouseButtonPress:   handleMousePress(static_cast<QMouseEvent*>(e));   return true;
    case QEvent::MouseButtonRelease: handleMouseRelease(static_cast<QMouseEvent*>(e)); return true;
    case QEvent::MouseMove:          handleMouseMove(static_cast<QMouseEvent*>(e));    return true;
    case QEvent::Wheel:              handleWheel(static_cast<QWheelEvent*>(e));        return true;
    default: return QWindow::event(e);
    }
}

void ViewportWindow::handleMousePress(QMouseEvent* e) {
    // Blender-style: any click in FPS mode drops back to orbit and is
    // swallowed (no pick, no camera state mutation).
    if (camera_mode_ == CameraMode::Fps) {
        exitFpsMode();
        active_button_ = Qt::NoButton;
        return;
    }
    active_button_ = e->button();
    last_mouse_pos_ = e->pos();
    box_select_armed_       = false;
    box_select_active_      = false;
    press_pick_id_          = 0;
    // Show the pivot indicator while the user holds the navigation
    // button(s) for the active preset — Blender uses MMB for both
    // orbit + pan, Revit too (with Shift swapping the role), Rhino
    // uses RMB.  Either way, the press of that button is what the
    // user means as "I'm about to navigate".
    {
        const NavBindings nb = currentNavBindings();
        if (e->button() == nb.orbit_button || e->button() == nb.pan_button) {
            setPivotIndicatorVisible(true);
            requestUpdate();
        }
    }
    if (section_tool_active_ && e->button() == Qt::LeftButton) {
        // First try to grab an existing plane's arrow gizmo.
        const int hit = hitTestSectionGizmo(e->pos().x(), e->pos().y());
        if (hit >= 0) {
            section_plane_selected_  = hit;
            section_drag_active_     = true;
            section_drag_index_      = hit;
            section_drag_start_origin_ = section_planes_[hit].origin;
            section_drag_start_mouse_  = e->pos();
            requestUpdate();
            return;
        }
        // Otherwise: create a new plane at the clicked surface.
        uint32_t  obj_id = 0;
        QVector3D pos, normal;
        if (pickSurfaceAt(e->pos().x(), e->pos().y(), obj_id, pos, normal)) {
            if (addSectionPlaneAtSurface(pos, normal)) {
                section_plane_selected_ = int(section_planes_.size()) - 1;
                requestUpdate();
            }
        } else {
            section_plane_selected_ = -1;
            requestUpdate();
        }
        return;
    }

    // Plain LMB: pick now so release can apply click semantics without
    // a second pick pass, and arm a potential box-select drag — promoted
    // to box_select_active_ at the first move past the click threshold.
    // Arming on every LMB press (not just empty-space presses) means a
    // drag that happens to start on an object still box-selects, which
    // matches user intuition: the start point shouldn't disqualify the
    // gesture.  Tool-mode LMB defers pick handling to surfacePickedInTool
    // on release for tools that consume clicks (Area / Length); Volume is
    // passive and routes input through normal selection.
    const bool tool_consumes_clicks =
        (tool_mode_ == ToolMode::Area || tool_mode_ == ToolMode::Length);
    if (e->button() == Qt::LeftButton && !tool_consumes_clicks) {
        press_pick_id_ = pickObjectAt(e->pos().x(), e->pos().y());
        box_select_start_pos_   = e->pos();
        box_select_current_pos_ = e->pos();
        box_select_press_mods_  = e->modifiers();
        box_select_armed_       = true;
    }
}
void ViewportWindow::handleMouseRelease(QMouseEvent* e) {
    if (camera_mode_ == CameraMode::Fps) return;
    if (section_drag_active_ && active_button_ == Qt::LeftButton) {
        section_drag_active_ = false;
        section_drag_index_  = -1;
        active_button_ = Qt::NoButton;
        return;
    }

    if (active_button_ == Qt::LeftButton && !section_tool_active_) {
        const bool was_drag =
            (e->pos() - box_select_start_pos_).manhattanLength() >= 5;

        if (box_select_active_) {
            // Finalize a box-select.  Modifier captured at press-time
            // decides commit semantics: plain → replace, Shift → add,
            // Ctrl → remove.  Box-select never sets the active id; the
            // last single-clicked object remains the active.
            const QRect rect = QRect(box_select_start_pos_,
                                     e->pos()).normalized();
            const auto picks = picksInRect(rect);
            const auto mods  = box_select_press_mods_;
            if (mods & Qt::ShiftModifier) {
                selection_.addToSelection(picks);
            } else if (mods & Qt::ControlModifier) {
                selection_.removeFromSelection(picks);
            } else {
                // Replace.  Active stays only if the new set still
                // contains it; SelectionState handles the coercion.
                selection_.setSelection(picks, selection_.activeObjectId());
            }
        } else if (!was_drag) {
            // Click — apply press-time pick + modifiers.  Area / Length
            // intercept clicks; None and Volume drive normal selection.
            const bool tool_consumes_clicks =
                (tool_mode_ == ToolMode::Area || tool_mode_ == ToolMode::Length);
            if (tool_consumes_clicks) {
                emit surfacePickedInTool(e->pos().x(), e->pos().y(),
                                         int(e->modifiers()));
            } else {
                const auto mods = e->modifiers();
                if (mods & (Qt::ShiftModifier | Qt::ControlModifier)) {
                    if (press_pick_id_ != 0) {
                        // Toggle.  No-op on empty space so a stray
                        // modifier-click doesn't blow away the set.
                        selection_.toggleInSelection(press_pick_id_);
                    }
                } else {
                    // Plain click — replace (or clear, for empty hit).
                    selection_.setSelectedObjectId(press_pick_id_);
                }
            }
        }
        // Drag from an object (no box select armed, movement > threshold)
        // is intentionally a no-op — keeps stray drags from rewriting
        // the selection.

        box_select_armed_  = false;
        box_select_active_ = false;
        press_pick_id_     = 0;
    }

    const NavBindings nb = currentNavBindings();
    const bool was_navigating =
        (active_button_ == nb.orbit_button || active_button_ == nb.pan_button);
    active_button_ = Qt::NoButton;
    if (was_navigating && pivot_indicator_visible_) {
        setPivotIndicatorVisible(false);
        requestUpdate();
    }
}
void ViewportWindow::handleMouseMove(QMouseEvent* e) {
    if (camera_mode_ == CameraMode::Fps) {
        // QCursor::setPos emits a synthetic MouseMove at the center; skip it.
        if (fps_ignore_next_mouse_move_) {
            fps_ignore_next_mouse_move_ = false;
            last_mouse_pos_ = e->pos();
            return;
        }
        const QPoint center(width() / 2, height() / 2);
        QPoint delta = e->pos() - center;
        if (delta.isNull()) return;

        camera_yaw_   -= delta.x() * 0.15f;
        camera_pitch_ += delta.y() * 0.15f;
        camera_pitch_ = qBound(-89.0f, camera_pitch_, 89.0f);

        // Pin target so the eye stays put during rotation.  Rebuild offset
        // from the new yaw/pitch and set target = eye - offset.  camera_eye_
        // is from the last rendered frame, which is fine: it's the eye the
        // user is currently seeing out of.
        const float yaw_rad   = qDegreesToRadians(camera_yaw_);
        const float pitch_rad = qDegreesToRadians(camera_pitch_);
        const QVector3D new_offset(camera_distance_ * cosf(pitch_rad) * cosf(yaw_rad),
                                   camera_distance_ * cosf(pitch_rad) * sinf(yaw_rad),
                                   camera_distance_ * sinf(pitch_rad));
        camera_target_ = camera_eye_ - new_offset;

        fps_ignore_next_mouse_move_ = true;
        recenterFpsCursor();
        have_cached_cull_ = false;
        requestUpdate();
        return;
    }

    if (section_drag_active_) {
        updateSectionDrag(e->pos().x(), e->pos().y());
        last_mouse_pos_ = e->pos();
        return;
    }

    // Promote an armed box-select to active once the cursor crosses the
    // 5-px click threshold, then keep updating the rect for the overlay.
    // last_mouse_pos_ stays at press position while armed so a fast tiny
    // drag isn't classified as a click on release.
    if (box_select_armed_ && active_button_ == Qt::LeftButton) {
        if (!box_select_active_
            && (e->pos() - box_select_start_pos_).manhattanLength() >= 5) {
            box_select_active_ = true;
        }
        if (box_select_active_) {
            box_select_current_pos_ = e->pos();
            requestUpdate();
        }
        return;
    }

    QPoint delta = e->pos() - last_mouse_pos_;
    last_mouse_pos_ = e->pos();

    // Dispatch nav drag against the active preset.  Modifier matching
    // is exact on the Shift bit only — the only modifier any of our
    // presets uses — so toggling Shift mid-drag flips between orbit
    // and pan when both share a button (Blender / Revit).
    const NavBindings nb = currentNavBindings();
    const bool shift_now      = (e->modifiers() & Qt::ShiftModifier) != 0;
    const bool shift_for_pan  = (nb.pan_mods   & Qt::ShiftModifier) != 0;
    const bool shift_for_orb  = (nb.orbit_mods & Qt::ShiftModifier) != 0;
    const bool is_pan   = (active_button_ == nb.pan_button)
                       && (shift_now == shift_for_pan);
    const bool is_orbit = !is_pan
                       && (active_button_ == nb.orbit_button)
                       && (shift_now == shift_for_orb);

    if (is_pan) {
        const float pan_speed = camera_distance_ * 0.002f;
        // Derive screen-right and screen-up from the actual camera basis
        // rather than yaw/pitch alone — the latter assumed up = world +Z,
        // which breaks at top/bottom views where updateCamera switches
        // the lookAt up vector to world +Y.
        const QVector3D forward = (camera_target_ - camera_eye_).normalized();
        const QVector3D up_ref = (std::abs(camera_pitch_) >= 89.0f)
                               ? QVector3D(0, 1, 0)
                               : QVector3D(0, 0, 1);
        const QVector3D right = QVector3D::crossProduct(forward, up_ref).normalized();
        const QVector3D up    = QVector3D::crossProduct(right, forward).normalized();
        camera_target_ -= right * delta.x() * pan_speed;
        camera_target_ += up    * delta.y() * pan_speed;
        requestUpdate();
    } else if (is_orbit) {
        camera_yaw_ -= delta.x() * 0.3f;
        camera_pitch_ += delta.y() * 0.3f;
        camera_pitch_ = qBound(-89.0f, camera_pitch_, 89.0f);
        requestUpdate();
    }
}
void ViewportWindow::handleWheel(QWheelEvent* e) {
    if (camera_mode_ == CameraMode::Fps) {
        float factor = e->angleDelta().y() > 0 ? 1.25f : 0.8f;
        fps_move_speed_ = qBound(0.05f, fps_move_speed_ * factor, 1000.0f);
        return;
    }
    float factor = e->angleDelta().y() > 0 ? 0.9f : 1.1f;
    camera_distance_ *= factor;
    camera_distance_ = qMax(0.1f, camera_distance_);
    setPivotIndicatorVisible(true, 750);
    requestUpdate();
}

// === Federation pipeline composition ===

void ViewportWindow::composeInstanceFromPlacement(InstanceCpu& inst,
                                                  const ModelGpuData& m) const {
    // Read placement_transformation as float-column-major and lift to double.
    using Mat4fCol = Eigen::Matrix<float, 4, 4, Eigen::ColMajor>;
    const Eigen::Matrix4d P =
        Eigen::Map<const Mat4fCol>(inst.placement_transformation).cast<double>();

    // FederatedFalseOrigin · ModelTransformation · CoordinateOperation · P.
    const Eigen::Matrix4d composed =
        federated_false_origin_meters_ *
        m.model_transformation_meters *
        m.coordinate_operation_meters *
        P;

    Eigen::Map<Mat4fCol> T_f(inst.transform);
    T_f = composed.cast<float>();

    // World AABB from the composed transform + the mesh's local AABB.
    if (inst.mesh_id < m.meshes.size()) {
        const MeshInfo& mi = m.meshes[inst.mesh_id];
        worldAabbFromLocalVp(mi.local_aabb_min, mi.local_aabb_max,
                             inst.transform,
                             inst.world_aabb_min, inst.world_aabb_max);
    } else {
        // Mesh not yet uploaded — leave AABB at zero.  The streamer's
        // contract emits MeshChunk before InstanceChunk so this branch
        // shouldn't fire under normal flow.
        for (int a = 0; a < 3; ++a) {
            inst.world_aabb_min[a] = 0.0f;
            inst.world_aabb_max[a] = 0.0f;
        }
    }
}

void ViewportWindow::recomposeAndUploadModel(uint32_t model_id) {
    if (!gl_initialized_) return;
    auto it = models_gpu_.find(model_id);
    if (it == models_gpu_.end()) return;
    ModelGpuData& m = it->second;
    if (m.instances.empty()) return;
    context_->makeCurrent(this);

    // Recompose every instance + refresh reflection flag.
    m.instance_reflected.resize(m.instances.size());
    std::vector<InstanceGpu> gpu(m.instances.size());
    for (size_t i = 0; i < m.instances.size(); ++i) {
        InstanceCpu& inst = m.instances[i];
        composeInstanceFromPlacement(inst, m);
        m.instance_reflected[i] = transformIsReflected(inst.transform) ? 1 : 0;

        InstanceGpu& dst = gpu[i];
        std::memcpy(dst.transform, inst.transform, sizeof(dst.transform));
        dst.object_id            = inst.object_id;
        dst.color_override_rgba8 = inst.color_override_rgba8;
        dst.mesh_id              = inst.mesh_id;
        dst._pad1                = 0;
    }

    // Re-upload the SSBO in one shot — its capacity already matches
    // m.instances.size() since the SSBO grew incrementally during stream.
    const size_t bytes = gpu.size() * sizeof(InstanceGpu);
    if (bytes > m.ssbo_capacity) {
        if (!growModelSsbo(m, bytes)) return;
    }
    if (bytes > 0) {
        gl_->glNamedBufferSubData(m.ssbo, 0, bytes, gpu.data());
    }
    m.ssbo_instance_count = static_cast<uint32_t>(gpu.size());

    // World AABBs changed, so the BVH is stale.
    buildBvhForModel(m, model_id);
    have_cached_cull_ = false;
    requestUpdate();
}

void ViewportWindow::setFederatedFalseOrigin(const Eigen::Matrix4d& matrix_meters) {
    if (federated_false_origin_meters_ == matrix_meters) return;
    federated_false_origin_meters_ = matrix_meters;
    for (auto& kv : models_gpu_) {
        recomposeAndUploadModel(kv.first);
    }
}

void ViewportWindow::setModelCoordinateOperation(uint32_t model_id,
                                                  const Eigen::Matrix4d& matrix_meters) {
    auto it = models_gpu_.find(model_id);
    if (it == models_gpu_.end()) return;
    if (it->second.coordinate_operation_meters == matrix_meters) return;
    it->second.coordinate_operation_meters = matrix_meters;
    recomposeAndUploadModel(model_id);
}

void ViewportWindow::setModelTransformation(uint32_t model_id,
                                            const Eigen::Matrix4d& matrix_meters) {
    auto it = models_gpu_.find(model_id);
    if (it == models_gpu_.end()) return;
    if (it->second.model_transformation_meters == matrix_meters) return;
    it->second.model_transformation_meters = matrix_meters;
    recomposeAndUploadModel(model_id);
}

void ViewportWindow::printSelectedObjectCoords() {
    const uint32_t target = selection_.activeObjectId();
    if (target == 0) {
        qInfo("printSelectedObjectCoords: no object selected");
        return;
    }
    if (!gl_initialized_) {
        qInfo("printSelectedObjectCoords: GL not initialised yet");
        return;
    }

    for (const auto& kv : models_gpu_) {
        const ModelGpuData& m = kv.second;
        for (size_t i = 0; i < m.instances.size(); ++i) {
            const InstanceCpu& inst = m.instances[i];
            if (inst.object_id != target) continue;

            qInfo("Selected object %u (model %u, mesh %u, instance %zu):",
                  inst.object_id, inst.model_id, inst.mesh_id, i);

            // Decode the first emitted vertex from the VBO.  Per
            // InstancedGeometry.h: pos is 3 x uint16 at offset 0, normalised
            // to [0,1] and dequantised against the mesh's local AABB.
            float vx = 0, vy = 0, vz = 0;
            bool have_vert = false;
            if (inst.mesh_id < m.meshes.size()) {
                const MeshInfo& mi = m.meshes[inst.mesh_id];
                if (mi.vertex_count > 0) {
                    context_->makeCurrent(this);
                    uint16_t pos_u16[3] = {0, 0, 0};
                    gl_->glGetNamedBufferSubData(
                        m.vbo, mi.vbo_byte_offset, sizeof(pos_u16), pos_u16);
                    auto lerp = [](float lo, float hi, float t) {
                        return lo + t * (hi - lo);
                    };
                    vx = lerp(mi.local_aabb_min[0], mi.local_aabb_max[0],
                              pos_u16[0] / 65535.0f);
                    vy = lerp(mi.local_aabb_min[1], mi.local_aabb_max[1],
                              pos_u16[1] / 65535.0f);
                    vz = lerp(mi.local_aabb_min[2], mi.local_aabb_max[2],
                              pos_u16[2] / 65535.0f);
                    have_vert = true;
                }
            }
            if (have_vert) {
                qInfo("  vertex (mesh-local m): (%g, %g, %g)", vx, vy, vz);
            } else {
                qInfo("  vertex: (no vertex data)");
            }

            using Mat4f = Eigen::Matrix<float, 4, 4, Eigen::ColMajor>;
            const Eigen::Matrix4d Pd =
                Eigen::Map<const Mat4f>(inst.placement_transformation).cast<double>();
            // global = CoordinateOperation · placement_transformation.
            // (FederatedFalseOrigin and ModelTransformation are user-side
            // tweaks; "global" here means the IFC's own georeferenced frame.)
            const Eigen::Matrix4d Gd = m.coordinate_operation_meters * Pd;

            auto print_mat = [](const char* label, const Eigen::Matrix4d& M) {
                qInfo("  %s (m):", label);
                for (int r = 0; r < 4; ++r) {
                    qInfo("    [% .9g % .9g % .9g % .9g]",
                          M(r, 0), M(r, 1), M(r, 2), M(r, 3));
                }
            };
            print_mat("placement_transformation", Pd);
            print_mat("global (CoordinateOperation . placement)", Gd);

            if (have_vert) {
                const Eigen::Vector4d vh(vx, vy, vz, 1.0);
                const Eigen::Vector3d after_p = (Pd * vh).head<3>();
                const Eigen::Vector3d after_g = (Gd * vh).head<3>();
                qInfo("  vertex after placement (m): (% .9g, % .9g, % .9g)",
                      after_p.x(), after_p.y(), after_p.z());
                qInfo("  vertex after global (m):    (% .9g, % .9g, % .9g)",
                      after_g.x(), after_g.y(), after_g.z());
            }
            return;
        }
    }
    qInfo("printSelectedObjectCoords: object_id %u not found in any model",
          target);
}

bool ViewportWindow::readbackMeshTriangles(uint32_t model_id, uint32_t mesh_id,
                                            MeshTriangles& out) {
    out.positions.clear();
    out.indices.clear();
    if (!gl_initialized_) return false;
    auto it = models_gpu_.find(model_id);
    if (it == models_gpu_.end()) return false;
    const ModelGpuData& m = it->second;
    if (!m.finalized) return false;
    if (mesh_id >= m.meshes.size()) return false;
    const MeshInfo& mi = m.meshes[mesh_id];
    if (mi.vertex_count == 0 || mi.index_count == 0) return false;

    context_->makeCurrent(this);

    // Vertices: stride is INSTANCED_VERTEX_STRIDE_BYTES (12), positions are
    // the first 6 bytes (3 x uint16) of each vertex.  Read the full range
    // and stride through it — simpler than chasing a position-only buffer.
    std::vector<uint8_t> raw(size_t(mi.vertex_count) * INSTANCED_VERTEX_STRIDE_BYTES);
    gl_->glGetNamedBufferSubData(m.vbo, mi.vbo_byte_offset,
                                 GLsizeiptr(raw.size()), raw.data());

    out.positions.resize(size_t(mi.vertex_count) * 3);
    const float ox = mi.local_aabb_min[0];
    const float oy = mi.local_aabb_min[1];
    const float oz = mi.local_aabb_min[2];
    const float sx = (mi.local_aabb_max[0] - mi.local_aabb_min[0]) / 65535.0f;
    const float sy = (mi.local_aabb_max[1] - mi.local_aabb_min[1]) / 65535.0f;
    const float sz = (mi.local_aabb_max[2] - mi.local_aabb_min[2]) / 65535.0f;
    for (uint32_t i = 0; i < mi.vertex_count; ++i) {
        const uint16_t* p = reinterpret_cast<const uint16_t*>(
            raw.data() + size_t(i) * INSTANCED_VERTEX_STRIDE_BYTES);
        out.positions[3 * i + 0] = ox + sx * float(p[0]);
        out.positions[3 * i + 1] = oy + sy * float(p[1]);
        out.positions[3 * i + 2] = oz + sz * float(p[2]);
    }

    // Indices: LOD0 only — measurements should use the full-resolution mesh.
    out.indices.resize(mi.index_count);
    gl_->glGetNamedBufferSubData(m.ebo, mi.ebo_byte_offset,
                                 GLsizeiptr(mi.index_count * sizeof(uint32_t)),
                                 out.indices.data());
    return true;
}

bool ViewportWindow::findInstance(uint32_t object_id, InstanceLookup& out) const {
    if (object_id == 0) return false;
    for (const auto& kv : models_gpu_) {
        const ModelGpuData& m = kv.second;
        for (const InstanceCpu& inst : m.instances) {
            if (inst.object_id != object_id) continue;
            out.model_id = inst.model_id;
            out.mesh_id  = inst.mesh_id;
            std::memcpy(out.placement_transformation,
                        inst.placement_transformation,
                        sizeof(out.placement_transformation));
            return true;
        }
    }
    return false;
}

bool ViewportWindow::pickMeshLocalAt(int x, int y, MeshLocalPick& out) {
    uint32_t  obj_id = 0;
    QVector3D world_pos, world_normal;
    if (!pickSurfaceAt(x, y, obj_id, world_pos, world_normal)) return false;

    for (const auto& kv : models_gpu_) {
        const ModelGpuData& m = kv.second;
        for (const InstanceCpu& inst : m.instances) {
            if (inst.object_id != obj_id) continue;
            using Mat4f = Eigen::Matrix<float, 4, 4, Eigen::ColMajor>;
            const Eigen::Matrix4f T  = Eigen::Map<const Mat4f>(inst.transform);
            const Eigen::Matrix4f Ti = T.inverse();
            const Eigen::Vector4f wp(world_pos.x(), world_pos.y(), world_pos.z(), 1.0f);
            const Eigen::Vector4f mp = Ti * wp;
            out.object_id     = obj_id;
            out.model_id      = inst.model_id;
            out.mesh_id       = inst.mesh_id;
            out.mesh_local[0] = mp.x();
            out.mesh_local[1] = mp.y();
            out.mesh_local[2] = mp.z();
            out.world_pos[0]  = world_pos.x();
            out.world_pos[1]  = world_pos.y();
            out.world_pos[2]  = world_pos.z();
            out.world_normal[0] = world_normal.x();
            out.world_normal[1] = world_normal.y();
            out.world_normal[2] = world_normal.z();
            std::memcpy(out.composed_transform, inst.transform,
                        sizeof(out.composed_transform));
            return true;
        }
    }
    return false;
}

void ViewportWindow::setToolMode(ToolMode mode) {
    if (tool_mode_ == mode) return;
    tool_mode_ = mode;
    emit toolModeChanged(mode);
}

namespace {

// Slab method ray-AABB.  inv_d is precomputed 1/dir per axis.
bool rayAabb(const float ro[3], const float inv_d[3],
             const float bmin[3], const float bmax[3]) {
    float tmin = 0.0f, tmax = std::numeric_limits<float>::infinity();
    for (int i = 0; i < 3; ++i) {
        float t1 = (bmin[i] - ro[i]) * inv_d[i];
        float t2 = (bmax[i] - ro[i]) * inv_d[i];
        tmin = std::max(tmin, std::min(t1, t2));
        tmax = std::min(tmax, std::max(t1, t2));
    }
    return tmax >= tmin && tmax >= 0.0f;
}

// Möller-Trumbore.  Returns true on hit; t is in dir-units.
bool rayTri(const float ro[3], const float rd[3],
            const float v0[3], const float v1[3], const float v2[3],
            float& t_out) {
    constexpr float EPS = 1e-7f;
    float e1[3] = {v1[0]-v0[0], v1[1]-v0[1], v1[2]-v0[2]};
    float e2[3] = {v2[0]-v0[0], v2[1]-v0[1], v2[2]-v0[2]};
    float h[3]  = {
        rd[1]*e2[2] - rd[2]*e2[1],
        rd[2]*e2[0] - rd[0]*e2[2],
        rd[0]*e2[1] - rd[1]*e2[0]
    };
    float a = e1[0]*h[0] + e1[1]*h[1] + e1[2]*h[2];
    if (a > -EPS && a < EPS) return false;
    float f = 1.0f / a;
    float s[3] = {ro[0]-v0[0], ro[1]-v0[1], ro[2]-v0[2]};
    float u = f * (s[0]*h[0] + s[1]*h[1] + s[2]*h[2]);
    if (u < 0.0f || u > 1.0f) return false;
    float q[3] = {
        s[1]*e1[2] - s[2]*e1[1],
        s[2]*e1[0] - s[0]*e1[2],
        s[0]*e1[1] - s[1]*e1[0]
    };
    float v = f * (rd[0]*q[0] + rd[1]*q[1] + rd[2]*q[2]);
    if (v < 0.0f || u + v > 1.0f) return false;
    float t = f * (e2[0]*q[0] + e2[1]*q[1] + e2[2]*q[2]);
    if (t > EPS) { t_out = t; return true; }
    return false;
}

// Stack-based BVH walk — collects the item indices of every leaf whose
// AABB the ray intersects.  Caller filters down to actual triangle hits.
void bvhCollectRayCandidates(const ModelBvh& bvh,
                              const float ro[3], const float inv_d[3],
                              std::vector<uint32_t>& out) {
    if (bvh.nodes.empty()) return;
    std::vector<uint32_t> stack;
    stack.reserve(64);
    stack.push_back(0);
    while (!stack.empty()) {
        uint32_t idx = stack.back(); stack.pop_back();
        const BvhNode& node = bvh.nodes[idx];
        if (!rayAabb(ro, inv_d, node.aabb_min, node.aabb_max)) continue;
        if (node.count > 0) {
            for (uint32_t i = 0; i < node.count; ++i) {
                out.push_back(bvh.item_indices[node.right_or_first + i]);
            }
        } else {
            stack.push_back(idx + 1);                    // left child
            stack.push_back(node.right_or_first);        // right child
        }
    }
}

} // namespace

bool ViewportWindow::raycast(const float origin[3], const float dir[3],
                              RaycastHit& out) {
    if (!gl_initialized_) return false;

    float inv_d[3];
    for (int i = 0; i < 3; ++i) {
        inv_d[i] = (std::abs(dir[i]) > 1e-12f)
            ? 1.0f / dir[i] : std::numeric_limits<float>::infinity();
    }

    // Local mesh-triangle cache shared across candidate instances of the
    // same mesh in this single call.  Avoids re-stalling the GL pipeline
    // for repeated readbacks of the same VBO/EBO range.
    std::unordered_map<uint64_t, MeshTriangles> mesh_cache;

    float closest_t = std::numeric_limits<float>::infinity();
    bool any_hit = false;
    using Mat4f = Eigen::Matrix<float, 4, 4, Eigen::ColMajor>;

    for (auto& kv : models_gpu_) {
        ModelGpuData& m = kv.second;
        if (!m.finalized || m.bvh.nodes.empty()) continue;

        std::vector<uint32_t> candidates;
        bvhCollectRayCandidates(m.bvh, origin, inv_d, candidates);
        if (candidates.empty()) continue;

        for (uint32_t inst_idx : candidates) {
            if (inst_idx >= m.instances.size()) continue;
            const InstanceCpu& inst = m.instances[inst_idx];

            // Transform ray into this instance's mesh-local space.  The
            // returned t parameter is identical in world and mesh-local
            // (both are along the same parametric line) so we don't need
            // to convert it back — assumes caller passed a unit world dir.
            const Eigen::Matrix4f T  = Eigen::Map<const Mat4f>(inst.transform);
            const Eigen::Matrix4f Ti = T.inverse();
            const Eigen::Vector4f wo(origin[0], origin[1], origin[2], 1.0f);
            const Eigen::Vector4f wd(dir[0],    dir[1],    dir[2],    0.0f);
            const Eigen::Vector4f lo = Ti * wo;
            const Eigen::Vector4f ld = Ti * wd;
            const float ro_l[3] = {lo.x(), lo.y(), lo.z()};
            const float rd_l[3] = {ld.x(), ld.y(), ld.z()};

            const uint64_t key = (uint64_t(inst.model_id) << 32) | inst.mesh_id;
            auto it = mesh_cache.find(key);
            if (it == mesh_cache.end()) {
                MeshTriangles tris;
                if (!readbackMeshTriangles(inst.model_id, inst.mesh_id, tris)) continue;
                it = mesh_cache.emplace(key, std::move(tris)).first;
            }
            const MeshTriangles& tris = it->second;

            // Möller-Trumbore against every triangle.  No per-mesh BVH
            // here yet — buildings rarely have meshes with > a few
            // thousand tris; if this becomes a hotspot we can add one.
            for (size_t i = 0; i + 2 < tris.indices.size(); i += 3) {
                const uint32_t ia = tris.indices[i + 0];
                const uint32_t ib = tris.indices[i + 1];
                const uint32_t ic = tris.indices[i + 2];
                const float* a = &tris.positions[3 * ia];
                const float* b = &tris.positions[3 * ib];
                const float* c = &tris.positions[3 * ic];
                float t;
                if (!rayTri(ro_l, rd_l, a, b, c, t)) continue;
                if (t >= closest_t) continue;
                closest_t = t;
                any_hit = true;
                out.object_id = inst.object_id;
                out.distance  = t;
                out.world_pos[0] = origin[0] + t * dir[0];
                out.world_pos[1] = origin[1] + t * dir[1];
                out.world_pos[2] = origin[2] + t * dir[2];
                // World normal: triangle normal in mesh-local, transformed
                // by inverse-transpose of the 3x3.
                const float e1x = b[0]-a[0], e1y = b[1]-a[1], e1z = b[2]-a[2];
                const float e2x = c[0]-a[0], e2y = c[1]-a[1], e2z = c[2]-a[2];
                Eigen::Vector3f n_local(
                    e1y*e2z - e1z*e2y,
                    e1z*e2x - e1x*e2z,
                    e1x*e2y - e1y*e2x);
                Eigen::Matrix3f N = T.block<3, 3>(0, 0).inverse().transpose();
                Eigen::Vector3f world_n = (N * n_local).normalized();
                out.world_normal[0] = world_n.x();
                out.world_normal[1] = world_n.y();
                out.world_normal[2] = world_n.z();
            }
        }
    }
    return any_hit;
}

void ViewportWindow::toggleAreaTool() {
    setToolMode(tool_mode_ == ToolMode::Area ? ToolMode::None : ToolMode::Area);
}

void ViewportWindow::toggleLengthTool() {
    setToolMode(tool_mode_ == ToolMode::Length ? ToolMode::None : ToolMode::Length);
}

void ViewportWindow::toggleVolumeTool() {
    setToolMode(tool_mode_ == ToolMode::Volume ? ToolMode::None : ToolMode::Volume);
}

ViewportWindow::NavBindings ViewportWindow::currentNavBindings() const {
    using NP = AppSettings::NavPreset;
    switch (AppSettings::instance().navPreset()) {
    case NP::Rhino:
        // Orbit RMB,        Pan Shift+RMB.
        return { Qt::RightButton,  Qt::NoModifier,
                 Qt::RightButton,  Qt::ShiftModifier };
    case NP::Revit:
        // Orbit Shift+MMB,  Pan MMB.
        return { Qt::MiddleButton, Qt::ShiftModifier,
                 Qt::MiddleButton, Qt::NoModifier };
    case NP::Blender:
    default:
        // Orbit MMB,        Pan Shift+MMB.
        return { Qt::MiddleButton, Qt::NoModifier,
                 Qt::MiddleButton, Qt::ShiftModifier };
    }
}

void ViewportWindow::setHighlightTriangles(const std::vector<float>& world_xyz,
                                            float r, float g, float b, float a) {
    if (!gl_initialized_) return;
    context_->makeCurrent(this);
    overlay_renderer_.setHighlightTriangles(world_xyz, r, g, b, a);
    requestUpdate();
}

void ViewportWindow::setOverlayLines(const std::vector<OverlayRenderer::LineGroup>& groups) {
    if (!gl_initialized_) return;
    context_->makeCurrent(this);
    overlay_renderer_.setOverlayLines(groups);
    requestUpdate();
}

void ViewportWindow::setOverlayPoints(const std::vector<float>& world_xyz,
                                       float r, float g, float b, float a,
                                       float pixel_size,
                                       float sr, float sg, float sb, float sa,
                                       float stroke_extra) {
    if (!gl_initialized_) return;
    context_->makeCurrent(this);
    overlay_renderer_.setOverlayPoints(world_xyz, r, g, b, a, pixel_size,
                                       sr, sg, sb, sa, stroke_extra);
    requestUpdate();
}

void ViewportWindow::setOverlayLabels(const std::vector<OverlayRenderer::Label>& labels) {
    overlay_renderer_.setOverlayLabels(labels);
    requestUpdate();
}

void ViewportWindow::setHudText(const QString& text) {
    overlay_renderer_.setHudText(text);
    requestUpdate();
}