1 files changed, 14 insertions, 2 deletions
diff --git a/stl.py b/stl.py
index 076c739..7989d43 100644
--- a/stl.py
+++ b/stl.py
@@ -1,7 +1,19 @@
 import numpy as np
+import string
 import struct
 
-def pull_vertices(filename):
+def read_stl(filename):
+    f = open(filename)
+    buf = f.read(200)
+    f.close()
+
+    for char in buf:
+        if char not in string.printable:
+            return read_binary_stl(filename)
+
+    return read_ascii_stl(filename)
+
+def read_ascii_stl(filename):
     f = open(filename)
 
     vertex = []
@@ -13,7 +25,7 @@ def pull_vertices(filename):
     f.close()
     return np.array(vertex)
 
-def pull_vertices_binary(filename):
+def read_binary_stl(filename):
     f = open(filename)
 
     f.read(80)