Scripts:
[blender.git] / release / scripts / slp_import.py
1 #!BPY
2
3 """
4 Name: 'Pro Engineer (.slp)...'
5 Blender: 232
6 Group: 'Import'
7 Tooltip: 'Import Pro Engineer (.slp) File Format'
8 """
9
10 __author__ = "Anthony D'Agostino (Scorpius)"
11 __url__ = ("blender", "elysiun",
12 "Author's homepage, http://www.redrival.com/scorpius")
13 __version__ = "Part of IOSuite 0.5"
14
15 __bpydoc__ = """\
16 This script imports Pro Engineer files to Blender.
17
18 This format can be exported from Pro/Engineer and most other CAD
19 applications. Written at the request of a Blender user. It is almost
20 identical to RAW format.
21
22 Usage:<br>
23         Execute this script from the "File->Import" menu and choose an SLP file to
24 open.
25
26 Notes:<br>
27         Generates the standard verts and faces lists, but without duplicate
28 verts. Only *exact* duplicates are removed, there is no way to specify a
29 tolerance.
30 """
31
32 # $Id$
33 #
34 # +---------------------------------------------------------+
35 # | Copyright (c) 2004 Anthony D'Agostino                   |
36 # | http://www.redrival.com/scorpius                        |
37 # | scorpius@netzero.com                                    |
38 # | May 3, 2004                                             |
39 # | Released under the Blender Artistic Licence (BAL)       |
40 # | Import Export Suite v0.5                                |
41 # +---------------------------------------------------------+
42 # | Read and write SLP Triangle File Format (*.slp)         |
43 # +---------------------------------------------------------+
44
45 import Blender, mod_meshtools
46 #import time
47
48 # ================================
49 # === Read SLP Triangle Format ===
50 # ================================
51 def read(filename):
52         #start = time.clock()
53         file = open(filename, "rb")
54
55         raw = []
56         for line in file.readlines():
57                 data = line.split()
58                 if data[0] == "vertex":
59                         vert = map(float, data[1:])
60                         raw.append(vert)
61
62         tri = []
63         for i in range(0, len(raw), 3):
64                 tri.append(raw[i] + raw[i+1] + raw[i+2])
65
66         #$import pprint; pprint.pprint(tri)
67
68         # Collect data from RAW format
69         faces = []
70         for line in tri:
71                 f1, f2, f3, f4, f5, f6, f7, f8, f9 = line
72                 faces.append([(f1, f2, f3), (f4, f5, f6), (f7, f8, f9)])
73
74         # Generate verts and faces lists, without duplicates
75         verts = []
76         coords = {}
77         index = 0
78         for i in range(len(faces)):
79                 for j in range(len(faces[i])):
80                         vertex = faces[i][j]
81                         if not coords.has_key(vertex):
82                                 coords[vertex] = index
83                                 index += 1
84                                 verts.append(vertex)
85                         faces[i][j] = coords[vertex]
86
87         objname = Blender.sys.splitext(Blender.sys.basename(filename))[0]
88
89         mod_meshtools.create_mesh(verts, faces, objname)
90         Blender.Window.DrawProgressBar(1.0, '')  # clear progressbar
91         file.close()
92         #end = time.clock()
93         #seconds = " in %.2f %s" % (end-start, "seconds")
94         message = "Successfully imported " + Blender.sys.basename(filename)# + seconds
95         mod_meshtools.print_boxed(message)
96
97 def fs_callback(filename):
98         read(filename)
99
100 Blender.Window.FileSelector(fs_callback, "Import SLP")