Scripts:
[blender.git] / release / scripts / raw_import.py
1 #!BPY
2
3 """
4 Name: 'Raw Triangle (.raw)...'
5 Blender: 232
6 Group: 'Import'
7 Tooltip: 'Import Raw Triangle File Format (.raw)'
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 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 # | Released under the Blender Artistic Licence (BAL)       |
42 # | Import Export Suite v0.5                                |
43 # +---------------------------------------------------------+
44 # | Read and write RAW Triangle File Format (*.raw)         |
45 # +---------------------------------------------------------+
46
47 import Blender, mod_meshtools
48 #import time
49
50 # ================================
51 # === Read RAW Triangle Format ===
52 # ================================
53 def read(filename):
54         #start = time.clock()
55         file = open(filename, "rb")
56
57         # Collect data from RAW format
58         faces = []
59         for line in file.readlines():
60                 try:
61                         f1, f2, f3, f4, f5, f6, f7, f8, f9 = map(float, line.split())
62                         faces.append([(f1, f2, f3), (f4, f5, f6), (f7, f8, f9)])
63                 except: # Quad
64                         f1, f2, f3, f4, f5, f6, f7, f8, f9, A, B, C = map(float, line.split())
65                         faces.append([(f1, f2, f3), (f4, f5, f6), (f7, f8, f9), (A, B, C)])
66
67         # Generate verts and faces lists, without duplicates
68         verts = []
69         coords = {}
70         index = 0
71         for i in range(len(faces)):
72                 for j in range(len(faces[i])):
73                         vertex = faces[i][j]
74                         if not coords.has_key(vertex):
75                                 coords[vertex] = index
76                                 index += 1
77                                 verts.append(vertex)
78                         faces[i][j] = coords[vertex]
79
80         objname = Blender.sys.splitext(Blender.sys.basename(filename))[0]
81
82         mod_meshtools.create_mesh(verts, faces, objname)
83         Blender.Window.DrawProgressBar(1.0, '')  # clear progressbar
84         file.close()
85         #end = time.clock()
86         #seconds = " in %.2f %s" % (end-start, "seconds")
87         message = "Successfully imported " + Blender.sys.basename(filename)# + seconds
88         mod_meshtools.print_boxed(message)
89
90 def fs_callback(filename):
91         read(filename)
92
93 Blender.Window.FileSelector(fs_callback, "Import Raw")