main.py

# This is where the magic happens!
# This file is executed on every boot (including wake-boot from deepsleep)
# Created By: Michael Pham

"""
Built for the PySquared FC Board
Version: 2.0.0
Published: Nov 19, 2024
"""

import time

import microcontroller

import lib.pysquared.pysquared as pysquared

print("=" * 70)
print("Hello World!")
print("PySquared FC Board Circuit Python Software Version: 2.0.0")
print("Published: November 19, 2024")
print("=" * 70)

loiter_time = 5

try:
    for i in range(loiter_time):
        print(f"Code Starting in {loiter_time-i} seconds")
        time.sleep(1)

    print("Initializing CubeSat")
    c = pysquared.Satellite()
    c.watchdog_pet()

    import gc  # Garbage collection
    import traceback

    import lib.pysquared.functions as functions
    from lib.pysquared.debugcolor import co

    def debug_print(statement):
        if c.debug:
            print(co(str(c.uptime) + "[MAIN]" + str(statement), "blue", "bold"))

    f = functions.functions(c)

    def initial_boot():
        c.watchdog_pet()
        f.beacon()
        c.watchdog_pet()
        f.listen()
        c.watchdog_pet()
        # f.state_of_health()
        # f.listen()
        # c.watchdog_pet()

    try:
        c.c_boot += 1  # Increment boot number
        debug_print("Boot number: " + str(c.c_boot))
        debug_print(str(gc.mem_free()) + " Bytes remaining")

        initial_boot()

    except Exception as e:
        debug_print("Error in Boot Sequence: " + "".join(traceback.format_exception(e)))
    finally:
        debug_print("Something went wrong!")

    def send_imu():
        debug_print("Looking to get imu data...")
        IMUData = []
        c.watchdog_pet()
        debug_print("IMU has baton")
        IMUData = f.get_imu_data()
        c.watchdog_pet()
        f.send(IMUData)

    def main():
        f.beacon()

        f.listen_loiter()

        f.state_of_health()

        f.listen_loiter()

        f.all_face_data()
        c.watchdog_pet()
        f.send_face()

        f.listen_loiter()

        send_imu()

        f.listen_loiter()

        f.joke()

        f.listen_loiter()

    def critical_power_operations():
        initial_boot()
        c.watchdog_pet()

        f.Long_Hybernate()

    def minimum_power_operations():
        initial_boot()
        c.watchdog_pet()

        f.Short_Hybernate()

    ######################### MAIN LOOP ##############################
    try:
        while True:
            # L0 automatic tasks no matter the battery level
            c.check_reboot()

            if c.power_mode == "critical":
                c.RGB = (0, 0, 0)
                critical_power_operations()

            elif c.power_mode == "minimum":
                c.RGB = (255, 0, 0)
                minimum_power_operations()

            elif c.power_mode == "normal":
                c.RGB = (255, 255, 0)
                main()

            elif c.power_mode == "maximum":
                c.RGB = (0, 255, 0)
                main()

            else:
                f.listen()

    except Exception as e:
        debug_print("Critical in Main Loop: " + "".join(traceback.format_exception(e)))
        time.sleep(10)
        microcontroller.on_next_reset(microcontroller.RunMode.NORMAL)
        microcontroller.reset()
    finally:
        debug_print("Going Neutral!")

        c.RGB = (0, 0, 0)
        c.hardware["WDT"] = False

except Exception as e:
    print(e)