PyZX/video.py at master · Q-Master/PyZX · 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
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
import pygame
import Z80

__show_fps__ = True

SCREEN_WIDTH = 256
SCREEN_HEIGHT = 192
FULL_SCREEN_WIDTH = 384
FULL_SCREEN_HEIGHT = 256
CAPTION = 'PyZX'

COLOR_ON_NORMAL = 205
COLOR_ON_BRIGHT = 255
COLORS = [
    (0, 0, 0),                                            # Black Bright Off
    (0, 0, COLOR_ON_NORMAL),                              # Blue Bright Off
    (COLOR_ON_NORMAL, 0, 0),                              # Red Bright Off
    (COLOR_ON_NORMAL, 0, COLOR_ON_NORMAL),                # Magenta Bright Off
    (0, COLOR_ON_NORMAL, 0),                              # Green Bright Off
    (0, COLOR_ON_NORMAL, COLOR_ON_NORMAL),                # Cyan Bright Off
    (COLOR_ON_NORMAL, COLOR_ON_NORMAL, 0),                # Yellow Bright Off
    (COLOR_ON_NORMAL, COLOR_ON_NORMAL, COLOR_ON_NORMAL),  # White Bright Off

    (0, 0, 0),                                            # Black Bright On
    (0, 0, COLOR_ON_BRIGHT),                              # Blue Bright On
    (COLOR_ON_BRIGHT, 0, 0),                              # Red Bright On
    (COLOR_ON_BRIGHT, 0, COLOR_ON_BRIGHT),                # Magenta Bright On
    (0, COLOR_ON_BRIGHT, 0),                              # Green Bright On
    (0, COLOR_ON_BRIGHT, COLOR_ON_BRIGHT),                # Cyan Bright On
    (COLOR_ON_BRIGHT, COLOR_ON_BRIGHT, 0),                # Yellow Bright On
    (COLOR_ON_BRIGHT, COLOR_ON_BRIGHT, COLOR_ON_BRIGHT)   # White Bright On
]

# Инициализация таблицы адресов пиксельных и аттрибутных линий
addr_pix = [(((line // 64) * 2048) + ((line % 8) * 256) + ((line & 56) * 4)) for line in range(192)]
addr_attr = [(6144 + ((line // 8) * 32)) for line in range(192)]
zxrowmap = [((coord_y & 0b111) << 3) + ((coord_y & 0b111000) >> 3) + (coord_y & 0b11000000) for coord_y in range(192)]
colormap = [((attr % 8) + (8 if attr & 64 else 0), (attr & 0b1111000) >> 3) for attr in range(256)]


pixelmap = bytearray(256*256*8)
pixelmap_m = memoryview(pixelmap)
STRIDE = 256*8

def init_pixelmap():
    for i in range(256):
        color_ink, color_paper = colormap[i]
        pixellist = pixelmap_m[i*STRIDE : i*STRIDE+STRIDE]
        for pix in range(256):
            pixels = pixellist[pix*8 : pix*8+8]
            for bit in range(8):
                pixels[7-bit] = color_ink if (pix & (1 << bit)) else color_paper


zx_screen = None
zx_screen_with_border = None
screen = None
ratio = 2

def init():
    global screen, zx_screen, zx_screen_with_border

    init_pixelmap()
    pygame.init()
    icon = pygame.image.load('icon.png')
    zx_screen = pygame.surface.Surface((SCREEN_WIDTH, SCREEN_HEIGHT), pygame.HWSURFACE, 8)
    zx_screen.set_palette(COLORS)
    zx_screen_with_border = pygame.surface.Surface((FULL_SCREEN_WIDTH, FULL_SCREEN_HEIGHT), pygame.HWSURFACE, 8)
    zx_screen_with_border.set_palette(COLORS)
    screen = pygame.display.set_mode((FULL_SCREEN_WIDTH*ratio, FULL_SCREEN_HEIGHT*ratio), pygame.HWSURFACE | pygame.DOUBLEBUF, 8)
    pygame.display.set_palette(COLORS)
    pygame.display.set_caption(CAPTION)
    pygame.display.set_icon(icon)
    pygame.display.flip()

    return


clock = pygame.time.Clock()
old_border = -1

def update():
    global __show_fps__, clock, old_border

    if __show_fps__:
        clock.tick()
        pygame.display.set_caption(f'{CAPTION} - {clock.get_fps():.2f} FPS')

    if Z80.ports.current_border != old_border:
        zx_screen_with_border.fill(Z80.ports.current_border)
        old_border = Z80.ports.current_border

    fill_screen_map()
    zx_screen_with_border.blit(zx_screen, (64, 32))
    pygame.transform.scale(zx_screen_with_border, (FULL_SCREEN_WIDTH*ratio, FULL_SCREEN_HEIGHT*ratio), screen)
    pygame.display.flip()

buffer = bytearray(SCREEN_WIDTH*SCREEN_HEIGHT)
buffer_m = memoryview(buffer)

def fill_screen_map():
    zx_videoram = Z80.memory.mem[16384:16384+6912]
    offs = 0

    for coord_y in range(SCREEN_HEIGHT):
        pix_addr = addr_pix[coord_y]
        attr_addr = addr_attr[coord_y]

        for i in range(0, 32):
            poffs = zx_videoram[attr_addr+i]*STRIDE+zx_videoram[pix_addr+i]*8
            buffer_m[offs : offs+8] = pixelmap_m[poffs : poffs+8]
            offs += 8

    buf = zx_screen.get_buffer()
    buf.write(buffer_m.tobytes())