This commit is contained in:
Piotr Zierhoffer 2022-07-24 20:39:43 -04:00 committed by GitHub
commit 4f0bcf3aac
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 205 additions and 2 deletions

200
docs/_ext/cell_list.py Normal file
View File

@ -0,0 +1,200 @@
#!/usr/bin/env python3
# -*- coding: utf-8 -*-
#
# Copyright 2020 SkyWater PDK Authors
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# https://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
#
# SPDX-License-Identifier: Apache-2.0
import argparse
import json
import os
import pathlib
import pprint
import sys
import textwrap
from docutils import nodes
from docutils.parsers.rst import Directive
from docutils.statemachine import ViewList
from sphinx.util.nodes import nested_parse_with_titles
from typing import Tuple, List, Dict
# using a list-table here to allow for easier line breaks in description
rst_header_line_char = '-'
rst_header = 'List of cells in :lib:`{libname}`'
rst_template ="""\
{header_line}
{header_underline}
.. list-table::
:header-rows: 1
* - Cell name
- Description
- Type
- Verilog name
{cell_list}
"""
cell_template = """\
* - {cell_name}
- {description}
- {type}
- {verilog_name}
"""
def collect(library_dir) -> Tuple[str, List[str]]:
"""Collect the available definitions for cells in a library
Parameters
----------
library_dir: str or pathlib.Path
Path to a library.
Returns
-------
lib : str
Library name
cells : list of pathlib.Path
definition files for cells in the library.
"""
if not isinstance(library_dir, pathlib.Path):
library_dir = pathlib.Path(library_dir)
libname = None
cells = set()
for p in library_dir.rglob("definition.json"):
if not p.is_file():
continue
cells.add(p)
if libname is None:
with open(str(p), "r") as sample_json:
sample_def = json.load(sample_json)
libname = sample_def['library']
assert len(libname) > 0
cells = list(sorted(cells))
return libname, cells
def generate_rst(library_dir, library_name, cells):
"""Generate the RST paragraph containing basic information about cells
Parameters
----------
library_dir: str or pathlib.Path
Path to a library.
library_name: str
Name of the library
cells: list of pathlib.Path
List of paths to JSON description files
Returns
-------
paragraph: str
Generated paragraph
"""
if not isinstance(library_dir, pathlib.Path):
library_dir = pathlib.Path(library_dir)
paragraph = ""
cell_list = ""
for cell in cells:
with open(str(cell), "r") as c:
cell_json = json.load(c)
cell_list = cell_list + cell_template.format(
cell_name = cell_json['name'],
#description = cell_json['description'].replace("\n", "\n "),
description = textwrap.indent(cell_json['description'], ' ').lstrip(),
type = cell_json['type'],
verilog_name = cell_json['verilog_name']
)
header = rst_header.format(libname = library_name)
paragraph = rst_template.format(
header_line = header,
header_underline = rst_header_line_char * len(header),
cell_list = cell_list
)
return paragraph
# --- Sphinx extension wrapper ---
class CellList(Directive):
def run(self):
env = self.state.document.settings.env
dirname = env.docname.rpartition('/')[0]
libname, cells = collect(dirname)
paragraph = generate_rst(dirname, libname, cells)
# parse rst string to docutils nodes
rst = ViewList()
for i,line in enumerate(paragraph.split('\n')):
rst.append(line, libname+"-cell-list.rst", i+1)
node = nodes.section()
node.document = self.state.document
nested_parse_with_titles(self.state, rst, node)
return node.children
def setup(app):
app.add_directive("cell_list", CellList)
return {
'version': '0.1',
'parallel_read_safe': True,
'parallel_write_safe': True,
}
# --- stand alone, command line operation ---
def main():
parser = argparse.ArgumentParser()
parser.add_argument(
"library_dir",
help="Path to the library.",
type=pathlib.Path,
nargs=1)
args = parser.parse_args()
lib = args.library_dir[0]
print(f"Analysing {lib}")
libname, cells = collect(lib)
print(f"Library name: {libname}, found {len(cells)} cells")
paragraph = generate_rst(lib, libname, cells)
library_dir = pathlib.Path(lib)
cell_list_file = pathlib.Path(library_dir, "cell-list.rst")
try:
with(open(str(cell_list_file), "w")) as c:
c.write(paragraph)
print(f'Generated {cell_list_file}')
except FileNotFoundError:
print(f"ERROR: Failed to create {str(cell_list_file)}", file=sys.stderr)
raise
if __name__ == "__main__":
sys.exit(main())

View File

@ -31,8 +31,9 @@
import docutils import docutils
import os import os
import re import re
# import sys import sys
# sys.path.insert(0, os.path.abspath('.')) # sys.path.insert(0, os.path.abspath('.'))
sys.path.insert(0, os.path.abspath('./_ext'))
# -- Project information ----------------------------------------------------- # -- Project information -----------------------------------------------------
@ -65,6 +66,7 @@ extensions = [
'sphinx.ext.napoleon', 'sphinx.ext.napoleon',
'sphinx.ext.todo', 'sphinx.ext.todo',
'sphinxcontrib_hdl_diagrams', 'sphinxcontrib_hdl_diagrams',
'cell_list',
] ]
# Add any paths that contain templates here, relative to this directory. # Add any paths that contain templates here, relative to this directory.
@ -127,6 +129,7 @@ exclude_patterns = [
'rules/summary/*-key.rst', 'rules/summary/*-key.rst',
'rules/layers/*-key.rst', 'rules/layers/*-key.rst',
'rules/hv/*-key.rst', 'rules/hv/*-key.rst',
'**/cell-list.rst',
] ]
# The name of the Pygments (syntax highlighting) style to use. # The name of the Pygments (syntax highlighting) style to use.

View File

@ -7,7 +7,7 @@ the SkyWater PDK.
It includes tools for decoding things like file names into human readable It includes tools for decoding things like file names into human readable
descriptions. descriptions.
It also includes tools for combining files together. It also includes tools for combining files together and generating list of cells in libraries for the documentation.
License License
======= =======