mapFolding 0.16.1__py3-none-any.whl → 0.16.4__py3-none-any.whl

mapFolding/someAssemblyRequired/makeJobTheorem2codon.py

@@ -4,59 +4,31 @@ https://docs.exaloop.io/start/install/
 """
 
 from astToolkit import (
-	Be, DOT, extractFunctionDef, Grab, identifierDotAttribute, IngredientsFunction, IngredientsModule, Make, NodeChanger,
-	NodeTourist, parseLogicalPath2astModule, Then)
+	Be, extractFunctionDef, Grab, identifierDotAttribute, IngredientsFunction, IngredientsModule, Make, NodeChanger,
+	parseLogicalPath2astModule, Then)
 from astToolkit.transformationTools import removeUnusedParameters, write_astModule
-from hunterMakesPy import autoDecodingRLE, raiseIfNone
-from mapFolding import DatatypeLeavesTotal, getPathFilenameFoldsTotal
+from hunterMakesPy import raiseIfNone
+from mapFolding import DatatypeLeavesTotal, getPathFilenameFoldsTotal, packageSettings
 from mapFolding.dataBaskets import MapFoldingState
-from mapFolding.someAssemblyRequired import IfThis
-from mapFolding.someAssemblyRequired.RecipeJob import RecipeJobTheorem2
-from mapFolding.syntheticModules.A007822.initializeState import transitionOnGroupsOfFolds
+from mapFolding.someAssemblyRequired import DatatypeConfiguration, default, IfThis
+from mapFolding.someAssemblyRequired.RecipeJob import (
+	customizeDatatypeViaImport, moveShatteredDataclass_arg2body, RecipeJobTheorem2)
+from mapFolding.syntheticModules.initializeState import transitionOnGroupsOfFolds
 from pathlib import Path, PurePosixPath
-from typing import cast, NamedTuple, TYPE_CHECKING
-import ast
+from typing import cast, TYPE_CHECKING
 import subprocess
 import sys
 
 if TYPE_CHECKING:
 	from io import TextIOBase
+	import ast
 
-class DatatypeConfiguration(NamedTuple):
-	"""Configuration for mapping framework datatypes to compiled datatypes.
+# TODO Converge with `makeJobTheorem2Numba`.
 
-	This configuration class defines how abstract datatypes used in the map folding framework should be replaced with compiled
-	datatypes during code generation. Each configuration specifies the source module, target type name, and optional import alias
-	for the transformation.
-
-	Attributes
-	----------
-	datatypeIdentifier : str
-		Framework datatype identifier to be replaced.
-	typeModule : identifierDotAttribute
-		Module containing the target datatype (e.g., 'codon', 'numpy').
-	typeIdentifier : str
-		Concrete type name in the target module.
-	type_asname : str | None = None
-		Optional import alias for the type.
-	"""
-
-	datatypeIdentifier: str
-	typeModule: identifierDotAttribute
-	typeIdentifier: str
-	type_asname: str | None = None
-
-# TODO replace with dynamic system. Probably use `Final` in the dataclass.
-listIdentifiersStaticValuesHARDCODED: list[str] = ['dimensionsTotal', 'leavesTotal']
-
-# TODO Dynamically calculate the bitwidth of each datatype.
 listDatatypeConfigurations: list[DatatypeConfiguration] = [
 	DatatypeConfiguration(datatypeIdentifier='DatatypeLeavesTotal', typeModule='numpy', typeIdentifier='uint8', type_asname='DatatypeLeavesTotal'),
 	DatatypeConfiguration(datatypeIdentifier='DatatypeElephino', typeModule='numpy', typeIdentifier='uint8', type_asname='DatatypeElephino'),
 	DatatypeConfiguration(datatypeIdentifier='DatatypeFoldsTotal', typeModule='numpy', typeIdentifier='int64', type_asname='DatatypeFoldsTotal'),
-]
-
-listNumPy_dtype: list[DatatypeConfiguration] = [
 	DatatypeConfiguration(datatypeIdentifier='Array1DLeavesTotal', typeModule='numpy', typeIdentifier='uint8', type_asname='Array1DLeavesTotal'),
 	DatatypeConfiguration(datatypeIdentifier='Array1DElephino', typeModule='numpy', typeIdentifier='uint8', type_asname='Array1DElephino'),
 	DatatypeConfiguration(datatypeIdentifier='Array3DLeavesTotal', typeModule='numpy', typeIdentifier='uint8', type_asname='Array3DLeavesTotal'),
@@ -78,55 +50,60 @@ def _addWriteFoldsTotal(ingredientsFunction: IngredientsFunction, job: RecipeJob
 
 	return ingredientsFunction
 
-def _datatypeDefinitions(ingredientsFunction: IngredientsFunction, ingredientsModule: IngredientsModule) -> tuple[IngredientsFunction, IngredientsModule]:
-	for datatypeConfig in [*listDatatypeConfigurations, *listNumPy_dtype]:
-		ingredientsFunction.imports.removeImportFrom(datatypeConfig.typeModule, None, datatypeConfig.datatypeIdentifier)
-		ingredientsFunction.imports.addImportFrom_asStr(datatypeConfig.typeModule, datatypeConfig.typeIdentifier, datatypeConfig.type_asname)
-
-	ingredientsFunction.imports.removeImportFromModule('mapFolding.dataBaskets')
+def _variableCompatibility(ingredientsFunction: IngredientsFunction, job: RecipeJobTheorem2) -> IngredientsFunction:
+	"""Ensure the variable is compiled to the correct type.
 
-	return ingredientsFunction, ingredientsModule
+	Add a type constructor to `identifier` to ensure compatibility if
+	- an incompatible type might be assigned to it,
+	- it might be compared with an incompatible type,
+	- it is used as an indexer but its type is not a valid indexer type.
 
-def _pythonCode2expr(string: str) -> ast.expr:
-	"""Convert *one* expression as a string of Python code to an `ast.expr`."""
-	return raiseIfNone(NodeTourist(Be.Expr, Then.extractIt(DOT.value)).captureLastMatch(ast.parse(string)))
+	Parameters
+	----------
+	ingredientsFunction : IngredientsFunction
+		Function to modify.
+	job : RecipeJobTheorem2
+		Configuration settings with identifiers and their type annotations.
 
-def _variableCompatibility(ingredientsFunction: IngredientsFunction, job: RecipeJobTheorem2) -> IngredientsFunction:
-	# On some assignment or comparison values, add a type constructor to ensure compatibility.
-	# On some values-as-indexer, add a type constructor to ensure indexing-method compatibility.
+	Returns
+	-------
+	ingredientsFunction : IngredientsFunction
+		Modified function.
+	"""
 	for ast_arg in job.shatteredDataclass.list_argAnnotated4ArgumentsSpecification:
-		identifier = ast_arg.arg
-		annotation = raiseIfNone(ast_arg.annotation)
-
-		# `identifier` in Augmented Assignment, or in Assignments and value is Constant.
-		NodeChanger(findThis=IfThis.isAnyOf(
-				Be.AugAssign.targetIs(IfThis.isNestedNameIdentifier(identifier))
-				, IfThis.isAllOf(
-					Be.Assign.targetsIs(Be.at(0, IfThis.isNestedNameIdentifier(identifier)))
-					, Be.Assign.valueIs(Be.Constant))
+		identifier: str = ast_arg.arg
+		annotation: ast.expr = raiseIfNone(ast_arg.annotation)
+
+	# ------- `identifier` is target of Augmented Assignment, or --------------
+	# ------- `identifier` is target of Assignment and value is Constant. -----
+		NodeChanger(
+			IfThis.isAnyOf(
+							Be.AugAssign.targetIs(IfThis.isNestedNameIdentifier(identifier))
+			,   IfThis.isAllOf(Be.Assign.targetsIs(Be.at(0, IfThis.isNestedNameIdentifier(identifier)))
+							,  Be.Assign.valueIs(Be.Constant))
 			)
 			, doThat=lambda node, annotation=annotation: Grab.valueAttribute(Then.replaceWith(Make.Call(annotation, listParameters=[node.value])))(node)
 		).visit(ingredientsFunction.astFunctionDef)
 
-		# `identifier` - 1.
+	# ------- `identifier` - 1. ----------------------------------------------
 		NodeChanger(Be.BinOp.leftIs(IfThis.isNestedNameIdentifier(identifier))
 			, doThat=lambda node, annotation=annotation: Grab.rightAttribute(Then.replaceWith(Make.Call(annotation, listParameters=[node.right])))(node)
 		).visit(ingredientsFunction.astFunctionDef)
 
-		# `identifier` in Comparison.
+	# ------- `identifier` in Comparison. -------------------------------------
 		NodeChanger(Be.Compare.leftIs(IfThis.isNestedNameIdentifier(identifier))
 			, doThat=lambda node, annotation=annotation: Grab.comparatorsAttribute(lambda at, annotation=annotation: Then.replaceWith([Make.Call(annotation, listParameters=[node.comparators[0]])])(at[0]))(node)
 		).visit(ingredientsFunction.astFunctionDef)
 
-		# `identifier` has exactly one index value.
+	# ------- `identifier` has exactly one index value. -----------------------
 		NodeChanger(IfThis.isAllOf(Be.Subscript.valueIs(IfThis.isNestedNameIdentifier(identifier))
 			, lambda node: not Be.Subscript.sliceIs(Be.Tuple)(node))
 			, doThat=lambda node: Grab.sliceAttribute(Then.replaceWith(Make.Call(Make.Name('int'), listParameters=[node.slice])))(node)
 		).visit(ingredientsFunction.astFunctionDef)
 
-		# `identifier` has multiple index values.
+	# ------- `identifier` has multiple index values. -------------------------
 		NodeChanger(IfThis.isAllOf(Be.Subscript.valueIs(IfThis.isNestedNameIdentifier(identifier))
-			, Be.Subscript.sliceIs(Be.Tuple))
+								,  Be.Subscript.sliceIs(Be.Tuple))
 			, doThat=lambda node: Grab.sliceAttribute(Grab.eltsAttribute(
 				Then.replaceWith([
 					Make.Call(Make.Name('int'), listParameters=[cast('ast.Tuple', node.slice).elts[index]])
@@ -135,19 +112,6 @@ def _variableCompatibility(ingredientsFunction: IngredientsFunction, job: Recipe
 
 	return ingredientsFunction
 
-def _move_arg2body(identifier: str, job: RecipeJobTheorem2) -> ast.AnnAssign | ast.Assign:
-	Ima___Assign, elementConstructor = job.shatteredDataclass.Z0Z_field2AnnAssign[identifier]
-	match elementConstructor:
-		case 'scalar':
-			cast('ast.Constant', cast('ast.Call', Ima___Assign.value).args[0]).value = int(job.state.__dict__[identifier])
-		case 'array':
-			dataAsStrRLE: str = autoDecodingRLE(job.state.__dict__[identifier], assumeAddSpaces=True)
-			dataAs_ast_expr: ast.expr = _pythonCode2expr(dataAsStrRLE)
-			cast('ast.Call', Ima___Assign.value).args = [dataAs_ast_expr]
-		case _:
-			pass
-	return Ima___Assign
-
 def makeJob(job: RecipeJobTheorem2) -> None:
 	"""Generate an optimized module for map folding calculations.
 
@@ -157,22 +121,21 @@ def makeJob(job: RecipeJobTheorem2) -> None:
 	Parameters
 	----------
 	job : RecipeJobTheorem2
-		Configuration recipe containing source locations, target paths, and state.
+		Configuration recipe containing source locations, target paths, raw materials, and state.
 
 	"""
 	ingredientsCount: IngredientsFunction = IngredientsFunction(raiseIfNone(extractFunctionDef(job.source_astModule, job.countCallable)))
 	ingredientsCount.astFunctionDef.decorator_list = []
 
 	# Replace identifiers-with-static-values with their values.
-	listIdentifiersStaticValues: list[str] = listIdentifiersStaticValuesHARDCODED
-	for identifier in listIdentifiersStaticValues:
+	for identifier in job.shatteredDataclass.listIdentifiersStaticScalars:
 		NodeChanger(IfThis.isNameIdentifier(identifier)
-			, Then.replaceWith(Make.Constant(int(job.state.__dict__[identifier])))
+			, Then.replaceWith(Make.Constant(int(eval(f"job.state.{identifier}"))))  # noqa: S307
 		).visit(ingredientsCount.astFunctionDef)
 
 	ingredientsCount.imports.update(job.shatteredDataclass.imports)
 	ingredientsCount = removeUnusedParameters(ingredientsCount)
-	NodeChanger(Be.arg, lambda removeIt: ingredientsCount.astFunctionDef.body.insert(0, _move_arg2body(removeIt.arg, job))).visit(ingredientsCount.astFunctionDef)
+	NodeChanger(Be.arg, lambda removeIt: ingredientsCount.astFunctionDef.body.insert(0, moveShatteredDataclass_arg2body(removeIt.arg, job))).visit(ingredientsCount.astFunctionDef)
 
 	ingredientsCount = _addWriteFoldsTotal(ingredientsCount, job)
 	ingredientsCount = _variableCompatibility(ingredientsCount, job)
@@ -181,11 +144,14 @@ def makeJob(job: RecipeJobTheorem2) -> None:
 		Make.If(Make.Compare(Make.Name('__name__'), [Make.Eq()], [Make.Constant('__main__')])
 			, body=[Make.Expr(Make.Call(Make.Name(job.countCallable)))])]))
 
-	ingredientsCount, ingredientsModule = _datatypeDefinitions(ingredientsCount, ingredientsModule)
+	ingredientsCount, ingredientsModule = customizeDatatypeViaImport(ingredientsCount, ingredientsModule, listDatatypeConfigurations)
+
+	ingredientsCount.imports.removeImportFromModule('mapFolding.dataBaskets')
 
 	ingredientsModule.appendIngredientsFunction(ingredientsCount)
 
 	if sys.platform == 'linux':
+		Path(job.pathFilenameModule).parent.mkdir(parents=True, exist_ok=True)
 		buildCommand: list[str] = ['codon', 'build', '--exe', '--release',
 			'--fast-math', '--enable-unsafe-fp-math', '--disable-exceptions',
 			'--mcpu=native',
@@ -200,6 +166,7 @@ def makeJob(job: RecipeJobTheorem2) -> None:
 		sys.stdout.write(f"sudo systemd-run --unit={job.moduleIdentifier} --nice=-10 --property=CPUAffinity=0 {job.pathFilenameModule.with_suffix('')}\n")
 	else:
 		write_astModule(ingredientsModule, pathFilename=job.pathFilenameModule, packageName=job.packageIdentifier)
+		sys.stdout.write(f"python {Path(job.pathFilenameModule)}\n")
 
 def fromMapShape(mapShape: tuple[DatatypeLeavesTotal, ...]) -> None:
 	"""Create a binary executable for a map-folding job from map dimensions.
@@ -214,14 +181,15 @@ def fromMapShape(mapShape: tuple[DatatypeLeavesTotal, ...]) -> None:
 		along one axis.
 
 	"""
-	state = transitionOnGroupsOfFolds(MapFoldingState(mapShape))
+	state: MapFoldingState = transitionOnGroupsOfFolds(MapFoldingState(mapShape))
 	pathModule = PurePosixPath(Path.home(), 'mapFolding', 'jobs')
-	source_astModule = parseLogicalPath2astModule('mapFolding.syntheticModules.theorem2A007822Numba')
+	logicalPath2astModule: identifierDotAttribute = f'{packageSettings.identifierPackage}.{default['logicalPath']['synthetic']}.theorem2Numba'
+	source_astModule: ast.Module = parseLogicalPath2astModule(logicalPath2astModule)
 	pathFilenameFoldsTotal = PurePosixPath(getPathFilenameFoldsTotal(state.mapShape, pathModule))
 	aJob = RecipeJobTheorem2(state, source_astModule=source_astModule, pathModule=pathModule, pathFilenameFoldsTotal=pathFilenameFoldsTotal)
 	makeJob(aJob)
 
 if __name__ == '__main__':
-	mapShape = (1, 3)
+	mapShape: tuple[DatatypeLeavesTotal, ...] = (1, 14)
 	fromMapShape(mapShape)
 

mapFolding/someAssemblyRequired/makingModules_count.py

@@ -9,41 +9,25 @@ from astToolkit import (
 from astToolkit.transformationTools import inlineFunctionDef, removeUnusedParameters, write_astModule
 from hunterMakesPy import raiseIfNone
 from mapFolding import packageSettings
-from mapFolding.someAssemblyRequired import (
-	identifierCallableSourceDEFAULT, identifierCountingDEFAULT, IfThis, ShatteredDataclass)
-from mapFolding.someAssemblyRequired.A007822.A007822rawMaterials import astExprCall_filterAsymmetricFoldsLeafBelow
-from mapFolding.someAssemblyRequired.toolkitMakeModules import findDataclass, getPathFilename
+from mapFolding.someAssemblyRequired import default, IfThis, ShatteredDataclass
+from mapFolding.someAssemblyRequired.toolkitMakeModules import findDataclass, getLogicalPath, getPathFilename
 from mapFolding.someAssemblyRequired.toolkitNumba import decorateCallableWithNumba, parametersNumbaLight
 from mapFolding.someAssemblyRequired.transformationTools import (
 	removeDataclassFromFunction, shatter_dataclassesDOTdataclass, unpackDataclassCallFunctionRepackDataclass)
-from os import PathLike
 from pathlib import PurePath
 from typing import cast
 import ast
 
-def makeDaoOfMapFoldingNumba(astModule: ast.Module, moduleIdentifier: str, callableIdentifier: str | None = None, logicalPathInfix: PathLike[str] | PurePath | identifierDotAttribute | None = None, sourceCallableDispatcher: str | None = None) -> PurePath:
-	"""Generate Numba-optimized sequential implementation of map folding algorithm.
-
-	(AI generated docstring)
-
-	Creates a high-performance sequential version of the map folding algorithm by
-	decomposing dataclass parameters into individual primitive values, removing
-	dataclass dependencies that are incompatible with Numba, applying Numba
-	decorators for just-in-time compilation, and optionally including a dispatcher
-	function for dataclass integration.
-
-	The generated module provides significant performance improvements over the
-	original dataclass-based implementation while maintaining algorithmic correctness.
-	The transformation preserves all computational logic while restructuring data
-	access patterns for optimal Numba compilation.
+def makeMapFoldingNumba(astModule: ast.Module, identifierModule: str, identifierCallable: str | None = None, logicalPathInfix: identifierDotAttribute | None = None, sourceCallableDispatcher: str | None = None) -> PurePath:
+	"""Generate Numba-optimized sequential implementation of an algorithm.
 
 	Parameters
 	----------
 	astModule : ast.Module
 		Source module containing the base algorithm.
-	moduleIdentifier : str
+	identifierModule : str
 		Name for the generated optimized module.
-	callableIdentifier : str | None = None
+	identifierCallable : str | None = None
 		Name for the main computational function.
 	logicalPathInfix : PathLike[str] | PurePath | str | None = None
 		Directory path for organizing the generated module.
@@ -56,9 +40,9 @@ def makeDaoOfMapFoldingNumba(astModule: ast.Module, moduleIdentifier: str, calla
 		Filesystem path where the optimized module was written.
 
 	"""
-	sourceCallableIdentifier: identifierDotAttribute = identifierCallableSourceDEFAULT
+	sourceCallableIdentifier: str = default['function']['counting']
 	ingredientsFunction = IngredientsFunction(inlineFunctionDef(sourceCallableIdentifier, astModule), LedgerOfImports(astModule))
-	ingredientsFunction.astFunctionDef.name = callableIdentifier or sourceCallableIdentifier
+	ingredientsFunction.astFunctionDef.name = identifierCallable or sourceCallableIdentifier
 
 	shatteredDataclass: ShatteredDataclass = shatter_dataclassesDOTdataclass(*findDataclass(ingredientsFunction))
 
@@ -88,46 +72,46 @@ def makeDaoOfMapFoldingNumba(astModule: ast.Module, moduleIdentifier: str, calla
 
 	ingredientsModule.removeImportFromModule('numpy')
 
-	pathFilename: PurePath = getPathFilename(packageSettings.pathPackage, logicalPathInfix, moduleIdentifier)
+	pathFilename: PurePath = getPathFilename(packageSettings.pathPackage, logicalPathInfix, identifierModule)
 
 	write_astModule(ingredientsModule, pathFilename, packageSettings.identifierPackage)
 
 	return pathFilename
 
-def makeTheorem2(astModule: ast.Module, moduleIdentifier: str, callableIdentifier: str | None = None, logicalPathInfix: PathLike[str] | PurePath | identifierDotAttribute | None = None, sourceCallableDispatcher: str | None = None) -> PurePath:
+def makeTheorem2(astModule: ast.Module, identifierModule: str, identifierCallable: str | None = None, logicalPathInfix: identifierDotAttribute | None = None, sourceCallableDispatcher: str | None = None) -> PurePath:
 	"""Generate module by applying optimization predicted by Theorem 2.
 
 	Parameters
 	----------
 	astModule : ast.Module
 		Source module containing the base algorithm.
-	moduleIdentifier : str
+	identifierModule : str
 		Name for the generated theorem-optimized module.
-	callableIdentifier : str | None = None
+	identifierCallable : str | None = None
 		Name for the optimized computational function.
 	logicalPathInfix : PathLike[str] | PurePath | str | None = None
 		Directory path for organizing the generated module.
 	sourceCallableDispatcher : str | None = None
-		Currently not implemented for this transformation.
+		Optional dispatcher function identifier.
 
 	Returns
 	-------
 	pathFilename : PurePath
 		Filesystem path where the theorem-optimized module was written.
+	"""
+	identifierCallableInitializeDataclassHARDCODED = 'transitionOnGroupsOfFolds'
+	identifierModuleInitializeDataclassHARDCODED = 'initializeState'
 
-	Raises
-	------
-	NotImplementedError
-		If `sourceCallableDispatcher` is provided.
+	identifierCallableInitializeDataclass = identifierCallableInitializeDataclassHARDCODED
+	identifierModuleInitializeDataclass = identifierModuleInitializeDataclassHARDCODED
 
-	"""
-	sourceCallableIdentifier = identifierCallableSourceDEFAULT
+	sourceCallableIdentifier = default['function']['counting']
 	ingredientsFunction = IngredientsFunction(inlineFunctionDef(sourceCallableIdentifier, astModule), LedgerOfImports(astModule))
-	ingredientsFunction.astFunctionDef.name = callableIdentifier or sourceCallableIdentifier
+	ingredientsFunction.astFunctionDef.name = identifierCallable or sourceCallableIdentifier
 
-	dataclassInstanceIdentifier: identifierDotAttribute = raiseIfNone(NodeTourist(Be.arg, Then.extractIt(DOT.arg)).captureLastMatch(ingredientsFunction.astFunctionDef))
+	dataclassInstanceIdentifier: str = raiseIfNone(NodeTourist(Be.arg, Then.extractIt(DOT.arg)).captureLastMatch(ingredientsFunction.astFunctionDef))
 
-	theCountingIdentifier: identifierDotAttribute = identifierCountingDEFAULT
+	theCountingIdentifier: str = default['variable']['counting']
 	doubleTheCount: ast.AugAssign = Make.AugAssign(Make.Attribute(Make.Name(dataclassInstanceIdentifier), theCountingIdentifier), Make.Mult(), Make.Constant(2))
 
 	NodeChanger(
@@ -171,40 +155,41 @@ def makeTheorem2(astModule: ast.Module, moduleIdentifier: str, callableIdentifie
 	ingredientsModule = IngredientsModule(ingredientsFunction)
 
 	if sourceCallableDispatcher is not None:
-		message = 'sourceCallableDispatcher is not implemented yet'
-		raise NotImplementedError(message)
+		ingredientsFunctionDispatcher: IngredientsFunction = astModuleToIngredientsFunction(astModule, sourceCallableDispatcher)
+		targetCallableIdentifier = ingredientsFunction.astFunctionDef.name
 
-	pathFilename: PurePath = getPathFilename(packageSettings.pathPackage, logicalPathInfix, moduleIdentifier)
+		# Update any calls to the original function name with the new target function name
+		NodeChanger(
+			findThis = Be.Call.funcIs(Be.Name.idIs(IfThis.isIdentifier(default['function']['counting'])))
+			, doThat = Grab.funcAttribute(Grab.idAttribute(Then.replaceWith(targetCallableIdentifier)))
+		).visit(ingredientsFunctionDispatcher.astFunctionDef)
 
-	write_astModule(ingredientsModule, pathFilename, packageSettings.identifierPackage)
+		AssignInitializedDataclass: ast.Assign = Make.Assign([Make.Name(dataclassInstanceIdentifier)], value=Make.Call(Make.Name(identifierCallableInitializeDataclass), [Make.Name(dataclassInstanceIdentifier)]))
 
-	return pathFilename
+		# Insert the transitionOnGroupsOfFolds call at the beginning of the function
+		ingredientsFunctionDispatcher.astFunctionDef.body.insert(0, AssignInitializedDataclass)
 
-def numbaOnTheorem2(astModule: ast.Module, moduleIdentifier: str, callableIdentifier: str | None = None, logicalPathInfix: PathLike[str] | PurePath | identifierDotAttribute | None = None, sourceCallableDispatcher: str | None = None) -> PurePath:  # noqa: ARG001
-	"""Generate Numba-accelerated Theorem 2 implementation with dataclass decomposition.
+		dotModule: identifierDotAttribute = getLogicalPath(packageSettings.identifierPackage, logicalPathInfix, identifierModuleInitializeDataclass)
+		ingredientsFunctionDispatcher.imports.addImportFrom_asStr(dotModule, identifierCallableInitializeDataclass)
 
-	(AI generated docstring)
+		ingredientsModule.appendIngredientsFunction(ingredientsFunctionDispatcher)
 
-	Creates a highly optimized version of the Theorem 2 algorithm by combining the
-	mathematical optimizations of Theorem 2 with Numba just-in-time compilation.
-	The transformation includes dataclass decomposition to convert structured
-	parameters into primitives, removal of Python object dependencies incompatible
-	with Numba, application of Numba decorators for maximum performance, and type
-	annotation optimization for efficient compilation.
+	pathFilename: PurePath = getPathFilename(packageSettings.pathPackage, logicalPathInfix, identifierModule)
 
-	This represents the highest level of optimization available for Theorem 2
-	implementations, providing both mathematical efficiency through theorem
-	application and computational efficiency through Numba acceleration.
-	The result is suitable for production use in high-performance computing
-	environments where maximum speed is required.
+	write_astModule(ingredientsModule, pathFilename, packageSettings.identifierPackage)
+
+	return pathFilename
+
+def numbaOnTheorem2(astModule: ast.Module, identifierModule: str, identifierCallable: str | None = None, logicalPathInfix: identifierDotAttribute | None = None, sourceCallableDispatcher: str | None = None) -> PurePath:
+	"""Generate Numba-accelerated Theorem 2 implementation with dataclass decomposition.
 
 	Parameters
 	----------
 	astModule : ast.Module
 		Source module containing the Theorem 2 implementation.
-	moduleIdentifier : str
+	identifierModule : str
 		Name for the generated Numba-accelerated module.
-	callableIdentifier : str | None = None
+	identifierCallable : str | None = None
 		Name for the accelerated computational function.
 	logicalPathInfix : PathLike[str] | PurePath | str | None = None
 		Directory path for organizing the generated module.
@@ -217,13 +202,13 @@ def numbaOnTheorem2(astModule: ast.Module, moduleIdentifier: str, callableIdenti
 		Filesystem path where the accelerated module was written.
 
 	"""
-	sourceCallableIdentifier = identifierCallableSourceDEFAULT
-	if callableIdentifier is None:
-		callableIdentifier = sourceCallableIdentifier
+	sourceCallableIdentifier = default['function']['counting']
 	ingredientsFunction = IngredientsFunction(inlineFunctionDef(sourceCallableIdentifier, astModule), LedgerOfImports(astModule))
-	ingredientsFunction.astFunctionDef.name = callableIdentifier
+	ingredientsFunction.astFunctionDef.name = identifierCallable or sourceCallableIdentifier
 
-	shatteredDataclass: ShatteredDataclass = shatter_dataclassesDOTdataclass(*findDataclass(ingredientsFunction))
+	logicalPathDataclass, identifierDataclass, identifierDataclassInstance = findDataclass(ingredientsFunction)
+
+	shatteredDataclass: ShatteredDataclass = shatter_dataclassesDOTdataclass(logicalPathDataclass, identifierDataclass, identifierDataclassInstance)
 
 	ingredientsFunction.imports.update(shatteredDataclass.imports)
 	ingredientsFunction: IngredientsFunction = removeDataclassFromFunction(ingredientsFunction, shatteredDataclass)
@@ -233,32 +218,40 @@ def numbaOnTheorem2(astModule: ast.Module, moduleIdentifier: str, callableIdenti
 	ingredientsModule = IngredientsModule(ingredientsFunction)
 	ingredientsModule.removeImportFromModule('numpy')
 
-	pathFilename: PurePath = getPathFilename(packageSettings.pathPackage, logicalPathInfix, moduleIdentifier)
+	if sourceCallableDispatcher is not None:
+		ingredientsFunctionDispatcher: IngredientsFunction = astModuleToIngredientsFunction(astModule, sourceCallableDispatcher)
+		ingredientsFunctionDispatcher.imports.update(shatteredDataclass.imports)
+		targetCallableIdentifier = ingredientsFunction.astFunctionDef.name
+		ingredientsFunctionDispatcher = unpackDataclassCallFunctionRepackDataclass(ingredientsFunctionDispatcher, targetCallableIdentifier, shatteredDataclass)
+		astTuple: ast.Tuple = cast('ast.Tuple', raiseIfNone(NodeTourist(Be.Return.valueIs(Be.Tuple)
+				, doThat=Then.extractIt(DOT.value)).captureLastMatch(ingredientsFunction.astFunctionDef)))
+		astTuple.ctx = Make.Store()
 
-	write_astModule(ingredientsModule, pathFilename, packageSettings.identifierPackage)
+		changeAssignCallToTarget = NodeChanger(
+			findThis = Be.Assign.valueIs(IfThis.isCallIdentifier(targetCallableIdentifier))
+			, doThat = Then.replaceWith(Make.Assign([astTuple], value=Make.Call(Make.Name(targetCallableIdentifier), astTuple.elts))))
+		changeAssignCallToTarget.visit(ingredientsFunctionDispatcher.astFunctionDef)
 
-	return pathFilename
+		ingredientsModule.appendIngredientsFunction(ingredientsFunctionDispatcher)
 
-def trimTheorem2(astModule: ast.Module, moduleIdentifier: str, callableIdentifier: str | None = None, logicalPathInfix: PathLike[str] | PurePath | identifierDotAttribute | None = None, sourceCallableDispatcher: str | None = None) -> PurePath:  # noqa: ARG001
-	"""Generate constrained Theorem 2 implementation by removing unnecessary logic.
+	ingredientsModule.removeImportFromModule('numpy')
+
+	pathFilename: PurePath = getPathFilename(packageSettings.pathPackage, logicalPathInfix, identifierModule)
 
-	(AI generated docstring)
+	write_astModule(ingredientsModule, pathFilename, packageSettings.identifierPackage)
 
-	Creates a trimmed version of the Theorem 2 implementation by eliminating conditional logic that is not needed under specific
-	constraint assumptions. This transformation removes checks for unconstrained dimensions, simplifying the algorithm for cases
-	where dimensional constraints are guaranteed to be satisfied by external conditions.
+	return pathFilename
 
-	The trimming operation is particularly valuable for generating lean implementations where the calling context ensures that
-	certain conditions will always be met, allowing the removal of defensive programming constructs that add computational
-	overhead without providing benefits in the constrained environment.
+def trimTheorem2(astModule: ast.Module, identifierModule: str, identifierCallable: str | None = None, logicalPathInfix: identifierDotAttribute | None = None, sourceCallableDispatcher: str | None = None) -> PurePath:
+	"""Generate constrained Theorem 2 implementation by removing unnecessary logic.
 
 	Parameters
 	----------
 	astModule : ast.Module
 		Source module containing the Theorem 2 implementation.
-	moduleIdentifier : str
+	identifierModule : str
 		Name for the generated trimmed module.
-	callableIdentifier : str | None = None
+	identifierCallable : str | None = None
 		Name for the trimmed computational function.
 	logicalPathInfix : PathLike[str] | PurePath | str | None = None
 		Directory path for organizing the generated module.
@@ -271,20 +264,33 @@ def trimTheorem2(astModule: ast.Module, moduleIdentifier: str, callableIdentifie
 		Filesystem path where the trimmed module was written.
 
 	"""
-	sourceCallableIdentifier = identifierCallableSourceDEFAULT
+	sourceCallableIdentifier: str = default['function']['counting']
 	ingredientsFunction = IngredientsFunction(inlineFunctionDef(sourceCallableIdentifier, astModule), LedgerOfImports(astModule))
-	ingredientsFunction.astFunctionDef.name = callableIdentifier or sourceCallableIdentifier
+	ingredientsFunction.astFunctionDef.name = identifierCallable or sourceCallableIdentifier
 
-	dataclassInstanceIdentifier: identifierDotAttribute = raiseIfNone(NodeTourist(Be.arg, Then.extractIt(DOT.arg)).captureLastMatch(ingredientsFunction.astFunctionDef))
+	identifierDataclassInstance: str = raiseIfNone(NodeTourist(Be.arg, Then.extractIt(DOT.arg)).captureLastMatch(ingredientsFunction.astFunctionDef))
 
-	findThis = IfThis.isIfUnaryNotAttributeNamespaceIdentifier(dataclassInstanceIdentifier, 'dimensionsUnconstrained')
-	doThat = Then.removeIt
-	NodeChanger(findThis, doThat).visit(ingredientsFunction.astFunctionDef)
+	NodeChanger(
+		findThis = IfThis.isIfUnaryNotAttributeNamespaceIdentifier(identifierDataclassInstance, 'dimensionsUnconstrained')
+		, doThat = Then.removeIt
+	).visit(ingredientsFunction.astFunctionDef)
 
 	ingredientsModule = IngredientsModule(ingredientsFunction)
 	ingredientsModule.removeImportFromModule('numpy')
 
-	pathFilename: PurePath = getPathFilename(packageSettings.pathPackage, logicalPathInfix, moduleIdentifier)
+	if sourceCallableDispatcher is not None:
+		ingredientsFunctionDispatcher: IngredientsFunction = astModuleToIngredientsFunction(astModule, sourceCallableDispatcher)
+		targetCallableIdentifier = ingredientsFunction.astFunctionDef.name
+
+		# Update any calls to the original function name with the new target function name
+		NodeChanger(
+			findThis = Be.Call.funcIs(Be.Name.idIs(IfThis.isIdentifier(default['function']['counting'])))
+			, doThat = Grab.funcAttribute(Grab.idAttribute(Then.replaceWith(targetCallableIdentifier)))
+		).visit(ingredientsFunctionDispatcher.astFunctionDef)
+
+		ingredientsModule.appendIngredientsFunction(ingredientsFunctionDispatcher)
+
+	pathFilename: PurePath = getPathFilename(packageSettings.pathPackage, logicalPathInfix, identifierModule)
 
 	write_astModule(ingredientsModule, pathFilename, packageSettings.identifierPackage)