Files
nuitka-mirror/nuitka/tree/InternalModule.py
Kay Hayen b4e603ab7c Optimization: Faster variable trace maintenance
* We now trace variables in trace collection as a dictionary
  per variable with a dictionary of the versions, this is
  closer to out frequent usage per variable.

* That makes it a lot easier to update variables after the
  tracing is finished to know their users and writers.

* Requires a lot less work, but also makes work less memory
  local such that the performance gain is relatively small
  despite less work being done.

* Also avoids that a set for the users is to be maintained.
2025-11-26 11:19:43 +00:00

97 lines
2.8 KiB
Python

# Copyright 2025, Kay Hayen, mailto:kay.hayen@gmail.com find license text at end of file
""" Internal module
This is a container for helper functions that are shared across modules. It
may not exist, and is treated specially in code generation. This avoids to
own these functions to a random module.
TODO: Clarify by renaming that the top module is now used, and these are
merely helpers to do it.
"""
from nuitka.ModuleRegistry import getRootTopModule
from nuitka.nodes.FunctionNodes import (
ExpressionFunctionPureBody,
ExpressionFunctionPureInlineConstBody,
)
from nuitka.SourceCodeReferences import makeSourceReferenceFromFilename
internal_source_ref = makeSourceReferenceFromFilename("internal").atInternal()
def once_decorator(func):
"""Cache result of a function call without arguments.
Used for all internal function accesses to become a singleton.
Note: This doesn't much specific anymore, but we are not having
this often enough to warrant reuse or generalization.
"""
func.cached_value = None
def replacement():
if func.cached_value is None:
func.cached_value = func()
return func.cached_value
return replacement
def getInternalModule():
"""Get the singleton internal module."""
return getRootTopModule()
_internal_helper_names = set()
def makeInternalHelperFunctionBody(name, parameters, inline_const_args=False):
# Make sure names of helpers are unique, the code names we choose require
# that to be true.
assert name not in _internal_helper_names
_internal_helper_names.add(name)
if inline_const_args:
node_class = ExpressionFunctionPureInlineConstBody
else:
node_class = ExpressionFunctionPureBody
result = node_class(
provider=getInternalModule(),
name=name,
code_object=None,
doc=None,
parameters=parameters,
flags=None,
auto_release=None,
code_prefix="helper_function",
source_ref=internal_source_ref,
)
for variable in parameters.getAllVariables():
result.removeVariableReleases(variable)
return result
# Part of "Nuitka", an optimizing Python compiler that is compatible and
# integrates with CPython, but also works on its own.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.