poketools/masuda.py at main · fdekeers/poketools · 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
"""
Automates egg hatching for shiny hunting with Masuda method.
"""

import sys
import os
import argparse
from math import ceil
from utils.controller import Controller
from configs import general
from utils import audio


#############
# CONSTANTS #
#############

STEPS_CYCLE = 257           # Number of steps in one egg cycle
EGG_HATCHING_DURATION = 10  # Egg hatching animation duration [s]
REC_DURATION = 4            # Game sound recording duration [s]
script_directory = os.path.dirname(os.path.abspath(__file__))  # current directory
SHINY_AUDIO_FILE = f"{script_directory}/template_sounds/shiny/template_cropped.wav"  # expected shiny sound to find


######################
# SCRIPT ENTRY POINT #
######################

if __name__ == "__main__":

    # Command line arguments
    parser = argparse.ArgumentParser()
    # Number of steps from the daycare desk to the right of the grandpa
    help = "Number of steps from the daycare desk to the right of the grandpa"
    parser.add_argument("steps_to_grandpa", type=int, help=help)
    # Egg cycles
    help = "Number of egg cycles before hatching for this specific Pokémon species."
    parser.add_argument("-e", "--egg-cycles", type=int, help=help)
    # Hatching steps
    help = "Number of steps before hatching for this specific Pokémon species."
    parser.add_argument("-s", "--egg-steps", type=int, help=help)
    # Number of steps
    # Presence of ability dividing the number of steps by two
    help = "Use this flag if a Pokémon in your party knows an ability dividing by 2 the number of steps to hatching."
    parser.add_argument("-a", "--ability", action="store_true", help=help)
    # Capture screenshot or video when shiny is found
    help = "Capture a video or screenshot once a shiny is found."
    parser.add_argument("-c", "--capture", help=help, choices=["video", "screenshot"])
    args = parser.parse_args()

    # Check if one and only one of the options -e and -s have been specified
    if (args.egg_cycles is None and args.egg_steps is None or
        args.egg_cycles is not None and args.egg_steps is not None):
        sys.stderr.write("Please specify one and only one of `-e` and `-s`.\n")
        exit(-1)

    # Initialize and connect virtual game controller, then go back to game
    controller = Controller()
    #controller.sync_and_go_back()
    controller.reconnect()

    # Main loop
    number_of_eggs = 1
    cycles = args.egg_steps / STEPS_CYCLE if args.egg_steps else args.egg_cycles
    cycles = ceil(cycles / 2) if args.ability else int(cycles)
    steps = cycles * STEPS_CYCLE - args.steps_to_grandpa
    is_shiny = False
    while not is_shiny:
        controller.macro(general.GET_ON_BIKE)
        controller.run_in_circles(steps)
        steps = cycles * STEPS_CYCLE
        controller.macro(general.EGG_HATCHING)
        # Shiny detection
        controller.busy_wait(EGG_HATCHING_DURATION)
        is_shiny, correlation = audio.record_and_check_shiny(SHINY_AUDIO_FILE, REC_DURATION)
        if is_shiny:
            print(f"Shiny found after {number_of_eggs} eggs.")
            # Shiny ! Put console in sleep mode
            if args.capture == "video":
                controller.macro(general.VIDEO)
                controller.busy_wait_b(8)
            elif args.capture == "screenshot":
                controller.macro(general.SCREENSHOT)
            controller.macro(general.SLEEP_MODE)
            exit(0)
        else:
            # Not shiny, release Pokemon and get new egg
            controller.macro(general.RELEASE_BREEDJECT)
            controller.macro(general.GET_NEW_EGG)
            number_of_eggs += 1