From 101f1b1cef749aea7d4b1a4897d74868eb77e6c6 Mon Sep 17 00:00:00 2001 From: Florent Kermarrec Date: Sun, 29 Sep 2019 14:22:26 +0200 Subject: [PATCH] soc/integration: add common.py and move helpers from soc_core to it --- litex/soc/integration/common.py | 70 +++++++++++++++++++++++++++++++ litex/soc/integration/soc_core.py | 70 +------------------------------ 2 files changed, 72 insertions(+), 68 deletions(-) create mode 100644 litex/soc/integration/common.py diff --git a/litex/soc/integration/common.py b/litex/soc/integration/common.py new file mode 100644 index 00000000..991680a6 --- /dev/null +++ b/litex/soc/integration/common.py @@ -0,0 +1,70 @@ +# This file is Copyright (c) 2019 Florent Kermarrec +# License: BSD + +import os +import math +import json +import time +import struct +import datetime + +from migen import * + +def mem_decoder(address, size=0x10000000): + address &= ~0x80000000 + size = 2**log2_int(size, False) + assert (address & (size - 1)) == 0 + address >>= 2 # bytes to words aligned + size >>= 2 # bytes to words aligned + return lambda a: (a[log2_int(size):-1] == (address >> log2_int(size))) + +def get_version(with_time=True): + if with_time: + return datetime.datetime.fromtimestamp( + time.time()).strftime("%Y-%m-%d %H:%M:%S") + else: + return datetime.datetime.fromtimestamp( + time.time()).strftime("%Y-%m-%d") + +def get_mem_data(filename_or_regions, endianness="big", mem_size=None): + # create memory regions + if isinstance(filename_or_regions, dict): + regions = filename_or_regions + else: + filename = filename_or_regions + _, ext = os.path.splitext(filename) + if ext == ".json": + f = open(filename, "r") + regions = json.load(f) + f.close() + else: + regions = {filename: "0x00000000"} + + # determine data_size + data_size = 0 + for filename, base in regions.items(): + data_size = max(int(base, 16) + os.path.getsize(filename), data_size) + assert data_size > 0 + if mem_size is not None: + assert data_size < mem_size, ( + "file is too big: {}/{} bytes".format( + data_size, mem_size)) + + # fill data + data = [0]*math.ceil(data_size/4) + for filename, base in regions.items(): + with open(filename, "rb") as f: + i = 0 + while True: + w = f.read(4) + if not w: + break + if len(w) != 4: + for _ in range(len(w), 4): + w += b'\x00' + if endianness == "little": + data[int(base, 16)//4 + i] = struct.unpack("I", w)[0] + i += 1 + return data diff --git a/litex/soc/integration/soc_core.py b/litex/soc/integration/soc_core.py index 21e8d005..171ef8a0 100644 --- a/litex/soc/integration/soc_core.py +++ b/litex/soc/integration/soc_core.py @@ -13,12 +13,7 @@ # License: BSD import os -import struct import inspect -import json -import math -import datetime -import time from operator import itemgetter from migen import * @@ -29,7 +24,7 @@ from litex.soc.cores import identifier, timer, uart from litex.soc.cores import cpu from litex.soc.interconnect.csr import * from litex.soc.interconnect import wishbone, csr_bus, wishbone2csr - +from litex.soc.integration.common import * __all__ = [ "mem_decoder", @@ -42,67 +37,6 @@ __all__ = [ "soc_mini_argdict", ] -# Helpers ------------------------------------------------------------------------------------------ - -def version(with_time=True): - if with_time: - return datetime.datetime.fromtimestamp( - time.time()).strftime("%Y-%m-%d %H:%M:%S") - else: - return datetime.datetime.fromtimestamp( - time.time()).strftime("%Y-%m-%d") - -def get_mem_data(filename_or_regions, endianness="big", mem_size=None): - # create memory regions - if isinstance(filename_or_regions, dict): - regions = filename_or_regions - else: - filename = filename_or_regions - _, ext = os.path.splitext(filename) - if ext == ".json": - f = open(filename, "r") - regions = json.load(f) - f.close() - else: - regions = {filename: "0x00000000"} - - # determine data_size - data_size = 0 - for filename, base in regions.items(): - data_size = max(int(base, 16) + os.path.getsize(filename), data_size) - assert data_size > 0 - if mem_size is not None: - assert data_size < mem_size, ( - "file is too big: {}/{} bytes".format( - data_size, mem_size)) - - # fill data - data = [0]*math.ceil(data_size/4) - for filename, base in regions.items(): - with open(filename, "rb") as f: - i = 0 - while True: - w = f.read(4) - if not w: - break - if len(w) != 4: - for _ in range(len(w), 4): - w += b'\x00' - if endianness == "little": - data[int(base, 16)//4 + i] = struct.unpack("I", w)[0] - i += 1 - return data - -def mem_decoder(address, size=0x10000000): - address &= ~0x80000000 - size = 2**log2_int(size, False) - assert (address & (size - 1)) == 0 - address >>= 2 # bytes to words aligned - size >>= 2 # bytes to words aligned - return lambda a: (a[log2_int(size):-1] == (address >> log2_int(size))) - # SoCController ------------------------------------------------------------------------------------ class SoCController(Module, AutoCSR): @@ -307,7 +241,7 @@ class SoCCore(Module): # Add Identifier if ident: if ident_version: - ident = ident + " " + version() + ident = ident + " " + get_version() self.submodules.identifier = identifier.Identifier(ident) self.add_csr("identifier_mem", allow_user_defined=True) self.config["CLOCK_FREQUENCY"] = int(clk_freq) -- 2.30.2