47 lines
1.6 KiB
Python
47 lines
1.6 KiB
Python
|
from typing import List
|
||
|
|
from ctypes import c_uint64, c_uint32, c_uint8
|
||
|
|
from enum import IntEnum
|
||
|
|
|
||
|
|
from UniTAP.libs.lib_tsi.tsi_types import TSI_R_MEMORY_SIZE, TSI_MEMORY_LAYOUT,\
|
||
|
|
TSI_MEMORY_BLOCK_INDEX, TSI_MEMORY_RESET_W, TSI_MEMORY_WRITE_W
|
||
|
|
from UniTAP.libs.lib_tsi.tsi import TSIX_TS_GetConfigItem, TSIX_TS_SetConfigItem
|
||
|
|
from UniTAP.libs.lib_tsi.tsi_io import DeviceIO
|
||
|
|
|
||
|
|
|
||
|
|
class MemoryManager:
|
||
|
|
class MemoryOwner(IntEnum):
|
||
|
|
MO_PatternGenerator = 0
|
||
|
|
MO_AudioGenerator = 1
|
||
|
|
MO_Events = 2
|
||
|
|
|
||
|
|
MAX_BLOCK_COUNT = 4
|
||
|
|
|
||
|
|
DEFAULT_MEMOTY_ALLOCATION = [534600704, 534600704, 534600704]
|
||
|
|
RESERVED_MEMORY_BYTES = 8192 * (5120 + 1) * 4 + 0x80000
|
||
|
|
|
||
|
|
def __init__(self, io: DeviceIO):
|
||
|
|
self.__io = io
|
||
|
|
|
||
|
|
def get_total_memory(self) -> int:
|
||
|
|
result = self.__io.get(TSI_R_MEMORY_SIZE, c_uint64)[1]
|
||
|
|
return result
|
||
|
|
|
||
|
|
def set_memory_layout(self, layout: List[int]):
|
||
|
|
self.__io.set(TSI_MEMORY_LAYOUT, layout, c_uint64, data_count=len(layout))
|
||
|
|
|
||
|
|
def get_memory_layout(self) -> List[int]:
|
||
|
|
return self.__io.get(TSI_MEMORY_LAYOUT, c_uint64, self.MAX_BLOCK_COUNT)
|
||
|
|
|
||
|
|
def set_memory_block_index(self, index: MemoryOwner):
|
||
|
|
self.__io.set(TSI_MEMORY_BLOCK_INDEX, index.value)
|
||
|
|
|
||
|
|
def make_default(self):
|
||
|
|
self.__io.set(TSI_MEMORY_LAYOUT, self.DEFAULT_MEMOTY_ALLOCATION, c_uint64,
|
||
|
|
data_count=len(self.DEFAULT_MEMOTY_ALLOCATION))
|
||
|
|
|
||
|
|
def reset(self):
|
||
|
|
self.__io.set(TSI_MEMORY_RESET_W, 0)
|
||
|
|
|
||
|
|
def memory_write(self, data, size):
|
||
|
|
self.__io.set(TSI_MEMORY_WRITE_W, bytearray(data), data_type=c_uint8, data_count=size)
|