boom.py

# fireworks_sprites.py - show a bunch of fireworks
# 4 Jul 2022 - @todbot / Tod Kurt
import time, random
import board
import busio
import displayio
import rainbowio
import adafruit_imageload
import adafruit_ili9341
import adafruit_tsc2007

fireworks_count = 5
sprite_fname = "fireworks_spritesheet.bmp"
sprite_count = 30*1
sprite_w,sprite_h = 64,64


# Release any resources currently in use for the displays
displayio.release_displays()

# i2c = board.I2C()
i2c = busio.I2C(board.SCL, board.SDA, frequency=200_000)

irq_dio = None
tsc = adafruit_tsc2007.TSC2007(i2c, irq=irq_dio)

# Use Hardware SPI
spi = board.SPI()

# Use Software SPI if you have a shield with pins 11-13 jumpered
# import busio
# spi = busio.SPI(board.D11, board.D13)

tft_cs = board.D10
tft_dc = board.D9

display_width = 320
display_height = 240

display_bus = displayio.FourWire(spi, command=tft_dc, chip_select=tft_cs)
display = adafruit_ili9341.ILI9341(display_bus, width=display_width, height=display_height)


# display = board.DISPLAY  # our board has built-in display
dw,dh = display_width, display_height  # convenience values

sprite_sheet, sprite_palette = adafruit_imageload.load(sprite_fname)

maingroup = displayio.Group()  # all fireworks go on 'maingroup'
display.show(maingroup)

def random_xy():
    return random.randint(0, dw-sprite_w), random.randint(dh//2, dh//2+20)

def touch_xy():
    tx = (point["y"])
    ty = (point["x"])
    print("touch x", tx, "touch y", ty)
    return(tx, ty)
    # return ( (dw - point["y"]//10)-60, (point["x"]//10)-120 )

def copy_palette(p):  # this is fastest way to copy a palette it seems
    new_p = displayio.Palette(len(p))
    for i in range(len(p)):  new_p[i] = p[i]
    return new_p

def new_colors(p):  # first two colors after bg seem to be the "tails"
    p[1] = rainbowio.colorwheel( random.randint(0,255) )
    p[2] = rainbowio.colorwheel( random.randint(0,255) )



# holds list of sprites,sprite_index,x,y,launch_time
fireworks = [(None,0,0,0, 0) ] * fireworks_count
# make our fireworks
for i in range(len(fireworks)):
    pal = copy_palette(sprite_palette)
    pal.make_transparent(0)  # make background color transparent
    x,y = random_xy()
    start_index = random.randint(0,sprite_count-1)  # pick random point in fireworks life to start
    fwsprite = displayio.TileGrid(sprite_sheet, pixel_shader=pal,
                            width = 1, height = 1,
                            tile_width = sprite_w, tile_height = sprite_h)
    fireworks[i]= (fwsprite, start_index, x, y, pal)  # add sprite, and initial index, x,y, palette
    maingroup.append(fwsprite)

touch_state = False


while True:
    if tsc.touched and not touch_state:
        point = tsc.touch
        touch_state = True
        if point["pressure"] < 200:  # ignore touches with no 'pressure' as false
            continue
        touch_xy()
        print("Touchpoint: (%d, %d, %d)" % (point["x"], point["y"], point["pressure"]))

        for i in range(len(fireworks)):
            (sprite, sprite_index, x,y, palette) = fireworks[i]  # get firework state
            # update firework state
            launching = sprite_index == 0 and random.random() < 0.9
            if launching:
                y = y - 3  # move towards top of screen
            else:
                sprite_index = sprite_index + 1  # explode!
            # firework finished
            if sprite_index == sprite_count:
                sprite_index = 0   # firework is reborn, make a new x,y for it
                x,y = touch_xy()  #random.randint(dw//4, 3*dw//4), random.randint(dh//2,dh//2+20)
                new_colors(palette)
            # update firework on screen
            sprite[0] = sprite_index  # set next sprite in animation
            sprite.x, sprite.y = x,y  # set its x,y
            # save the firework state
            fireworks[i] = (sprite, sprite_index, x,y, palette)
        time.sleep(0.008)  # determines our framerate

    if not tsc.touched and touch_state:
        touch_state = False