Add Graphviz test files (#1368)

lutzroeder · Sep 29, 2024 · d168752 · d168752
1 parent dadc151
commit d168752
Show file tree

Hide file tree

Showing 3 changed files with 61 additions and 0 deletions.
diff --git a/source/graphviz.js b/source/graphviz.js
@@ -0,0 +1,36 @@
+
+const graphviz = {};
+
+graphviz.ModelFactory = class {
+
+    match(context) {
+        const reader = context.read('text', 0x10000);
+        if (reader) {
+            try {
+                const line = reader.read('\n');
+                if (line === undefined) {
+                    return;
+                }
+                if (line.indexOf('digraph') !== -1) {
+                    context.type = 'graphviz.dot';
+                }
+            } catch {
+                // continue regardless of error
+            }
+        }
+    }
+
+    async open(/* context */) {
+        throw new graphviz.Error('Invalid file content. File contains Graphviz data.');
+    }
+};
+
+graphviz.Error = class extends Error {
+
+    constructor(message) {
+        super(message);
+        this.name = 'Error loading Graphviz model.';
+    }
+};
+
+export const ModelFactory = graphviz.ModelFactory;
diff --git a/source/view.js b/source/view.js
@@ -5778,6 +5778,7 @@ view.ModelFactoryService = class {
         this.register('./safetensors', ['.safetensors', '.safetensors.index.json']);
         this.register('./tvm', ['.json', '.params']);
         this.register('./modular', ['.maxviz']);
+        this.register('./graphviz', ['.dot']);
         this.register('./catboost', ['.cbm']);
         this.register('./weka', ['.model']);
         this.register('./qnn', ['.json', '.bin', '.serialized']);

diff --git a/test/models.json b/test/models.json
@@ -2088,6 +2088,30 @@
     "format":   "Flux",
     "link":     "https://github.com/lutzroeder/netron/issues/334"
   },
+  {
+    "type":     "graphviz",
+    "target":   "EfficientDet-d0.onnx.dot",
+    "source":   "https://github.com/user-attachments/files/17180084/EfficientDet-d0.onnx.dot.zip[EfficientDet-d0.onnx.dot]",
+    "format":   "Graphviz",
+    "error":    "Invalid file content. File contains Graphviz data.",
+    "link":     "https://github.com/lutzroeder/netron/issues/1368"
+  },
+  {
+    "type":     "graphviz",
+    "target":   "gpt2-10.onnx.dot",
+    "source":   "https://github.com/user-attachments/files/17180085/gpt2-10.onnx.dot.zip[gpt2-10.onnx.dot]",
+    "format":   "Graphviz",
+    "error":    "Invalid file content. File contains Graphviz data.",
+    "link":     "https://github.com/lutzroeder/netron/issues/1368"
+  },
+  {
+    "type":     "graphviz",
+    "target":   "resnet18.dot",
+    "source":   "https://github.com/user-attachments/files/17175694/resnet18.dot.zip[resnet18.dot]",
+    "format":   "Graphviz",
+    "error":    "Invalid file content. File contains Graphviz data.",
+    "link":     "https://github.com/lutzroeder/netron/issues/1368"
+  },
   {
     "type":     "gguf",
     "target":   "mixtral-8x7b-v0.1.Q4_K_M.gguf",