-
Notifications
You must be signed in to change notification settings - Fork 0
/
hmconvert.py
490 lines (439 loc) · 14.9 KB
/
hmconvert.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
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
472
473
474
475
476
477
478
479
480
481
482
483
484
485
486
487
488
489
490
"""
HyperModelConvert / HyperMC
convert.py
Tool to bulk convert Maya Binary and/or Panda3D egg/bam files.
Offers legacy support for bulk conversion of older bam file versions.
Author: Loonatic
Date: 6/26/20
"""
import subprocess
import os
import time
import sys
import psutil
import argparse
# todo:
# custom maya arg(s) incl. file convert (bam2maya, etc.)
# support for panda args, i.e. with bam2egg -h
# warn the user if the maya2egg_server / client is missing so we dont run into an infinite loop
"""
# Argument Handler #
Example usage: convert.py --egg2maya --mayaver 2016 -r -v
"""
parser = argparse.ArgumentParser()
parser.add_argument(
'--all-phases',
'--all_phases',
action = 'store_true',
help = 'Convert all phase files folders. (3 to 14)'
)
parser.add_argument(
'--selected_phases',
'--phase',
action = 'extend',
nargs = '+',
type = str,
metavar = '3 3.5 4',
help = 'List phase files folders to convert.'
)
parser.add_argument(
'--selected_folders',
'--folder',
'--folders',
action = 'extend',
nargs = '+',
type = str,
metavar = 'models/',
help = 'List of folders with subfolders to convert.'
)
parser.add_argument(
'--rootonly',
'--root',
action = 'store_true',
help = 'Just convert files in the root of the working directory.'
)
parser.add_argument(
'--bindir',
'-bin',
'-b',
action = 'store',
type = str,
help = 'Set folder path to desired Panda3D bin location',
default = "bin/"
)
parser.add_argument(
'--fromfile',
'--file',
'-f',
action = 'store_true',
help = 'Use this flag to read the path from the PANDA_BIN_PATH folder.'
)
## Bam
parser.add_argument(
'--bam2egg',
'--to_egg',
'--to-egg',
action = 'store_true',
help = 'Convert BAM file(s) into EGG file(s).'
)
parser.add_argument(
'--egg2bam',
'--to_bam',
'--to-bam',
action = 'store_true',
help = 'Convert EGG file(s) into BAM file(s).'
)
## Maya
parser.add_argument(
'--egg2maya',
action = 'store_true',
help = 'Convert EGG file(s) into Maya Binary files.'
)
parser.add_argument(
'--egg2maya_legacy',
action = 'store_true',
help = 'Convert EGG file(s) into Maya Binary files. [LEGACY]'
)
parser.add_argument(
'--maya2egg',
action = 'store_true',
help = 'Convert Maya Binary file(s) into EGG files.'
)
parser.add_argument(
'--maya2egg_legacy',
action = 'store_true',
help = 'Convert Maya Binary files into EGG file(s). [LEGACY]'
)
parser.add_argument(
'--mayaver',
'--mayaversion',
'-mv',
action = 'store',
nargs = '?',
type = str,
default = '2016',
metavar = 'MayaVersion',
help = 'Use specific maya version. (Default is 2016)'
)
## Obj
parser.add_argument(
'--obj2egg',
action = 'store_true',
help = 'Convert OBJ files into EGG files.'
)
parser.add_argument(
'--egg2obj',
action = 'store_true',
help = 'Convert EGG files into OBJ files.'
)
## Fbx
parser.add_argument(
'--fbx2egg',
action = 'store_true',
help = 'Convert FBX files into EGG files.'
)
parser.add_argument(
'--egg2fbx',
action = 'store_true',
help = 'Convert EGG files into FBX files.'
)
## Misc
parser.add_argument(
'--verbose',
'-v',
action = 'store_true',
help = 'Enable verbose output.'
)
parser.add_argument(
'--overwrite',
'-o',
action = 'store_true',
help = 'Overwrite preexisting files.'
)
parser.add_argument(
'--recursive',
'-r',
action = 'store_true',
help = 'Convert all folders in the directory, recursively.'
' Typically used if there are models outside of "phase" folders.'
)
# Deprecating this feature. Go get your own Panda!
# parser.add_argument('--legacy', '--use-legacy', type=str, choices=['panda105', 'panda150', 'panda162', 'panda172',
# 'panda181'], action='store', help='Use Panda3D 1.0.5 or Panda3D 1.5.0 instead to convert LEGACY bams.')
# parser.add_argument('--panda_args', '--pargs', action='extend', nargs='+', type=str, help='Optional Panda3D args to
# pass. To get a full list, run `convert.py --pargs help`.')
# Note: Removing pargs cause I dunno how to make this pretty at the moment
args = parser.parse_args()
### End of argparse configuring ###
# If the user gave us particular folders, let's not consider them to be phase files.
selectedFolders = args.selected_folders
# Config #
# mayaArgs = "-a -m" ?
allFiles = []
if not selectedFolders:
if args.all_phases:
args.selected_phases = ['3', '3.5', '4', '5', '5.5', '6', '7', '8', '9', '10', '11', '12', '13', '14']
else:
args.selected_phases = None
args.all_phases = None
# Tool settings
settings = None # [ inputFile, outputFile, tool ]
## Bam
bam2egg = ['.bam', '.egg', 'bam2egg.exe']
egg2bam = ['.egg', '.bam', 'egg2bam.exe']
## Maya
egg2maya_legacy = ['.egg', '.mb', 'egg2maya%s.exe' % args.mayaver]
egg2maya = ['.egg', '.mb', 'egg2maya_client.exe']
maya2egg_legacy = ['.mb', '.egg', 'maya2egg%s.exe' % args.mayaver]
maya2egg = ['.mb', '.egg', 'maya2egg_client.exe']
## Obj
obj2egg = ['.obj', '.egg', 'obj2egg.exe']
egg2obj = ['.egg', '.obj', 'egg2obj.exe']
## Fbx
fbx2egg = ['.fbx', '.egg', 'fbx2egg.exe']
egg2fbx = ['.egg', '.fbx', 'egg2fbx.exe']
# for now, egg is only accepted. we can allow for bams later.
if args.bam2egg:
settings = bam2egg
elif args.egg2bam:
settings = egg2bam
elif args.egg2maya_legacy:
settings = egg2maya_legacy
elif args.egg2maya:
settings = egg2maya
elif args.maya2egg_legacy:
settings = maya2egg_legacy
elif args.maya2egg:
settings = maya2egg
elif args.obj2egg: # obj->egg
settings = obj2egg
elif args.egg2obj: # egg->obj
settings = egg2obj
elif args.fbx2egg: # fbx->egg
settings = fbx2egg
elif args.egg2fbx: # egg->fbx
settings = egg2obj
# Optional args (unfinished)
# optionalArgs = []
# if args.panda_args is not None:
# args.panda_args = [" -" + arg for arg in args.panda_args]
# Grr this is so hacky.
# optionalArgs = (args.panda_args)
####
# Overwrite arg
overwriteArg = []
if args.overwrite:
overwriteArg.append('-o')
####
if args.fromfile:
if not os.path.isfile("PANDA_BIN_PATH"):
print(
"Error: Cannot read from the PANDA_BIN_PATH file."
" Please make sure the file exists in this directory and try again."
)
sys.exit()
path = open("PANDA_BIN_PATH", 'r').readline()
path = path.replace("\\", "/")
defaultBin = path
else:
defaultBin = args.bindir
# Make sure a conversion method was properly inputted
if settings is None:
# Bug: causes the script to immaturely crash if you do not pass in any arguments
# if args.panda_args is not None:
# print("#######################################################################################################")
# subprocess.run(['%s/%s' % (defaultBin, 'egg2bam'), '-h'])
# print("#######################################################################################################")
# subprocess.run(['%s/%s' % (defaultBin, 'bam2egg'), '-h'])
# print("#######################################################################################################")
parser.print_help()
sys.exit()
inputFile, outputFile, tool = settings
verbose = args.verbose
recursive = args.recursive
selectedPhases = args.selected_phases
rootOnly = args.rootonly
if (not recursive) and (selectedPhases or selectedFolders):
recursive = True # we're gonna do recursion on the phases anyway
if rootOnly and (selectedPhases or selectedFolders):
print("Error: You can't specify rootOnly and selected phases or folders.")
sys.exit()
# Let's list the folders we're gonna iterate through if we wanna be verbose.
if verbose and selectedPhases:
print(selectedPhases)
elif verbose and selectedFolders:
print(selectedFolders)
# Maya configuration
global maya_mode
maya_mode = args.egg2maya or args.maya2egg
# We globalize maya_legacy to rollback if maya server fails.
# We probably don't *Need* to do this though..?
global maya_legacy
maya_legacy = args.egg2maya_legacy or args.maya2egg_legacy
# Not defined in the checkMayaServer function to ensure it gets properly cleaned up after the conversion.
global mayaProcess
mayaProcess = tool[0:8]
global mayaServer
mayaServer = str(mayaProcess + args.mayaver + "_bin.exe") # maya2egg20xx_bin.exe
####
def checkMayaServer():
DETACHED_PROCESS = 0x00000008 # A tad hacky, but we need this for the Maya service.
global mayaProcess
global maya_legacy
global mayaServer
if not mayaServer in (p.name() for p in psutil.process_iter()):
if verbose:
print("Attempting to start the Maya server...")
mayaServerLocation = str('%s/%s%s' % (defaultBin, mayaProcess, args.mayaver)) + ".exe"
if verbose:
print("Checking for " + mayaServerLocation)
if not os.path.exists(mayaServerLocation):
print("ERROR: Can't find the Maya server process!")
sys.exit()
# I would rather *not* kill the script and revert to maya legacy instead, but for now this'll do.
# print("ERROR: Can't find the Maya server process! Automatically reverting to legacy mode...")
# maya_legacy = True
# WIP: will this work? If it doesn't, then we will unfortunately skip the first model entered.
# tool = str(mayaProcess + args.mayaver)
return
subprocess.Popen(
[mayaServerLocation, "-server"], creationflags = DETACHED_PROCESS
)
if verbose:
print("Sleeping for 5 seconds...")
time.sleep(5) # give it some time
checkMayaServer()
else:
return # It's running, no problem.
def convertPhases(phases):
global maya_legacy
global maya_mode
if recursive: # Recursion time!
for phase in phases:
if not os.path.exists('phase_%s' % phase):
continue
for root, _, files in os.walk('phase_%s' % phase):
for file in files:
if not file.endswith(inputFile): # Input file
if verbose:
print("Skipping %s" % file)
continue
if verbose:
print("Adding %s" % file)
file = os.path.join(root, file)
allFiles.append(file)
else:
for file in os.listdir('.'):
if not file.endswith(inputFile):
if verbose:
print("Skipping %s" % file)
continue
if verbose:
print("Adding in %s" % file)
allFiles.append(file)
for file in allFiles:
newFile = file.replace(inputFile, outputFile)
if os.path.exists(newFile) and not args.overwrite:
print('Warning: %s already exists' % newFile)
continue
if verbose:
print("Converting %s..." % file)
if maya_mode and not maya_legacy:
if verbose:
print("Checking to see if we have a Maya server up and running...")
checkMayaServer() # Check & run for the Maya server.
# 'bin/panda105' / 'bam2egg[.exe]' optionalArgs file.bam overwriteArg newFile.egg
subprocess.run(['%s/%s' % (defaultBin, tool), file] + overwriteArg + [newFile])
def convertFolders(folders):
global maya_mode
global maya_legacy
if recursive: # Recursion time!
for folder in folders:
if not os.path.exists(folder):
continue
for root, _, files in os.walk(folder):
for file in files:
if not file.endswith(inputFile): # Input file
if verbose:
print("Skipping %s" % file)
continue
if verbose:
print("Adding %s" % file)
file = os.path.join(root, file)
allFiles.append(file)
else:
for file in os.listdir('.'):
if not file.endswith(inputFile):
if verbose:
print("Skipping %s" % file)
continue
if verbose:
print("Adding in %s" % file)
allFiles.append(file)
for file in allFiles:
newFile = file.replace(inputFile, outputFile)
if os.path.exists(newFile) and not args.overwrite:
print('Warning: %s already exists' % newFile)
continue
if verbose:
print("Converting %s..." % file)
if maya_mode and not maya_legacy:
checkMayaServer() # Check & run for the Maya server.
# 'bin/panda105' / 'bam2egg[.exe]' optionalArgs file.bam overwriteArg newFile.egg
subprocess.run(['%s/%s' % (defaultBin, tool), file] + overwriteArg + [newFile])
def convertRoot():
global maya_mode
global maya_legacy
if recursive: # Recursion time!
for root, _, files in os.walk('.'):
for file in files:
if not file.endswith(inputFile): # Input file
if verbose:
print("Skipping %s" % file)
continue
if verbose:
print("Adding %s" % file)
file = os.path.join(root, file)
allFiles.append(file)
else:
for file in os.listdir('.'):
if not file.endswith(inputFile):
if verbose:
print("Skipping %s" % file)
continue
if verbose:
print("Adding in %s" % file)
allFiles.append(file)
for file in allFiles:
newFile = file.replace(inputFile, outputFile)
if os.path.exists(newFile) and not args.overwrite:
print('Warning: %s already exists' % newFile)
continue
if verbose:
print("Converting %s..." % file)
if maya_mode and not maya_legacy:
checkMayaServer() # Check & run for the Maya server.
# 'bin/panda105' / 'bam2egg[.exe]' optionalArgs file.bam overwriteArg newFile.egg
subprocess.run(['%s/%s' % (defaultBin, tool), file] + overwriteArg + [newFile])
# Startup #
# Uses milliseconds for now.
start = int(round(time.time() * 1000))
# Which operation are we gonna run?
if selectedPhases:
convertPhases(selectedPhases)
elif selectedFolders:
convertFolders(selectedFolders)
elif rootOnly:
convertRoot()
else:
# Uhm, user should not get here. Probably a good idea to yell at 'em for invalid arguments.
print("Error: You need to include either the selectedPhases or selectedFolders arg, but not both!")
# We can safely call sys.exit() as we never called the maya server to init.
sys.exit()
# Cleanup #
# One more thing, let's clean up the maya server.
if maya_mode and not maya_legacy:
for p in psutil.process_iter():
if p.name() is mayaServer:
p.kill()
print("Conversion complete. Total time elapsed: %d ms" % (int(round(time.time() * 1000)) - start))