2.5: Top Menu
[blender.git] / release / scripts / raw_import.py
1 #!BPY
2
3 """
4 Name: 'Raw Faces (.raw)...'
5 Blender: 242
6 Group: 'Import'
7 Tooltip: 'Import Raw Triangle File Format (.raw)'
8 """
9
10 __author__ = "Anthony D'Agostino (Scorpius)"
11 __url__ = ("blender", "blenderartists.org",
12 "Author's homepage, http://www.redrival.com/scorpius")
13 __version__ = "Part of IOSuite 0.5"
14
15 __bpydoc__ = """\
16 This script imports Raw Triangle File format files to Blender.
17
18 The raw triangle format is very simple; it has no verts or faces lists.
19 It's just a simple ascii text file with the vertices of each triangle
20 listed on each line. There were some very old utilities (when the PovRay
21 forum was in existence on CompuServe) that preformed operations on these
22 files.
23
24 Usage:<br>
25         Execute this script from the "File->Import" menu and choose a Raw file to
26 open.
27
28 Notes:<br>
29         Generates the standard verts and faces lists, but without duplicate
30 verts. Only *exact* duplicates are removed, there is no way to specify a
31 tolerance.
32 """
33
34 # $Id$
35 #
36 # +---------------------------------------------------------+
37 # | Copyright (c) 2002 Anthony D'Agostino                   |
38 # | http://www.redrival.com/scorpius                        |
39 # | scorpius@netzero.com                                    |
40 # | April 28, 2002                                          |
41 # | Read and write RAW Triangle File Format (*.raw)         |
42 # +---------------------------------------------------------+
43
44 # ***** BEGIN GPL LICENSE BLOCK *****
45 #
46 # This program is free software; you can redistribute it and/or
47 # modify it under the terms of the GNU General Public License
48 # as published by the Free Software Foundation; either version 2
49 # of the License, or (at your option) any later version.
50 #
51 # This program is distributed in the hope that it will be useful,
52 # but WITHOUT ANY WARRANTY; without even the implied warranty of
53 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
54 # GNU General Public License for more details.
55 #
56 # You should have received a copy of the GNU General Public License
57 # along with this program; if not, write to the Free Software Foundation,
58 # Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
59 #
60 # ***** END GPL LICENCE BLOCK *****
61
62 import Blender
63
64 # ================================
65 # === Read RAW Triangle Format ===
66 # ================================
67 def read(filename):
68         t = Blender.sys.time()
69         file = open(filename, "rb")
70
71         # Collect data from RAW format
72         def line_to_face(line):
73                 # Each triplet is an xyz float
74                 line_split= map(float, line.split())
75                 if len(line_split)==9: # Tri
76                         f1, f2, f3, f4, f5, f6, f7, f8, f9 = line_split
77                         return [(f1, f2, f3), (f4, f5, f6), (f7, f8, f9)]
78                 if len(line_split)==12: # Quad
79                         f1, f2, f3, f4, f5, f6, f7, f8, f9, A, B, C = line_split
80                         return [(f1, f2, f3), (f4, f5, f6), (f7, f8, f9), (A, B, C)]
81         
82         faces = [ line_to_face(line) for line in file.readlines()]
83         file.close()
84         
85         # Generate verts and faces lists, without duplicates
86         verts = []
87         coords = {}
88         index = 0
89         
90         for f in faces:
91                 if f: # Line might be blank
92                         for i, v in enumerate(f):
93                                 try:
94                                         f[i]= coords[v]
95                                 except:
96                                         f[i]= coords[v] = index
97                                         index += 1
98                                         verts.append(v)
99         
100         me= Blender.Mesh.New()
101         me.verts.extend(verts)
102         me.faces.extend(faces)
103         
104         
105         scn= Blender.Scene.GetCurrent()
106         for obj in scn.objects:
107                 obj.sel= 0
108         
109         me.name= Blender.sys.splitext(Blender.sys.basename(filename))[0]
110         ob = scn.objects.new(me)
111         Blender.Redraw()
112         
113         print 'Successfully imported "%s" in %.4f seconds' % (Blender.sys.basename(filename), Blender.sys.time()-t)
114
115
116 def main():
117         Blender.Window.FileSelector(read, 'RAW Import', Blender.sys.makename(ext='.raw'))
118
119 if __name__=='__main__':
120         main()