forked from theonlydude/RandomMetroidSolver
-
Notifications
You must be signed in to change notification settings - Fork 0
/
randomizer.py
executable file
·921 lines (868 loc) · 47.9 KB
/
randomizer.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
491
492
493
494
495
496
497
498
499
500
501
502
503
504
505
506
507
508
509
510
511
512
513
514
515
516
517
518
519
520
521
522
523
524
525
526
527
528
529
530
531
532
533
534
535
536
537
538
539
540
541
542
543
544
545
546
547
548
549
550
551
552
553
554
555
556
557
558
559
560
561
562
563
564
565
566
567
568
569
570
571
572
573
574
575
576
577
578
579
580
581
582
583
584
585
586
587
588
589
590
591
592
593
594
595
596
597
598
599
600
601
602
603
604
605
606
607
608
609
610
611
612
613
614
615
616
617
618
619
620
621
622
623
624
625
626
627
628
629
630
631
632
633
634
635
636
637
638
639
640
641
642
643
644
645
646
647
648
649
650
651
652
653
654
655
656
657
658
659
660
661
662
663
664
665
666
667
668
669
670
671
672
673
674
675
676
677
678
679
680
681
682
683
684
685
686
687
688
689
690
691
692
693
694
695
696
697
698
699
700
701
702
703
704
705
706
707
708
709
710
711
712
713
714
715
716
717
718
719
720
721
722
723
724
725
726
727
728
729
730
731
732
733
734
735
736
737
738
739
740
741
742
743
744
745
746
747
748
749
750
751
752
753
754
755
756
757
758
759
760
761
762
763
764
765
766
767
768
769
770
771
772
773
774
775
776
777
778
779
780
781
782
783
784
785
786
787
788
789
790
791
792
793
794
795
796
797
798
799
800
801
802
803
804
805
806
807
808
809
810
811
812
813
814
815
816
817
818
819
820
821
822
823
824
825
826
827
828
829
830
831
832
833
834
835
836
837
838
839
840
841
842
843
844
845
846
847
848
849
850
851
852
853
854
855
856
857
858
859
860
861
862
863
864
865
866
867
868
869
870
871
872
873
874
875
876
877
878
879
880
881
882
883
884
885
886
887
888
889
890
891
892
893
894
895
896
897
898
899
900
901
902
903
904
905
906
907
908
909
910
911
912
913
914
915
916
917
918
919
920
921
#!/usr/bin/python3
import argparse, os.path, json, sys, shutil, random
from rando.RandoSettings import RandoSettings, GraphSettings
from rando.RandoExec import RandoExec
from rom.PaletteRando import PaletteRando
from graph.graph_utils import vanillaTransitions, vanillaBossesTransitions, GraphUtils, getAccessPoint
from utils.parameters import Knows, easy, medium, hard, harder, hardcore, mania, infinity, text2diff, diff2text, appDir
from rom.rom_patches import RomPatches
from rom.rompatcher import RomPatcher, MusicPatcher, RomTypeForMusic
from rom.rom import FakeROM
from utils.utils import PresetLoader, loadRandoPreset, getDefaultMultiValues, getPresetDir
from utils.version import displayedVersion
from logic.smbool import SMBool
from utils.doorsmanager import DoorsManager
from logic.logic import Logic
from utils.objectives import Objectives
import utils.log
import utils.db as db
# we need to know the logic before doing anything else
def getLogic():
# check if --logic is there
logic = 'vanilla'
for i, param in enumerate(sys.argv):
if param == '--logic' and i+1 < len(sys.argv):
logic = sys.argv[i+1]
return logic
Logic.factory(getLogic())
defaultMultiValues = getDefaultMultiValues()
speeds = defaultMultiValues['progressionSpeed']
energyQties = defaultMultiValues['energyQty']
progDiffs = defaultMultiValues['progressionDifficulty']
morphPlacements = defaultMultiValues['morphPlacement']
majorsSplits = defaultMultiValues['majorsSplit']
gravityBehaviours = defaultMultiValues['gravityBehaviour']
objectives = defaultMultiValues['objective']
def randomMulti(args, param, defaultMultiValues):
value = args[param]
isRandom = False
if value == "random":
isRandom = True
if args[param+"List"] != None:
# use provided list
choices = args[param+"List"].split(',')
value = random.choice(choices)
else:
# use default list
value = random.choice(defaultMultiValues)
return (isRandom, value)
def dumpErrorMsg(outFileName, msg):
print("DIAG: " + msg)
if outFileName is not None:
with open(outFileName, 'w') as jsonFile:
json.dump({"errorMsg": msg}, jsonFile)
def dumpErrorMsgs(outFileName, msgs):
dumpErrorMsg(outFileName, joinErrorMsgs(msgs))
def joinErrorMsgs(msgs):
return '\n'.join(msgs)
def restricted_float(x):
x = float(x)
if x < 0.0 or x > 9.0:
raise argparse.ArgumentTypeError("%r not in range [1.0, 9.0]"%(x,))
return x
if __name__ == "__main__":
parser = argparse.ArgumentParser(description="Random Metroid Randomizer")
parser.add_argument('--patchOnly',
help="only apply patches, do not perform any randomization", action='store_true',
dest='patchOnly', default=False)
parser.add_argument('--param', '-p', help="the input parameters",
default=None, dest='paramsFileName')
parser.add_argument('--dir',
help="output directory for ROM and dot files",
dest='directory', nargs='?', default='.')
parser.add_argument('--dot',
help="generate dot file with area graph",
action='store_true',dest='dot', default=False)
parser.add_argument('--area', help="area mode",
dest='area', nargs='?', const=True, default=False)
parser.add_argument('--areaLayoutBase',
help="use simple layout patch for area mode", action='store_true',
dest='areaLayoutBase', default=False)
parser.add_argument('--lightArea', help="keep number of transitions between vanilla areas", action='store_true',
dest='lightArea', default=False)
parser.add_argument('--escapeRando',
help="Randomize the escape sequence",
dest='escapeRando', nargs='?', const=True, default=False)
parser.add_argument('--noRemoveEscapeEnemies',
help="Do not remove enemies during escape sequence", action='store_true',
dest='noRemoveEscapeEnemies', default=False)
parser.add_argument('--bosses', help="randomize bosses",
dest='bosses', nargs='?', const=True, default=False)
parser.add_argument('--minimizer', help="minimizer mode: area and boss mixed together. arg is number of non boss locations",
dest='minimizerN', nargs='?', const=35, default=None,
choices=[str(i) for i in range(30,101)]+["random"])
parser.add_argument('--startLocation', help="Name of the Access Point to start from",
dest='startLocation', nargs='?', default="Landing Site",
choices=['random'] + GraphUtils.getStartAccessPointNames())
parser.add_argument('--startLocationList', help="list to choose from when random",
dest='startLocationList', nargs='?', default=None)
parser.add_argument('--debug', '-d', help="activate debug logging", dest='debug',
action='store_true')
parser.add_argument('--maxDifficulty', '-t',
help="the maximum difficulty generated seed will be for given parameters",
dest='maxDifficulty', nargs='?', default=None,
choices=['easy', 'medium', 'hard', 'harder', 'hardcore', 'mania', 'random'])
parser.add_argument('--minDifficulty',
help="the minimum difficulty generated seed will be for given parameters (speedrun prog speed required)",
dest='minDifficulty', nargs='?', default=None,
choices=['easy', 'medium', 'hard', 'harder', 'hardcore', 'mania'])
parser.add_argument('--seed', '-s', help="randomization seed to use", dest='seed',
nargs='?', default=0, type=int)
parser.add_argument('--rom', '-r',
help="the vanilla ROM",
dest='rom', nargs='?', default=None)
parser.add_argument('--output',
help="to choose the name of the generated json (for the webservice)",
dest='output', nargs='?', default=None)
parser.add_argument('--preset',
help="the name of the preset (for the webservice)",
dest='preset', nargs='?', default=None)
parser.add_argument('--patch', '-c',
help="optional patches to add",
dest='patches', nargs='?', default=[], action='append',
choices=['itemsounds.ips', 'random_music.ips',
'fast_doors.ips', 'elevators_speed.ips', 'elevators_doors_speed.ips',
'spinjumprestart.ips', 'rando_speed.ips', 'No_Music', 'AimAnyButton.ips',
'max_ammo_display.ips', 'supermetroid_msu1.ips', 'Infinite_Space_Jump',
'refill_before_save.ips', 'remove_elevators_speed.ips',
'remove_fast_doors.ips', 'remove_Infinite_Space_Jump.ips',
'remove_rando_speed.ips', 'remove_spinjumprestart.ips',
'remove_itemsounds.ips', 'vanilla_music.ips', 'custom_ship.ips',
'Ship_Takeoff_Disable_Hide_Samus', 'widescreen.ips',
'hell.ips', 'lava_acid_physics.ips'])
parser.add_argument('--missileQty', '-m',
help="quantity of missiles",
dest='missileQty', nargs='?', default=3,
type=restricted_float)
parser.add_argument('--superQty', '-q',
help="quantity of super missiles",
dest='superQty', nargs='?', default=2,
type=restricted_float)
parser.add_argument('--powerBombQty', '-w',
help="quantity of power bombs",
dest='powerBombQty', nargs='?', default=1,
type=restricted_float)
parser.add_argument('--minorQty', '-n',
help="quantity of minors",
dest='minorQty', nargs='?', default=100,
choices=[str(i) for i in range(0,101)])
parser.add_argument('--energyQty', '-g',
help="quantity of ETanks/Reserve Tanks",
dest='energyQty', nargs='?', default='vanilla',
choices=energyQties + ['random'])
parser.add_argument('--energyQtyList', help="list to choose from when random",
dest='energyQtyList', nargs='?', default=None)
parser.add_argument('--strictMinors',
help="minors quantities values will be strictly followed instead of being probabilities",
dest='strictMinors', nargs='?', const=True, default=False)
parser.add_argument('--majorsSplit',
help="how to split majors/minors: Full, FullWithHUD, Major, Chozo, Scavenger",
dest='majorsSplit', nargs='?', choices=majorsSplits + ['random'], default='Full')
parser.add_argument('--majorsSplitList', help="list to choose from when random",
dest='majorsSplitList', nargs='?', default=None)
parser.add_argument('--scavNumLocs',
help="For Scavenger split, number of major locations in the mandatory route",
dest='scavNumLocs', nargs='?', default=10,
choices=["0"]+[str(i) for i in range(4,18)])
parser.add_argument('--scavRandomized',
help="For Scavenger split, decide whether mandatory major locs will have non-vanilla items",
dest='scavRandomized', nargs='?', const=True, default=False)
parser.add_argument('--suitsRestriction',
help="no suits in early game",
dest='suitsRestriction', nargs='?', const=True, default=False)
parser.add_argument('--morphPlacement',
help="morph placement",
dest='morphPlacement', nargs='?', default='early',
choices=morphPlacements + ['random'])
parser.add_argument('--morphPlacementList', help="list to choose from when random",
dest='morphPlacementList', nargs='?', default=None)
parser.add_argument('--hideItems', help="Like in dessy's rando hide half of the items",
dest="hideItems", nargs='?', const=True, default=False)
parser.add_argument('--progressionSpeed', '-i',
help="progression speed, from " + str(speeds) + ". 'random' picks a random speed from these. Pick a random speed from a subset using comma-separated values, like 'slow,medium,fast'.",
dest='progressionSpeed', nargs='?', default='medium', choices=speeds+['random'])
parser.add_argument('--progressionSpeedList', help="list to choose from when random",
dest='progressionSpeedList', nargs='?', default=None)
parser.add_argument('--progressionDifficulty',
help="",
dest='progressionDifficulty', nargs='?', default='normal',
choices=progDiffs + ['random'])
parser.add_argument('--progressionDifficultyList', help="list to choose from when random",
dest='progressionDifficultyList', nargs='?', default=None)
parser.add_argument('--superFun',
help="randomly remove major items from the pool for maximum enjoyment",
dest='superFun', nargs='?', default=[], action='append',
choices=['Movement', 'Combat', 'Suits', 'MovementRandom', 'CombatRandom', 'SuitsRandom'])
parser.add_argument('--animals',
help="randomly change the save the animals room",
dest='animals', action='store_true', default=False)
parser.add_argument('--nolayout',
help="do not include total randomizer layout patches",
dest='noLayout', action='store_true', default=False)
parser.add_argument('--gravityBehaviour',
help="varia/gravity suits behaviour",
dest='gravityBehaviour', nargs='?', default='Balanced', choices=gravityBehaviours+['random'])
parser.add_argument('--gravityBehaviourList', help="list to choose from when random",
dest='gravityBehaviourList', nargs='?', default=None)
parser.add_argument('--nerfedCharge',
help="apply nerfed charge patch",
dest='nerfedCharge', action='store_true', default=False)
parser.add_argument('--novariatweaks',
help="do not include VARIA randomizer tweaks",
dest='noVariaTweaks', action='store_true', default=False)
parser.add_argument('--controls',
help="specify controls, comma-separated, in that order: Shoot,Jump,Dash,ItemSelect,ItemCancel,AngleUp,AngleDown. Possible values: A,B,X,Y,L,R,Select,None",
dest='controls')
parser.add_argument('--moonwalk',
help="Enables moonwalk by default",
dest='moonWalk', action='store_true', default=False)
parser.add_argument('--runtime',
help="Maximum runtime limit in seconds. If 0 or negative, no runtime limit. Default is 30.",
dest='runtimeLimit_s', nargs='?', default=30, type=int)
parser.add_argument('--race', help="Race mode magic number", dest='raceMagic',
type=int)
parser.add_argument('--vcr', help="Generate VCR output file", dest='vcr', action='store_true')
parser.add_argument('--palette', help="Randomize the palettes", dest='palette', action='store_true')
parser.add_argument('--individual_suit_shift', help="palette param", action='store_true',
dest='individual_suit_shift', default=False)
parser.add_argument('--individual_tileset_shift', help="palette param", action='store_true',
dest='individual_tileset_shift', default=False)
parser.add_argument('--no_match_ship_and_power', help="palette param", action='store_false',
dest='match_ship_and_power', default=True)
parser.add_argument('--seperate_enemy_palette_groups', help="palette param", action='store_true',
dest='seperate_enemy_palette_groups', default=False)
parser.add_argument('--no_match_room_shift_with_boss', help="palette param", action='store_false',
dest='match_room_shift_with_boss', default=True)
parser.add_argument('--no_shift_tileset_palette', help="palette param", action='store_false',
dest='shift_tileset_palette', default=True)
parser.add_argument('--no_shift_boss_palettes', help="palette param", action='store_false',
dest='shift_boss_palettes', default=True)
parser.add_argument('--no_shift_suit_palettes', help="palette param", action='store_false',
dest='shift_suit_palettes', default=True)
parser.add_argument('--no_shift_enemy_palettes', help="palette param", action='store_false',
dest='shift_enemy_palettes', default=True)
parser.add_argument('--no_shift_beam_palettes', help="palette param", action='store_false',
dest='shift_beam_palettes', default=True)
parser.add_argument('--no_shift_ship_palette', help="palette param", action='store_false',
dest='shift_ship_palette', default=True)
parser.add_argument('--min_degree', help="min hue shift", dest='min_degree', nargs='?', default=-180, type=int)
parser.add_argument('--max_degree', help="max hue shift", dest='max_degree', nargs='?', default=180, type=int)
parser.add_argument('--no_global_shift', help="", action='store_false', dest='global_shift', default=True)
parser.add_argument('--invert', help="invert color range", dest='invert', action='store_true', default=False)
parser.add_argument('--no_blue_door_palette', help="palette param", action='store_true',
dest='no_blue_door_palette', default=False)
parser.add_argument('--ext_stats', help="dump extended stats SQL", nargs='?', default=None, dest='extStatsFilename')
parser.add_argument('--randoPreset', help="rando preset file", dest="randoPreset", nargs='?', default=None)
parser.add_argument('--fakeRandoPreset', help="for prog speed stats", dest="fakeRandoPreset", nargs='?', default=None)
parser.add_argument('--plandoRando', help="json string with already placed items/locs", dest="plandoRando",
nargs='?', default=None)
parser.add_argument('--sprite', help='use a custom sprite for Samus', dest='sprite', default=None)
parser.add_argument('--no_spin_attack', help='when using a custom sprite, use the same animation for screw attack with or without Space Jump', dest='noSpinAttack', action='store_true', default=False)
parser.add_argument('--customItemNames', help='add custom item names for some of them, related to the custom sprite',
dest='customItemNames', action='store_true', default=False)
parser.add_argument('--ship', help='use a custom sprite for Samus ship', dest='ship', default=None)
parser.add_argument('--seedIps', help='ips generated from previous seed', dest='seedIps', default=None)
parser.add_argument('--jm,', help="display data used by jm for its stats", dest='jm', action='store_true', default=False)
parser.add_argument('--doorsColorsRando', help='randomize color of colored doors', dest='doorsColorsRando',
nargs='?', const=True, default=False)
parser.add_argument('--allowGreyDoors', help='add grey color in doors colors pool', dest='allowGreyDoors',
nargs='?', const=True, default=False)
parser.add_argument('--logic', help='logic to use', dest='logic', nargs='?', default="varia", choices=["varia", "rotation"])
parser.add_argument('--hud', help='Enable VARIA hud', dest='hud',
nargs='?', const=True, default=False)
parser.add_argument('--music',
help="JSON file for music replacement mapping",
dest='music', nargs='?', default=None)
parser.add_argument('--objective',
help="objectives to open G4",
dest='objective', nargs='?', default=[], action='append',
choices=Objectives.getAllGoals()+["random"]+[str(i) for i in range(6)])
parser.add_argument('--objectiveList', help="list to choose from when random",
dest='objectiveList', nargs='?', default=None)
parser.add_argument('--tourian', help="Tourian mode",
dest='tourian', nargs='?', default='Vanilla',
choices=['Vanilla', 'Fast', 'Disabled'])
parser.add_argument('--hellrun', help="Hellrun damage rate in %, between 0 and 400 (default 100)",
dest='hellrunRate', default=100, type=int)
parser.add_argument('--etanks', help="Additional ETanks, between 0 (default) and 18",
dest='additionalEtanks', default=0, type=int)
# parse args
args = parser.parse_args()
if args.output is None and args.rom is None:
print("Need --output or --rom parameter")
sys.exit(-1)
elif args.output is not None and args.rom is not None:
print("Can't have both --output and --rom parameters")
sys.exit(-1)
if args.plandoRando != None and args.output == None:
print("plandoRando param requires output param")
sys.exit(-1)
if args.additionalEtanks < 0 or args.additionalEtanks > 18:
print("additionalEtanks must be between 0 and 18")
sys.exit(-1)
if args.hellrunRate < 0 or args.hellrunRate > 400:
print("hellrunRate must be between 0 and 400")
sys.exit(-1)
utils.log.init(args.debug)
logger = utils.log.get('Rando')
Logic.factory(args.logic)
# service to force an argument value and notify it
argDict = vars(args)
forcedArgs = {}
optErrMsgs = [ ]
def forceArg(arg, value, msg, altValue=None, webArg=None, webValue=None):
okValues = [value]
if altValue is not None:
okValues.append(altValue)
if argDict[arg] not in okValues:
argDict[arg] = value
forcedArgs[webArg if webArg != None else arg] = webValue if webValue != None else value
print(msg)
optErrMsgs.append(msg)
# if rando preset given, load it first
if args.randoPreset != None:
preset = loadRandoPreset(args.randoPreset, args)
# use the skill preset from the rando preset
if preset is not None and args.paramsFileName is None:
args.paramsFileName = '{}/{}/{}.json'.format(appDir, getPresetDir(preset), preset)
# if diff preset given, load it
if args.paramsFileName is not None:
PresetLoader.factory(args.paramsFileName).load()
preset = os.path.splitext(os.path.basename(args.paramsFileName))[0]
if args.preset is not None:
preset = args.preset
else:
preset = 'default'
logger.debug("preset: {}".format(preset))
# if no seed given, choose one
if args.seed == 0:
seed = random.randrange(sys.maxsize)
else:
seed = args.seed
logger.debug("seed: {}".format(seed))
seed4rand = seed
if args.raceMagic is not None:
seed4rand = seed ^ args.raceMagic
random.seed(seed4rand)
# if no max diff, set it very high
if args.maxDifficulty:
if args.maxDifficulty == 'random':
diffs = ['easy', 'medium', 'hard', 'harder', 'hardcore', 'mania']
maxDifficulty = text2diff[random.choice(diffs)]
else:
maxDifficulty = text2diff[args.maxDifficulty]
else:
maxDifficulty = infinity
# same as solver, increase max difficulty
threshold = maxDifficulty
epsilon = 0.001
if maxDifficulty <= easy:
threshold = medium - epsilon
elif maxDifficulty <= medium:
threshold = hard - epsilon
elif maxDifficulty <= hard:
threshold = harder - epsilon
elif maxDifficulty <= harder:
threshold = hardcore - epsilon
elif maxDifficulty <= hardcore:
threshold = mania - epsilon
maxDifficulty = threshold
logger.debug("maxDifficulty: {}".format(maxDifficulty))
# handle random parameters with dynamic pool of values
(_, progSpeed) = randomMulti(args.__dict__, "progressionSpeed", speeds)
(_, progDiff) = randomMulti(args.__dict__, "progressionDifficulty", progDiffs)
(majorsSplitRandom, args.majorsSplit) = randomMulti(args.__dict__, "majorsSplit", majorsSplits)
(_, gravityBehaviour) = randomMulti(args.__dict__, "gravityBehaviour", gravityBehaviours)
if args.minDifficulty:
minDifficulty = text2diff[args.minDifficulty]
if progSpeed != "speedrun":
optErrMsgs.append("Minimum difficulty setting ignored, as prog speed is not speedrun")
else:
minDifficulty = 0
if args.area == True and args.bosses == True and args.minimizerN is not None:
if args.minimizerN == "random":
minimizerN = random.randint(30, 60)
logger.debug("minimizerN: {}".format(minimizerN))
else:
minimizerN = int(args.minimizerN)
if minimizerN < 100:
forceArg('majorsSplit', 'Full', "'Majors Split' forced to Full. Use 100 locations on your minimizer to use a non-Full split.", altValue='FullWithHUD')
else:
minimizerN = None
areaRandom = False
if args.area == 'random':
areaRandom = True
args.area = bool(random.getrandbits(1))
logger.debug("area: {}".format(args.area))
doorsColorsRandom = False
if args.doorsColorsRando == 'random':
doorsColorsRandom = True
args.doorsColorsRando = bool(random.getrandbits(1))
logger.debug("doorsColorsRando: {}".format(args.doorsColorsRando))
bossesRandom = False
if args.bosses == 'random':
bossesRandom = True
args.bosses = bool(random.getrandbits(1))
logger.debug("bosses: {}".format(args.bosses))
if args.escapeRando == 'random':
args.escapeRando = bool(random.getrandbits(1))
logger.debug("escapeRando: {}".format(args.escapeRando))
if args.suitsRestriction != False and minimizerN is not None:
forceArg('suitsRestriction', False, "'Suits restriction' forced to off", webValue='off')
if args.suitsRestriction == 'random':
if args.morphPlacement == 'late' and args.area == True:
forceArg('suitsRestriction', False, "'Suits restriction' forced to off", webValue='off')
else:
args.suitsRestriction = bool(random.getrandbits(1))
logger.debug("suitsRestriction: {}".format(args.suitsRestriction))
if args.hideItems == 'random':
args.hideItems = bool(random.getrandbits(1))
if args.morphPlacement == 'random':
if args.morphPlacementList != None:
morphPlacements = args.morphPlacementList.split(',')
args.morphPlacement = random.choice(morphPlacements)
# Scavenger Hunt constraints
if args.majorsSplit == 'Scavenger':
forceArg('progressionSpeed', 'speedrun', "'Progression speed' forced to speedrun")
progSpeed = "speedrun"
forceArg('hud', True, "'VARIA HUD' forced to on", webValue='on')
if not GraphUtils.isStandardStart(args.startLocation):
forceArg('startLocation', "Landing Site", "Start Location forced to Landing Site because of Scavenger mode")
if args.morphPlacement == 'late':
forceArg('morphPlacement', 'normal', "'Morph Placement' forced to normal instead of late")
# use escape rando for auto escape trigger
if args.tourian == 'Disabled':
forceArg('escapeRando', True, "'Escape randomization' forced to on", webValue='on')
forceArg('noRemoveEscapeEnemies', True, "Enemies enabled during escape sequence", webArg='removeEscapeEnemies', webValue='off')
# random fill makes certain options unavailable
if (progSpeed == 'speedrun' or progSpeed == 'basic') and args.majorsSplit != 'Scavenger':
forceArg('progressionDifficulty', 'normal', "'Progression difficulty' forced to normal")
progDiff = args.progressionDifficulty
logger.debug("progressionDifficulty: {}".format(progDiff))
if args.strictMinors == 'random':
args.strictMinors = bool(random.getrandbits(1))
# in plando rando we know that the start ap is ok
if not GraphUtils.isStandardStart(args.startLocation) and args.plandoRando is None:
if args.majorsSplit in ['Major', "Chozo"]:
forceArg('hud', True, "'VARIA HUD' forced to on", webValue='on')
forceArg('noVariaTweaks', False, "'VARIA tweaks' forced to on", webValue='on')
forceArg('noLayout', False, "'Anti-softlock layout patches' forced to on", webValue='on')
forceArg('suitsRestriction', False, "'Suits restriction' forced to off", webValue='off')
forceArg('areaLayoutBase', False, "'Additional layout patches for easier navigation' forced to on", webValue='on')
possibleStartAPs, reasons = GraphUtils.getPossibleStartAPs(args.area, maxDifficulty, args.morphPlacement)
if args.startLocation == 'random':
if args.startLocationList != None:
# to be able to give the list in jm we had to replace ' ' with '_', do the opposite operation
startLocationList = args.startLocationList.replace('_', ' ')
startLocationList = startLocationList.split(',')
# intersection between user whishes and reality
possibleStartAPs = sorted(list(set(possibleStartAPs).intersection(set(startLocationList))))
if len(possibleStartAPs) == 0:
optErrMsgs += ["%s : %s" % (apName, cause) for apName, cause in reasons.items() if apName in startLocationList]
optErrMsgs.append('Invalid start locations list with your settings.')
dumpErrorMsgs(args.output, optErrMsgs)
sys.exit(-1)
args.startLocation = random.choice(possibleStartAPs)
elif args.startLocation not in possibleStartAPs:
optErrMsgs.append('Invalid start location: {}. {}'.format(args.startLocation, reasons[args.startLocation]))
optErrMsgs.append('Possible start locations with these settings: {}'.format(possibleStartAPs))
dumpErrorMsgs(args.output, optErrMsgs)
sys.exit(-1)
ap = getAccessPoint(args.startLocation)
if 'forcedEarlyMorph' in ap.Start and ap.Start['forcedEarlyMorph'] == True:
forceArg('morphPlacement', 'early', "'Morph Placement' forced to early for custom start location")
else:
if progSpeed == 'speedrun':
if args.morphPlacement == 'late':
forceArg('morphPlacement', 'normal', "'Morph Placement' forced to normal instead of late")
elif (not GraphUtils.isStandardStart(args.startLocation)) and args.morphPlacement != 'normal':
forceArg('morphPlacement', 'normal', "'Morph Placement' forced to normal for custom start location")
if args.majorsSplit == 'Chozo' and args.morphPlacement == "late":
forceArg('morphPlacement', 'normal', "'Morph Placement' forced to normal for Chozo")
if args.patchOnly == False:
print("SEED: " + str(seed))
objectivesManager = Objectives(args.tourian != 'Disabled')
addedObjectives = 0
if args.majorsSplit == "Scavenger":
objectivesManager.setScavengerHunt()
addedObjectives = 1
if args.objective:
maxActiveGoals = Objectives.maxActiveGoals - addedObjectives
try:
nbObjectives = int(args.objective[0])
except:
nbObjectives = 0 if "random" in args.objective else None
if nbObjectives is not None:
availableObjectives = args.objectiveList.replace('_', ' ').split(',') if args.objectiveList is not None else objectives
if nbObjectives > 0:
nbObjectives = min(nbObjectives, maxActiveGoals, len(availableObjectives))
else:
nbObjectives = random.randint(1, min(maxActiveGoals, len(availableObjectives)))
objectivesManager.setRandom(nbObjectives, availableObjectives)
else:
if len(args.objective) > maxActiveGoals:
args.objective = args.objective[0:maxActiveGoals]
for goal in args.objective:
objectivesManager.addGoal(goal)
objectivesManager.expandGoals()
else:
objectivesManager.setVanilla()
if any(goal for goal in Objectives.activeGoals if goal.area is not None):
forceArg('hud', True, "'VARIA HUD' forced to on", webValue='on')
# fill restrictions dict
restrictions = { 'Suits' : args.suitsRestriction, 'Morph' : args.morphPlacement, "doors": "normal" if not args.doorsColorsRando else "late" }
restrictions['MajorMinor'] = 'Full' if args.majorsSplit == 'FullWithHUD' else args.majorsSplit
if restrictions["MajorMinor"] == "Scavenger":
scavNumLocs = int(args.scavNumLocs)
if scavNumLocs == 0:
scavNumLocs = random.randint(4,16)
restrictions["ScavengerParams"] = {'numLocs':scavNumLocs, 'vanillaItems':not args.scavRandomized}
restrictions["EscapeTrigger"] = args.tourian == 'Disabled'
seedCode = 'X'
if majorsSplitRandom == False:
if restrictions['MajorMinor'] == 'Full':
seedCode = 'FX'
elif restrictions['MajorMinor'] == 'Chozo':
seedCode = 'ZX'
elif restrictions['MajorMinor'] == 'Major':
seedCode = 'MX'
elif restrictions['MajorMinor'] == 'Scavenger':
seedCode = 'SX'
if args.bosses == True and bossesRandom == False:
seedCode = 'B'+seedCode
if args.doorsColorsRando == True and doorsColorsRandom == False:
seedCode = 'D'+seedCode
if args.area == True and areaRandom == False:
seedCode = 'A'+seedCode
# output ROM name
if args.patchOnly == False:
fileName = 'VARIA_Randomizer_' + seedCode + str(seed) + '_' + preset
if args.progressionSpeed != "random":
fileName += "_" + args.progressionSpeed
else:
fileName = 'VARIA' # TODO : find better way to name the file (argument?)
seedName = fileName
if args.directory != '.':
fileName = args.directory + '/' + fileName
if args.noLayout == True:
RomPatches.ActivePatches = RomPatches.TotalBase
else:
RomPatches.ActivePatches = RomPatches.Total
RomPatches.ActivePatches.remove(RomPatches.BlueBrinstarBlueDoor)
RomPatches.ActivePatches += GraphUtils.getGraphPatches(args.startLocation)
if gravityBehaviour != "Balanced":
RomPatches.ActivePatches.remove(RomPatches.NoGravityEnvProtection)
if gravityBehaviour == "Progressive":
RomPatches.ActivePatches.append(RomPatches.ProgressiveSuits)
if args.nerfedCharge == True:
RomPatches.ActivePatches.append(RomPatches.NerfedCharge)
if args.noVariaTweaks == False:
RomPatches.ActivePatches += RomPatches.VariaTweaks
if minimizerN is not None:
RomPatches.ActivePatches.append(RomPatches.NoGadoras)
if args.tourian == 'Fast':
RomPatches.ActivePatches += RomPatches.MinimizerTourian
elif args.tourian == 'Disabled':
RomPatches.ActivePatches.append(RomPatches.NoTourian)
missileQty = float(args.missileQty)
superQty = float(args.superQty)
powerBombQty = float(args.powerBombQty)
minorQty = int(args.minorQty)
energyQty = args.energyQty
if missileQty < 1:
missileQty = random.randint(1, 9)
if superQty < 1:
superQty = random.randint(1, 9)
if powerBombQty < 1:
powerBombQty = random.randint(1, 9)
if minorQty < 1:
minorQty = random.randint(25, 100)
if energyQty == 'random':
if args.energyQtyList != None:
# with jm can't have a list with space in it
energyQtyList = args.energyQtyList.replace('_', ' ')
energyQties = energyQtyList.split(',')
energyQty = random.choice(energyQties)
if energyQty == 'ultra sparse':
# add nerfed rainbow beam patch
RomPatches.ActivePatches.append(RomPatches.NerfedRainbowBeam)
qty = {'energy': energyQty,
'minors': minorQty,
'ammo': { 'Missile': missileQty,
'Super': superQty,
'PowerBomb': powerBombQty },
'strictMinors' : args.strictMinors }
logger.debug("quantities: {}".format(qty))
if len(args.superFun) > 0:
superFun = []
for fun in args.superFun:
if fun.find('Random') != -1:
if bool(random.getrandbits(1)) == True:
superFun.append(fun[0:fun.find('Random')])
else:
superFun.append(fun)
args.superFun = superFun
logger.debug("superFun: {}".format(args.superFun))
ctrlDict = None
if args.controls:
ctrlList = args.controls.split(',')
if len(ctrlList) != 7:
raise ValueError("Invalid control list size")
ctrlKeys = ["Shot", "Jump", "Dash", "ItemSelect", "ItemCancel", "AngleUp", "AngleDown"]
ctrlDict = {}
i = 0
for k in ctrlKeys:
b = ctrlList[i]
if b in RomPatcher.buttons:
ctrlDict[k] = b
i += 1
else:
raise ValueError("Invalid button name : " + str(b))
plandoSettings = None
if args.plandoRando is not None:
forceArg('progressionSpeed', 'speedrun', "'Progression Speed' forced to speedrun")
progSpeed = 'speedrun'
forceArg('majorsSplit', 'Full', "'Majors Split' forced to Full")
forceArg('morphPlacement', 'normal', "'Morph Placement' forced to normal")
forceArg('progressionDifficulty', 'normal', "'Progression difficulty' forced to normal")
progDiff = 'normal'
args.plandoRando = json.loads(args.plandoRando)
RomPatches.ActivePatches = args.plandoRando["patches"]
DoorsManager.unserialize(args.plandoRando["doors"])
plandoSettings = {"locsItems": args.plandoRando['locsItems'], "forbiddenItems": args.plandoRando['forbiddenItems']}
randoSettings = RandoSettings(maxDifficulty, progSpeed, progDiff, qty,
restrictions, args.superFun, args.runtimeLimit_s,
plandoSettings, minDifficulty)
# print some parameters for jm's stats
if args.jm == True:
print("startLocation:{}".format(args.startLocation))
print("progressionSpeed:{}".format(progSpeed))
print("majorsSplit:{}".format(args.majorsSplit))
print("morphPlacement:{}".format(args.morphPlacement))
print("gravity:{}".format(gravityBehaviour))
print("maxDifficulty:{}".format(maxDifficulty))
dotFile = None
if args.area == True:
if args.dot == True:
dotFile = args.directory + '/' + seedName + '.dot'
RomPatches.ActivePatches += RomPatches.AreaBaseSet
if args.areaLayoutBase == False:
RomPatches.ActivePatches += RomPatches.AreaComfortSet
if args.doorsColorsRando == True:
RomPatches.ActivePatches.append(RomPatches.RedDoorsMissileOnly)
graphSettings = GraphSettings(args.startLocation, args.area, args.lightArea, args.bosses,
args.escapeRando, minimizerN, dotFile, args.doorsColorsRando, args.allowGreyDoors,
args.plandoRando["transitions"] if args.plandoRando != None else None)
if args.plandoRando is None:
DoorsManager.setDoorsColor()
if not args.patchOnly:
try:
randoExec = RandoExec(seedName, args.vcr, randoSettings, graphSettings)
(stuck, itemLocs, progItemLocs) = randoExec.randomize()
# if we couldn't find an area layout then the escape graph is not created either
# and getDoorConnections will crash if random escape is activated.
if not stuck or args.vcr == True:
doors = GraphUtils.getDoorConnections(randoExec.areaGraph,
args.area, args.bosses,
args.escapeRando if not stuck else False)
escapeAttr = randoExec.areaGraph.EscapeAttributes if args.escapeRando else None
if escapeAttr is not None:
if "patches" not in escapeAttr:
escapeAttr['patches'] = []
if args.noRemoveEscapeEnemies == True:
escapeAttr['patches'].append("Escape_Rando_Enable_Enemies")
if args.tourian == 'Disabled':
escapeAttr['patches'].append('Escape_Trigger')
if args.majorsSplit == 'Scavenger' and any(il for il in progItemLocs if il.Location.Name == "Ridley"):
args.patches.append("Blinking[RidleyRoomIn]")
except Exception as e:
import traceback
traceback.print_exc(file=sys.stdout)
dumpErrorMsg(args.output, "Error: {}".format(e))
sys.exit(-1)
else:
stuck = False
itemLocs = []
progItemLocs = None
if stuck == True:
dumpErrorMsg(args.output, randoExec.errorMsg)
print("Can't generate " + fileName + " with the given parameters: {}".format(randoExec.errorMsg))
# in vcr mode we still want the seed to be generated to analyze it
if args.vcr == False:
sys.exit(-1)
if args.patchOnly == False:
randoExec.postProcessItemLocs(itemLocs, args.hideItems)
# choose on animal patch
if args.animals == True:
animalsPatches = ['animal_enemies.ips', 'animals.ips', 'draygonimals.ips', 'escapimals.ips',
'gameend.ips', 'grey_door_animals.ips', 'low_timer.ips', 'metalimals.ips',
'phantoonimals.ips', 'ridleyimals.ips']
if args.escapeRando == False:
args.patches.append(random.choice(animalsPatches))
args.patches.append("Escape_Animals_Change_Event")
else:
optErrMsgs.append("Ignored animals surprise because of escape randomization")
# transform itemLocs in our usual dict(location, item), exclude minors, we'll get them with the solver
locsItems = {}
for itemLoc in itemLocs:
locName = itemLoc.Location.Name
itemType = itemLoc.Item.Type
if itemType in ['Missile', 'Super', 'PowerBomb']:
continue
locsItems[locName] = itemType
if args.debug == True:
for loc in sorted(locsItems.keys()):
print('{:>50}: {:>16} '.format(loc, locsItems[loc]))
if args.plandoRando != None:
with open(args.output, 'w') as jsonFile:
json.dump({"itemLocs": [il.json() for il in itemLocs], "errorMsg": randoExec.errorMsg}, jsonFile)
sys.exit(0)
# generate extended stats
if args.extStatsFilename != None:
with open(args.extStatsFilename, 'a') as extStatsFile:
skillPreset = os.path.splitext(os.path.basename(args.paramsFileName))[0]
if args.fakeRandoPreset is not None:
randoPreset = args.fakeRandoPreset
else:
randoPreset = os.path.splitext(os.path.basename(args.randoPreset))[0]
db.DB.dumpExtStatsItems(skillPreset, randoPreset, locsItems, extStatsFile)
try:
# args.rom is not None: generate local rom named filename.sfc with args.rom as source
# args.output is not None: generate local json named args.output
if args.rom is not None:
# patch local rom
romFileName = args.rom
outFileName = fileName + '.sfc'
shutil.copyfile(romFileName, outFileName)
romPatcher = RomPatcher(outFileName, magic=args.raceMagic)
else:
outFileName = args.output
romPatcher = RomPatcher(magic=args.raceMagic)
musicPatcher = None
if args.music is not None:
args.patches.append('custom_music.ips')
romType = 0
with open(args.music, "r") as f:
music = json.load(f)
musicParams = music.get('params', {})
musicMapping = music.get('mapping', {})
if args.patchOnly == False:
romType |= RomTypeForMusic.VariaSeed
if args.area == True:
romType |= RomTypeForMusic.AreaSeed
if args.bosses == True:
romType |= RomTypeForMusic.BossSeed
else:
variaSeed = musicParams.get('varia', False)
areaSeed = musicParams.get('area', False)
bossSeed = musicParams.get('boss', False)
if variaSeed == True:
romType |= RomTypeForMusic.VariaSeed
if areaSeed == True:
romType |= RomTypeForMusic.AreaSeed
if bossSeed == True:
romType |= RomTypeForMusic.BossSeed
musicPatcher = MusicPatcher(romPatcher.romFile, romType)
if args.hud == True or args.majorsSplit == "FullWithHUD":
args.patches.append("varia_hud.ips")
if args.debug == True:
args.patches.append("Disable_Clear_Save_Boot")
if args.patchOnly == False:
romPatcher.applyIPSPatches(args.startLocation, args.patches,
args.noLayout, gravityBehaviour,
args.area, args.bosses, args.areaLayoutBase,
args.noVariaTweaks, args.nerfedCharge, energyQty == 'ultra sparse',
escapeAttr, minimizerN, args.tourian == 'Fast',
args.doorsColorsRando)
else:
# from customizer permalink, apply previously generated seed ips first
if args.seedIps != None:
romPatcher.applyIPSPatch(args.seedIps)
romPatcher.addIPSPatches(args.patches)
if args.sprite is not None:
purge = args.ship is not None
romPatcher.customSprite(args.sprite, args.customItemNames, args.noSpinAttack, purge) # adds another IPS
if args.ship is not None:
romPatcher.customShip(args.ship) # adds another IPS
# don't color randomize custom ships
args.shift_ship_palette = False
# we have to write ips to ROM before doing our direct modifications which will rewrite some parts (like in credits)
romPatcher.commitIPS()
if args.patchOnly == False:
romPatcher.writeObjectives(objectivesManager, itemLocs)
romPatcher.writeItemsLocs(itemLocs)
romPatcher.writeSplitLocs(args.majorsSplit, itemLocs, progItemLocs)
romPatcher.writeItemsNumber()
romPatcher.writeSeed(seed) # lol if race mode
romPatcher.writeSpoiler(itemLocs, progItemLocs)
romPatcher.writeRandoSettings(randoSettings, itemLocs)
romPatcher.writeDoorConnections(doors)
romPatcher.writeVersion(displayedVersion)
if ctrlDict is not None:
romPatcher.writeControls(ctrlDict)
if args.moonWalk == True:
romPatcher.enableMoonWalk()
if args.patchOnly == False:
romPatcher.writeMagic()
romPatcher.writeMajorsSplit(args.majorsSplit)
romPatcher.writeAdditionalETanks(args.additionalEtanks)
romPatcher.writeHellrunRate(args.hellrunRate)
if args.palette == True:
paletteSettings = {
"global_shift": None,
"individual_suit_shift": None,
"individual_tileset_shift": None,
"match_ship_and_power": None,
"seperate_enemy_palette_groups": None,
"match_room_shift_with_boss": None,
"shift_tileset_palette": None,
"shift_boss_palettes": None,
"shift_suit_palettes": None,
"shift_enemy_palettes": None,
"shift_beam_palettes": None,
"shift_ship_palette": None,
"min_degree": None,
"max_degree": None,
"invert": None,
"no_blue_door_palette": None
}
for param in paletteSettings:
paletteSettings[param] = getattr(args, param)
PaletteRando(romPatcher, paletteSettings, args.sprite).randomize()
if musicPatcher is not None:
musicPatcher.replace(musicMapping,
updateReferences=musicParams.get('room_states', True),
output=musicParams.get("output", None))
romPatcher.end()
if args.patchOnly == False:
if len(optErrMsgs) > 0:
optErrMsgs.append(randoExec.errorMsg)
msg = joinErrorMsgs(optErrMsgs)
else:
msg = randoExec.errorMsg
else:
msg = ''
if args.rom is None: # web mode
data = romPatcher.romFile.data
fileName = '{}.sfc'.format(fileName)
data["fileName"] = fileName
# error msg in json to be displayed by the web site
data["errorMsg"] = msg
# replaced parameters to update stats in database
if len(forcedArgs) > 0:
data["forcedArgs"] = forcedArgs
with open(outFileName, 'w') as jsonFile:
json.dump(data, jsonFile)
else: # CLI mode
if msg != "":
print(msg)
except Exception as e:
import traceback
traceback.print_exc(file=sys.stdout)
msg = "Error patching {}: ({}: {})".format(outFileName, type(e).__name__, e)
dumpErrorMsg(args.output, msg)
sys.exit(-1)
if stuck == True:
print("Rom generated for debug purpose: {}".format(fileName))
else:
print("Rom generated: {}".format(fileName))