Skip to content

Code Generation

FunctionBase dataclass

Bases: ABC

Abstract base class for all code-generatable functions.

Subclasses must provide the header/source code fragments and any codegen constants needed to emit a complete C++ module via generate_module.

Source code in src/anvil/codegen/common.py 
 96
 97
 98
 99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
@dataclass(frozen=True)
class FunctionBase(ABC):
  """Abstract base class for all code-generatable functions.

  Subclasses must provide the header/source code fragments and any codegen constants
  needed to emit a complete C++ module via `generate_module`.
  """

  name: str

  @abstractmethod
  def __call__(self, *args, **kwargs) -> Any:
    pass

  def dependencies(self) -> "list[FunctionBase]":
    return []

  @property
  @abstractmethod
  def codegen_constants(self) -> set[CodegenIntConstant]: ...

  @property
  @abstractmethod
  def header_includes(self) -> set[str]: ...

  @property
  @abstractmethod
  def header_code(self) -> str: ...

  @property
  @abstractmethod
  def source_includes(self) -> set[str]: ...

  @property
  @abstractmethod
  def source_code(self) -> str: ...

FloatArray = NDArray[np.floating]