1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777177817791780178117821783178417851786178717881789179017911792179317941795179617971798179918001801180218031804180518061807180818091810181118121813181418151816181718181819182018211822182318241825182618271828182918301831183218331834183518361837183818391840184118421843184418451846184718481849185018511852185318541855185618571858185918601861186218631864186518661867186818691870187118721873187418751876187718781879188018811882188318841885188618871888188918901891189218931894189518961897189818991900190119021903190419051906190719081909191019111912191319141915191619171918191919201921192219231924192519261927192819291930193119321933193419351936193719381939194019411942194319441945194619471948194919501951195219531954195519561957195819591960196119621963196419651966196719681969197019711972197319741975197619771978197919801981198219831984198519861987198819891990199119921993199419951996199719981999200020012002200320042005200620072008200920102011201220132014201520162017201820192020202120222023202420252026202720282029203020312032203320342035203620372038203920402041204220432044204520462047204820492050205120522053205420552056205720582059206020612062206320642065206620672068206920702071207220732074207520762077207820792080208120822083208420852086208720882089209020912092209320942095209620972098209921002101210221032104210521062107210821092110211121122113211421152116211721182119212021212122212321242125212621272128212921302131213221332134213521362137213821392140214121422143214421452146214721482149215021512152215321542155215621572158215921602161216221632164216521662167216821692170217121722173217421752176217721782179218021812182218321842185218621872188218921902191219221932194219521962197219821992200220122022203220422052206220722082209221022112212221322142215221622172218221922202221222222232224222522262227222822292230223122322233223422352236223722382239224022412242224322442245224622472248224922502251225222532254225522562257225822592260226122622263226422652266226722682269227022712272227322742275227622772278227922802281228222832284228522862287228822892290229122922293229422952296229722982299230023012302230323042305230623072308230923102311231223132314231523162317231823192320232123222323232423252326232723282329233023312332233323342335233623372338233923402341234223432344234523462347234823492350235123522353235423552356235723582359236023612362236323642365236623672368236923702371237223732374237523762377237823792380238123822383238423852386238723882389 |
- #!/usr/bin/env python3
- # This script makes RST files from the XML class reference for use with the online docs.
- import argparse
- import os
- import platform
- import re
- import sys
- import xml.etree.ElementTree as ET
- from collections import OrderedDict
- from typing import List, Dict, TextIO, Tuple, Optional, Any, Union
- # Import hardcoded version information from version.py
- root_directory = os.path.join(os.path.dirname(os.path.abspath(__file__)), "../../")
- sys.path.append(root_directory) # Include the root directory
- import version
- # $DOCS_URL/path/to/page.html(#fragment-tag)
- GODOT_DOCS_PATTERN = re.compile(r"^\$DOCS_URL/(.*)\.html(#.*)?$")
- # Based on reStructuredText inline markup recognition rules
- # https://docutils.sourceforge.io/docs/ref/rst/restructuredtext.html#inline-markup-recognition-rules
- MARKUP_ALLOWED_PRECEDENT = " -:/'\"<([{"
- MARKUP_ALLOWED_SUBSEQUENT = " -.,:;!?\\/'\")]}>"
- # Used to translate section headings and other hardcoded strings when required with
- # the --lang argument. The BASE_STRINGS list should be synced with what we actually
- # write in this script (check `translate()` uses), and also hardcoded in
- # `scripts/extract_classes.py` (godotengine/godot-editor-l10n repo) to include them in the source POT file.
- BASE_STRINGS = [
- "All classes",
- "Globals",
- "Nodes",
- "Resources",
- "Editor-only",
- "Other objects",
- "Variant types",
- "Description",
- "Tutorials",
- "Properties",
- "Constructors",
- "Methods",
- "Operators",
- "Theme Properties",
- "Signals",
- "Enumerations",
- "Constants",
- "Annotations",
- "Property Descriptions",
- "Constructor Descriptions",
- "Method Descriptions",
- "Operator Descriptions",
- "Theme Property Descriptions",
- "Inherits:",
- "Inherited By:",
- "(overrides %s)",
- "Default",
- "Setter",
- "value",
- "Getter",
- "This method should typically be overridden by the user to have any effect.",
- "This method has no side effects. It doesn't modify any of the instance's member variables.",
- "This method accepts any number of arguments after the ones described here.",
- "This method is used to construct a type.",
- "This method doesn't need an instance to be called, so it can be called directly using the class name.",
- "This method describes a valid operator to use with this type as left-hand operand.",
- "This value is an integer composed as a bitmask of the following flags.",
- "There is currently no description for this class. Please help us by :ref:`contributing one <doc_updating_the_class_reference>`!",
- "There is currently no description for this signal. Please help us by :ref:`contributing one <doc_updating_the_class_reference>`!",
- "There is currently no description for this annotation. Please help us by :ref:`contributing one <doc_updating_the_class_reference>`!",
- "There is currently no description for this property. Please help us by :ref:`contributing one <doc_updating_the_class_reference>`!",
- "There is currently no description for this constructor. Please help us by :ref:`contributing one <doc_updating_the_class_reference>`!",
- "There is currently no description for this method. Please help us by :ref:`contributing one <doc_updating_the_class_reference>`!",
- "There is currently no description for this operator. Please help us by :ref:`contributing one <doc_updating_the_class_reference>`!",
- "There is currently no description for this theme property. Please help us by :ref:`contributing one <doc_updating_the_class_reference>`!",
- "There are notable differences when using this API with C#. See :ref:`doc_c_sharp_differences` for more information.",
- ]
- strings_l10n: Dict[str, str] = {}
- STYLES: Dict[str, str] = {}
- CLASS_GROUPS: Dict[str, str] = {
- "global": "Globals",
- "node": "Nodes",
- "resource": "Resources",
- "object": "Other objects",
- "editor": "Editor-only",
- "variant": "Variant types",
- }
- CLASS_GROUPS_BASE: Dict[str, str] = {
- "node": "Node",
- "resource": "Resource",
- "object": "Object",
- "variant": "Variant",
- }
- # Sync with editor\register_editor_types.cpp
- EDITOR_CLASSES: List[str] = [
- "FileSystemDock",
- "ScriptCreateDialog",
- "ScriptEditor",
- "ScriptEditorBase",
- ]
- # Sync with the types mentioned in https://docs.godotengine.org/en/stable/tutorials/scripting/c_sharp/c_sharp_differences.html
- CLASSES_WITH_CSHARP_DIFFERENCES: List[str] = [
- "@GlobalScope",
- "String",
- "StringName",
- "NodePath",
- "Signal",
- "Callable",
- "RID",
- "Basis",
- "Transform2D",
- "Transform3D",
- "Rect2",
- "Rect2i",
- "AABB",
- "Quaternion",
- "Projection",
- "Color",
- "Array",
- "Dictionary",
- "PackedByteArray",
- "PackedColorArray",
- "PackedFloat32Array",
- "PackedFloat64Array",
- "PackedInt32Array",
- "PackedInt64Array",
- "PackedStringArray",
- "PackedVector2Array",
- "PackedVector3Array",
- "Variant",
- ]
- class State:
- def __init__(self) -> None:
- self.num_errors = 0
- self.num_warnings = 0
- self.classes: OrderedDict[str, ClassDef] = OrderedDict()
- self.current_class: str = ""
- def parse_class(self, class_root: ET.Element, filepath: str) -> None:
- class_name = class_root.attrib["name"]
- self.current_class = class_name
- class_def = ClassDef(class_name)
- self.classes[class_name] = class_def
- class_def.filepath = filepath
- inherits = class_root.get("inherits")
- if inherits is not None:
- class_def.inherits = inherits
- brief_desc = class_root.find("brief_description")
- if brief_desc is not None and brief_desc.text:
- class_def.brief_description = brief_desc.text
- desc = class_root.find("description")
- if desc is not None and desc.text:
- class_def.description = desc.text
- properties = class_root.find("members")
- if properties is not None:
- for property in properties:
- assert property.tag == "member"
- property_name = property.attrib["name"]
- if property_name in class_def.properties:
- print_error(f'{class_name}.xml: Duplicate property "{property_name}".', self)
- continue
- type_name = TypeName.from_element(property)
- setter = property.get("setter") or None # Use or None so '' gets turned into None.
- getter = property.get("getter") or None
- default_value = property.get("default") or None
- if default_value is not None:
- default_value = f"``{default_value}``"
- overrides = property.get("overrides") or None
- property_def = PropertyDef(
- property_name, type_name, setter, getter, property.text, default_value, overrides
- )
- class_def.properties[property_name] = property_def
- constructors = class_root.find("constructors")
- if constructors is not None:
- for constructor in constructors:
- assert constructor.tag == "constructor"
- method_name = constructor.attrib["name"]
- qualifiers = constructor.get("qualifiers")
- return_element = constructor.find("return")
- if return_element is not None:
- return_type = TypeName.from_element(return_element)
- else:
- return_type = TypeName("void")
- params = self.parse_params(constructor, "constructor")
- desc_element = constructor.find("description")
- method_desc = None
- if desc_element is not None:
- method_desc = desc_element.text
- method_def = MethodDef(method_name, return_type, params, method_desc, qualifiers)
- method_def.definition_name = "constructor"
- if method_name not in class_def.constructors:
- class_def.constructors[method_name] = []
- class_def.constructors[method_name].append(method_def)
- methods = class_root.find("methods")
- if methods is not None:
- for method in methods:
- assert method.tag == "method"
- method_name = method.attrib["name"]
- qualifiers = method.get("qualifiers")
- return_element = method.find("return")
- if return_element is not None:
- return_type = TypeName.from_element(return_element)
- else:
- return_type = TypeName("void")
- params = self.parse_params(method, "method")
- desc_element = method.find("description")
- method_desc = None
- if desc_element is not None:
- method_desc = desc_element.text
- method_def = MethodDef(method_name, return_type, params, method_desc, qualifiers)
- if method_name not in class_def.methods:
- class_def.methods[method_name] = []
- class_def.methods[method_name].append(method_def)
- operators = class_root.find("operators")
- if operators is not None:
- for operator in operators:
- assert operator.tag == "operator"
- method_name = operator.attrib["name"]
- qualifiers = operator.get("qualifiers")
- return_element = operator.find("return")
- if return_element is not None:
- return_type = TypeName.from_element(return_element)
- else:
- return_type = TypeName("void")
- params = self.parse_params(operator, "operator")
- desc_element = operator.find("description")
- method_desc = None
- if desc_element is not None:
- method_desc = desc_element.text
- method_def = MethodDef(method_name, return_type, params, method_desc, qualifiers)
- method_def.definition_name = "operator"
- if method_name not in class_def.operators:
- class_def.operators[method_name] = []
- class_def.operators[method_name].append(method_def)
- constants = class_root.find("constants")
- if constants is not None:
- for constant in constants:
- assert constant.tag == "constant"
- constant_name = constant.attrib["name"]
- value = constant.attrib["value"]
- enum = constant.get("enum")
- is_bitfield = constant.get("is_bitfield") == "true"
- constant_def = ConstantDef(constant_name, value, constant.text, is_bitfield)
- if enum is None:
- if constant_name in class_def.constants:
- print_error(f'{class_name}.xml: Duplicate constant "{constant_name}".', self)
- continue
- class_def.constants[constant_name] = constant_def
- else:
- if enum in class_def.enums:
- enum_def = class_def.enums[enum]
- else:
- enum_def = EnumDef(enum, TypeName("int", enum), is_bitfield)
- class_def.enums[enum] = enum_def
- enum_def.values[constant_name] = constant_def
- annotations = class_root.find("annotations")
- if annotations is not None:
- for annotation in annotations:
- assert annotation.tag == "annotation"
- annotation_name = annotation.attrib["name"]
- qualifiers = annotation.get("qualifiers")
- params = self.parse_params(annotation, "annotation")
- desc_element = annotation.find("description")
- annotation_desc = None
- if desc_element is not None:
- annotation_desc = desc_element.text
- annotation_def = AnnotationDef(annotation_name, params, annotation_desc, qualifiers)
- if annotation_name not in class_def.annotations:
- class_def.annotations[annotation_name] = []
- class_def.annotations[annotation_name].append(annotation_def)
- signals = class_root.find("signals")
- if signals is not None:
- for signal in signals:
- assert signal.tag == "signal"
- signal_name = signal.attrib["name"]
- if signal_name in class_def.signals:
- print_error(f'{class_name}.xml: Duplicate signal "{signal_name}".', self)
- continue
- params = self.parse_params(signal, "signal")
- desc_element = signal.find("description")
- signal_desc = None
- if desc_element is not None:
- signal_desc = desc_element.text
- signal_def = SignalDef(signal_name, params, signal_desc)
- class_def.signals[signal_name] = signal_def
- theme_items = class_root.find("theme_items")
- if theme_items is not None:
- for theme_item in theme_items:
- assert theme_item.tag == "theme_item"
- theme_item_name = theme_item.attrib["name"]
- theme_item_data_name = theme_item.attrib["data_type"]
- theme_item_id = "{}_{}".format(theme_item_data_name, theme_item_name)
- if theme_item_id in class_def.theme_items:
- print_error(
- f'{class_name}.xml: Duplicate theme item "{theme_item_name}" of type "{theme_item_data_name}".',
- self,
- )
- continue
- default_value = theme_item.get("default") or None
- if default_value is not None:
- default_value = f"``{default_value}``"
- theme_item_def = ThemeItemDef(
- theme_item_name,
- TypeName.from_element(theme_item),
- theme_item_data_name,
- theme_item.text,
- default_value,
- )
- class_def.theme_items[theme_item_name] = theme_item_def
- tutorials = class_root.find("tutorials")
- if tutorials is not None:
- for link in tutorials:
- assert link.tag == "link"
- if link.text is not None:
- class_def.tutorials.append((link.text.strip(), link.get("title", "")))
- self.current_class = ""
- def parse_params(self, root: ET.Element, context: str) -> List["ParameterDef"]:
- param_elements = root.findall("param")
- params: Any = [None] * len(param_elements)
- for param_index, param_element in enumerate(param_elements):
- param_name = param_element.attrib["name"]
- index = int(param_element.attrib["index"])
- type_name = TypeName.from_element(param_element)
- default = param_element.get("default")
- if param_name.strip() == "" or param_name.startswith("_unnamed_arg"):
- print_error(
- f'{self.current_class}.xml: Empty argument name in {context} "{root.attrib["name"]}" at position {param_index}.',
- self,
- )
- params[index] = ParameterDef(param_name, type_name, default)
- cast: List[ParameterDef] = params
- return cast
- def sort_classes(self) -> None:
- self.classes = OrderedDict(sorted(self.classes.items(), key=lambda t: t[0].lower()))
- class TagState:
- def __init__(self, raw: str, name: str, arguments: List[str], closing: bool) -> None:
- self.raw = raw
- self.name = name
- self.arguments = arguments
- self.closing = closing
- class TypeName:
- def __init__(self, type_name: str, enum: Optional[str] = None, is_bitfield: bool = False) -> None:
- self.type_name = type_name
- self.enum = enum
- self.is_bitfield = is_bitfield
- def to_rst(self, state: State) -> str:
- if self.enum is not None:
- return make_enum(self.enum, self.is_bitfield, state)
- elif self.type_name == "void":
- return "void"
- else:
- return make_type(self.type_name, state)
- @classmethod
- def from_element(cls, element: ET.Element) -> "TypeName":
- return cls(element.attrib["type"], element.get("enum"), element.get("is_bitfield") == "true")
- class DefinitionBase:
- def __init__(
- self,
- definition_name: str,
- name: str,
- ) -> None:
- self.definition_name = definition_name
- self.name = name
- class PropertyDef(DefinitionBase):
- def __init__(
- self,
- name: str,
- type_name: TypeName,
- setter: Optional[str],
- getter: Optional[str],
- text: Optional[str],
- default_value: Optional[str],
- overrides: Optional[str],
- ) -> None:
- super().__init__("property", name)
- self.type_name = type_name
- self.setter = setter
- self.getter = getter
- self.text = text
- self.default_value = default_value
- self.overrides = overrides
- class ParameterDef(DefinitionBase):
- def __init__(self, name: str, type_name: TypeName, default_value: Optional[str]) -> None:
- super().__init__("parameter", name)
- self.type_name = type_name
- self.default_value = default_value
- class SignalDef(DefinitionBase):
- def __init__(self, name: str, parameters: List[ParameterDef], description: Optional[str]) -> None:
- super().__init__("signal", name)
- self.parameters = parameters
- self.description = description
- class AnnotationDef(DefinitionBase):
- def __init__(
- self,
- name: str,
- parameters: List[ParameterDef],
- description: Optional[str],
- qualifiers: Optional[str],
- ) -> None:
- super().__init__("annotation", name)
- self.parameters = parameters
- self.description = description
- self.qualifiers = qualifiers
- class MethodDef(DefinitionBase):
- def __init__(
- self,
- name: str,
- return_type: TypeName,
- parameters: List[ParameterDef],
- description: Optional[str],
- qualifiers: Optional[str],
- ) -> None:
- super().__init__("method", name)
- self.return_type = return_type
- self.parameters = parameters
- self.description = description
- self.qualifiers = qualifiers
- class ConstantDef(DefinitionBase):
- def __init__(self, name: str, value: str, text: Optional[str], bitfield: bool) -> None:
- super().__init__("constant", name)
- self.value = value
- self.text = text
- self.is_bitfield = bitfield
- class EnumDef(DefinitionBase):
- def __init__(self, name: str, type_name: TypeName, bitfield: bool) -> None:
- super().__init__("enum", name)
- self.type_name = type_name
- self.values: OrderedDict[str, ConstantDef] = OrderedDict()
- self.is_bitfield = bitfield
- class ThemeItemDef(DefinitionBase):
- def __init__(
- self, name: str, type_name: TypeName, data_name: str, text: Optional[str], default_value: Optional[str]
- ) -> None:
- super().__init__("theme item", name)
- self.type_name = type_name
- self.data_name = data_name
- self.text = text
- self.default_value = default_value
- class ClassDef(DefinitionBase):
- def __init__(self, name: str) -> None:
- super().__init__("class", name)
- self.constants: OrderedDict[str, ConstantDef] = OrderedDict()
- self.enums: OrderedDict[str, EnumDef] = OrderedDict()
- self.properties: OrderedDict[str, PropertyDef] = OrderedDict()
- self.constructors: OrderedDict[str, List[MethodDef]] = OrderedDict()
- self.methods: OrderedDict[str, List[MethodDef]] = OrderedDict()
- self.operators: OrderedDict[str, List[MethodDef]] = OrderedDict()
- self.signals: OrderedDict[str, SignalDef] = OrderedDict()
- self.annotations: OrderedDict[str, List[AnnotationDef]] = OrderedDict()
- self.theme_items: OrderedDict[str, ThemeItemDef] = OrderedDict()
- self.inherits: Optional[str] = None
- self.brief_description: Optional[str] = None
- self.description: Optional[str] = None
- self.tutorials: List[Tuple[str, str]] = []
- # Used to match the class with XML source for output filtering purposes.
- self.filepath: str = ""
- # Entry point for the RST generator.
- def main() -> None:
- # Enable ANSI escape code support on Windows 10 and later (for colored console output).
- # <https://bugs.python.org/issue29059>
- if platform.system().lower() == "windows":
- from ctypes import windll, c_int, byref # type: ignore
- stdout_handle = windll.kernel32.GetStdHandle(c_int(-11))
- mode = c_int(0)
- windll.kernel32.GetConsoleMode(c_int(stdout_handle), byref(mode))
- mode = c_int(mode.value | 4)
- windll.kernel32.SetConsoleMode(c_int(stdout_handle), mode)
- parser = argparse.ArgumentParser()
- parser.add_argument("path", nargs="+", help="A path to an XML file or a directory containing XML files to parse.")
- parser.add_argument("--filter", default="", help="The filepath pattern for XML files to filter.")
- parser.add_argument("--lang", "-l", default="en", help="Language to use for section headings.")
- parser.add_argument(
- "--color",
- action="store_true",
- help="If passed, force colored output even if stdout is not a TTY (useful for continuous integration).",
- )
- group = parser.add_mutually_exclusive_group()
- group.add_argument("--output", "-o", default=".", help="The directory to save output .rst files in.")
- group.add_argument(
- "--dry-run",
- action="store_true",
- help="If passed, no output will be generated and XML files are only checked for errors.",
- )
- args = parser.parse_args()
- should_color = args.color or (hasattr(sys.stdout, "isatty") and sys.stdout.isatty())
- STYLES["red"] = "\x1b[91m" if should_color else ""
- STYLES["green"] = "\x1b[92m" if should_color else ""
- STYLES["yellow"] = "\x1b[93m" if should_color else ""
- STYLES["bold"] = "\x1b[1m" if should_color else ""
- STYLES["regular"] = "\x1b[22m" if should_color else ""
- STYLES["reset"] = "\x1b[0m" if should_color else ""
- # Retrieve heading translations for the given language.
- if not args.dry_run and args.lang != "en":
- lang_file = os.path.join(
- os.path.dirname(os.path.realpath(__file__)), "..", "translations", "{}.po".format(args.lang)
- )
- if os.path.exists(lang_file):
- try:
- import polib # type: ignore
- except ImportError:
- print("Base template strings localization requires `polib`.")
- exit(1)
- pofile = polib.pofile(lang_file)
- for entry in pofile.translated_entries():
- if entry.msgid in BASE_STRINGS:
- strings_l10n[entry.msgid] = entry.msgstr
- else:
- print(f'No PO file at "{lang_file}" for language "{args.lang}".')
- print("Checking for errors in the XML class reference...")
- file_list: List[str] = []
- for path in args.path:
- # Cut off trailing slashes so os.path.basename doesn't choke.
- if path.endswith("/") or path.endswith("\\"):
- path = path[:-1]
- if os.path.basename(path) in ["modules", "platform"]:
- for subdir, dirs, _ in os.walk(path):
- if "doc_classes" in dirs:
- doc_dir = os.path.join(subdir, "doc_classes")
- class_file_names = (f for f in os.listdir(doc_dir) if f.endswith(".xml"))
- file_list += (os.path.join(doc_dir, f) for f in class_file_names)
- elif os.path.isdir(path):
- file_list += (os.path.join(path, f) for f in os.listdir(path) if f.endswith(".xml"))
- elif os.path.isfile(path):
- if not path.endswith(".xml"):
- print(f'Got non-.xml file "{path}" in input, skipping.')
- continue
- file_list.append(path)
- classes: Dict[str, Tuple[ET.Element, str]] = {}
- state = State()
- for cur_file in file_list:
- try:
- tree = ET.parse(cur_file)
- except ET.ParseError as e:
- print_error(f"{cur_file}: Parse error while reading the file: {e}", state)
- continue
- doc = tree.getroot()
- name = doc.attrib["name"]
- if name in classes:
- print_error(f'{cur_file}: Duplicate class "{name}".', state)
- continue
- classes[name] = (doc, cur_file)
- for name, data in classes.items():
- try:
- state.parse_class(data[0], data[1])
- except Exception as e:
- print_error(f"{name}.xml: Exception while parsing class: {e}", state)
- state.sort_classes()
- pattern = re.compile(args.filter)
- # Create the output folder recursively if it doesn't already exist.
- os.makedirs(args.output, exist_ok=True)
- print("Generating the RST class reference...")
- grouped_classes: Dict[str, List[str]] = {}
- for class_name, class_def in state.classes.items():
- if args.filter and not pattern.search(class_def.filepath):
- continue
- state.current_class = class_name
- make_rst_class(class_def, state, args.dry_run, args.output)
- group_name = get_class_group(class_def, state)
- if group_name not in grouped_classes:
- grouped_classes[group_name] = []
- grouped_classes[group_name].append(class_name)
- if is_editor_class(class_def):
- if "editor" not in grouped_classes:
- grouped_classes["editor"] = []
- grouped_classes["editor"].append(class_name)
- print("")
- print("Generating the index file...")
- make_rst_index(grouped_classes, args.dry_run, args.output)
- print("")
- # Print out warnings and errors, or lack thereof, and exit with an appropriate code.
- if state.num_warnings >= 2:
- print(
- f'{STYLES["yellow"]}{state.num_warnings} warnings were found in the class reference XML. Please check the messages above.{STYLES["reset"]}'
- )
- elif state.num_warnings == 1:
- print(
- f'{STYLES["yellow"]}1 warning was found in the class reference XML. Please check the messages above.{STYLES["reset"]}'
- )
- if state.num_errors >= 2:
- print(
- f'{STYLES["red"]}{state.num_errors} errors were found in the class reference XML. Please check the messages above.{STYLES["reset"]}'
- )
- elif state.num_errors == 1:
- print(
- f'{STYLES["red"]}1 error was found in the class reference XML. Please check the messages above.{STYLES["reset"]}'
- )
- if state.num_warnings == 0 and state.num_errors == 0:
- print(f'{STYLES["green"]}No warnings or errors found in the class reference XML.{STYLES["reset"]}')
- if not args.dry_run:
- print(f"Wrote reStructuredText files for each class to: {args.output}")
- else:
- exit(1)
- # Common helpers.
- def print_error(error: str, state: State) -> None:
- print(f'{STYLES["red"]}{STYLES["bold"]}ERROR:{STYLES["regular"]} {error}{STYLES["reset"]}')
- state.num_errors += 1
- def print_warning(warning: str, state: State) -> None:
- print(f'{STYLES["yellow"]}{STYLES["bold"]}WARNING:{STYLES["regular"]} {warning}{STYLES["reset"]}')
- state.num_warnings += 1
- def translate(string: str) -> str:
- """Translate a string based on translations sourced from `doc/translations/*.po`
- for a language if defined via the --lang command line argument.
- Returns the original string if no translation exists.
- """
- return strings_l10n.get(string, string)
- def get_git_branch() -> str:
- if hasattr(version, "docs") and version.docs != "latest":
- return version.docs
- return "master"
- def get_class_group(class_def: ClassDef, state: State) -> str:
- group_name = "variant"
- class_name = class_def.name
- if class_name.startswith("@"):
- group_name = "global"
- elif class_def.inherits:
- inherits = class_def.inherits.strip()
- while inherits in state.classes:
- if inherits == "Node":
- group_name = "node"
- break
- if inherits == "Resource":
- group_name = "resource"
- break
- if inherits == "Object":
- group_name = "object"
- break
- inode = state.classes[inherits].inherits
- if inode:
- inherits = inode.strip()
- else:
- break
- return group_name
- def is_editor_class(class_def: ClassDef) -> bool:
- class_name = class_def.name
- if class_name.startswith("Editor"):
- return True
- if class_name in EDITOR_CLASSES:
- return True
- return False
- # Generator methods.
- def make_rst_class(class_def: ClassDef, state: State, dry_run: bool, output_dir: str) -> None:
- class_name = class_def.name
- if dry_run:
- f = open(os.devnull, "w", encoding="utf-8")
- else:
- f = open(os.path.join(output_dir, f"class_{class_name.lower()}.rst"), "w", encoding="utf-8")
- # Remove the "Edit on Github" button from the online docs page.
- f.write(":github_url: hide\n\n")
- # Warn contributors not to edit this file directly.
- # Also provide links to the source files for reference.
- git_branch = get_git_branch()
- source_xml_path = os.path.relpath(class_def.filepath, root_directory).replace("\\", "/")
- source_github_url = f"https://github.com/godotengine/godot/tree/{git_branch}/{source_xml_path}"
- generator_github_url = f"https://github.com/godotengine/godot/tree/{git_branch}/doc/tools/make_rst.py"
- f.write(".. DO NOT EDIT THIS FILE!!!\n")
- f.write(".. Generated automatically from Godot engine sources.\n")
- f.write(f".. Generator: {generator_github_url}.\n")
- f.write(f".. XML source: {source_github_url}.\n\n")
- # Document reference id and header.
- f.write(f".. _class_{class_name}:\n\n")
- f.write(make_heading(class_name, "=", False))
- ### INHERITANCE TREE ###
- # Ascendants
- if class_def.inherits:
- inherits = class_def.inherits.strip()
- f.write(f'**{translate("Inherits:")}** ')
- first = True
- while inherits in state.classes:
- if not first:
- f.write(" **<** ")
- else:
- first = False
- f.write(make_type(inherits, state))
- inode = state.classes[inherits].inherits
- if inode:
- inherits = inode.strip()
- else:
- break
- f.write("\n\n")
- # Descendants
- inherited: List[str] = []
- for c in state.classes.values():
- if c.inherits and c.inherits.strip() == class_name:
- inherited.append(c.name)
- if len(inherited):
- f.write(f'**{translate("Inherited By:")}** ')
- for i, child in enumerate(inherited):
- if i > 0:
- f.write(", ")
- f.write(make_type(child, state))
- f.write("\n\n")
- ### INTRODUCTION ###
- has_description = False
- # Brief description
- if class_def.brief_description is not None and class_def.brief_description.strip() != "":
- has_description = True
- f.write(f"{format_text_block(class_def.brief_description.strip(), class_def, state)}\n\n")
- # Class description
- if class_def.description is not None and class_def.description.strip() != "":
- has_description = True
- f.write(".. rst-class:: classref-introduction-group\n\n")
- f.write(make_heading("Description", "-"))
- f.write(f"{format_text_block(class_def.description.strip(), class_def, state)}\n\n")
- if not has_description:
- f.write(".. container:: contribute\n\n\t")
- f.write(
- translate(
- "There is currently no description for this class. Please help us by :ref:`contributing one <doc_updating_the_class_reference>`!"
- )
- + "\n\n"
- )
- if class_def.name in CLASSES_WITH_CSHARP_DIFFERENCES:
- f.write(".. note::\n\n\t")
- f.write(
- translate(
- "There are notable differences when using this API with C#. See :ref:`doc_c_sharp_differences` for more information."
- )
- + "\n\n"
- )
- # Online tutorials
- if len(class_def.tutorials) > 0:
- f.write(".. rst-class:: classref-introduction-group\n\n")
- f.write(make_heading("Tutorials", "-"))
- for url, title in class_def.tutorials:
- f.write(f"- {make_link(url, title)}\n\n")
- ### REFERENCE TABLES ###
- # Reused container for reference tables.
- ml: List[Tuple[Optional[str], ...]] = []
- # Properties reference table
- if len(class_def.properties) > 0:
- f.write(".. rst-class:: classref-reftable-group\n\n")
- f.write(make_heading("Properties", "-"))
- ml = []
- for property_def in class_def.properties.values():
- type_rst = property_def.type_name.to_rst(state)
- default = property_def.default_value
- if default is not None and property_def.overrides:
- ref = f":ref:`{property_def.overrides}<class_{property_def.overrides}_property_{property_def.name}>`"
- # Not using translate() for now as it breaks table formatting.
- ml.append((type_rst, property_def.name, f"{default} (overrides {ref})"))
- else:
- ref = f":ref:`{property_def.name}<class_{class_name}_property_{property_def.name}>`"
- ml.append((type_rst, ref, default))
- format_table(f, ml, True)
- # Constructors, Methods, Operators reference tables
- if len(class_def.constructors) > 0:
- f.write(".. rst-class:: classref-reftable-group\n\n")
- f.write(make_heading("Constructors", "-"))
- ml = []
- for method_list in class_def.constructors.values():
- for m in method_list:
- ml.append(make_method_signature(class_def, m, "constructor", state))
- format_table(f, ml)
- if len(class_def.methods) > 0:
- f.write(".. rst-class:: classref-reftable-group\n\n")
- f.write(make_heading("Methods", "-"))
- ml = []
- for method_list in class_def.methods.values():
- for m in method_list:
- ml.append(make_method_signature(class_def, m, "method", state))
- format_table(f, ml)
- if len(class_def.operators) > 0:
- f.write(".. rst-class:: classref-reftable-group\n\n")
- f.write(make_heading("Operators", "-"))
- ml = []
- for method_list in class_def.operators.values():
- for m in method_list:
- ml.append(make_method_signature(class_def, m, "operator", state))
- format_table(f, ml)
- # Theme properties reference table
- if len(class_def.theme_items) > 0:
- f.write(".. rst-class:: classref-reftable-group\n\n")
- f.write(make_heading("Theme Properties", "-"))
- ml = []
- for theme_item_def in class_def.theme_items.values():
- ref = f":ref:`{theme_item_def.name}<class_{class_name}_theme_{theme_item_def.data_name}_{theme_item_def.name}>`"
- ml.append((theme_item_def.type_name.to_rst(state), ref, theme_item_def.default_value))
- format_table(f, ml, True)
- ### DETAILED DESCRIPTIONS ###
- # Signal descriptions
- if len(class_def.signals) > 0:
- f.write(make_separator(True))
- f.write(".. rst-class:: classref-descriptions-group\n\n")
- f.write(make_heading("Signals", "-"))
- index = 0
- for signal in class_def.signals.values():
- if index != 0:
- f.write(make_separator())
- # Create signal signature and anchor point.
- f.write(f".. _class_{class_name}_signal_{signal.name}:\n\n")
- f.write(".. rst-class:: classref-signal\n\n")
- _, signature = make_method_signature(class_def, signal, "", state)
- f.write(f"{signature}\n\n")
- # Add signal description, or a call to action if it's missing.
- if signal.description is not None and signal.description.strip() != "":
- f.write(f"{format_text_block(signal.description.strip(), signal, state)}\n\n")
- else:
- f.write(".. container:: contribute\n\n\t")
- f.write(
- translate(
- "There is currently no description for this signal. Please help us by :ref:`contributing one <doc_updating_the_class_reference>`!"
- )
- + "\n\n"
- )
- index += 1
- # Enumeration descriptions
- if len(class_def.enums) > 0:
- f.write(make_separator(True))
- f.write(".. rst-class:: classref-descriptions-group\n\n")
- f.write(make_heading("Enumerations", "-"))
- index = 0
- for e in class_def.enums.values():
- if index != 0:
- f.write(make_separator())
- # Create enumeration signature and anchor point.
- f.write(f".. _enum_{class_name}_{e.name}:\n\n")
- f.write(".. rst-class:: classref-enumeration\n\n")
- if e.is_bitfield:
- f.write(f"flags **{e.name}**:\n\n")
- else:
- f.write(f"enum **{e.name}**:\n\n")
- for value in e.values.values():
- # Also create signature and anchor point for each enum constant.
- f.write(f".. _class_{class_name}_constant_{value.name}:\n\n")
- f.write(".. rst-class:: classref-enumeration-constant\n\n")
- f.write(f"{e.type_name.to_rst(state)} **{value.name}** = ``{value.value}``\n\n")
- # Add enum constant description.
- if value.text is not None and value.text.strip() != "":
- f.write(f"{format_text_block(value.text.strip(), value, state)}")
- f.write("\n\n")
- index += 1
- # Constant descriptions
- if len(class_def.constants) > 0:
- f.write(make_separator(True))
- f.write(".. rst-class:: classref-descriptions-group\n\n")
- f.write(make_heading("Constants", "-"))
- for constant in class_def.constants.values():
- # Create constant signature and anchor point.
- f.write(f".. _class_{class_name}_constant_{constant.name}:\n\n")
- f.write(".. rst-class:: classref-constant\n\n")
- f.write(f"**{constant.name}** = ``{constant.value}``\n\n")
- # Add enum constant description.
- if constant.text is not None and constant.text.strip() != "":
- f.write(f"{format_text_block(constant.text.strip(), constant, state)}")
- f.write("\n\n")
- # Annotation descriptions
- if len(class_def.annotations) > 0:
- f.write(make_separator(True))
- f.write(make_heading("Annotations", "-"))
- index = 0
- for method_list in class_def.annotations.values(): # type: ignore
- for i, m in enumerate(method_list):
- if index != 0:
- f.write(make_separator())
- # Create annotation signature and anchor point.
- if i == 0:
- f.write(f".. _class_{class_name}_annotation_{m.name}:\n\n")
- f.write(".. rst-class:: classref-annotation\n\n")
- _, signature = make_method_signature(class_def, m, "", state)
- f.write(f"{signature}\n\n")
- # Add annotation description, or a call to action if it's missing.
- if m.description is not None and m.description.strip() != "":
- f.write(f"{format_text_block(m.description.strip(), m, state)}\n\n")
- else:
- f.write(".. container:: contribute\n\n\t")
- f.write(
- translate(
- "There is currently no description for this annotation. Please help us by :ref:`contributing one <doc_updating_the_class_reference>`!"
- )
- + "\n\n"
- )
- index += 1
- # Property descriptions
- if any(not p.overrides for p in class_def.properties.values()) > 0:
- f.write(make_separator(True))
- f.write(".. rst-class:: classref-descriptions-group\n\n")
- f.write(make_heading("Property Descriptions", "-"))
- index = 0
- for property_def in class_def.properties.values():
- if property_def.overrides:
- continue
- if index != 0:
- f.write(make_separator())
- # Create property signature and anchor point.
- f.write(f".. _class_{class_name}_property_{property_def.name}:\n\n")
- f.write(".. rst-class:: classref-property\n\n")
- property_default = ""
- if property_def.default_value is not None:
- property_default = f" = {property_def.default_value}"
- f.write(f"{property_def.type_name.to_rst(state)} **{property_def.name}**{property_default}\n\n")
- # Create property setter and getter records.
- property_setget = ""
- if property_def.setter is not None and not property_def.setter.startswith("_"):
- property_setter = make_setter_signature(class_def, property_def, state)
- property_setget += f"- {property_setter}\n"
- if property_def.getter is not None and not property_def.getter.startswith("_"):
- property_getter = make_getter_signature(class_def, property_def, state)
- property_setget += f"- {property_getter}\n"
- if property_setget != "":
- f.write(".. rst-class:: classref-property-setget\n\n")
- f.write(property_setget)
- f.write("\n")
- # Add property description, or a call to action if it's missing.
- if property_def.text is not None and property_def.text.strip() != "":
- f.write(f"{format_text_block(property_def.text.strip(), property_def, state)}\n\n")
- else:
- f.write(".. container:: contribute\n\n\t")
- f.write(
- translate(
- "There is currently no description for this property. Please help us by :ref:`contributing one <doc_updating_the_class_reference>`!"
- )
- + "\n\n"
- )
- index += 1
- # Constructor, Method, Operator descriptions
- if len(class_def.constructors) > 0:
- f.write(make_separator(True))
- f.write(".. rst-class:: classref-descriptions-group\n\n")
- f.write(make_heading("Constructor Descriptions", "-"))
- index = 0
- for method_list in class_def.constructors.values():
- for i, m in enumerate(method_list):
- if index != 0:
- f.write(make_separator())
- # Create constructor signature and anchor point.
- if i == 0:
- f.write(f".. _class_{class_name}_constructor_{m.name}:\n\n")
- f.write(".. rst-class:: classref-constructor\n\n")
- ret_type, signature = make_method_signature(class_def, m, "", state)
- f.write(f"{ret_type} {signature}\n\n")
- # Add constructor description, or a call to action if it's missing.
- if m.description is not None and m.description.strip() != "":
- f.write(f"{format_text_block(m.description.strip(), m, state)}\n\n")
- else:
- f.write(".. container:: contribute\n\n\t")
- f.write(
- translate(
- "There is currently no description for this constructor. Please help us by :ref:`contributing one <doc_updating_the_class_reference>`!"
- )
- + "\n\n"
- )
- index += 1
- if len(class_def.methods) > 0:
- f.write(make_separator(True))
- f.write(".. rst-class:: classref-descriptions-group\n\n")
- f.write(make_heading("Method Descriptions", "-"))
- index = 0
- for method_list in class_def.methods.values():
- for i, m in enumerate(method_list):
- if index != 0:
- f.write(make_separator())
- # Create method signature and anchor point.
- if i == 0:
- method_qualifier = ""
- if m.name.startswith("_"):
- method_qualifier = "private_"
- f.write(f".. _class_{class_name}_{method_qualifier}method_{m.name}:\n\n")
- f.write(".. rst-class:: classref-method\n\n")
- ret_type, signature = make_method_signature(class_def, m, "", state)
- f.write(f"{ret_type} {signature}\n\n")
- # Add method description, or a call to action if it's missing.
- if m.description is not None and m.description.strip() != "":
- f.write(f"{format_text_block(m.description.strip(), m, state)}\n\n")
- else:
- f.write(".. container:: contribute\n\n\t")
- f.write(
- translate(
- "There is currently no description for this method. Please help us by :ref:`contributing one <doc_updating_the_class_reference>`!"
- )
- + "\n\n"
- )
- index += 1
- if len(class_def.operators) > 0:
- f.write(make_separator(True))
- f.write(".. rst-class:: classref-descriptions-group\n\n")
- f.write(make_heading("Operator Descriptions", "-"))
- index = 0
- for method_list in class_def.operators.values():
- for i, m in enumerate(method_list):
- if index != 0:
- f.write(make_separator())
- # Create operator signature and anchor point.
- operator_anchor = f".. _class_{class_name}_operator_{sanitize_operator_name(m.name, state)}"
- for parameter in m.parameters:
- operator_anchor += f"_{parameter.type_name.type_name}"
- operator_anchor += f":\n\n"
- f.write(operator_anchor)
- f.write(".. rst-class:: classref-operator\n\n")
- ret_type, signature = make_method_signature(class_def, m, "", state)
- f.write(f"{ret_type} {signature}\n\n")
- # Add operator description, or a call to action if it's missing.
- if m.description is not None and m.description.strip() != "":
- f.write(f"{format_text_block(m.description.strip(), m, state)}\n\n")
- else:
- f.write(".. container:: contribute\n\n\t")
- f.write(
- translate(
- "There is currently no description for this operator. Please help us by :ref:`contributing one <doc_updating_the_class_reference>`!"
- )
- + "\n\n"
- )
- index += 1
- # Theme property descriptions
- if len(class_def.theme_items) > 0:
- f.write(make_separator(True))
- f.write(".. rst-class:: classref-descriptions-group\n\n")
- f.write(make_heading("Theme Property Descriptions", "-"))
- index = 0
- for theme_item_def in class_def.theme_items.values():
- if index != 0:
- f.write(make_separator())
- # Create theme property signature and anchor point.
- f.write(f".. _class_{class_name}_theme_{theme_item_def.data_name}_{theme_item_def.name}:\n\n")
- f.write(".. rst-class:: classref-themeproperty\n\n")
- theme_item_default = ""
- if theme_item_def.default_value is not None:
- theme_item_default = f" = {theme_item_def.default_value}"
- f.write(f"{theme_item_def.type_name.to_rst(state)} **{theme_item_def.name}**{theme_item_default}\n\n")
- # Add theme property description, or a call to action if it's missing.
- if theme_item_def.text is not None and theme_item_def.text.strip() != "":
- f.write(f"{format_text_block(theme_item_def.text.strip(), theme_item_def, state)}\n\n")
- else:
- f.write(".. container:: contribute\n\n\t")
- f.write(
- translate(
- "There is currently no description for this theme property. Please help us by :ref:`contributing one <doc_updating_the_class_reference>`!"
- )
- + "\n\n"
- )
- index += 1
- f.write(make_footer())
- def make_type(klass: str, state: State) -> str:
- if klass.find("*") != -1: # Pointer, ignore
- return klass
- link_type = klass
- if link_type.endswith("[]"): # Typed array, strip [] to link to contained type.
- link_type = link_type[:-2]
- if link_type in state.classes:
- return f":ref:`{klass}<class_{link_type}>`"
- print_error(f'{state.current_class}.xml: Unresolved type "{klass}".', state)
- return klass
- def make_enum(t: str, is_bitfield: bool, state: State) -> str:
- p = t.find(".")
- if p >= 0:
- c = t[0:p]
- e = t[p + 1 :]
- # Variant enums live in GlobalScope but still use periods.
- if c == "Variant":
- c = "@GlobalScope"
- e = "Variant." + e
- else:
- c = state.current_class
- e = t
- if c in state.classes and e not in state.classes[c].enums:
- c = "@GlobalScope"
- if c in state.classes and e in state.classes[c].enums:
- if is_bitfield:
- if not state.classes[c].enums[e].is_bitfield:
- print_error(f'{state.current_class}.xml: Enum "{t}" is not bitfield.', state)
- return f"|bitfield|\\<:ref:`{e}<enum_{c}_{e}>`\\>"
- else:
- return f":ref:`{e}<enum_{c}_{e}>`"
- # Don't fail for `Vector3.Axis`, as this enum is a special case which is expected not to be resolved.
- if f"{c}.{e}" != "Vector3.Axis":
- print_error(f'{state.current_class}.xml: Unresolved enum "{t}".', state)
- return t
- def make_method_signature(
- class_def: ClassDef, definition: Union[AnnotationDef, MethodDef, SignalDef], ref_type: str, state: State
- ) -> Tuple[str, str]:
- ret_type = ""
- if isinstance(definition, MethodDef):
- ret_type = definition.return_type.to_rst(state)
- qualifiers = None
- if isinstance(definition, (MethodDef, AnnotationDef)):
- qualifiers = definition.qualifiers
- out = ""
- if isinstance(definition, MethodDef) and ref_type != "":
- if ref_type == "operator":
- op_name = definition.name.replace("<", "\\<") # So operator "<" gets correctly displayed.
- out += f":ref:`{op_name}<class_{class_def.name}_{ref_type}_{sanitize_operator_name(definition.name, state)}"
- for parameter in definition.parameters:
- out += f"_{parameter.type_name.type_name}"
- out += f">` "
- elif ref_type == "method":
- ref_type_qualifier = ""
- if definition.name.startswith("_"):
- ref_type_qualifier = "private_"
- out += f":ref:`{definition.name}<class_{class_def.name}_{ref_type_qualifier}{ref_type}_{definition.name}>` "
- else:
- out += f":ref:`{definition.name}<class_{class_def.name}_{ref_type}_{definition.name}>` "
- else:
- out += f"**{definition.name}** "
- out += "**(**"
- for i, arg in enumerate(definition.parameters):
- if i > 0:
- out += ", "
- else:
- out += " "
- out += f"{arg.type_name.to_rst(state)} {arg.name}"
- if arg.default_value is not None:
- out += f"={arg.default_value}"
- if qualifiers is not None and "vararg" in qualifiers:
- if len(definition.parameters) > 0:
- out += ", ..."
- else:
- out += " ..."
- out += " **)**"
- if qualifiers is not None:
- # Use substitutions for abbreviations. This is used to display tooltips on hover.
- # See `make_footer()` for descriptions.
- for qualifier in qualifiers.split():
- out += f" |{qualifier}|"
- return ret_type, out
- def make_setter_signature(class_def: ClassDef, property_def: PropertyDef, state: State) -> str:
- if property_def.setter is None:
- return ""
- # If setter is a method available as a method definition, we use that.
- if property_def.setter in class_def.methods:
- setter = class_def.methods[property_def.setter][0]
- # Otherwise we fake it with the information we have available.
- else:
- setter_params: List[ParameterDef] = []
- setter_params.append(ParameterDef("value", property_def.type_name, None))
- setter = MethodDef(property_def.setter, TypeName("void"), setter_params, None, None)
- ret_type, signature = make_method_signature(class_def, setter, "", state)
- return f"{ret_type} {signature}"
- def make_getter_signature(class_def: ClassDef, property_def: PropertyDef, state: State) -> str:
- if property_def.getter is None:
- return ""
- # If getter is a method available as a method definition, we use that.
- if property_def.getter in class_def.methods:
- getter = class_def.methods[property_def.getter][0]
- # Otherwise we fake it with the information we have available.
- else:
- getter_params: List[ParameterDef] = []
- getter = MethodDef(property_def.getter, property_def.type_name, getter_params, None, None)
- ret_type, signature = make_method_signature(class_def, getter, "", state)
- return f"{ret_type} {signature}"
- def make_heading(title: str, underline: str, l10n: bool = True) -> str:
- if l10n:
- new_title = translate(title)
- if new_title != title:
- title = new_title
- underline *= 2 # Double length to handle wide chars.
- return f"{title}\n{(underline * len(title))}\n\n"
- def make_footer() -> str:
- # Generate reusable abbreviation substitutions.
- # This way, we avoid bloating the generated rST with duplicate abbreviations.
- virtual_msg = translate("This method should typically be overridden by the user to have any effect.")
- const_msg = translate("This method has no side effects. It doesn't modify any of the instance's member variables.")
- vararg_msg = translate("This method accepts any number of arguments after the ones described here.")
- constructor_msg = translate("This method is used to construct a type.")
- static_msg = translate(
- "This method doesn't need an instance to be called, so it can be called directly using the class name."
- )
- operator_msg = translate("This method describes a valid operator to use with this type as left-hand operand.")
- bitfield_msg = translate("This value is an integer composed as a bitmask of the following flags.")
- return (
- f".. |virtual| replace:: :abbr:`virtual ({virtual_msg})`\n"
- f".. |const| replace:: :abbr:`const ({const_msg})`\n"
- f".. |vararg| replace:: :abbr:`vararg ({vararg_msg})`\n"
- f".. |constructor| replace:: :abbr:`constructor ({constructor_msg})`\n"
- f".. |static| replace:: :abbr:`static ({static_msg})`\n"
- f".. |operator| replace:: :abbr:`operator ({operator_msg})`\n"
- f".. |bitfield| replace:: :abbr:`BitField ({bitfield_msg})`\n"
- )
- def make_separator(section_level: bool = False) -> str:
- separator_class = "item"
- if section_level:
- separator_class = "section"
- return f".. rst-class:: classref-{separator_class}-separator\n\n----\n\n"
- def make_link(url: str, title: str) -> str:
- match = GODOT_DOCS_PATTERN.search(url)
- if match:
- groups = match.groups()
- if match.lastindex == 2:
- # Doc reference with fragment identifier: emit direct link to section with reference to page, for example:
- # `#calling-javascript-from-script in Exporting For Web`
- # Or use the title if provided.
- if title != "":
- return f"`{title} <../{groups[0]}.html{groups[1]}>`__"
- return f"`{groups[1]} <../{groups[0]}.html{groups[1]}>`__ in :doc:`../{groups[0]}`"
- elif match.lastindex == 1:
- # Doc reference, for example:
- # `Math`
- if title != "":
- return f":doc:`{title} <../{groups[0]}>`"
- return f":doc:`../{groups[0]}`"
- # External link, for example:
- # `http://enet.bespin.org/usergroup0.html`
- if title != "":
- return f"`{title} <{url}>`__"
- return f"`{url} <{url}>`__"
- def make_rst_index(grouped_classes: Dict[str, List[str]], dry_run: bool, output_dir: str) -> None:
- if dry_run:
- f = open(os.devnull, "w", encoding="utf-8")
- else:
- f = open(os.path.join(output_dir, "index.rst"), "w", encoding="utf-8")
- # Remove the "Edit on Github" button from the online docs page, and disallow user-contributed notes
- # on the index page. User-contributed notes are allowed on individual class pages.
- f.write(":github_url: hide\n:allow_comments: False\n\n")
- # Warn contributors not to edit this file directly.
- # Also provide links to the source files for reference.
- git_branch = get_git_branch()
- generator_github_url = f"https://github.com/godotengine/godot/tree/{git_branch}/doc/tools/make_rst.py"
- f.write(".. DO NOT EDIT THIS FILE!!!\n")
- f.write(".. Generated automatically from Godot engine sources.\n")
- f.write(f".. Generator: {generator_github_url}.\n\n")
- f.write(".. _doc_class_reference:\n\n")
- f.write(make_heading("All classes", "="))
- for group_name in CLASS_GROUPS:
- if group_name in grouped_classes:
- f.write(make_heading(CLASS_GROUPS[group_name], "="))
- f.write(".. toctree::\n")
- f.write(" :maxdepth: 1\n")
- f.write(f" :name: toc-class-ref-{group_name}s\n")
- f.write("\n")
- if group_name in CLASS_GROUPS_BASE:
- f.write(f" class_{CLASS_GROUPS_BASE[group_name].lower()}\n")
- for class_name in grouped_classes[group_name]:
- if group_name in CLASS_GROUPS_BASE and CLASS_GROUPS_BASE[group_name].lower() == class_name.lower():
- continue
- f.write(f" class_{class_name.lower()}\n")
- f.write("\n")
- # Formatting helpers.
- RESERVED_FORMATTING_TAGS = ["i", "b", "u", "code", "kbd", "center", "url", "br"]
- RESERVED_LAYOUT_TAGS = ["codeblocks"]
- RESERVED_CODEBLOCK_TAGS = ["codeblock", "gdscript", "csharp"]
- RESERVED_CROSSLINK_TAGS = [
- "method",
- "constructor",
- "operator",
- "member",
- "signal",
- "constant",
- "enum",
- "annotation",
- "theme_item",
- "param",
- ]
- def is_in_tagset(tag_text: str, tagset: List[str]) -> bool:
- for tag in tagset:
- # Complete match.
- if tag_text == tag:
- return True
- # Tag with arguments.
- if tag_text.startswith(tag + " "):
- return True
- # Tag with arguments, special case for [url].
- if tag_text.startswith(tag + "="):
- return True
- return False
- def get_tag_and_args(tag_text: str) -> TagState:
- tag_name = tag_text
- arguments: List[str] = []
- assign_pos = tag_text.find("=")
- if assign_pos >= 0:
- tag_name = tag_text[:assign_pos]
- arguments = [tag_text[assign_pos + 1 :].strip()]
- else:
- space_pos = tag_text.find(" ")
- if space_pos >= 0:
- tag_name = tag_text[:space_pos]
- arguments = [tag_text[space_pos + 1 :].strip()]
- closing = False
- if tag_name.startswith("/"):
- tag_name = tag_name[1:]
- closing = True
- return TagState(tag_text, tag_name, arguments, closing)
- def parse_link_target(link_target: str, state: State, context_name: str) -> List[str]:
- if link_target.find(".") != -1:
- return link_target.split(".")
- else:
- return [state.current_class, link_target]
- def format_text_block(
- text: str,
- context: Union[DefinitionBase, None],
- state: State,
- ) -> str:
- # Linebreak + tabs in the XML should become two line breaks unless in a "codeblock"
- pos = 0
- while True:
- pos = text.find("\n", pos)
- if pos == -1:
- break
- pre_text = text[:pos]
- indent_level = 0
- while pos + 1 < len(text) and text[pos + 1] == "\t":
- pos += 1
- indent_level += 1
- post_text = text[pos + 1 :]
- # Handle codeblocks
- if (
- post_text.startswith("[codeblock]")
- or post_text.startswith("[codeblock ")
- or post_text.startswith("[gdscript]")
- or post_text.startswith("[gdscript ")
- or post_text.startswith("[csharp]")
- or post_text.startswith("[csharp ")
- ):
- tag_text = post_text[1:].split("]", 1)[0]
- tag_state = get_tag_and_args(tag_text)
- result = format_codeblock(tag_state, post_text, indent_level, state)
- if result is None:
- return ""
- text = f"{pre_text}{result[0]}"
- pos += result[1] - indent_level
- # Handle normal text
- else:
- text = f"{pre_text}\n\n{post_text}"
- pos += 2 - indent_level
- next_brac_pos = text.find("[")
- text = escape_rst(text, next_brac_pos)
- context_name = format_context_name(context)
- # Handle [tags]
- inside_code = False
- inside_code_tag = ""
- inside_code_tabs = False
- ignore_code_warnings = False
- code_warning_if_intended_string = "If this is intended, use [code skip-lint]...[/code]."
- pos = 0
- tag_depth = 0
- while True:
- pos = text.find("[", pos)
- if pos == -1:
- break
- endq_pos = text.find("]", pos + 1)
- if endq_pos == -1:
- break
- pre_text = text[:pos]
- post_text = text[endq_pos + 1 :]
- tag_text = text[pos + 1 : endq_pos]
- escape_pre = False
- escape_post = False
- # Tag is a reference to a class.
- if tag_text in state.classes and not inside_code:
- if tag_text == state.current_class:
- # Don't create a link to the same class, format it as strong emphasis.
- tag_text = f"**{tag_text}**"
- else:
- tag_text = make_type(tag_text, state)
- escape_pre = True
- escape_post = True
- # Tag is a cross-reference or a formatting directive.
- else:
- tag_state = get_tag_and_args(tag_text)
- # Anything identified as a tag inside of a code block is valid,
- # unless it's a matching closing tag.
- if inside_code:
- # Exiting codeblocks and inline code tags.
- if tag_state.closing and tag_state.name == inside_code_tag:
- if is_in_tagset(tag_state.name, RESERVED_CODEBLOCK_TAGS):
- tag_text = ""
- tag_depth -= 1
- inside_code = False
- ignore_code_warnings = False
- # Strip newline if the tag was alone on one
- if pre_text[-1] == "\n":
- pre_text = pre_text[:-1]
- elif is_in_tagset(tag_state.name, ["code"]):
- tag_text = "``"
- tag_depth -= 1
- inside_code = False
- ignore_code_warnings = False
- escape_post = True
- else:
- if not ignore_code_warnings and tag_state.closing:
- print_warning(
- f'{state.current_class}.xml: Found a code string that looks like a closing tag "[{tag_state.raw}]" in {context_name}. {code_warning_if_intended_string}',
- state,
- )
- tag_text = f"[{tag_text}]"
- # Entering codeblocks and inline code tags.
- elif tag_state.name == "codeblocks":
- if tag_state.closing:
- tag_depth -= 1
- tag_text = ""
- inside_code_tabs = False
- else:
- tag_depth += 1
- tag_text = "\n.. tabs::"
- inside_code_tabs = True
- elif is_in_tagset(tag_state.name, RESERVED_CODEBLOCK_TAGS):
- tag_depth += 1
- if tag_state.name == "gdscript":
- if not inside_code_tabs:
- print_error(
- f"{state.current_class}.xml: GDScript code block is used outside of [codeblocks] in {context_name}.",
- state,
- )
- tag_text = "\n .. code-tab:: gdscript\n"
- elif tag_state.name == "csharp":
- if not inside_code_tabs:
- print_error(
- f"{state.current_class}.xml: C# code block is used outside of [codeblocks] in {context_name}.",
- state,
- )
- tag_text = "\n .. code-tab:: csharp\n"
- else:
- tag_text = "\n::\n"
- inside_code = True
- inside_code_tag = tag_state.name
- ignore_code_warnings = "skip-lint" in tag_state.arguments
- elif is_in_tagset(tag_state.name, ["code"]):
- tag_text = "``"
- tag_depth += 1
- inside_code = True
- inside_code_tag = "code"
- ignore_code_warnings = "skip-lint" in tag_state.arguments
- escape_pre = True
- if not ignore_code_warnings:
- endcode_pos = text.find("[/code]", endq_pos + 1)
- if endcode_pos == -1:
- print_error(
- f"{state.current_class}.xml: Tag depth mismatch for [code]: no closing [/code] in {context_name}.",
- state,
- )
- break
- inside_code_text = text[endq_pos + 1 : endcode_pos]
- if inside_code_text.endswith("()"):
- # It's formatted like a call for some reason, may still be a mistake.
- inside_code_text = inside_code_text[:-2]
- if inside_code_text in state.classes:
- print_warning(
- f'{state.current_class}.xml: Found a code string "{inside_code_text}" that matches one of the known classes in {context_name}. {code_warning_if_intended_string}',
- state,
- )
- target_class_name, target_name, *rest = parse_link_target(inside_code_text, state, context_name)
- if len(rest) == 0 and target_class_name in state.classes:
- class_def = state.classes[target_class_name]
- if target_name in class_def.methods:
- print_warning(
- f'{state.current_class}.xml: Found a code string "{inside_code_text}" that matches the {target_class_name}.{target_name} method in {context_name}. {code_warning_if_intended_string}',
- state,
- )
- elif target_name in class_def.constructors:
- print_warning(
- f'{state.current_class}.xml: Found a code string "{inside_code_text}" that matches the {target_class_name}.{target_name} constructor in {context_name}. {code_warning_if_intended_string}',
- state,
- )
- elif target_name in class_def.operators:
- print_warning(
- f'{state.current_class}.xml: Found a code string "{inside_code_text}" that matches the {target_class_name}.{target_name} operator in {context_name}. {code_warning_if_intended_string}',
- state,
- )
- elif target_name in class_def.properties:
- print_warning(
- f'{state.current_class}.xml: Found a code string "{inside_code_text}" that matches the {target_class_name}.{target_name} member in {context_name}. {code_warning_if_intended_string}',
- state,
- )
- elif target_name in class_def.signals:
- print_warning(
- f'{state.current_class}.xml: Found a code string "{inside_code_text}" that matches the {target_class_name}.{target_name} signal in {context_name}. {code_warning_if_intended_string}',
- state,
- )
- elif target_name in class_def.annotations:
- print_warning(
- f'{state.current_class}.xml: Found a code string "{inside_code_text}" that matches the {target_class_name}.{target_name} annotation in {context_name}. {code_warning_if_intended_string}',
- state,
- )
- elif target_name in class_def.theme_items:
- print_warning(
- f'{state.current_class}.xml: Found a code string "{inside_code_text}" that matches the {target_class_name}.{target_name} theme item in {context_name}. {code_warning_if_intended_string}',
- state,
- )
- elif target_name in class_def.constants:
- print_warning(
- f'{state.current_class}.xml: Found a code string "{inside_code_text}" that matches the {target_class_name}.{target_name} constant in {context_name}. {code_warning_if_intended_string}',
- state,
- )
- else:
- for enum in class_def.enums.values():
- if target_name in enum.values:
- print_warning(
- f'{state.current_class}.xml: Found a code string "{inside_code_text}" that matches the {target_class_name}.{target_name} enum value in {context_name}. {code_warning_if_intended_string}',
- state,
- )
- break
- valid_param_context = isinstance(context, (MethodDef, SignalDef, AnnotationDef))
- if valid_param_context:
- context_params: List[ParameterDef] = context.parameters # type: ignore
- for param_def in context_params:
- if param_def.name == inside_code_text:
- print_warning(
- f'{state.current_class}.xml: Found a code string "{inside_code_text}" that matches one of the parameters in {context_name}. {code_warning_if_intended_string}',
- state,
- )
- break
- # Cross-references to items in this or other class documentation pages.
- elif is_in_tagset(tag_state.name, RESERVED_CROSSLINK_TAGS):
- link_target: str = tag_state.arguments[0] if len(tag_state.arguments) > 0 else ""
- if link_target == "":
- print_error(
- f'{state.current_class}.xml: Empty cross-reference link "[{tag_state.raw}]" in {context_name}.',
- state,
- )
- tag_text = ""
- else:
- if (
- tag_state.name == "method"
- or tag_state.name == "constructor"
- or tag_state.name == "operator"
- or tag_state.name == "member"
- or tag_state.name == "signal"
- or tag_state.name == "annotation"
- or tag_state.name == "theme_item"
- or tag_state.name == "constant"
- ):
- target_class_name, target_name, *rest = parse_link_target(link_target, state, context_name)
- if len(rest) > 0:
- print_error(
- f'{state.current_class}.xml: Bad reference "{link_target}" in {context_name}.',
- state,
- )
- # Default to the tag command name. This works by default for most tags,
- # but method, member, and theme_item have special cases.
- ref_type = "_{}".format(tag_state.name)
- if target_class_name in state.classes:
- class_def = state.classes[target_class_name]
- if tag_state.name == "method":
- if target_name.startswith("_"):
- ref_type = "_private_method"
- if target_name not in class_def.methods:
- print_error(
- f'{state.current_class}.xml: Unresolved method reference "{link_target}" in {context_name}.',
- state,
- )
- elif tag_state.name == "constructor" and target_name not in class_def.constructors:
- print_error(
- f'{state.current_class}.xml: Unresolved constructor reference "{link_target}" in {context_name}.',
- state,
- )
- elif tag_state.name == "operator" and target_name not in class_def.operators:
- print_error(
- f'{state.current_class}.xml: Unresolved operator reference "{link_target}" in {context_name}.',
- state,
- )
- elif tag_state.name == "member":
- ref_type = "_property"
- if target_name not in class_def.properties:
- print_error(
- f'{state.current_class}.xml: Unresolved member reference "{link_target}" in {context_name}.',
- state,
- )
- elif tag_state.name == "signal" and target_name not in class_def.signals:
- print_error(
- f'{state.current_class}.xml: Unresolved signal reference "{link_target}" in {context_name}.',
- state,
- )
- elif tag_state.name == "annotation" and target_name not in class_def.annotations:
- print_error(
- f'{state.current_class}.xml: Unresolved annotation reference "{link_target}" in {context_name}.',
- state,
- )
- elif tag_state.name == "theme_item":
- if target_name not in class_def.theme_items:
- print_error(
- f'{state.current_class}.xml: Unresolved theme item reference "{link_target}" in {context_name}.',
- state,
- )
- else:
- # Needs theme data type to be properly linked, which we cannot get without a class.
- name = class_def.theme_items[target_name].data_name
- ref_type = f"_theme_{name}"
- elif tag_state.name == "constant":
- found = False
- # Search in the current class
- search_class_defs = [class_def]
- if link_target.find(".") == -1:
- # Also search in @GlobalScope as a last resort if no class was specified
- search_class_defs.append(state.classes["@GlobalScope"])
- for search_class_def in search_class_defs:
- if target_name in search_class_def.constants:
- target_class_name = search_class_def.name
- found = True
- else:
- for enum in search_class_def.enums.values():
- if target_name in enum.values:
- target_class_name = search_class_def.name
- found = True
- break
- if not found:
- print_error(
- f'{state.current_class}.xml: Unresolved constant reference "{link_target}" in {context_name}.',
- state,
- )
- else:
- print_error(
- f'{state.current_class}.xml: Unresolved type reference "{target_class_name}" in method reference "{link_target}" in {context_name}.',
- state,
- )
- repl_text = target_name
- if target_class_name != state.current_class:
- repl_text = f"{target_class_name}.{target_name}"
- tag_text = f":ref:`{repl_text}<class_{target_class_name}{ref_type}_{target_name}>`"
- escape_pre = True
- escape_post = True
- elif tag_state.name == "enum":
- tag_text = make_enum(link_target, False, state)
- escape_pre = True
- escape_post = True
- elif tag_state.name == "param":
- valid_param_context = isinstance(context, (MethodDef, SignalDef, AnnotationDef))
- if not valid_param_context:
- print_error(
- f'{state.current_class}.xml: Argument reference "{link_target}" used outside of method, signal, or annotation context in {context_name}.',
- state,
- )
- else:
- context_params: List[ParameterDef] = context.parameters # type: ignore
- found = False
- for param_def in context_params:
- if param_def.name == link_target:
- found = True
- break
- if not found:
- print_error(
- f'{state.current_class}.xml: Unresolved argument reference "{link_target}" in {context_name}.',
- state,
- )
- tag_text = f"``{link_target}``"
- escape_pre = True
- escape_post = True
- # Formatting directives.
- elif is_in_tagset(tag_state.name, ["url"]):
- url_target = tag_state.arguments[0] if len(tag_state.arguments) > 0 else ""
- if url_target == "":
- print_error(
- f'{state.current_class}.xml: Misformatted [url] tag "[{tag_state.raw}]" in {context_name}.',
- state,
- )
- else:
- # Unlike other tags, URLs are handled in full here, as we need to extract
- # the optional link title to use `make_link`.
- endurl_pos = text.find("[/url]", endq_pos + 1)
- if endurl_pos == -1:
- print_error(
- f"{state.current_class}.xml: Tag depth mismatch for [url]: no closing [/url] in {context_name}.",
- state,
- )
- break
- link_title = text[endq_pos + 1 : endurl_pos]
- tag_text = make_link(url_target, link_title)
- pre_text = text[:pos]
- post_text = text[endurl_pos + 6 :]
- if pre_text and pre_text[-1] not in MARKUP_ALLOWED_PRECEDENT:
- pre_text += "\\ "
- if post_text and post_text[0] not in MARKUP_ALLOWED_SUBSEQUENT:
- post_text = "\\ " + post_text
- text = pre_text + tag_text + post_text
- pos = len(pre_text) + len(tag_text)
- continue
- elif tag_state.name == "br":
- # Make a new paragraph instead of a linebreak, rst is not so linebreak friendly
- tag_text = "\n\n"
- # Strip potential leading spaces
- while post_text[0] == " ":
- post_text = post_text[1:]
- elif tag_state.name == "center":
- if tag_state.closing:
- tag_depth -= 1
- else:
- tag_depth += 1
- tag_text = ""
- elif tag_state.name == "i":
- if tag_state.closing:
- tag_depth -= 1
- escape_post = True
- else:
- tag_depth += 1
- escape_pre = True
- tag_text = "*"
- elif tag_state.name == "b":
- if tag_state.closing:
- tag_depth -= 1
- escape_post = True
- else:
- tag_depth += 1
- escape_pre = True
- tag_text = "**"
- elif tag_state.name == "u":
- if tag_state.closing:
- tag_depth -= 1
- escape_post = True
- else:
- tag_depth += 1
- escape_pre = True
- tag_text = ""
- elif tag_state.name == "kbd":
- tag_text = "`"
- if tag_state.closing:
- tag_depth -= 1
- escape_post = True
- else:
- tag_text = ":kbd:" + tag_text
- tag_depth += 1
- escape_pre = True
- # Invalid syntax.
- else:
- if tag_state.closing:
- print_error(
- f'{state.current_class}.xml: Unrecognized closing tag "[{tag_state.raw}]" in {context_name}.',
- state,
- )
- tag_text = f"[{tag_text}]"
- else:
- print_error(
- f'{state.current_class}.xml: Unrecognized opening tag "[{tag_state.raw}]" in {context_name}.',
- state,
- )
- tag_text = f"``{tag_text}``"
- escape_pre = True
- escape_post = True
- # Properly escape things like `[Node]s`
- if escape_pre and pre_text and pre_text[-1] not in MARKUP_ALLOWED_PRECEDENT:
- pre_text += "\\ "
- if escape_post and post_text and post_text[0] not in MARKUP_ALLOWED_SUBSEQUENT:
- post_text = "\\ " + post_text
- next_brac_pos = post_text.find("[", 0)
- iter_pos = 0
- while not inside_code:
- iter_pos = post_text.find("*", iter_pos, next_brac_pos)
- if iter_pos == -1:
- break
- post_text = f"{post_text[:iter_pos]}\\*{post_text[iter_pos + 1 :]}"
- iter_pos += 2
- iter_pos = 0
- while not inside_code:
- iter_pos = post_text.find("_", iter_pos, next_brac_pos)
- if iter_pos == -1:
- break
- if not post_text[iter_pos + 1].isalnum(): # don't escape within a snake_case word
- post_text = f"{post_text[:iter_pos]}\\_{post_text[iter_pos + 1 :]}"
- iter_pos += 2
- else:
- iter_pos += 1
- text = pre_text + tag_text + post_text
- pos = len(pre_text) + len(tag_text)
- if tag_depth > 0:
- print_error(
- f"{state.current_class}.xml: Tag depth mismatch: too many (or too few) open/close tags in {context_name}.",
- state,
- )
- return text
- def format_context_name(context: Union[DefinitionBase, None]) -> str:
- context_name: str = "unknown context"
- if context is not None:
- context_name = f'{context.definition_name} "{context.name}" description'
- return context_name
- def escape_rst(text: str, until_pos: int = -1) -> str:
- # Escape \ character, otherwise it ends up as an escape character in rst
- pos = 0
- while True:
- pos = text.find("\\", pos, until_pos)
- if pos == -1:
- break
- text = f"{text[:pos]}\\\\{text[pos + 1 :]}"
- pos += 2
- # Escape * character to avoid interpreting it as emphasis
- pos = 0
- while True:
- pos = text.find("*", pos, until_pos)
- if pos == -1:
- break
- text = f"{text[:pos]}\\*{text[pos + 1 :]}"
- pos += 2
- # Escape _ character at the end of a word to avoid interpreting it as an inline hyperlink
- pos = 0
- while True:
- pos = text.find("_", pos, until_pos)
- if pos == -1:
- break
- if not text[pos + 1].isalnum(): # don't escape within a snake_case word
- text = f"{text[:pos]}\\_{text[pos + 1 :]}"
- pos += 2
- else:
- pos += 1
- return text
- def format_codeblock(
- tag_state: TagState, post_text: str, indent_level: int, state: State
- ) -> Union[Tuple[str, int], None]:
- end_pos = post_text.find("[/" + tag_state.name + "]")
- if end_pos == -1:
- print_error(
- f"{state.current_class}.xml: Tag depth mismatch for [{tag_state.name}]: no closing [/{tag_state.name}].",
- state,
- )
- return None
- opening_formatted = tag_state.name
- if len(tag_state.arguments) > 0:
- opening_formatted += " " + " ".join(tag_state.arguments)
- code_text = post_text[len(f"[{opening_formatted}]") : end_pos]
- post_text = post_text[end_pos:]
- # Remove extraneous tabs
- code_pos = 0
- while True:
- code_pos = code_text.find("\n", code_pos)
- if code_pos == -1:
- break
- to_skip = 0
- while code_pos + to_skip + 1 < len(code_text) and code_text[code_pos + to_skip + 1] == "\t":
- to_skip += 1
- if to_skip > indent_level:
- print_error(
- f"{state.current_class}.xml: Four spaces should be used for indentation within [{tag_state.name}].",
- state,
- )
- if len(code_text[code_pos + to_skip + 1 :]) == 0:
- code_text = f"{code_text[:code_pos]}\n"
- code_pos += 1
- else:
- code_text = f"{code_text[:code_pos]}\n {code_text[code_pos + to_skip + 1 :]}"
- code_pos += 5 - to_skip
- return (f"\n[{opening_formatted}]{code_text}{post_text}", len(f"\n[{opening_formatted}]{code_text}"))
- def format_table(f: TextIO, data: List[Tuple[Optional[str], ...]], remove_empty_columns: bool = False) -> None:
- if len(data) == 0:
- return
- f.write(".. table::\n")
- f.write(" :widths: auto\n\n")
- # Calculate the width of each column first, we will use this information
- # to properly format RST-style tables.
- column_sizes = [0] * len(data[0])
- for row in data:
- for i, text in enumerate(row):
- text_length = len(text or "")
- if text_length > column_sizes[i]:
- column_sizes[i] = text_length
- # Each table row is wrapped in two separators, consecutive rows share the same separator.
- # All separators, or rather borders, have the same shape and content. We compose it once,
- # then reuse it.
- sep = ""
- for size in column_sizes:
- if size == 0 and remove_empty_columns:
- continue
- sep += "+" + "-" * (size + 2) # Content of each cell is padded by 1 on each side.
- sep += "+\n"
- # Draw the first separator.
- f.write(f" {sep}")
- # Draw each row and close it with a separator.
- for row in data:
- row_text = "|"
- for i, text in enumerate(row):
- if column_sizes[i] == 0 and remove_empty_columns:
- continue
- row_text += f' {(text or "").ljust(column_sizes[i])} |'
- row_text += "\n"
- f.write(f" {row_text}")
- f.write(f" {sep}")
- f.write("\n")
- def sanitize_operator_name(dirty_name: str, state: State) -> str:
- clear_name = dirty_name.replace("operator ", "")
- if clear_name == "!=":
- clear_name = "neq"
- elif clear_name == "==":
- clear_name = "eq"
- elif clear_name == "<":
- clear_name = "lt"
- elif clear_name == "<=":
- clear_name = "lte"
- elif clear_name == ">":
- clear_name = "gt"
- elif clear_name == ">=":
- clear_name = "gte"
- elif clear_name == "+":
- clear_name = "sum"
- elif clear_name == "-":
- clear_name = "dif"
- elif clear_name == "*":
- clear_name = "mul"
- elif clear_name == "/":
- clear_name = "div"
- elif clear_name == "%":
- clear_name = "mod"
- elif clear_name == "**":
- clear_name = "pow"
- elif clear_name == "unary+":
- clear_name = "unplus"
- elif clear_name == "unary-":
- clear_name = "unminus"
- elif clear_name == "<<":
- clear_name = "bwsl"
- elif clear_name == ">>":
- clear_name = "bwsr"
- elif clear_name == "&":
- clear_name = "bwand"
- elif clear_name == "|":
- clear_name = "bwor"
- elif clear_name == "^":
- clear_name = "bwxor"
- elif clear_name == "~":
- clear_name = "bwnot"
- elif clear_name == "[]":
- clear_name = "idx"
- else:
- clear_name = "xxx"
- print_error(f'Unsupported operator type "{dirty_name}", please add the missing rule.', state)
- return clear_name
- if __name__ == "__main__":
- main()
|