-
Notifications
You must be signed in to change notification settings - Fork 49
/
importldraw.py
400 lines (342 loc) · 16.3 KB
/
importldraw.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
# -*- coding: utf-8 -*-
"""Import LDraw GPLv2 license.
This program is free software; you can redistribute it and/or
modify it under the terms of the GNU General Public License
as published by the Free Software Foundation; either version 2
of the License, or (at your option) any later version.
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with this program; if not, write to the Free Software Foundation,
Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
"""
"""
Import LDraw
This file defines the importer for Blender.
It stores and recalls preferences for the importer.
The execute() function kicks off the import process.
The python module loadldraw does the actual work.
"""
import configparser
import os
import bpy
from bpy.props import (StringProperty,
FloatProperty,
EnumProperty,
BoolProperty
)
from bpy_extras.io_utils import ImportHelper
from .loadldraw import loadldraw
"""
Example preferences file:
[DEFAULT]
[importldraw]
ldrawDirectory = ""
realScale = 0.0002
resolution = "Standard"
smoothShading = True
useLook = "normal"
useColourScheme = "lgeo"
gaps = True
realGapWidth = 0.0002
createInstances = True
numberNodes = True
positionObjectOnGroundAtOrigin = True
flattenHierarchy = False
useUnofficialParts = True
useLogoStuds = False
instanceStuds = False
curvedWalls = True
(etc)
"""
class Preferences():
"""Import LDraw - Preferences"""
__sectionName = 'importldraw'
def __init__(self):
self.__ldPath = None
self.__prefsPath = os.path.dirname(__file__)
self.__prefsFilepath = os.path.join(self.__prefsPath, "ImportLDrawPreferences.ini")
self.__config = configparser.RawConfigParser()
self.__prefsRead = self.__config.read(self.__prefsFilepath)
if self.__prefsRead and not self.__config[Preferences.__sectionName]:
self.__prefsRead = False
def get(self, option, default):
if not self.__prefsRead:
return default
if type(default) is bool:
return self.__config.getboolean(Preferences.__sectionName, option, fallback=default)
elif type(default) is float:
return self.__config.getfloat(Preferences.__sectionName, option, fallback=default)
elif type(default) is int:
return self.__config.getint(Preferences.__sectionName, option, fallback=default)
else:
return self.__config.get(Preferences.__sectionName, option, fallback=default)
def set(self, option, value):
if not (Preferences.__sectionName in self.__config):
self.__config[Preferences.__sectionName] = {}
self.__config[Preferences.__sectionName][option] = str(value)
def save(self):
try:
with open(self.__prefsFilepath, 'w') as configfile:
self.__config.write(configfile)
return True
except Exception:
# Fail gracefully
e = sys.exc_info()[0]
debugPrint("WARNING: Could not save preferences. {0}".format(e))
return False
class ImportLDrawOps(bpy.types.Operator, ImportHelper):
"""Import LDraw - Import Operator."""
bl_idname = "import_scene.importldraw"
bl_description = "Import LDraw models (.io/.mpd/.ldr/.l3b/.dat)"
bl_label = "Import LDraw Models"
bl_space_type = "PROPERTIES"
bl_region_type = "WINDOW"
bl_options = {'REGISTER', 'UNDO', 'PRESET'}
# Instance the preferences system
prefs = Preferences()
# File type filter in file browser
filename_ext = ".ldr"
filter_glob: StringProperty(
default="*.io;*.mpd;*.ldr;*.l3b;*.dat",
options={'HIDDEN'}
)
ldrawPath: StringProperty(
name="",
description="Full filepath to the LDraw Parts Library (download from http://www.ldraw.org)",
default=prefs.get("ldrawDirectory", loadldraw.Configure.findDefaultLDrawDirectory())
)
realScale: FloatProperty(
name="Scale",
description="Sets a scale for the model (1.0 = real life scale)",
default=prefs.get("realScale", 1.0)
)
resPrims: EnumProperty(
name="Resolution of part primitives",
description="Resolution of part primitives, ie. how much geometry they have",
default=prefs.get("resolution", "Standard"),
items=(
("Standard", "Standard primitives", "Import using standard resolution primitives."),
("High", "High resolution primitives", "Import using high resolution primitives."),
("Low", "Low resolution primitives", "Import using low resolution primitives.")
)
)
smoothParts: BoolProperty(
name="Smooth faces and edge-split",
description="Smooth faces and add an edge-split modifier",
default=prefs.get("smoothShading", True)
)
look: EnumProperty(
name="Overall Look",
description="Realism or Schematic look",
default=prefs.get("useLook", "normal"),
items=(
("normal", "Realistic Look", "Render to look realistic."),
("instructions", "Lego Instructions Look", "Render to look like the instruction book pictures."),
)
)
colourScheme: EnumProperty(
name="Colour scheme options",
description="Colour scheme options",
default=prefs.get("useColurScheme", "lgeo"),
items=(
("lgeo", "Realistic colours", "Uses the LGEO colour scheme for realistic colours."),
("ldraw", "Original LDraw colours", "Uses the standard LDraw colour scheme. Looks good with the Instructions Look."),
("alt", "Alternate LDraw colours", "Uses the alternate LDraw colour scheme. Looks good with the Instructions Look."),
)
)
addGaps: BoolProperty(
name="Add space between each part:",
description="Add a small space between each part",
default=prefs.get("gaps", False)
)
gapWidthMM: FloatProperty(
name="Space",
description="Amount of space between each part (default 0.2mm)",
default=1000 * prefs.get("realGapWidth", 0.0002)
)
curvedWalls: BoolProperty(
name="Use curved wall normals",
description="Makes surfaces look slightly concave, for interesting reflections",
default=prefs.get("curvedWalls", True)
)
importCameras: BoolProperty(
name="Import cameras",
description="Import camera definitions (from models authored in LeoCAD)",
default=prefs.get("importCameras", True)
)
linkParts: BoolProperty(
name="Link identical parts",
description="Identical parts (of the same type and colour) share the same mesh",
default=prefs.get("linkParts", True)
)
numberNodes: BoolProperty(
name="Number each object",
description="Each object has a five digit prefix eg. 00001_car. This keeps the list in it's proper order",
default=prefs.get("numberNodes", True)
)
positionOnGround: BoolProperty(
name="Put model on ground at origin",
description="The object is centred at the origin, and on the ground plane",
default=prefs.get("positionObjectOnGroundAtOrigin", True)
)
flatten: BoolProperty(
name="Flatten tree",
description="In Scene Outliner, all parts are placed directly below the root - there's no tree of submodels",
default=prefs.get("flattenHierarchy", False)
)
minifigHierarchy: BoolProperty(
name="Parent Minifigs",
description="Add a parent/child hierarchy (tree) for Minifigs",
default=prefs.get("minifigHierarchy", True)
)
useUnofficialParts: BoolProperty(
name="Include unofficial parts",
description="Additionally searches for parts in the <ldraw-dir>/unofficial/ directory",
default=prefs.get("useUnofficialParts", True)
)
useLogoStuds: BoolProperty(
name="Show 'LEGO' logo on studs",
description="Shows the LEGO logo on each stud (at the expense of some extra geometry and import time)",
default=prefs.get("useLogoStuds", False)
)
instanceStuds: BoolProperty(
name="Make individual studs",
description="Creates a Blender Object for each and every stud (WARNING: can be slow to import and edit in Blender if there are lots of studs)",
default=prefs.get("instanceStuds", False)
)
resolveNormals: EnumProperty(
name="Resolve ambiguous normals option",
description="Some older LDraw parts have faces with ambiguous normals, this specifies what do do with them",
default=prefs.get("resolveNormals", "guess"),
items=(
("guess", "Recalculate Normals", "Uses Blender's Recalculate Normals to get a consistent set of normals."),
("double", "Two faces back to back", "Two faces are added with their normals pointing in opposite directions."),
)
)
bevelEdges: BoolProperty(
name="Bevel edges",
description="Adds a Bevel modifier for rounding off sharp edges",
default=prefs.get("bevelEdges", True)
)
bevelWidth: FloatProperty(
name="Bevel Width",
description="Width of the bevelled edges",
default=prefs.get("bevelWidth", 0.5)
)
addEnvironment: BoolProperty(
name="Add Environment",
description="Adds a ground plane and environment texture (for realistic look only)",
default=prefs.get("addEnvironment", True)
)
positionCamera: BoolProperty(
name="Position the camera",
description="Position the camera to show the whole model",
default=prefs.get("positionCamera", True)
)
cameraBorderPercentage: FloatProperty(
name="Camera Border %",
description="When positioning the camera, include a (percentage) border leeway around the model in the rendered image",
default=prefs.get("cameraBorderPercentage", 5.0)
)
def draw(self, context):
"""Display import options."""
layout = self.layout
layout.use_property_split = True # Active single-column layout
box = layout.box()
box.label(text="Import Options", icon='PREFERENCES')
box.label(text="LDraw filepath:", icon='FILEBROWSER')
box.prop(self, "ldrawPath")
box.prop(self, "realScale")
box.prop(self, "look", expand=True)
box.prop(self, "addEnvironment")
box.prop(self, "positionCamera")
box.prop(self, "cameraBorderPercentage")
box.prop(self, "colourScheme", expand=True)
box.prop(self, "resPrims", expand=True)
box.prop(self, "smoothParts")
box.prop(self, "bevelEdges")
box.prop(self, "bevelWidth")
box.prop(self, "addGaps")
box.prop(self, "gapWidthMM")
box.prop(self, "curvedWalls")
box.prop(self, "importCameras")
box.prop(self, "linkParts")
box.prop(self, "useUnofficialParts")
box.prop(self, "useLogoStuds")
box.prop(self, "instanceStuds")
box.prop(self, "positionOnGround")
box.prop(self, "numberNodes")
box.prop(self, "flatten")
box.prop(self, "minifigHierarchy")
box.label(text="Resolve Ambiguous Normals:", icon='ORIENTATION_NORMAL')
box.prop(self, "resolveNormals", expand=True)
def execute(self, context):
"""Start the import process."""
# Read current preferences from the UI and save them
ImportLDrawOps.prefs.set("ldrawDirectory", self.ldrawPath)
ImportLDrawOps.prefs.set("realScale", self.realScale)
ImportLDrawOps.prefs.set("resolution", self.resPrims)
ImportLDrawOps.prefs.set("smoothShading", self.smoothParts)
ImportLDrawOps.prefs.set("bevelEdges", self.bevelEdges)
ImportLDrawOps.prefs.set("bevelWidth", self.bevelWidth)
ImportLDrawOps.prefs.set("useLook", self.look)
ImportLDrawOps.prefs.set("useColourScheme", self.colourScheme)
ImportLDrawOps.prefs.set("gaps", self.addGaps)
ImportLDrawOps.prefs.set("realGapWidth", self.gapWidthMM / 1000)
ImportLDrawOps.prefs.set("curvedWalls", self.curvedWalls)
ImportLDrawOps.prefs.set("importCameras", self.importCameras)
ImportLDrawOps.prefs.set("linkParts", self.linkParts)
ImportLDrawOps.prefs.set("numberNodes", self.numberNodes)
ImportLDrawOps.prefs.set("positionObjectOnGroundAtOrigin", self.positionOnGround)
ImportLDrawOps.prefs.set("flattenHierarchy", self.flatten)
ImportLDrawOps.prefs.set("minifigHierarchy", self.minifigHierarchy)
ImportLDrawOps.prefs.set("useUnofficialParts", self.useUnofficialParts)
ImportLDrawOps.prefs.set("useLogoStuds", self.useLogoStuds)
ImportLDrawOps.prefs.set("instanceStuds", self.instanceStuds)
ImportLDrawOps.prefs.set("resolveNormals", self.resolveNormals)
ImportLDrawOps.prefs.set("addEnvironment", self.addEnvironment)
ImportLDrawOps.prefs.set("positionCamera", self.positionCamera)
ImportLDrawOps.prefs.set("cameraBorderPercentage",self.cameraBorderPercentage)
ImportLDrawOps.prefs.save()
# Set bpy related variables here since it isn't available immediately on Blender startup
loadldraw.hasCollections = hasattr(bpy.data, "collections")
# Set import options and import
loadldraw.Options.ldrawDirectory = self.ldrawPath
loadldraw.Options.realScale = self.realScale
loadldraw.Options.useUnofficialParts = self.useUnofficialParts
loadldraw.Options.resolution = self.resPrims
loadldraw.Options.defaultColour = "4"
loadldraw.Options.createInstances = self.linkParts
loadldraw.Options.instructionsLook = self.look == "instructions"
loadldraw.Options.useColourScheme = self.colourScheme
loadldraw.Options.numberNodes = self.numberNodes
loadldraw.Options.removeDoubles = True
loadldraw.Options.smoothShading = self.smoothParts
loadldraw.Options.edgeSplit = self.smoothParts # Edge split is appropriate only if we are smoothing
loadldraw.Options.gaps = self.addGaps
loadldraw.Options.realGapWidth = self.gapWidthMM / 1000
loadldraw.Options.curvedWalls = self.curvedWalls
loadldraw.Options.importCameras = self.importCameras
loadldraw.Options.positionObjectOnGroundAtOrigin = self.positionOnGround
loadldraw.Options.flattenHierarchy = self.flatten
loadldraw.Options.minifigHierarchy = self.minifigHierarchy
loadldraw.Options.useLogoStuds = self.useLogoStuds
loadldraw.Options.logoStudVersion = "4"
loadldraw.Options.instanceStuds = self.instanceStuds
loadldraw.Options.useLSynthParts = True
loadldraw.Options.LSynthDirectory = os.path.join(os.path.dirname(__file__), "lsynth")
loadldraw.Options.studLogoDirectory = os.path.join(os.path.dirname(__file__), "studs")
loadldraw.Options.resolveAmbiguousNormals = self.resolveNormals
loadldraw.Options.overwriteExistingMaterials = False
loadldraw.Options.overwriteExistingMeshes = False
loadldraw.Options.addBevelModifier = self.bevelEdges and not loadldraw.Options.instructionsLook
loadldraw.Options.bevelWidth = self.bevelWidth
loadldraw.Options.addWorldEnvironmentTexture = self.addEnvironment
loadldraw.Options.addGroundPlane = self.addEnvironment
loadldraw.Options.positionCamera = self.positionCamera
loadldraw.Options.cameraBorderPercent = self.cameraBorderPercentage / 100.0
loadldraw.loadFromFile(self, self.filepath)
return {'FINISHED'}