Refactor HelixViewer extend format support

Author: emako

QuickLook.Plugin/QuickLook.Plugin.HelixViewer/Handler.cs

@@ -15,8 +15,10 @@
 // You should have received a copy of the GNU General Public License
 // along with this program.  If not, see <http://www.gnu.org/licenses/>.
 
+using System;
 using System.IO;
 using System.Linq;
+using System.Text;
 
 namespace QuickLook.Plugin.HelixViewer;
 
@@ -26,6 +28,7 @@ public static bool CanHandle(string path)
     {
         var ext = Path.GetExtension(path).ToLower();
 
+        // Simple solution to doubts
         if (ext == ".obj")
         {
             var firstLines = File.ReadLines(path).Take(10);
@@ -38,9 +41,27 @@ public static bool CanHandle(string path)
                 }
             }
         }
+#if S_DXF
+        else if (ext == ".dxf")
+        {
+            using var s = new FileStream(path, FileMode.Open, FileAccess.Read, FileShare.ReadWrite);
+            const int bufferLength = 16 * 1024;
+            var buffer = new byte[bufferLength];
+            int size = s.Read(buffer, 0, buffer.Length);
+
+            for (int i = 0; i < size - 1; i++)
+            {
+                if (buffer[i] == (byte)'3' && buffer[i + 1] == (byte)'D')
+                {
+                    return true;
+                }
+            }
+        }
+#endif
         else
         {
-            return true; // Assume other formats are supported
+            // Assume other formats are supported
+            return true;
         }
 
         return false;

QuickLook.Plugin/QuickLook.Plugin.HelixViewer/HelixPanel.Import.cs

@@ -0,0 +1,104 @@
+// Copyright © 2017-2025 QL-Win Contributors
+//
+// This file is part of QuickLook program.
+//
+// This program is free software: you can redistribute it and/or modify
+// it under the terms of the GNU General Public License as published by
+// the Free Software Foundation, either version 3 of the License, or
+// (at your option) any later version.
+//
+// This program 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
+// GNU General Public License for more details.
+//
+// You should have received a copy of the GNU General Public License
+// along with this program.  If not, see <http://www.gnu.org/licenses/>.
+
+using Assimp;
+using HelixToolkit.Wpf;
+using System;
+using System.IO;
+using System.Linq;
+using System.Windows.Media;
+using System.Windows.Media.Media3D;
+
+namespace QuickLook.Plugin.HelixViewer;
+
+public partial class HelixPanel
+{
+    private void Load()
+    {
+        var importerType = Importer.GetImporterType(_path);
+
+        try
+        {
+            if (importerType == ImporterType.Extended)
+            {
+                var context = new AssimpContext();
+                var scene = context.ImportFile(_path, PostProcessSteps.Triangulate);
+
+                foreach (var mesh in scene.Meshes)
+                {
+                    var geometry = new MeshGeometry3D()
+                    {
+                        Positions = [.. mesh.Vertices.Select(v => new Point3D(v.X, v.Y, v.Z))],
+                        TriangleIndices = [.. mesh.GetIndices()],
+                    };
+                    var model = new GeometryModel3D()
+                    {
+                        Geometry = geometry,
+                        Material = Materials.Gray,
+                    };
+
+                    modelVisual.Content = model;
+                }
+            }
+            else
+            {
+                var modelImporter = new ModelImporter();
+                var model3DGroup = modelImporter.Load(_path);
+                var diffuseMaterial = new DiffuseMaterial(new SolidColorBrush(Color.FromRgb(0xA0, 0xA0, 0xA0)));
+
+                foreach (GeometryModel3D child in model3DGroup.Children.Cast<GeometryModel3D>())
+                {
+                    child.Material = diffuseMaterial;
+                    child.BackMaterial = diffuseMaterial;
+                }
+
+                modelVisual.Content = model3DGroup;
+            }
+        }
+        catch (Exception ex)
+        {
+            errorInfo.Text = $"[{nameof(ImporterType)}.{importerType}] {ex}";
+            errorInfo.Visibility = System.Windows.Visibility.Visible;
+            viewer.Visibility = System.Windows.Visibility.Collapsed;
+        }
+    }
+}
+
+file static class Importer
+{
+    public static ImporterType GetImporterType(string path)
+    {
+        if (string.IsNullOrEmpty(path))
+            return ImporterType.Unknown;
+
+        return Path.GetExtension(path).ToLower() switch
+        {
+            ".stl" or ".obj" or ".3ds" or ".lwo" or ".ply" => ImporterType.Default,
+            ".fbx" or ".3mf" or ".glb" or ".gltf" or ".dae" or ".dxf" => ImporterType.Extended,
+            ".pmx" => ImporterType.Extended_MMD,
+            _ => ImporterType.Unknown,
+        };
+    }
+}
+
+file enum ImporterType
+{
+    Unknown,
+    Default,
+    Extended,
+    Extended_MMD,
+}

QuickLook.Plugin/QuickLook.Plugin.HelixViewer/HelixPanel.xaml

@@ -9,6 +9,12 @@
              d:DesignWidth="800"
              mc:Ignorable="d">
     <Grid>
+        <TextBlock x:Name="errorInfo"
+                   HorizontalAlignment="Left"
+                   VerticalAlignment="Center"
+                   FontSize="12"
+                   TextWrapping="Wrap"
+                   Visibility="Collapsed" />
         <helix:HelixViewport3D x:Name="viewer"
                                CameraRotationMode="Turnball"
                                CoordinateSystemHorizontalPosition="Left"

QuickLook.Plugin/QuickLook.Plugin.HelixViewer/HelixPanel.xaml.cs

@@ -15,13 +15,7 @@
 // You should have received a copy of the GNU General Public License
 // along with this program.  If not, see <http://www.gnu.org/licenses/>.
 
-using Assimp;
-using HelixToolkit.Wpf;
-using System;
-using System.Linq;
 using System.Windows.Controls;
-using System.Windows.Media;
-using System.Windows.Media.Media3D;
 
 namespace QuickLook.Plugin.HelixViewer;
 
@@ -39,43 +33,4 @@ public HelixPanel()
     {
         InitializeComponent();
     }
-
-    private void Load()
-    {
-        if (_path.EndsWith(".fbx", StringComparison.OrdinalIgnoreCase))
-        {
-            var context = new AssimpContext();
-            var scene = context.ImportFile(_path, PostProcessSteps.Triangulate);
-
-            foreach (var mesh in scene.Meshes)
-            {
-                var geometry = new MeshGeometry3D()
-                {
-                    Positions = [.. mesh.Vertices.Select(v => new Point3D(v.X, v.Y, v.Z))],
-                    TriangleIndices = [.. mesh.GetIndices()],
-                };
-                var model = new GeometryModel3D()
-                {
-                    Geometry = geometry,
-                    Material = Materials.Gray,
-                };
-
-                modelVisual.Content = model;
-            }
-        }
-        else
-        {
-            var modelImporter = new ModelImporter();
-            var model3DGroup = modelImporter.Load(_path);
-            var diffuseMaterial = new DiffuseMaterial(new SolidColorBrush(Color.FromRgb(0xA0, 0xA0, 0xA0)));
-
-            foreach (GeometryModel3D child in model3DGroup.Children.Cast<GeometryModel3D>())
-            {
-                child.Material = diffuseMaterial;
-                child.BackMaterial = diffuseMaterial;
-            }
-
-            modelVisual.Content = model3DGroup;
-        }
-    }
 }

QuickLook.Plugin/QuickLook.Plugin.HelixViewer/Plugin.cs

@@ -19,21 +19,34 @@
 using System;
 using System.Collections.Generic;
 using System.IO;
+using System.Linq;
 using System.Windows;
 
 namespace QuickLook.Plugin.HelixViewer;
 
 public class Plugin : IViewer
 {
+    /// <summary>
+    /// <seealso cref="ImporterType"/>
+    /// </summary>
     private static readonly HashSet<string> WellKnownExtensions = new(
     [
+        // Default
         ".stl", ".obj", ".3ds", ".lwo", ".ply",
-        ".fbx",
+
+        // Extended
+        ".fbx", ".3mf", ".blend", ".glb", ".gltf", ".dae",
+#if S_DXF
+        ".dxf",
+#endif
+
+        // Extended_MMD
+        ".pmx",
     ]);
 
     private HelixPanel _hp;
 
-    public int Priority => 0;
+    public int Priority => -5;
 
     public void Init()
     {
@@ -42,7 +55,7 @@ public void Init()
     public bool CanHandle(string path)
     {
         return !Directory.Exists(path)
-            && WellKnownExtensions.Contains(Path.GetExtension(path.ToLower()))
+            && WellKnownExtensions.Any(ext => path.EndsWith(ext, StringComparison.OrdinalIgnoreCase))
             && Handler.CanHandle(path);
     }