#!/usr/bin/env python3 # SPDX-License-Identifier: MIT import sys, pathlib sys.path.append(str(pathlib.Path(__file__).resolve().parents[1])) from m1n1.setup import * from m1n1.hw.dart8110 import DART8110, DART8110Regs from m1n1.hw.prores import * from m1n1.utils import * import time def divroundup(val, div): return (val + div - 1) // div def bswp16(x): return (x >> 8) | ((x & 0xFF) << 8) # ffmpeg -y -i prores-encode-large.png -c:v rawvideo -pix_fmt nv24 prores-encode-large.yuv im_W = 1920 im_H = 1080 with open('prores-encode-large.yuv', 'rb') as f: im_data = f.read() assert len(im_data) == (im_W*im_H) * 3 image_data_luma = im_data[:im_W*im_H] image_data_chroma = im_data[im_W*im_H:] p.pmgr_adt_clocks_enable(f'/arm-io/dart-apr0') p.pmgr_adt_clocks_enable(f'/arm-io/apr0') dart = DART8110.from_adt(u, f'/arm-io/dart-apr0') dart.initialize() apr_base, _ = u.adt[f'/arm-io/apr0'].get_reg(0) apr = ProResRegs(u, apr_base) print(f"Register 0 (ID?) {apr.REG_0x0}") # TUNABLES apr.MODE = 0x0 apr.REG_0x118 = apr.REG_0x118.val & ~0x07FF07FF | 0x00000600 apr.REG_0x148 = apr.REG_0x148.val & ~0x00000001 | 0x00000001 apr.REG_0x160 = apr.REG_0x160.val & ~0x800F3FFF | 0x800A04FF apr.REG_0x164 = apr.REG_0x164.val & ~0x07FF07FF | 0x07800080 apr.REG_0x170 = apr.REG_0x170.val & ~0x800F3FFF | 0x800404FF apr.REG_0x174 = apr.REG_0x174.val & ~0x07FF07FF | 0x06000080 apr.REG_0x180 = apr.REG_0x180.val & ~0x800F3FFF | 0x800504FF apr.REG_0x184 = apr.REG_0x184.val & ~0x07FF07FF | 0x06800080 apr.REG_0x190 = apr.REG_0x190.val & ~0x800F3FFF | 0x800004FF apr.REG_0x194 = apr.REG_0x194.val & ~0x000000FF | 0x00000040 apr.REG_0x1a0 = apr.REG_0x1a0.val & ~0x800F3FFF | 0x800104FF apr.REG_0x1a4 = apr.REG_0x1a4.val & ~0x000000FF | 0x00000080 apr.REG_0x1b0 = apr.REG_0x1b0.val & ~0x800F3FFF | 0x800204FF apr.REG_0x1b4 = apr.REG_0x1b4.val & ~0x000000FF | 0x00000040 apr.REG_0x1c0 = apr.REG_0x1c0.val & ~0x800F3FFF | 0x800304FF apr.REG_0x1c4 = apr.REG_0x1c4.val & ~0x000000FF | 0x00000040 apr.REG_0x1d0 = apr.REG_0x1d0.val & ~0xBC00FF86 | 0xA4000786 apr.REG_0x1d4 = apr.REG_0x1d4.val & ~0x000000FF | 0x00000020 apr.REG_0x1d8 = apr.REG_0x1d8.val & ~0x000000FF | 0x000000FF apr.REG_0x1dc = apr.REG_0x1dc.val & ~0x00FFFFFF | 0x00928170 apr.REG_0x270 = apr.REG_0x270.val & ~0x800F3FFF | 0x800B08FF apr.REG_0x274 = apr.REG_0x274.val & ~0x07FF07FF | 0x07000080 apr.REG_0x280 = apr.REG_0x280.val & ~0xFFFFFFC0 | 0x00180000 apr.REG_0x290 = apr.REG_0x290.val & ~0x800F3FFF | 0x800004FF apr.REG_0x294 = apr.REG_0x294.val & ~0x000000FF | 0x00000080 apr.REG_0x2a0 = apr.REG_0x2a0.val & ~0x800F3FFF | 0x800104FF apr.REG_0x2a4 = apr.REG_0x2a4.val & ~0x000000FF | 0x00000080 apr.REG_0x2b0 = apr.REG_0x2b0.val & ~0x800F3FFF | 0x800204FF apr.REG_0x2b4 = apr.REG_0x2b4.val & ~0x000000FF | 0x00000040 apr.REG_0x2c0 = apr.REG_0x2c0.val & ~0x800F3FFF | 0x800304FF apr.REG_0x2c4 = apr.REG_0x2c4.val & ~0x000000FF | 0x00000040 apr.REG_0x2d0 = apr.REG_0x2d0.val & ~0x802FF04C | 0x80070040 apr.REG_0x2d4 = apr.REG_0x2d4.val & ~0x00000001 | 0x00000000 apr.REG_0x2d8 = apr.REG_0x2d8.val & ~0xFFFF0003 | 0x00FF0003 apr.REG_0x2e0 = apr.REG_0x2e0.val & ~0x07FF07FF | 0x06000040 apr.REG_0x2f8 = apr.REG_0x2f8.val & ~0x802FF04C | 0x80081040 apr.REG_0x2fc = apr.REG_0x2fc.val & ~0x00000001 | 0x00000000 apr.REG_0x300 = apr.REG_0x300.val & ~0xFFFF0003 | 0x00FF0003 apr.REG_0x308 = apr.REG_0x308.val & ~0x07FF07FF | 0x06400040 apr.REG_0x320 = apr.REG_0x320.val & ~0x802FF04C | 0x80092040 apr.REG_0x324 = apr.REG_0x324.val & ~0x00000001 | 0x00000000 apr.REG_0x328 = apr.REG_0x328.val & ~0xFFFF0003 | 0x00FF0003 apr.REG_0x330 = apr.REG_0x330.val & ~0x07FF07FF | 0x06800040 apr.REG_0x350 = apr.REG_0x350.val & ~0x800F3FFF | 0x800B08FF apr.REG_0x354 = apr.REG_0x354.val & ~0x07FF07FF | 0x076000A0 apr.REG_0x360 = apr.REG_0x360.val & ~0xFFFFFFC0 | 0x00180000 apr.REG_0x370 = apr.REG_0x370.val & ~0x800F3FFF | 0x800604FF apr.REG_0x374 = apr.REG_0x374.val & ~0x07FF07FF | 0x06C000A0 print("Applied tunables") # XXX test wrap around behavior DESC_RING_SZ = 0x4000 desc_ring_phys = u.heap.memalign(0x4000, DESC_RING_SZ) desc_ring_iova = dart.iomap(0, desc_ring_phys, DESC_RING_SZ) print(f"Descriptor ring @ phys {desc_ring_phys:016X} iova {desc_ring_iova:016X}") apr.DR_HEAD = 0 apr.DR_TAIL = 0 apr.DR_SIZE = DESC_RING_SZ apr.DR_ADDR_LO = desc_ring_iova & 0xFFFFFFFF apr.DR_ADDR_HI = desc_ring_iova >> 32 apr.MODE = 0xd # FIXME: dunno what this means # MATRICES apr.QUANT_LUMA_EHQ[0].val = 0x802802 apr.QUANT_CHROMA_EHQ[0].val = 0x804804 apr.QUANT_LUMA_EHQ[1].val = 0x802802 apr.QUANT_CHROMA_EHQ[1].val = 0x804804 apr.QUANT_LUMA_EHQ[2].val = 0x802802 apr.QUANT_CHROMA_EHQ[2].val = 0x804804 apr.QUANT_LUMA_EHQ[3].val = 0x802802 apr.QUANT_CHROMA_EHQ[3].val = 0x804804 apr.QUANT_LUMA_EHQ[4].val = 0x802802 apr.QUANT_CHROMA_EHQ[4].val = 0x804804 apr.QUANT_LUMA_EHQ[5].val = 0x802802 apr.QUANT_CHROMA_EHQ[5].val = 0x804804 apr.QUANT_LUMA_EHQ[6].val = 0x802802 apr.QUANT_CHROMA_EHQ[6].val = 0x804804 apr.QUANT_LUMA_EHQ[7].val = 0x802802 apr.QUANT_CHROMA_EHQ[7].val = 0x804804 apr.QUANT_LUMA_EHQ[8].val = 0x802802 apr.QUANT_CHROMA_EHQ[8].val = 0x804804 apr.QUANT_LUMA_EHQ[9].val = 0x802802 apr.QUANT_CHROMA_EHQ[9].val = 0x804804 apr.QUANT_LUMA_EHQ[10].val = 0x802802 apr.QUANT_CHROMA_EHQ[10].val = 0x804804 apr.QUANT_LUMA_EHQ[11].val = 0x802802 apr.QUANT_CHROMA_EHQ[11].val = 0x804804 apr.QUANT_LUMA_EHQ[12].val = 0x802802 apr.QUANT_CHROMA_EHQ[12].val = 0x804804 apr.QUANT_LUMA_EHQ[13].val = 0x802802 apr.QUANT_CHROMA_EHQ[13].val = 0x804804 apr.QUANT_LUMA_EHQ[14].val = 0x802802 apr.QUANT_CHROMA_EHQ[14].val = 0x804804 apr.QUANT_LUMA_EHQ[15].val = 0x803802 apr.QUANT_CHROMA_EHQ[15].val = 0x805804 apr.QUANT_LUMA_EHQ[16].val = 0x802802 apr.QUANT_CHROMA_EHQ[16].val = 0x804804 apr.QUANT_LUMA_EHQ[17].val = 0x802802 apr.QUANT_CHROMA_EHQ[17].val = 0x804804 apr.QUANT_LUMA_EHQ[18].val = 0x802802 apr.QUANT_CHROMA_EHQ[18].val = 0x804804 apr.QUANT_LUMA_EHQ[19].val = 0x803803 apr.QUANT_CHROMA_EHQ[19].val = 0x805805 apr.QUANT_LUMA_EHQ[20].val = 0x802802 apr.QUANT_CHROMA_EHQ[20].val = 0x804804 apr.QUANT_LUMA_EHQ[21].val = 0x802802 apr.QUANT_CHROMA_EHQ[21].val = 0x804804 apr.QUANT_LUMA_EHQ[22].val = 0x803802 apr.QUANT_CHROMA_EHQ[22].val = 0x805804 apr.QUANT_LUMA_EHQ[23].val = 0x803803 apr.QUANT_CHROMA_EHQ[23].val = 0x806805 apr.QUANT_LUMA_EHQ[24].val = 0x802802 apr.QUANT_CHROMA_EHQ[24].val = 0x804804 apr.QUANT_LUMA_EHQ[25].val = 0x802802 apr.QUANT_CHROMA_EHQ[25].val = 0x804804 apr.QUANT_LUMA_EHQ[26].val = 0x803803 apr.QUANT_CHROMA_EHQ[26].val = 0x805805 apr.QUANT_LUMA_EHQ[27].val = 0x804803 apr.QUANT_CHROMA_EHQ[27].val = 0x807806 apr.QUANT_LUMA_EHQ[28].val = 0x802802 apr.QUANT_CHROMA_EHQ[28].val = 0x804804 apr.QUANT_LUMA_EHQ[29].val = 0x802802 apr.QUANT_CHROMA_EHQ[29].val = 0x804804 apr.QUANT_LUMA_EHQ[30].val = 0x803803 apr.QUANT_CHROMA_EHQ[30].val = 0x806805 apr.QUANT_LUMA_EHQ[31].val = 0x804804 apr.QUANT_CHROMA_EHQ[31].val = 0x807807 apr.QUANT_LUMA_HQ[0].val = 0x804804 apr.QUANT_CHROMA_HQ[0].val = 0x804804 apr.QUANT_LUMA_HQ[1].val = 0x804804 apr.QUANT_CHROMA_HQ[1].val = 0x804804 apr.QUANT_LUMA_HQ[2].val = 0x804804 apr.QUANT_CHROMA_HQ[2].val = 0x804804 apr.QUANT_LUMA_HQ[3].val = 0x804804 apr.QUANT_CHROMA_HQ[3].val = 0x804804 apr.QUANT_LUMA_HQ[4].val = 0x804804 apr.QUANT_CHROMA_HQ[4].val = 0x804804 apr.QUANT_LUMA_HQ[5].val = 0x804804 apr.QUANT_CHROMA_HQ[5].val = 0x804804 apr.QUANT_LUMA_HQ[6].val = 0x804804 apr.QUANT_CHROMA_HQ[6].val = 0x804804 apr.QUANT_LUMA_HQ[7].val = 0x804804 apr.QUANT_CHROMA_HQ[7].val = 0x804804 apr.QUANT_LUMA_HQ[8].val = 0x804804 apr.QUANT_CHROMA_HQ[8].val = 0x804804 apr.QUANT_LUMA_HQ[9].val = 0x804804 apr.QUANT_CHROMA_HQ[9].val = 0x804804 apr.QUANT_LUMA_HQ[10].val = 0x804804 apr.QUANT_CHROMA_HQ[10].val = 0x804804 apr.QUANT_LUMA_HQ[11].val = 0x804804 apr.QUANT_CHROMA_HQ[11].val = 0x804804 apr.QUANT_LUMA_HQ[12].val = 0x804804 apr.QUANT_CHROMA_HQ[12].val = 0x804804 apr.QUANT_LUMA_HQ[13].val = 0x804804 apr.QUANT_CHROMA_HQ[13].val = 0x804804 apr.QUANT_LUMA_HQ[14].val = 0x804804 apr.QUANT_CHROMA_HQ[14].val = 0x804804 apr.QUANT_LUMA_HQ[15].val = 0x805804 apr.QUANT_CHROMA_HQ[15].val = 0x805804 apr.QUANT_LUMA_HQ[16].val = 0x804804 apr.QUANT_CHROMA_HQ[16].val = 0x804804 apr.QUANT_LUMA_HQ[17].val = 0x804804 apr.QUANT_CHROMA_HQ[17].val = 0x804804 apr.QUANT_LUMA_HQ[18].val = 0x804804 apr.QUANT_CHROMA_HQ[18].val = 0x804804 apr.QUANT_LUMA_HQ[19].val = 0x805805 apr.QUANT_CHROMA_HQ[19].val = 0x805805 apr.QUANT_LUMA_HQ[20].val = 0x804804 apr.QUANT_CHROMA_HQ[20].val = 0x804804 apr.QUANT_LUMA_HQ[21].val = 0x804804 apr.QUANT_CHROMA_HQ[21].val = 0x804804 apr.QUANT_LUMA_HQ[22].val = 0x805804 apr.QUANT_CHROMA_HQ[22].val = 0x805804 apr.QUANT_LUMA_HQ[23].val = 0x806805 apr.QUANT_CHROMA_HQ[23].val = 0x806805 apr.QUANT_LUMA_HQ[24].val = 0x804804 apr.QUANT_CHROMA_HQ[24].val = 0x804804 apr.QUANT_LUMA_HQ[25].val = 0x804804 apr.QUANT_CHROMA_HQ[25].val = 0x804804 apr.QUANT_LUMA_HQ[26].val = 0x805805 apr.QUANT_CHROMA_HQ[26].val = 0x805805 apr.QUANT_LUMA_HQ[27].val = 0x807806 apr.QUANT_CHROMA_HQ[27].val = 0x807806 apr.QUANT_LUMA_HQ[28].val = 0x804804 apr.QUANT_CHROMA_HQ[28].val = 0x804804 apr.QUANT_LUMA_HQ[29].val = 0x804804 apr.QUANT_CHROMA_HQ[29].val = 0x804804 apr.QUANT_LUMA_HQ[30].val = 0x806805 apr.QUANT_CHROMA_HQ[30].val = 0x806805 apr.QUANT_LUMA_HQ[31].val = 0x807807 apr.QUANT_CHROMA_HQ[31].val = 0x807807 apr.QUANT_LUMA_NQ[0].val = 0x804804 apr.QUANT_CHROMA_NQ[0].val = 0x804804 apr.QUANT_LUMA_NQ[1].val = 0x805805 apr.QUANT_CHROMA_NQ[1].val = 0x805805 apr.QUANT_LUMA_NQ[2].val = 0x807806 apr.QUANT_CHROMA_NQ[2].val = 0x807806 apr.QUANT_LUMA_NQ[3].val = 0x809807 apr.QUANT_CHROMA_NQ[3].val = 0x809807 apr.QUANT_LUMA_NQ[4].val = 0x804804 apr.QUANT_CHROMA_NQ[4].val = 0x804804 apr.QUANT_LUMA_NQ[5].val = 0x806805 apr.QUANT_CHROMA_NQ[5].val = 0x806805 apr.QUANT_LUMA_NQ[6].val = 0x807807 apr.QUANT_CHROMA_NQ[6].val = 0x807807 apr.QUANT_LUMA_NQ[7].val = 0x809809 apr.QUANT_CHROMA_NQ[7].val = 0x809809 apr.QUANT_LUMA_NQ[8].val = 0x805805 apr.QUANT_CHROMA_NQ[8].val = 0x805805 apr.QUANT_LUMA_NQ[9].val = 0x807806 apr.QUANT_CHROMA_NQ[9].val = 0x807806 apr.QUANT_LUMA_NQ[10].val = 0x809807 apr.QUANT_CHROMA_NQ[10].val = 0x809807 apr.QUANT_LUMA_NQ[11].val = 0x80a809 apr.QUANT_CHROMA_NQ[11].val = 0x80a809 apr.QUANT_LUMA_NQ[12].val = 0x805805 apr.QUANT_CHROMA_NQ[12].val = 0x805805 apr.QUANT_LUMA_NQ[13].val = 0x807806 apr.QUANT_CHROMA_NQ[13].val = 0x807806 apr.QUANT_LUMA_NQ[14].val = 0x809807 apr.QUANT_CHROMA_NQ[14].val = 0x809807 apr.QUANT_LUMA_NQ[15].val = 0x80a809 apr.QUANT_CHROMA_NQ[15].val = 0x80a809 apr.QUANT_LUMA_NQ[16].val = 0x806805 apr.QUANT_CHROMA_NQ[16].val = 0x806805 apr.QUANT_LUMA_NQ[17].val = 0x807807 apr.QUANT_CHROMA_NQ[17].val = 0x807807 apr.QUANT_LUMA_NQ[18].val = 0x809808 apr.QUANT_CHROMA_NQ[18].val = 0x809808 apr.QUANT_LUMA_NQ[19].val = 0x80c80a apr.QUANT_CHROMA_NQ[19].val = 0x80c80a apr.QUANT_LUMA_NQ[20].val = 0x807806 apr.QUANT_CHROMA_NQ[20].val = 0x807806 apr.QUANT_LUMA_NQ[21].val = 0x808807 apr.QUANT_CHROMA_NQ[21].val = 0x808807 apr.QUANT_LUMA_NQ[22].val = 0x80a809 apr.QUANT_CHROMA_NQ[22].val = 0x80a809 apr.QUANT_LUMA_NQ[23].val = 0x80f80c apr.QUANT_CHROMA_NQ[23].val = 0x80f80c apr.QUANT_LUMA_NQ[24].val = 0x807806 apr.QUANT_CHROMA_NQ[24].val = 0x807806 apr.QUANT_LUMA_NQ[25].val = 0x809807 apr.QUANT_CHROMA_NQ[25].val = 0x809807 apr.QUANT_LUMA_NQ[26].val = 0x80b80a apr.QUANT_CHROMA_NQ[26].val = 0x80b80a apr.QUANT_LUMA_NQ[27].val = 0x81180e apr.QUANT_CHROMA_NQ[27].val = 0x81180e apr.QUANT_LUMA_NQ[28].val = 0x807807 apr.QUANT_CHROMA_NQ[28].val = 0x807807 apr.QUANT_LUMA_NQ[29].val = 0x80a809 apr.QUANT_CHROMA_NQ[29].val = 0x80a809 apr.QUANT_LUMA_NQ[30].val = 0x80e80b apr.QUANT_CHROMA_NQ[30].val = 0x80e80b apr.QUANT_LUMA_NQ[31].val = 0x815811 apr.QUANT_CHROMA_NQ[31].val = 0x815811 apr.QUANT_LUMA_LT[0].val = 0x805804 apr.QUANT_CHROMA_LT[0].val = 0x805804 apr.QUANT_LUMA_LT[1].val = 0x807806 apr.QUANT_CHROMA_LT[1].val = 0x807806 apr.QUANT_LUMA_LT[2].val = 0x80b809 apr.QUANT_CHROMA_LT[2].val = 0x80b809 apr.QUANT_LUMA_LT[3].val = 0x80f80d apr.QUANT_CHROMA_LT[3].val = 0x80f80d apr.QUANT_LUMA_LT[4].val = 0x805805 apr.QUANT_CHROMA_LT[4].val = 0x805805 apr.QUANT_LUMA_LT[5].val = 0x808807 apr.QUANT_CHROMA_LT[5].val = 0x808807 apr.QUANT_LUMA_LT[6].val = 0x80d80b apr.QUANT_CHROMA_LT[6].val = 0x80d80b apr.QUANT_LUMA_LT[7].val = 0x81180f apr.QUANT_CHROMA_LT[7].val = 0x81180f apr.QUANT_LUMA_LT[8].val = 0x807806 apr.QUANT_CHROMA_LT[8].val = 0x807806 apr.QUANT_LUMA_LT[9].val = 0x80b809 apr.QUANT_CHROMA_LT[9].val = 0x80b809 apr.QUANT_LUMA_LT[10].val = 0x80f80d apr.QUANT_CHROMA_LT[10].val = 0x80f80d apr.QUANT_LUMA_LT[11].val = 0x81180f apr.QUANT_CHROMA_LT[11].val = 0x81180f apr.QUANT_LUMA_LT[12].val = 0x807807 apr.QUANT_CHROMA_LT[12].val = 0x807807 apr.QUANT_LUMA_LT[13].val = 0x80b809 apr.QUANT_CHROMA_LT[13].val = 0x80b809 apr.QUANT_LUMA_LT[14].val = 0x80f80d apr.QUANT_CHROMA_LT[14].val = 0x80f80d apr.QUANT_LUMA_LT[15].val = 0x813811 apr.QUANT_CHROMA_LT[15].val = 0x813811 apr.QUANT_LUMA_LT[16].val = 0x809807 apr.QUANT_CHROMA_LT[16].val = 0x809807 apr.QUANT_LUMA_LT[17].val = 0x80d80b apr.QUANT_CHROMA_LT[17].val = 0x80d80b apr.QUANT_LUMA_LT[18].val = 0x81080e apr.QUANT_CHROMA_LT[18].val = 0x81080e apr.QUANT_LUMA_LT[19].val = 0x817813 apr.QUANT_CHROMA_LT[19].val = 0x817813 apr.QUANT_LUMA_LT[20].val = 0x80b809 apr.QUANT_CHROMA_LT[20].val = 0x80b809 apr.QUANT_LUMA_LT[21].val = 0x80e80d apr.QUANT_CHROMA_LT[21].val = 0x80e80d apr.QUANT_LUMA_LT[22].val = 0x813810 apr.QUANT_CHROMA_LT[22].val = 0x813810 apr.QUANT_LUMA_LT[23].val = 0x81d817 apr.QUANT_CHROMA_LT[23].val = 0x81d817 apr.QUANT_LUMA_LT[24].val = 0x80b809 apr.QUANT_CHROMA_LT[24].val = 0x80b809 apr.QUANT_LUMA_LT[25].val = 0x80f80d apr.QUANT_CHROMA_LT[25].val = 0x80f80d apr.QUANT_LUMA_LT[26].val = 0x815811 apr.QUANT_CHROMA_LT[26].val = 0x815811 apr.QUANT_LUMA_LT[27].val = 0x82381c apr.QUANT_CHROMA_LT[27].val = 0x82381c apr.QUANT_LUMA_LT[28].val = 0x80d80b apr.QUANT_CHROMA_LT[28].val = 0x80d80b apr.QUANT_LUMA_LT[29].val = 0x811810 apr.QUANT_CHROMA_LT[29].val = 0x811810 apr.QUANT_LUMA_LT[30].val = 0x81c815 apr.QUANT_CHROMA_LT[30].val = 0x81c815 apr.QUANT_LUMA_LT[31].val = 0x829823 apr.QUANT_CHROMA_LT[31].val = 0x829823 apr.QUANT_LUMA_PROXY[0].val = 0x807804 apr.QUANT_CHROMA_PROXY[0].val = 0x807804 apr.QUANT_LUMA_PROXY[1].val = 0x80b809 apr.QUANT_CHROMA_PROXY[1].val = 0x80b809 apr.QUANT_LUMA_PROXY[2].val = 0x80e80d apr.QUANT_CHROMA_PROXY[2].val = 0x80e80d apr.QUANT_LUMA_PROXY[3].val = 0xfff83f apr.QUANT_CHROMA_PROXY[3].val = 0xfff83f apr.QUANT_LUMA_PROXY[4].val = 0x807807 apr.QUANT_CHROMA_PROXY[4].val = 0x807807 apr.QUANT_LUMA_PROXY[5].val = 0x80c80b apr.QUANT_CHROMA_PROXY[5].val = 0x80c80b apr.QUANT_LUMA_PROXY[6].val = 0x83f80e apr.QUANT_CHROMA_PROXY[6].val = 0x83f80e apr.QUANT_LUMA_PROXY[7].val = 0xffffff apr.QUANT_CHROMA_PROXY[7].val = 0xffffff apr.QUANT_LUMA_PROXY[8].val = 0x80b809 apr.QUANT_CHROMA_PROXY[8].val = 0x80b809 apr.QUANT_LUMA_PROXY[9].val = 0x80e80d apr.QUANT_CHROMA_PROXY[9].val = 0x80e80d apr.QUANT_LUMA_PROXY[10].val = 0xfff83f apr.QUANT_CHROMA_PROXY[10].val = 0xfff83f apr.QUANT_LUMA_PROXY[11].val = 0xffffff apr.QUANT_CHROMA_PROXY[11].val = 0xffffff apr.QUANT_LUMA_PROXY[12].val = 0x80b80b apr.QUANT_CHROMA_PROXY[12].val = 0x80b80b apr.QUANT_LUMA_PROXY[13].val = 0x80e80d apr.QUANT_CHROMA_PROXY[13].val = 0x80e80d apr.QUANT_LUMA_PROXY[14].val = 0xffffff apr.QUANT_CHROMA_PROXY[14].val = 0xffffff apr.QUANT_LUMA_PROXY[15].val = 0xffffff apr.QUANT_CHROMA_PROXY[15].val = 0xffffff apr.QUANT_LUMA_PROXY[16].val = 0x80d80b apr.QUANT_CHROMA_PROXY[16].val = 0x80d80b apr.QUANT_LUMA_PROXY[17].val = 0xfff80e apr.QUANT_CHROMA_PROXY[17].val = 0xfff80e apr.QUANT_LUMA_PROXY[18].val = 0xffffff apr.QUANT_CHROMA_PROXY[18].val = 0xffffff apr.QUANT_LUMA_PROXY[19].val = 0xffffff apr.QUANT_CHROMA_PROXY[19].val = 0xffffff apr.QUANT_LUMA_PROXY[20].val = 0x80e80d apr.QUANT_CHROMA_PROXY[20].val = 0x80e80d apr.QUANT_LUMA_PROXY[21].val = 0xffffff apr.QUANT_CHROMA_PROXY[21].val = 0xffffff apr.QUANT_LUMA_PROXY[22].val = 0xffffff apr.QUANT_CHROMA_PROXY[22].val = 0xffffff apr.QUANT_LUMA_PROXY[23].val = 0xffffff apr.QUANT_CHROMA_PROXY[23].val = 0xffffff apr.QUANT_LUMA_PROXY[24].val = 0xfff80d apr.QUANT_CHROMA_PROXY[24].val = 0xfff80d apr.QUANT_LUMA_PROXY[25].val = 0xffffff apr.QUANT_CHROMA_PROXY[25].val = 0xffffff apr.QUANT_LUMA_PROXY[26].val = 0xffffff apr.QUANT_CHROMA_PROXY[26].val = 0xffffff apr.QUANT_LUMA_PROXY[27].val = 0xffffff apr.QUANT_CHROMA_PROXY[27].val = 0xffffff apr.QUANT_LUMA_PROXY[28].val = 0xffffff apr.QUANT_CHROMA_PROXY[28].val = 0xffffff apr.QUANT_LUMA_PROXY[29].val = 0xffffff apr.QUANT_CHROMA_PROXY[29].val = 0xffffff apr.QUANT_LUMA_PROXY[30].val = 0xffffff apr.QUANT_CHROMA_PROXY[30].val = 0xffffff apr.QUANT_LUMA_PROXY[31].val = 0xffffff apr.QUANT_CHROMA_PROXY[31].val = 0xffffff apr.DC_QUANT_SCALE[0].val = 0x401 apr.DC_QUANT_SCALE[1].val = 0x803 apr.DC_QUANT_SCALE[2].val = 0xc05 apr.DC_QUANT_SCALE[3].val = 0x1007 apr.DC_QUANT_SCALE[4].val = 0x1409 apr.DC_QUANT_SCALE[5].val = 0x180b apr.DC_QUANT_SCALE[6].val = 0x1c0d apr.DC_QUANT_SCALE[7].val = 0x200f apr.DC_QUANT_SCALE[8].val = 0x2411 apr.DC_QUANT_SCALE[9].val = 0x2813 apr.DC_QUANT_SCALE[10].val = 0x2c15 apr.DC_QUANT_SCALE[11].val = 0x3017 apr.DC_QUANT_SCALE[12].val = 0x3419 apr.DC_QUANT_SCALE[13].val = 0x381b apr.DC_QUANT_SCALE[14].val = 0x3c1d apr.DC_QUANT_SCALE[15].val = 0x401f apr.DC_QUANT_SCALE[16].val = 0x4421 apr.DC_QUANT_SCALE[17].val = 0x4823 apr.DC_QUANT_SCALE[18].val = 0x4c25 apr.DC_QUANT_SCALE[19].val = 0x5027 apr.DC_QUANT_SCALE[20].val = 0x5429 apr.DC_QUANT_SCALE[21].val = 0x582b apr.DC_QUANT_SCALE[22].val = 0x5c2d apr.DC_QUANT_SCALE[23].val = 0x602f apr.DC_QUANT_SCALE[24].val = 0x6431 apr.DC_QUANT_SCALE[25].val = 0x6833 apr.DC_QUANT_SCALE[26].val = 0x6c35 apr.DC_QUANT_SCALE[27].val = 0x7037 apr.DC_QUANT_SCALE[28].val = 0x7439 apr.DC_QUANT_SCALE[29].val = 0x783b apr.DC_QUANT_SCALE[30].val = 0x7c3d apr.DC_QUANT_SCALE[31].val = 0x803f apr.DC_QUANT_SCALE[32].val = 0x8441 apr.DC_QUANT_SCALE[33].val = 0x8843 apr.DC_QUANT_SCALE[34].val = 0x8c45 apr.DC_QUANT_SCALE[35].val = 0x9047 apr.DC_QUANT_SCALE[36].val = 0x9449 apr.DC_QUANT_SCALE[37].val = 0x984b apr.DC_QUANT_SCALE[38].val = 0x9c4d apr.DC_QUANT_SCALE[39].val = 0xa04f apr.DC_QUANT_SCALE[40].val = 0xa451 apr.DC_QUANT_SCALE[41].val = 0xa853 apr.DC_QUANT_SCALE[42].val = 0xac55 apr.DC_QUANT_SCALE[43].val = 0xb057 apr.DC_QUANT_SCALE[44].val = 0xb459 apr.DC_QUANT_SCALE[45].val = 0xb85b apr.DC_QUANT_SCALE[46].val = 0xbc5d apr.DC_QUANT_SCALE[47].val = 0xc05f apr.DC_QUANT_SCALE[48].val = 0xc461 apr.DC_QUANT_SCALE[49].val = 0xc863 apr.DC_QUANT_SCALE[50].val = 0xcc65 apr.DC_QUANT_SCALE[51].val = 0xd067 apr.DC_QUANT_SCALE[52].val = 0xd469 apr.DC_QUANT_SCALE[53].val = 0xd86b apr.DC_QUANT_SCALE[54].val = 0xdc6d apr.DC_QUANT_SCALE[55].val = 0xe06f apr.DC_QUANT_SCALE[56].val = 0xe471 apr.DC_QUANT_SCALE[57].val = 0xe873 apr.DC_QUANT_SCALE[58].val = 0xec75 apr.DC_QUANT_SCALE[59].val = 0xf077 apr.DC_QUANT_SCALE[60].val = 0xf479 apr.DC_QUANT_SCALE[61].val = 0xf87b apr.DC_QUANT_SCALE[62].val = 0xfc7d apr.DC_QUANT_SCALE[63].val = 0x1007f apr.DC_QUANT_SCALE[64].val = 0x11084 apr.DC_QUANT_SCALE[65].val = 0x1208c apr.DC_QUANT_SCALE[66].val = 0x13094 apr.DC_QUANT_SCALE[67].val = 0x1409c apr.DC_QUANT_SCALE[68].val = 0x150a4 apr.DC_QUANT_SCALE[69].val = 0x160ac apr.DC_QUANT_SCALE[70].val = 0x170b4 apr.DC_QUANT_SCALE[71].val = 0x180bc apr.DC_QUANT_SCALE[72].val = 0x190c4 apr.DC_QUANT_SCALE[73].val = 0x1a0cc apr.DC_QUANT_SCALE[74].val = 0x1b0d4 apr.DC_QUANT_SCALE[75].val = 0x1c0dc apr.DC_QUANT_SCALE[76].val = 0x1d0e4 apr.DC_QUANT_SCALE[77].val = 0x1e0ec apr.DC_QUANT_SCALE[78].val = 0x1f0f4 apr.DC_QUANT_SCALE[79].val = 0x200fc apr.DC_QUANT_SCALE[80].val = 0x21104 apr.DC_QUANT_SCALE[81].val = 0x2210c apr.DC_QUANT_SCALE[82].val = 0x23114 apr.DC_QUANT_SCALE[83].val = 0x2411c apr.DC_QUANT_SCALE[84].val = 0x25124 apr.DC_QUANT_SCALE[85].val = 0x2612c apr.DC_QUANT_SCALE[86].val = 0x27134 apr.DC_QUANT_SCALE[87].val = 0x2813c apr.DC_QUANT_SCALE[88].val = 0x29144 apr.DC_QUANT_SCALE[89].val = 0x2a14c apr.DC_QUANT_SCALE[90].val = 0x2b154 apr.DC_QUANT_SCALE[91].val = 0x2c15c apr.DC_QUANT_SCALE[92].val = 0x2d164 apr.DC_QUANT_SCALE[93].val = 0x2e16c apr.DC_QUANT_SCALE[94].val = 0x2f174 apr.DC_QUANT_SCALE[95].val = 0x3017c apr.DC_QUANT_SCALE[96].val = 0x31184 apr.DC_QUANT_SCALE[97].val = 0x3218c apr.DC_QUANT_SCALE[98].val = 0x33194 apr.DC_QUANT_SCALE[99].val = 0x3419c apr.DC_QUANT_SCALE[100].val = 0x351a4 apr.DC_QUANT_SCALE[101].val = 0x361ac apr.DC_QUANT_SCALE[102].val = 0x371b4 apr.DC_QUANT_SCALE[103].val = 0x381bc apr.DC_QUANT_SCALE[104].val = 0x391c4 apr.DC_QUANT_SCALE[105].val = 0x3a1cc apr.DC_QUANT_SCALE[106].val = 0x3b1d4 apr.DC_QUANT_SCALE[107].val = 0x3c1dc apr.DC_QUANT_SCALE[108].val = 0x3d1e4 apr.DC_QUANT_SCALE[109].val = 0x3e1ec apr.DC_QUANT_SCALE[110].val = 0x3f1f4 apr.DC_QUANT_SCALE[111].val = 0x1fc print("Set matrices") # dunno how this gets calculated OUT_SZ = 0x1000000 out_buf_phys = u.heap.memalign(0x4000, OUT_SZ) iface.writemem(out_buf_phys, b'\xAA' * OUT_SZ) out_buf_iova = dart.iomap(0, out_buf_phys, OUT_SZ) print(f"Output buffer @ phys {out_buf_phys:016X} iova {out_buf_iova:016X}") IN_SZ_LUMA = align_up(im_W*im_H) in_buf_luma_phys = u.heap.memalign(0x4000, IN_SZ_LUMA) iface.writemem(in_buf_luma_phys, image_data_luma + b'\xaa' * (IN_SZ_LUMA - len(image_data_luma))) in_buf_luma_iova = dart.iomap(0, in_buf_luma_phys, IN_SZ_LUMA) print(f"Input buffer luma @ phys {in_buf_luma_phys:016X} iova {in_buf_luma_iova:016X}") IN_SZ_CHROMA = align_up(im_W*2*im_H) in_buf_chroma_phys = u.heap.memalign(0x4000, IN_SZ_CHROMA) iface.writemem(in_buf_chroma_phys, image_data_chroma + b'\xaa' * (IN_SZ_CHROMA - len(image_data_chroma))) in_buf_chroma_iova = dart.iomap(0, in_buf_chroma_phys, IN_SZ_CHROMA) print(f"Input buffer chroma @ phys {in_buf_chroma_phys:016X} iova {in_buf_chroma_iova:016X}") dart.dump_all() desc = EncodeNotRawDescriptor( flags=0x373c, flags2=0, output_iova=out_buf_iova, max_out_sz=OUT_SZ, offset_x=0, offset_y=0, # this is the important set pix_surface_w_2_=im_W, pix_surface_h_2_=im_H, # changing this doesn't seem to break anything pix_surface_w=im_W, pix_surface_h=im_H, # XXX how does the div work exactly? it's different in "tiled" mode luma_stride=divroundup(im_W, 64), chroma_stride=divroundup(im_W*2, 64), alpha_stride=divroundup(im_W, 64), unk_pad_0x26_=b'\x00\x00', luma_iova=in_buf_luma_iova, pix_plane0_tileheader_thing_=0, chroma_iova=in_buf_chroma_iova, pix_plane1_tileheader_thing_=0, alpha_iova=in_buf_luma_iova, pix_plane2_tileheader_thing_=0, # changing this does add extra 0 bytes frame_header_sz=bswp16(0x94), unk_pad_0x5a_=b'\x00', bitstream_version=0, encoder_identifier=0xcafeface, # cannot change arbitrarily, will break pix_surface_w_byteswap_=bswp16(im_W), pix_surface_h_byteswap_=bswp16(im_H), # seemingly can change arbitrarily chroma_format_interlace_mode=0xc0, aspect_ratio_frame_rate=0, color_primaries=2, transfer_characteristic=2, matrix_coefficients=1, alpha_channel_type=1, # tables will still be output even if bits not set here frame_hdr_reserved14=b'\x00\x03', unk_pad_0x6c_=b'\x00' * 128, deprecated_number_of_slices=0, # this one affects the encoding not just the header log2_desired_slice_size_in_mb=0x30, quantization_index=0x2, # this impacts the quality somehow, not quite understood # might be a target bitrate unk_0xf0_=0xffff, unk_0xf2_=0xffff, # none of this stuff is understood, and it never seems to change unk_0xf4_=0x8000402015100c0c, unk_0xfc_=0x2c8080, unk_0x100_0_=0x880080, unk_0x100_1_=0x4e00c5, unk_0x100_2_=0x9000d0, unk_0x100_3_=0x200122, unk_0x110_0_=0x400200, # looks like a quant table, but ??? not used ??? unk_0x110_1_=0x400200, unk_0x110_2_=0x400200, unk_0x110_3_=0x400200, unk_0x110_4_=0x400200, unk_0x110_5_=0x400200, unk_0x110_6_=0x400200, unk_0x110_7_=0x400200, unk_0x110_8_=0x400200, unk_0x110_9_=0x400200, unk_0x110_10_=0x400200, unk_0x110_11_=0x400200, unk_0x110_12_=0x400200, unk_0x110_13_=0x400200, unk_0x110_14_=0x400200, unk_0x110_15_=0x400200, quant_table_sel=0x23, unk_pad_0x154_=b'\x00' * 44, ) desc_bytes = struct.pack(ENCODE_NOT_RAW_STRUCT, *desc) chexdump(desc_bytes) iface.writemem(desc_ring_phys, desc_bytes) # let's go apr.DR_HEAD = len(desc_bytes) start_time = time.time() while apr.IRQ_STATUS.val == 0: if time.time() - start_time > 5: print("TIMED OUT!!!") break print(f"Done, IRQ status is {apr.IRQ_STATUS}") print(f"ST0 = {apr.ST0}") print(f"ST1 = {apr.ST1}") print(f"REG_0x1c = {apr.REG_0x1c}") print(f"REG_0x3c = {apr.REG_0x3c}") print(f"REG_0x44 = {apr.REG_0x44}") print(f"DR_HEAD = {apr.DR_HEAD}") print(f"DR_TAIL = {apr.DR_TAIL}") print(f"unk REG_0x38 = {apr.REG_0x38}") print(f"unk REG_0x40 = {apr.REG_0x40}") print(f"unk REG_0x48 = {apr.REG_0x48}") print(f"unk REG_0x50 = {apr.REG_0x50}") print(f"unk REG_0x54 = {apr.REG_0x54}") apr.IRQ_STATUS = apr.IRQ_STATUS.val dr_memory_new = iface.readmem(desc_ring_phys, DESC_RING_SZ) chexdump(dr_memory_new) out_buf_new = iface.readmem(out_buf_phys, OUT_SZ) with open('prores.bin', 'wb') as f: f.write(out_buf_new) outlen = struct.unpack(">I", out_buf_new[:4])[0] if outlen <= len(out_buf_new): with open('prores.mov', 'wb') as f: f.write(b'\x00\x00\x00\x14\x66\x74\x79\x70\x71\x74\x20\x20\x00\x00\x02\x00\x71\x74\x20\x20\x00\x00\x00\x08\x77\x69\x64\x65') f.write(struct.pack(">I", outlen + 8)) f.write(b'mdat') f.write(out_buf_new[:outlen]) f.write(b'\x00\x00\x03\x1e\x6d\x6f\x6f\x76\x00\x00\x00\x6c\x6d\x76\x68\x64\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x03\xe8\x00\x00\x00\x11\x00\x01\x00\x00\x01\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x01\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x01\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x40\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x02\x00\x00\x02\x89\x74\x72\x61\x6b\x00\x00\x00\x5c\x74\x6b\x68\x64\x00\x00\x00\x03\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x01\x00\x00\x00\x00\x00\x00\x00\x11\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x01\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x01\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x40\x00\x00\x00\x07\x80\x00\x00\x04\x38\x00\x00\x00\x00\x00\x24\x65\x64\x74\x73\x00\x00\x00\x1c\x65\x6c\x73\x74\x00\x00\x00\x00\x00\x00\x00\x01\x00\x00\x00\x11\x00\x00\x00\x00\x00\x01\x00\x00\x00\x00\x02\x01\x6d\x64\x69\x61\x00\x00\x00\x20\x6d\x64\x68\x64\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x3c\x00\x00\x00\x01\x00\x7f\xff\x00\x00\x00\x00\x00\x2d\x68\x64\x6c\x72\x00\x00\x00\x00\x6d\x68\x6c\x72\x76\x69\x64\x65\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x0c\x56\x69\x64\x65\x6f\x48\x61\x6e\x64\x6c\x65\x72\x00\x00\x01\xac\x6d\x69\x6e\x66\x00\x00\x00\x14\x76\x6d\x68\x64\x00\x00\x00\x01\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x2c\x68\x64\x6c\x72\x00\x00\x00\x00\x64\x68\x6c\x72\x75\x72\x6c\x20\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x0b\x44\x61\x74\x61\x48\x61\x6e\x64\x6c\x65\x72\x00\x00\x00\x24\x64\x69\x6e\x66\x00\x00\x00\x1c\x64\x72\x65\x66\x00\x00\x00\x00\x00\x00\x00\x01\x00\x00\x00\x0c\x75\x72\x6c\x20\x00\x00\x00\x01\x00\x00\x01\x40\x73\x74\x62\x6c\x00\x00\x00\xdc\x73\x74\x73\x64\x00\x00\x00\x00\x00\x00\x00\x01\x00\x00\x00\xcc\x61\x70\x63\x6e\x00\x00\x00\x00\x00\x00\x00\x01\x00\x00\x00\x00\x46\x46\x4d\x50\x00\x00\x02\x00\x00\x00\x02\x00\x07\x80\x04\x38\x00\x48\x00\x00\x00\x48\x00\x00\x00\x00\x00\x00\x00\x01\x1f\x4c\x61\x76\x63\x35\x39\x2e\x32\x35\x2e\x31\x30\x30\x20\x70\x72\x6f\x72\x65\x73\x5f\x76\x69\x64\x65\x6f\x74\x6f\x6f\x6c\x62\x00\x18\xff\xff\x00\x00\x00\x6c\x67\x6c\x62\x6c\x00\x00\x00\x64\x61\x70\x63\x6e\x00\x00\x00\x00\x00\x00\xff\xff\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x03\xff\x07\x80\x04\x38\x00\x48\x00\x00\x00\x48\x00\x00\x00\x00\x00\x00\x00\x01\x10\x41\x70\x70\x6c\x65\x20\x50\x72\x6f\x52\x65\x73\x20\x34\x32\x32\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x18\xff\xff\x00\x00\x00\x0a\x66\x69\x65\x6c\x01\x00\x00\x00\x00\x00\x00\x00\x00\x0a\x66\x69\x65\x6c\x01\x00\x00\x00\x00\x18\x73\x74\x74\x73\x00\x00\x00\x00\x00\x00\x00\x01\x00\x00\x00\x01\x00\x00\x01\x00\x00\x00\x00\x1c\x73\x74\x73\x63\x00\x00\x00\x00\x00\x00\x00\x01\x00\x00\x00\x01\x00\x00\x00\x01\x00\x00\x00\x01\x00\x00\x00\x14\x73\x74\x73\x7a\x00\x00\x00\x00') f.write(struct.pack(">I", outlen)) f.write(b'\x00\x00\x00\x01\x00\x00\x00\x14\x73\x74\x63\x6f\x00\x00\x00\x00\x00\x00\x00\x01\x00\x00\x00\x24\x00\x00\x00\x21\x75\x64\x74\x61\x00\x00\x00\x19\xa9\x73\x77\x72\x00\x0d\x55\xc4\x4c\x61\x76\x66\x35\x39\x2e\x32\x30\x2e\x31\x30\x31') # ffmpeg -i prores.mov prores-dec%d.png