q3dviewer 1.2.1__py3-none-any.whl → 1.2.2__py3-none-any.whl

q3dviewer/custom_items/mesh_item.py

@@ -10,7 +10,7 @@ import numpy as np
 from q3dviewer.base_item import BaseItem
 from OpenGL.GL import *
 from OpenGL.GL import shaders
-from q3dviewer.Qt.QtWidgets import QLabel, QCheckBox, QDoubleSpinBox, QSlider, QHBoxLayout, QLineEdit
+from q3dviewer.Qt.QtWidgets import QLabel, QCheckBox, QDoubleSpinBox, QSlider, QHBoxLayout, QLineEdit, QComboBox
 import matplotlib.colors as mcolors
 import os
 from q3dviewer.utils import set_uniform, text_to_rgba
@@ -18,7 +18,53 @@ from q3dviewer.utils import set_uniform, text_to_rgba
 
 
 class MeshItem(BaseItem):
-    def __init__(self, color='lightblue', wireframe=False, enable_lighting=True):
+    """
+    An OpenGL mesh item for rendering triangulated 3D surfaces.
+
+    Attributes:
+        color (str or tuple): The flat color to use when `color_mode` is 'FLAT'. 
+            Accepts any valid matplotlib color (e.g., 'lightblue', 'red', '#FF4500', (1.0, 0.5, 0.0)).
+        wireframe (bool): If True, render the mesh in wireframe mode (edges only). 
+            If False, render filled triangles. Default is False.
+        enable_lighting (bool): Whether to enable Phong lighting for the mesh. 
+            If True, the mesh will be shaded based on light direction and material properties.
+            If False, the mesh will use flat shading with object colors only. Default is True.
+        color_mode (str): The coloring mode for mesh vertices.
+            - 'FLAT': Single flat color for all vertices (uses the `color` attribute).
+            - 'I': Color by intensity channel from per-vertex colors (rainbow gradient).
+            - 'RGB': Per-vertex RGB color from per-vertex color data.
+        alpha (float): The transparency of the mesh, in the range [0, 1], 
+            where 0 is fully transparent and 1 is fully opaque. Default is 1.0.
+        line_width (float): The width of lines when rendering in wireframe mode. 
+            Range is typically 0.5 to 5.0. Default is 1.0.
+        
+    Material Properties (Phong Lighting):
+        ambient_strength (float): Ambient light contribution [0.0-1.0]. Default is 0.1.
+        diffuse_strength (float): Diffuse light contribution [0.0-2.0]. Default is 1.2.
+        specular_strength (float): Specular highlight contribution [0.0-2.0]. Default is 0.1.
+        shininess (float): Specular shininess exponent [1-256]. Higher values = smaller highlights. Default is 32.0.
+        
+    Methods:
+        set_data(verts, faces, colors=None): Set mesh geometry and optional per-vertex colors.
+            - verts: np.ndarray of shape (N, 3) - vertex positions
+            - faces: np.ndarray of shape (M, 3) with uint32 indices - triangle indices
+            - colors: np.ndarray of shape (N,) with uint32 IRGB format (optional)
+                     uint32 format: I (bits 24-31), R (bits 16-23), G (bits 8-15), B (bits 0-7)
+                     
+    Example:
+        # Create a simple triangle mesh with per-vertex colors
+        verts = np.array([[0,0,0], [1,0,0], [0,1,0]], dtype=np.float32)
+        faces = np.array([[0,1,2]], dtype=np.uint32)
+        colors = np.array([
+            (255 << 24) | (255 << 16) | (0 << 8) | 0,  # Red, intensity=255
+            (200 << 24) | (0 << 16) | (255 << 8) | 0,  # Green, intensity=200
+            (150 << 24) | (0 << 16) | (0 << 8) | 255   # Blue, intensity=150
+        ], dtype=np.uint32)
+        
+        mesh = q3d.MeshItem(color='lightblue', color_mode='RGB', enable_lighting=True)
+        mesh.set_data(verts, faces, colors)
+    """
+    def __init__(self, color='lightblue', wireframe=False, enable_lighting=True, color_mode='FLAT'):
         super(MeshItem, self).__init__()
         self.color = color
         self.flat_rgb = text_to_rgba(color, flat=True)
@@ -28,12 +74,18 @@ class MeshItem(BaseItem):
         # Mesh data
         self.triangles = None
         self.normals = None
+        self.vertex_colors = None  # Per-vertex colors (uint32 IRGB format)
+        
+        self.mode_table = {'FLAT': 0, 'I': 1, 'RGB': 2}
+        self.color_mode = self.mode_table[color_mode]
+        self.vmin = 0
+        self.vmax = 255
         
         # OpenGL objects
         self.vao = None
         self.vbo_vertices = None
         self.vbo_normals = None
-        # self.vbo_colors = None
+        self.vbo_colors = None
         self.program = None
         
         # Rendering parameters
@@ -90,6 +142,17 @@ class MeshItem(BaseItem):
         layout.addWidget(alpha_box)
         
 
+        # Color mode selection
+        label_color = QLabel("Color Mode:")
+        layout.addWidget(label_color)
+        self.combo_color = QComboBox()
+        self.combo_color.addItem("flat color")
+        self.combo_color.addItem("intensity")
+        self.combo_color.addItem("RGB")
+        self.combo_color.setCurrentIndex(self.color_mode)
+        self.combo_color.currentIndexChanged.connect(self._on_color_mode)
+        layout.addWidget(self.combo_color)
+
         label_rgb = QLabel("Color:")
         label_rgb.setToolTip("Use hex color, i.e. #FF4500, or named color, i.e. 'red'")
         layout.addWidget(label_rgb)
@@ -158,6 +221,11 @@ class MeshItem(BaseItem):
         except ValueError:
             pass
 
+    def _on_color_mode(self, index):
+        print(f"Color mode1 : {index}")
+        self.color_mode = index
+        self.edit_rgb.setVisible(index == self.mode_table['FLAT'])
+        self.need_update_setting = True
 
     def update_wireframe(self, value):
         self.wireframe = value
@@ -191,14 +259,26 @@ class MeshItem(BaseItem):
         self.alpha = float(value)
         self.need_update_setting = True
             
-    def set_data(self, verts, faces):
+    def set_data(self, verts, faces, colors=None):
         """
         verts: np.ndarray of shape (N, 3)
         faces: np.ndarray of shape (M, 3) with uint32 indices
+        colors: np.ndarray of shape (N,) with uint32 IRGB format (optional)
+                uint32 contains: I (bits 24-31), R (bits 16-23), G (bits 8-15), B (bits 0-7)
         """
         verts = np.asarray(verts, dtype=np.float32)
         faces = np.asarray(faces, dtype=np.uint32)
         triangles = verts[faces.flatten()]
+
+        if colors is not None:
+            colors = np.asarray(colors, dtype=np.uint32)
+            if len(colors) == len(verts):
+                # Expand per-vertex colors to per-triangle-vertex
+                self.vertex_colors = colors[faces.flatten()]
+            else:
+                self.vertex_colors = None
+        else:
+            self.vertex_colors = None
             
         self.triangles = np.asarray(triangles, dtype=np.float32)
         self.normals = self.calculate_normals()
@@ -256,7 +336,7 @@ class MeshItem(BaseItem):
             self.vao = glGenVertexArrays(1)
             self.vbo_vertices = glGenBuffers(1)
             self.vbo_normals = glGenBuffers(1)
-            # self.vbo_colors = glGenBuffers(1)
+            self.vbo_colors = glGenBuffers(1)
         
         glBindVertexArray(self.vao)
         
@@ -273,13 +353,13 @@ class MeshItem(BaseItem):
             glEnableVertexAttribArray(1)
             glVertexAttribPointer(1, 3, GL_FLOAT, GL_FALSE, 0, None)
         
-        # Color buffer
-        # if self.flat_rgbs is not None:
-        #     glBindBuffer(GL_ARRAY_BUFFER, self.vbo_colors)
-        #     glBufferData(GL_ARRAY_BUFFER, self.flat_rgbs.nbytes, self.flat_rgbs, GL_STATIC_DRAW)
-        #     glEnableVertexAttribArray(2)
-        #     glVertexAttribPointer(2, 1, GL_FLOAT, GL_FALSE, 0, None)
-        # 
+        # Color buffer (uint32 IRGB format)
+        if self.vertex_colors is not None:
+            glBindBuffer(GL_ARRAY_BUFFER, self.vbo_colors)
+            glBufferData(GL_ARRAY_BUFFER, self.vertex_colors.nbytes, self.vertex_colors, GL_STATIC_DRAW)
+            glEnableVertexAttribArray(2)
+            glVertexAttribIPointer(2, 1, GL_UNSIGNED_INT, 0, None)
+        
         glBindVertexArray(0)
         self.need_update_buffer = False
         
@@ -295,6 +375,9 @@ class MeshItem(BaseItem):
         set_uniform(self.program, float(self.shininess), 'shininess')
         set_uniform(self.program, float(self.alpha), 'alpha')
         set_uniform(self.program, int(self.flat_rgb), 'flat_rgb')
+        set_uniform(self.program, int(self.color_mode), 'color_mode')
+        set_uniform(self.program, float(self.vmin), 'vmin')
+        set_uniform(self.program, float(self.vmax), 'vmax')
         self.need_update_setting = False
 
     def paint(self):

q3dviewer/shaders/mesh_vert.glsl

@@ -1,7 +1,7 @@
 #version 330 core
 layout (location = 0) in vec3 aPos;
 layout (location = 1) in vec3 aNormal;
-layout (location = 2) in float aColor;
+layout (location = 2) in uint aColor;
 
 out vec3 FragPos;
 out vec3 Normal;
@@ -10,19 +10,55 @@ out vec3 objectColor;
 uniform mat4 view;
 uniform mat4 projection;
 uniform int flat_rgb;
+uniform int color_mode;  // 0: FLAT, 1: Intensity, 2: RGB
+uniform float vmin;
+uniform float vmax;
+
+vec3 getRainbowColor(uint value_raw) {
+    float range = vmax - vmin;
+    float value = 1.0 - (float(value_raw) - vmin) / range;
+    value = clamp(value, 0.0, 1.0);
+    float hue = value * 5.0 + 1.0;
+    int i = int(floor(hue));
+    float f = hue - float(i);
+    if (mod(i, 2) == 0) f = 1.0 - f;
+    float n = 1.0 - f;
+
+    vec3 color;
+    if (i <= 1) color = vec3(n, 0.0, 1.0);
+    else if (i == 2) color = vec3(0.0, n, 1.0);
+    else if (i == 3) color = vec3(0.0, 1.0, n);
+    else if (i == 4) color = vec3(n, 1.0, 0.0);
+    else color = vec3(1.0, n, 0.0);
+    return color;
+}
 
 void main()
 {
     FragPos = aPos;
-    
     Normal = aNormal;
     
-    int color_int = int(aColor);
-    objectColor = vec3(
-        float((uint(flat_rgb) & uint(0x00FF0000)) >> 16)/255.,
-        float((uint(flat_rgb) & uint(0x0000FF00)) >> 8)/255.,
-        float( uint(flat_rgb) & uint(0x000000FF))/255.
-    ) ;
+    vec3 c = vec3(1.0, 1.0, 1.0);
+    
+    if (color_mode == 0) {
+        // FLAT: use uniform flat color
+        c.z = float( uint(flat_rgb) & uint(0x000000FF))/255.;
+        c.y = float((uint(flat_rgb) & uint(0x0000FF00)) >> 8)/255.;
+        c.x = float((uint(flat_rgb) & uint(0x00FF0000)) >> 16)/255.;
+    }
+    else if (color_mode == 1) {
+        // Intensity: use intensity channel (bits 24-31) for rainbow color
+        uint intensity = aColor >> 24;
+        c = getRainbowColor(intensity);
+    }
+    else if (color_mode == 2) {
+        // RGB: use RGB channels (bits 0-23)
+        c.z = float(aColor & uint(0x000000FF))/255.;
+        c.y = float((aColor & uint(0x0000FF00)) >> 8)/255.;
+        c.x = float((aColor & uint(0x00FF0000)) >> 16)/255.;
+    }
+    
+    objectColor = c;
     
     // Final vertex position (apply view/projection)
     gl_Position = projection * view * vec4(aPos, 1.0);

q3dviewer/tools/cloud_viewer.py

@@ -58,7 +58,12 @@ class FileLoaderThread(QThread):
             if url.toLocalFile().lower().endswith(('.stl')):
                 from q3dviewer.utils.cloud_io import load_stl
                 verts, faces = load_stl(file_path)
-                mesh_item.set_data(verts=verts, faces=faces)
+                # create colors, N, array
+                color = np.zeros((verts.shape[0],), dtype=np.uint32)
+                # random uint32 colors in IRGB format
+                color = np.random.randint(0, 0xFFFFFFFF, size=(verts.shape[0],), dtype=np.uint32)
+                mesh_item.set_data(verts=verts, faces=faces, colors=color)
+                break
             else:
                 cloud = cloud_item.load(file_path, append=(i > 0))
                 center = np.nanmean(cloud['xyz'].astype(np.float64), axis=0)
@@ -186,18 +191,20 @@ def main():
     viewer = CloudViewer(name='Cloud Viewer')
     cloud_item = q3d.CloudIOItem(size=1, alpha=0.1)
     axis_item = q3d.AxisItem(size=0.5, width=5)
+    axis_item.disable_setting()
     grid_item = q3d.GridItem(size=1000, spacing=20)
     marker_item = q3d.Text3DItem()  # Changed from CloudItem to Text3DItem
     text_item = q3d.Text2DItem(pos=(20, 40), text="", color='lime', size=16)
+    text_item.disable_setting()
     mesh_item = q3d.MeshItem()  # Added MeshIOItem for mesh support
 
     viewer.add_items(
         {'marker': marker_item, 
-         'cloud': cloud_item, 
+         'cloud': cloud_item,
+         'mesh': mesh_item,
          'grid': grid_item, 
          'axis': axis_item, 
-         'text': text_item,
-         'mesh': mesh_item})
+         'text': text_item,})
 
     if args.path:
         pcd_fn = args.path

{q3dviewer-1.2.1.dist-info → q3dviewer-1.2.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: q3dviewer
-Version: 1.2.1
+Version: 1.2.2
 Summary: A library designed for quickly deploying a 3D viewer.
 Home-page: https://github.com/scomup/q3dviewer
 Author: Liu Yang

{q3dviewer-1.2.1.dist-info → q3dviewer-1.2.2.dist-info}/RECORD

@@ -13,7 +13,7 @@ q3dviewer/custom_items/gaussian_item.py,sha256=JMubpahkTPh0E8ShL3FLTahv0e35ODzjg
 q3dviewer/custom_items/grid_item.py,sha256=LDB_MYACoxld-xvz01_MfAf12vLcRkH7R_WtGHHdSgk,4945
 q3dviewer/custom_items/image_item.py,sha256=k7HNTqdL2ckTbxMx7A7eKaP4aksZ85-pBjNdbpm6PXM,5355
 q3dviewer/custom_items/line_item.py,sha256=rel-lx8AgjDY7qyIecHxHQZzaswRn2ZTiOIjB_0Mrqo,4444
-q3dviewer/custom_items/mesh_item.py,sha256=LfiF6A6XTbMVyaP5E6GpgiR-wPjwxNLXKdWEOzYs-TU,13342
+q3dviewer/custom_items/mesh_item.py,sha256=Ds3GyYzvFF5VpjG-_LU9SNZiBoLmztCZrkcalEMjub8,17890
 q3dviewer/custom_items/text3d_item.py,sha256=DYBPXnCmMEzWDE1y523YsWSl91taXAdu0kdnhUcwE4A,5524
 q3dviewer/custom_items/text_item.py,sha256=toeGjBu7RtT8CMUuaDWnmXPnA1UKHhnCzUNeonGczSo,2703
 q3dviewer/shaders/cloud_frag.glsl,sha256=psKVt9qI6BW0bCqOk4lcKqUd6XgYGtdFigyN9OdYSNI,609
@@ -22,10 +22,10 @@ q3dviewer/shaders/gau_frag.glsl,sha256=vWt5I3Ojrc2PCxRlBJGyJhujbveSicMA54T01Fk29
 q3dviewer/shaders/gau_prep.glsl,sha256=0BiWhYCQGeX2iN-e7m3dy1xWXqWrErErRAzHlcmWHF0,7218
 q3dviewer/shaders/gau_vert.glsl,sha256=_rkm51zaWgPDJ-otJL-WX12fDvnPBOTooVfqo21Rexs,1666
 q3dviewer/shaders/mesh_frag.glsl,sha256=i9ljnO2kjLNGaR1TPQIK4-4iJ-JppJ5bCsOHg1730gQ,1997
-q3dviewer/shaders/mesh_vert.glsl,sha256=oF96vp2Et_ueY5slBtyuuJp4ZPJGoKKE-9xu0vlFORc,683
+q3dviewer/shaders/mesh_vert.glsl,sha256=tj9pbaWUYoKn1CtYahahPVzXNDTEtaQanq3S3VphmPg,1896
 q3dviewer/shaders/sort_by_key.glsl,sha256=M5RK6uRDp40vVH6XtBIrdJTcYatqXyZwd6kCzEa2DZg,1097
 q3dviewer/tools/__init__.py,sha256=01wG7BGM6VX0QyFBKsqPmyf2e-vrmV_N3-mo-VQ1VBg,20
-q3dviewer/tools/cloud_viewer.py,sha256=ZDdyHDPump8BKOwK-8yxgJ8cuVfGrcSOEkyaLR-Kk4o,7431
+q3dviewer/tools/cloud_viewer.py,sha256=sZ16uOt5J_3E7H0ZW3NHq0mTkWuoBkRRtwEQgcEv-Io,7794
 q3dviewer/tools/example_viewer.py,sha256=C867mLnCBjawS6LGgRsJ_c6-6wztfL9vOBQt85KbbdU,572
 q3dviewer/tools/film_maker.py,sha256=xLFgRhFWoMQ37qlvcu1lXWaTWXMNRYlRcZFfHW5JtmQ,16676
 q3dviewer/tools/gaussian_viewer.py,sha256=vIwWmiFhjNmknrEkBLzt2yiegeH7LP3OeNjnGM6GzaI,1633
@@ -39,9 +39,9 @@ q3dviewer/utils/gl_helper.py,sha256=dRY_kUqyPMr7NTcupUr6_VTvgnj53iE2C0Lk0-oFYsI,
 q3dviewer/utils/helpers.py,sha256=SqR4YTQZi13FKbkVUYgodXce1JJ_YmrHEIRkUmnIUas,3085
 q3dviewer/utils/maths.py,sha256=zHaPtvVZIuo8xepIXCMeSL9tpx8FahUrq0l4K1oXrBk,8834
 q3dviewer/utils/range_slider.py,sha256=Cs_xrwt6FCDVxGxan7r-ARd5ySwQ50xnCzcmz0dB_X0,4215
-q3dviewer-1.2.1.dist-info/LICENSE,sha256=81cMOyNfw8KLb1JnPYngGHJ5W83gSbZEBU9MEP3tl-E,1124
-q3dviewer-1.2.1.dist-info/METADATA,sha256=BOYtmAeK-XjasJ_GTJcnJ6q3N2XJQnsfvqeHhm4P8kI,8049
-q3dviewer-1.2.1.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
-q3dviewer-1.2.1.dist-info/entry_points.txt,sha256=EOjker7XYaBk70ffvNB_knPcfA33Bnlg21ZjEeM1EyI,362
-q3dviewer-1.2.1.dist-info/top_level.txt,sha256=HFFDCbGu28txcGe2HPc46A7EPaguBa_b5oH7bufmxHM,10
-q3dviewer-1.2.1.dist-info/RECORD,,
+q3dviewer-1.2.2.dist-info/LICENSE,sha256=81cMOyNfw8KLb1JnPYngGHJ5W83gSbZEBU9MEP3tl-E,1124
+q3dviewer-1.2.2.dist-info/METADATA,sha256=pZQwcgig77nSIonNFabIUAptK75BfyLqrydzgc7FwNg,8049
+q3dviewer-1.2.2.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
+q3dviewer-1.2.2.dist-info/entry_points.txt,sha256=EOjker7XYaBk70ffvNB_knPcfA33Bnlg21ZjEeM1EyI,362
+q3dviewer-1.2.2.dist-info/top_level.txt,sha256=HFFDCbGu28txcGe2HPc46A7EPaguBa_b5oH7bufmxHM,10
+q3dviewer-1.2.2.dist-info/RECORD,,