Model-Rando/Randomize.py at main · KH2FM-Mods-Num/Model-Rando · GitHub

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
OnPC = True
PCBlacklist = {'Roxas/Chain/P_EX100_NM',
               'Roxas/Chain/P_EX100_NM_BTLF',
               'Roxas/Chain/P_EX100_NM_MAGF',
               'Roxas/Chain/P_EX100_NM_ULTF',
               'Roxas/Coat/P_EX100_NM_KH1F',
               'Shananas/B_EX150',
               'Dee-Ayy/B_EX420',}
PCRemaster = ('P_EX020_CM','P_EX020_DC','P_EX030_CM','P_EX030_DC',
              'W_EX010'   ,'W_EX010_NM'   ,'W_EX010_TR'   ,
              'W_EX010_00','W_EX010_NM_00','W_EX010_TR_00',
              'W_EX010_X0','W_EX010_NM_X0','W_EX010_TR_X0',
              'W_EX010_Y0','W_EX010_NM_Y0','W_EX010_TR_Y0',
              'W_EX010_Z0','W_EX010_NM_Z0','W_EX010_TR_Z0',)
#Useless, figure out what to do later

import sys
import os
import yaml
import random

#Functions to write stuff
def writemodel(old,new):
    old = 'obj/' + old
    new = 'obj/' + new
    #A.FM
    if not OnPC: #Don't edit on PC due to a.us pax being remastered-reliant
        if os.path.isfile(new+'.a.fm') and 'ROXAS' not in old: #Stolen Keyblades' PAX are different
            f.write('- name: '+old+'.a.fm\n')
            f.write('  method: copy\n')
            f.write('  source:\n')
            f.write('  - name: '+new+'.a.fm\n')
        elif os.path.isfile(new+'.imd') or os.path.isfile(new+'.sqd'):
            f.write('- name: '+old+'.a.fm\n')
            f.write('  method: binarc\n')
            f.write('  source:\n')
            if os.path.isfile(new+'.imd'):
                f.write('  - name: face\n')
                f.write('    type: imgd\n')
                f.write('    method: copy\n')
                f.write('    source:\n')
                f.write('    - name: '+new+'.imd\n')
            if os.path.isfile(new+'.sqd'):
                f.write('  - name: face\n')
                f.write('    type: seqd\n')
                f.write('    method: copy\n')
                f.write('    source:\n')
                f.write('    - name: '+new+'.sqd\n')
    #MDLX
    if os.path.isfile(new+'.model') or os.path.isfile(new+'.tim'):
        f.write('- name: '+old+'.mdlx\n')
        f.write('  method: binarc\n')
        f.write('  source:\n')
        if os.path.isfile(new+'.model'):
            subfile = old[4:8].lower()
            if '_PLAYER' in old:
                subfile = 'p_ex'
            f.write('  - name: '+subfile+'\n')
            f.write('    type: model\n')
            f.write('    method: copy\n')
            f.write('    source:\n')
            f.write('    - name: '+new+'.model\n')
        if os.path.isfile(new+'.tim'):
            f.write('  - name: tim_\n')
            f.write('    type: modeltexture\n')
            f.write('    method: copy\n')
            f.write('    source:\n')
            f.write('    - name: '+new+'.tim\n')
    '''This entire code block doesn't work without bfmm update
    #Write useless file to prevent using HD textures (mess up the file count)
    if OnPC and old[4:] in PCRemaster:
        f.write('- name: remastered/'+old+'.mdlx/-69.dds\n')
        f.write('  method: copy\n')
        f.write('  source:\n')
        f.write('  - name: randomize.py\n')'''

#Get KH2 model filenames
currentDir = sys.argv[0].replace((sys.argv[0].split('\\')[-1]),'')
objs = yaml.safe_load(open(currentDir+'modellist.yml'))

#Write the mod.yml
f = open(currentDir+'mod.yml','w',encoding='utf-8')
f.write('description: Credits for PandaPyre, Shananas, GeminiHero, Napstio, DA, Oathseeker, Zurph, and FrederykStrife for providing their respective textures.\n')
f.write('assets:\n')
for objtype in objs:
    obj = objs[objtype]
    for models in obj.values():
        Base          = models['Base']
        ExtraDomain   = models['ExtraDomain']
        ExtraCodomain = models['ExtraCodomain']
        #Remove blacklisted models
        if OnPC:
            Base          = [x for x in Base          if not x in PCBlacklist]
            ExtraDomain   = [x for x in ExtraDomain   if not x in PCBlacklist]
            ExtraCodomain = [x for x in ExtraCodomain if not x in PCBlacklist]
        #Get eligible elements
        oldmodels = Base + ExtraDomain
        newmodels = Base + ExtraCodomain
        while len(oldmodels) >= len(newmodels):
            newmodels += (Base + ExtraCodomain)
        random.shuffle(newmodels)
        for i in range(len(oldmodels)):
            if oldmodels[i] == newmodels[i]:
                continue
            writemodel(oldmodels[i],newmodels[i])

f.close()