Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add a ROT13 workshop example #273

Open
wants to merge 4 commits into
base: master
Choose a base branch
from
Open
Show file tree
Hide file tree
Changes from 3 commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
7 changes: 7 additions & 0 deletions docs/migen.rst
Original file line number Diff line number Diff line change
Expand Up @@ -206,3 +206,10 @@ your ``build/csr.csv`` says) and see them take effect immediately.

You can see that it takes very little code to take a Signal from HDL and
expose it on the Wishbone bus.

Tying Signals together
^^^^^^^^^^^^^^^^^^^^^^

See ``workshop_rot13.py`` for another example, where two CSRs are connected
(one as an input, one as an output) in order to create a one-character ROT13
encoder/decoder.
92 changes: 92 additions & 0 deletions litex/workshop_rot13.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,92 @@
#!/usr/bin/env python3
# This variable defines all the external programs that this module
# relies on. lxbuildenv reads this variable in order to ensure
# the build will finish without exiting due to missing third-party
# programs.
LX_DEPENDENCIES = ["icestorm", "yosys", "nextpnr-ice40"]
#LX_CONFIG = "skip-git" # This can be useful for workshops

# Import lxbuildenv to integrate the deps/ directory
import os,os.path,shutil,sys,subprocess
sys.path.insert(0, os.path.dirname(__file__))
import lxbuildenv

# Disable pylint's E1101, which breaks completely on migen
#pylint:disable=E1101

from migen import *
from migen.genlib.resetsync import AsyncResetSynchronizer

from litex.soc.integration.soc_core import SoCCore
from litex.soc.integration.builder import Builder
from litex.soc.interconnect.csr import AutoCSR, CSRStatus, CSRStorage, CSRField

from litex_boards.partner.targets.fomu import BaseSoC, add_dfu_suffix

from valentyusb.usbcore import io as usbio
from valentyusb.usbcore.cpu import dummyusb

import argparse

# ROT13 input CSR. Doesn't need to do anything special.
class FomuROT13In(Module, AutoCSR):
def __init__(self):
# 8-Bit CSR (one ASCII character)
self.csr = CSRStorage(8)

# ROT13 output CSR, plus the wiring to automatically create the output from the input CSR.
class FomuROT13Out(Module, AutoCSR):
def __init__(self, rot13_in):
# Set up an 8-bit CSR (one ASCII character)
self.csr = CSRStorage(8)
# There are three cases:
# 1. It's "A" through "M": Add 13, to make the letters "N" through "Z".
# 2. It's "N" through "Z": Remove 13, to make the letters "A" through "M".
# 3. It's something else, so leave it alone.
#
# In all three cases, we want to wire up the "output" signal (self.csr.storage)
# to be equal to something based on the input signal (rot13_in.csr.storage).
self.sync += If( # A-M, a-m
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I think this could do with some more explanation :-)

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Added comments in 5a3c933, let me know if that's what you're looking for.

Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Yeap starting to look good! I do think you might want a comment which says "we only rotate the alphabetic characters" (what about the numbers? :-) and maybe a link to the ROT13 wikipedia page?

Could be good to add a test bench in the future too....

From https://en.wikipedia.org/wiki/ROT13 it looks like memfrob() in the GNU C library could be an excellent target for this type of acceleration :-P

The GNU C library, a set of standard routines available for use in computer programming, contains a function—memfrob()[15]—which has a similar purpose to ROT13, although it is intended for use with arbitrary binary data. The function operates by combining each byte with the binary pattern 00101010 (42) using the exclusive or (XOR) operation.

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Linked the wiki page, added a brief explanation of the overall cipher, mentioned that it's limited to the English alphabet.

(rot13_in.csr.storage >= ord('A')) & (rot13_in.csr.storage <= ord('M')) | (rot13_in.csr.storage >= ord('a')) & (rot13_in.csr.storage <= ord('m')),
self.csr.storage.eq(rot13_in.csr.storage + 13)
).Elif( # N-Z, n-z
(rot13_in.csr.storage >= ord('N')) & (rot13_in.csr.storage <= ord('Z')) | (rot13_in.csr.storage >= ord('n')) & (rot13_in.csr.storage <= ord('z')),
self.csr.storage.eq(rot13_in.csr.storage - 13)
).Else(
self.csr.storage.eq(rot13_in.csr.storage)
)

def main():
parser = argparse.ArgumentParser(
description="Build Fomu Main Gateware")
parser.add_argument(
"--seed", default=0, help="seed to use in nextpnr"
)
parser.add_argument(
"--placer", default="heap", choices=["sa", "heap"], help="which placer to use in nextpnr"
)
parser.add_argument(
"--board", choices=["evt", "pvt", "hacker"], required=True,
help="build for a particular hardware board"
)
args = parser.parse_args()

soc = BaseSoC(args.board, pnr_seed=args.seed, pnr_placer=args.placer, usb_bridge=True)

# Create a CSR-based ROT13 input and output, export the CSRs
rot13_in = FomuROT13In()
soc.submodules.fomu_rot13_in = rot13_in
soc.submodules.fomu_rot13_out = FomuROT13Out(rot13_in)
soc.add_csr("fomu_rot13_in")
soc.add_csr("fomu_rot13_out")

builder = Builder(soc,
output_dir="build", csr_csv="build/csr.csv",
compile_software=False)
vns = builder.build()
soc.do_exit(vns)
add_dfu_suffix(os.path.join('build', 'gateware', 'top.bin'))


if __name__ == "__main__":
main()