upsilon/gateware/soc.py

"""
##########################################################################
# Portions of this file incorporate code licensed under the
# BSD 2-Clause License.
#
# Copyright (c) 2015-2019 Florent Kermarrec <florent@enjoy-digital.fr>
# Copyright (c) 2020 Antmicro <www.antmicro.com>
# Copyright (c) 2022 Victor Suarez Rovere <suarezvictor@gmail.com>
# BSD 2-Clause License
# 
# Copyright (c) Copyright 2012-2022 Enjoy-Digital.
# Copyright (c) Copyright 2012-2022 / LiteX-Hub community.
# All rights reserved.
# 
# Redistribution and use in source and binary forms, with or without
# modification, are permitted provided that the following conditions are met:
# 
# 1. Redistributions of source code must retain the above copyright notice, this
#    list of conditions and the following disclaimer.
# 
# 2. Redistributions in binary form must reproduce the above copyright notice,
#    this list of conditions and the following disclaimer in the documentation
#    and/or other materials provided with the distribution.
# 
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
# AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
# DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE
# FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
# DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
# SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
# CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
# OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
##########################################################################
# Copyright 2023 (C) Peter McGoron
#
# This file is a part of Upsilon, a free and open source software project.
# For license terms, refer to the files in `doc/copying` in the Upsilon
# source distribution.
"""

# There is nothing fundamental about the Arty A7(35|100)T to this
# design, but another eval board will require some porting.
from migen import *
import litex_boards.platforms.digilent_arty as board_spec
from litex.soc.cores.gpio import GPIOTristate
from litex.soc.integration.builder import Builder
from litex.build.generic_platform import IOStandard, Pins, Subsignal
from litex.soc.integration.soc_core import SoCCore
from litex.soc.cores.clock import S7PLL, S7IDELAYCTRL
from litex.soc.interconnect.csr import AutoCSR, Module, CSRStorage, CSRStatus

from litedram.phy import s7ddrphy
from litedram.modules import MT41K128M16
from litedram.frontend.dma import LiteDRAMDMAReader
from liteeth.phy.mii import LiteEthPHYMII

"""
Refer to `A7-constraints.xdc` for pin names.
DAC: SS MOSI MISO SCK
  0:  1    2    3   4 (PMOD A top, right to left)
  1:  1    2    3   4 (PMOD A bottom, right to left)
  2:  1    2    3   4 (PMOD B top, right to left)
  3:  0    1    2   3 (Analog header)
  4:  0    1    2   3 (PMOD C top, right to left)
  5:  4    5    6   8 (Analog header)
  6:  1    2    3   4 (PMOD D top, right to left)
  7:  1    2    3   4 (PMOD D bottom, right to left)


Outer chip header (C=CONV, K=SCK, D=SDO, XX=not connected)
26  27  28  29  30  31  32  33  34  35  36  37  38  39  40  41
C4  K4  D4  C5  K5  D5  XX  XX  C6  K6  D6  C7  K7  D7  XX  XX
C0  K0  D0  C1  K1  D1  XX  XX  C2  K2  D2  C3  K3  D3
0   1   2   3   4   5   6   7   8   9   10  11  12  13

The `io` list maps hardware pins to names used by the SoC
generator. These pins are then connected to Verilog modules.

If there is more than one pin in the Pins string, the resulting
name will be a vector of pins.
"""
io = [
	("differntial_output_low", 0, Pins("J17 J18 K15 J15 U14 V14 T13 U13 B6 E5 A3"), IOStandard("LVCMOS33")),
	("dac_ss_L", 0, Pins("G13 D13 E15 F5 U12 D7 D4 E2"), IOStandard("LVCMOS33")),
	("dac_mosi", 0, Pins("B11 B18 E16 D8 V12 D5 D3 D2"), IOStandard("LVCMOS33")),
	("dac_miso", 0, Pins("A11 A18 D15 C7 V10 B7 F4 H2"), IOStandard("LVCMOS33")),
	("dac_sck", 0, Pins("D12 K16 C15 E7 V11 E6 F3 G2"), IOStandard("LVCMOS33")),
	("adc_conv", 0, Pins("V15 T11 N15 U18 U11 R10 R16 U17"), IOStandard("LVCMOS33")),
	("adc_sck", 0, Pins("U16 R12 M16 R17 V16 R11 N16 T18"), IOStandard("LVCMOS33")),
	("adc_sdo", 0, Pins("P14 T14 V17 P17 M13 R13 N14 R18"), IOStandard("LVCMOS33")),
	("module_reset", 0, Pins("D9"), IOStandard("LVCMOS33")),
	("test_clock", 0, Pins("P18"), IOStandard("LVCMOS33"))
]

# TODO: Assign widths to ADCs here using parameters

class Base(Module, AutoCSR):
	""" The subclass AutoCSR will automatically make CSRs related
	to this class when those CSRs are attributes (i.e. accessed by
	`self.csr_name`) of instances of this class. (CSRs are MMIO,
    they are NOT RISC-V CSRs!)

	Since there are a lot of input and output wires, the CSRs are
	assigned using `setattr()`.

	CSRs are for input wires (`CSRStorage`) or output wires
	(`CSRStatus`).	The first argument to the CSR constructor is
	the amount of bits the CSR takes. The `name` keyword argument
	is required since the constructor needs the name of the attribute.
	The `description` keyword is used for documentation.

	In LiteX, modules in separate Verilog files are instantiated as
	    self.specials += Instance(
	        "module_name",
	        PARAMETER_NAME=value,
	        i_input = input_port,
	        o_output = output_port,
	        ...
	    )

	Since the "base" module has a bunch of repeated input and output
	pins that have to be connected to CSRs, the LiteX wrapper uses
	keyword arguments to pass all the arguments.
	"""

	def _make_csr(self, name, csrclass, csrlen, description, num=None):
		""" Add a CSR for a pin `f"{name}_{num}"` with CSR type
		`csrclass`. This will automatically handle the `i_` and
		`o_` prefix in the keyword arguments.

        This function is used to automate the creation of memory mapped
        IO pins for all the converters on the device.

        `csrclass` must be CSRStorage (Read-Write) or CSRStatus (Read only).
        `csrlen` is the length in bits of the MMIO register. LiteX automatically
        takes care of byte alignment, etc. so the length can be any positive
        number.

        Description is optional but recommended for debugging.
		"""

		if name not in self.csrdict.keys():
			self.csrdict[name] = csrlen
		if num is not None:
			name = f"{name}_{num}"

		csr = csrclass(csrlen, name=name, description=description)
		setattr(self, name, csr)

		if csrclass is CSRStorage:
			self.kwargs[f'i_{name}'] = csr.storage
		elif csrclass is CSRStatus:
			self.kwargs[f'o_{name}'] = csr.status
		else:
			raise Exception(f"Unknown class {csrclass}")

	def __init__(self, clk, sdram, platform):
		self.kwargs = {}
		self.csrdict = {}

		for i in range(0,8):
			self._make_csr("adc_sel", CSRStorage, 3, f"Select ADC {i} Output", num=i)
			self._make_csr("dac_sel", CSRStorage, 3, f"Select DAC {i} Output", num=i)
			self._make_csr("dac_finished", CSRStatus, 1, f"DAC {i} Transmission Finished Flag", num=i)
			self._make_csr("dac_arm", CSRStorage, 1, f"DAC {i} Arm Flag", num=i)
			self._make_csr("from_dac", CSRStatus, 24, f"DAC {i} Received Data", num=i)
			self._make_csr("to_dac", CSRStorage, 24, f"DAC {i} Data to Send", num=i)
#			self._make_csr("wf_arm", CSRStorage, 1, f"Waveform {i} Arm Flag", num=i)
#			self._make_csr("wf_halt_on_finish", CSRStorage, 1, f"Waveform {i} Halt on Finish Flag", num=i)
#			self._make_csr("wf_finished", CSRStatus, 1, f"Waveform {i} Finished Flag", num=i)
#			self._make_csr("wf_running", CSRStatus, 1, f"Waveform {i} Running Flag", num=i)
#			self._make_csr("wf_time_to_wait", CSRStorage, 16, f"Waveform {i} Wait Time", num=i)
#			self._make_csr("wf_refresh_start", CSRStorage, 1, f"Waveform {i} Data Refresh Start Flag", num=i)
#			self._make_csr("wf_refresh_finished", CSRStatus, 1, f"Waveform {i} Data Refresh Finished Flag", num=i)
#			self._make_csr("wf_start_addr", CSRStorage, 32, f"Waveform {i} Data Addr", num=i)
#
#			port = sdram.crossbar.get_port()
#			setattr(self, f"wf_sdram_{i}", LiteDRAMDMAReader(port))
#			cur_sdram = getattr(self, f"wf_sdram_{i}")
#
#			self.kwargs[f"o_wf_ram_dma_addr_{i}"] = cur_sdram.sink.address
#			self.kwargs[f"i_wf_ram_word_{i}"] = cur_sdram.source.data
#			self.kwargs[f"o_wf_ram_read_{i}"] = cur_sdram.sink.valid
#			self.kwargs[f"i_wf_ram_valid_{i}"] = cur_sdram.source.valid

			self._make_csr("adc_finished", CSRStatus, 1, f"ADC {i} Finished Flag", num=i)
			self._make_csr("adc_arm", CSRStorage, 1, f"ADC {i} Arm Flag", num=i)
			self._make_csr("from_adc", CSRStatus, 32, f"ADC {i} Received Data", num=i)

		self._make_csr("cl_in_loop", CSRStatus, 1, "Control Loop Loop Enabled Flag")
		self._make_csr("cl_cmd", CSRStorage, 8, "Control Loop Command Input")
		self._make_csr("cl_word_in", CSRStorage, 64, "Control Loop Data Input")
		self._make_csr("cl_word_out", CSRStatus, 64, "Control Loop Data Output")
		self._make_csr("cl_start_cmd", CSRStorage, 1, "Control Loop Command Start Flag")
		self._make_csr("cl_finish_cmd", CSRStatus, 1, "Control Loop Command Finished Flag")

		self.kwargs["i_clk"] = clk
		self.kwargs["i_rst_L"] = ~platform.request("module_reset")
		self.kwargs["i_dac_miso"] = platform.request("dac_miso")
		self.kwargs["o_dac_mosi"] = platform.request("dac_mosi")
		self.kwargs["o_dac_sck"] = platform.request("dac_sck")
		self.kwargs["o_dac_ss_L"] = platform.request("dac_ss_L")
		self.kwargs["o_adc_conv"] = platform.request("adc_conv")
		self.kwargs["i_adc_sdo"] = platform.request("adc_sdo")
		self.kwargs["o_adc_sck"] = platform.request("adc_sck")
		self.kwargs["o_test_clock"] = platform.request("test_clock")
		self.kwargs["o_set_low"] = platform.request("differntial_output_low")

		""" Dump all MMIO pins to a JSON file with their exact bit widths. """
		with open("csr_bitwidth.json", mode='w') as f:
			import json
			json.dump(self.csrdict, f)

		self.specials += Instance("base", **self.kwargs)

# Clock and Reset Generator
# I don't know how this works, I only know that it does.
class _CRG(Module):
	def __init__(self, platform, sys_clk_freq, with_dram, rst_pin):
		self.rst = Signal()
		self.clock_domains.cd_sys	   = ClockDomain()
		self.clock_domains.cd_eth	   = ClockDomain()
		if with_dram:
			self.clock_domains.cd_sys4x	 = ClockDomain()
			self.clock_domains.cd_sys4x_dqs = ClockDomain()
			self.clock_domains.cd_idelay	= ClockDomain()

		# Clk/Rst.
		clk100 = platform.request("clk100")
		rst	= ~rst_pin if rst_pin is not None else 0

		# PLL.
		self.submodules.pll = pll = S7PLL(speedgrade=-1)
		self.comb += pll.reset.eq(rst | self.rst)
		pll.register_clkin(clk100, 100e6)
		pll.create_clkout(self.cd_sys, sys_clk_freq)
		pll.create_clkout(self.cd_eth, 25e6)
		self.comb += platform.request("eth_ref_clk").eq(self.cd_eth.clk)
		platform.add_false_path_constraints(self.cd_sys.clk, pll.clkin) # Ignore sys_clk to pll.clkin path created by SoC's rst.
		if with_dram:
			pll.create_clkout(self.cd_sys4x,	 4*sys_clk_freq)
			pll.create_clkout(self.cd_sys4x_dqs, 4*sys_clk_freq, phase=90)
			pll.create_clkout(self.cd_idelay,	200e6)

		# IdelayCtrl.
		if with_dram:
			self.submodules.idelayctrl = S7IDELAYCTRL(self.cd_idelay)

class UpsilonSoC(SoCCore):
	def __init__(self, variant):
		sys_clk_freq = int(100e6)
		platform = board_spec.Platform(variant=variant, toolchain="f4pga")
		rst = platform.request("cpu_reset")
		self.submodules.crg = _CRG(platform, sys_clk_freq, True, rst)
		"""
		These source files need to be sorted so that modules
		that rely on another module come later. For instance,
		`control_loop` depends on `control_loop_math`, so
		control_loop_math.v comes before control_loop.v

		If you want to add a new verilog file to the design, look at the
		modules that it refers to and place it the files with those modules.

		Since Yosys doesn't support modern Verilog, only put preprocessed
		(if applicable) files here.
		"""
		platform.add_source("rtl/spi/spi_switch_preprocessed.v")
		platform.add_source("rtl/spi/spi_master_preprocessed.v")
		platform.add_source("rtl/spi/spi_master_no_write_preprocessed.v")
		platform.add_source("rtl/spi/spi_master_no_read_preprocessed.v")
		platform.add_source("rtl/spi/spi_master_ss_preprocessed.v")
		platform.add_source("rtl/spi/spi_master_ss_no_write_preprocessed.v")
		platform.add_source("rtl/spi/spi_master_ss_no_read_preprocessed.v")
		platform.add_source("rtl/control_loop/sign_extend.v")
		platform.add_source("rtl/control_loop/intsat.v")
		platform.add_source("rtl/control_loop/boothmul_preprocessed.v")
		platform.add_source("rtl/control_loop/control_loop_math.v")
		platform.add_source("rtl/control_loop/control_loop.v")
#		platform.add_source("rtl/waveform/bram_interface_preprocessed.v")
#		platform.add_source("rtl/waveform/waveform_preprocessed.v")
		platform.add_source("rtl/base/base.v")

		# SoCCore does not have sane defaults (no integrated rom)
		SoCCore.__init__(self,
				clk_freq=sys_clk_freq,
				toolchain="symbiflow", 
				platform = platform,
				bus_standard = "wishbone",
				ident = f"Arty-{variant} F4PGA LiteX VexRiscV Zephyr - Upsilon",
				bus_data_width = 32,
				bus_address_width = 32,
				bus_timeout = int(1e6),
				cpu_type = "vexriscv_smp",
                cpu_count = 1,
                cpu_variant="linux",
				integrated_rom_size=0x20000,
				integrated_sram_size = 0x2000,
				csr_data_width=32,
				csr_address_width=14,
				csr_paging=0x800,
				csr_ordering="big",
				local_ip='192.168.1.50',
				remote_ip='192.168.1.100',
				timer_uptime = True)
		# This initializes the connection to the physical DRAM interface.
		self.submodules.ddrphy = s7ddrphy.A7DDRPHY(platform.request("ddram"),
			memtype		= "DDR3",
			nphases		= 4,
			sys_clk_freq   = sys_clk_freq)
		# Synchronous dynamic ram. This is what controls all access to RAM.
		# This houses the "crossbar", which negotiates all RAM accesses to different
		# modules, including the verilog interfaces (waveforms etc.)
		self.add_sdram("sdram",
			phy		   = self.ddrphy,
			module		= MT41K128M16(sys_clk_freq, "1:4"),
			l2_cache_size = 8192
		)
		self.submodules.ethphy = LiteEthPHYMII(
			clock_pads = platform.request("eth_clocks"),
			pads	   = platform.request("eth"))
		self.add_ethernet(phy=self.ethphy, dynamic_ip=True)

		platform.add_extension(io)
		self.submodules.base = Base(ClockSignal(), self.sdram, platform)

def main():
	soc =UpsilonSoC("a7-100")
	builder = Builder(soc, csr_json="csr.json", compile_software=True)
	builder.build()

if __name__ == "__main__":
	main()