from pybricks.tools import wait, StopWatch
from pybricks.parameters import Port
from pybricks import version
import OtherFunctions as debug
import usys


class HubDiagnostics:
    def __init__(self, hub):
        self.hub = hub
        self.port_map = {
            "A": Port.A,
            "B": Port.B,
            "C": Port.C,
            "D": Port.D,
            "E": Port.E,
            "F": Port.F,
        }
    def printAbout(self):
        print("Pybricks version information:", version)
        print("MicroPython information:", usys.implementation)
        print("MicroPython version:", usys.version)
    def testLightSources(self, verbose):
        v = verbose
        self.hub.display.off()
        for x in range(5):
            for y in range(5):
                debug.log(f"Turning on pixel at position {x}, {y}...", v)
                self.hub.display.pixel(x, y, brightness=100)
                wait(100)
                debug.log(f"Turning off pixel at position {x}, {y}...", v)
                self.hub.display.pixel(x, y, brightness=0)
    def printAll(self):
        self.printAbout()
        self.testLightSources(False)