fix code_to_adapted_unit

pthom · Dec 15, 2023 · 4b4006a · 4b4006a
1 parent 9a2f7d4
commit 4b4006a
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 6 deletions.
diff --git a/packages/litgen/__init__.py b/packages/litgen/__init__.py
@@ -1,7 +1,7 @@
 from __future__ import annotations
 from litgen.options import LitgenOptions, TemplateClassOptions, TemplateFunctionsOptions
 
-from litgen.code_to_adapted_unit import code_to_adapted_unit
+from litgen.code_to_adapted_unit import code_to_adapted_unit, code_to_adapted_unit_in_context
 from litgen.internal.cpp_to_python import (
     standard_type_replacements,
     standard_value_replacements,
@@ -25,6 +25,7 @@
     "TemplateFunctionsOptions",
     "generate_code",
     "code_to_adapted_unit",
+    "code_to_adapted_unit_in_context",
     "write_generated_code_for_file",
     "write_generated_code_for_files",
     # When it is needed to have different options per c++ header file

diff --git a/packages/litgen/code_to_adapted_unit.py b/packages/litgen/code_to_adapted_unit.py
@@ -8,8 +8,10 @@
     LitgenContext,
 )
 
+from litgen.options import LitgenOptions
 
-def code_to_adapted_unit(
+
+def code_to_adapted_unit_in_context(
     lg_context: LitgenContext,
     code: Optional[str] = None,
     filename: Optional[str] = None,
@@ -19,3 +21,10 @@ def code_to_adapted_unit(
     adapted_unit = AdaptedUnit(lg_context, cpp_unit)
 
     return adapted_unit
+
+
+def code_to_adapted_unit(
+    options: LitgenOptions, code: Optional[str] = None, filename: Optional[str] = None
+) -> AdaptedUnit:
+    lg_context = LitgenContext(options)
+    return code_to_adapted_unit_in_context(lg_context, code, filename)
diff --git a/packages/litgen/litgen_generator.py b/packages/litgen/litgen_generator.py
@@ -8,7 +8,7 @@
 from srcmlcpp.internal.code_to_srcml import _SRCML_CALLER
 
 from litgen import LitgenOptions
-from litgen.code_to_adapted_unit import code_to_adapted_unit
+from litgen.code_to_adapted_unit import code_to_adapted_unit_in_context
 from litgen.internal import boxed_python_type, cpp_to_python
 from litgen.internal.context.litgen_context import LitgenContext
 
@@ -130,9 +130,9 @@ def _boxed_types_cpp_code(self) -> CppCode:
 
     def process_cpp_code(self, code: str, filename: str) -> None:
         if len(filename) > 0:
-            adapted_unit = code_to_adapted_unit(self.lg_context, code, filename)
+            adapted_unit = code_to_adapted_unit_in_context(self.lg_context, code, filename)
         else:
-            adapted_unit = code_to_adapted_unit(self.lg_context, code)
+            adapted_unit = code_to_adapted_unit_in_context(self.lg_context, code)
         stub_code = adapted_unit.str_stub()
         pydef_code = adapted_unit.str_pydef()
         self.lg_context.clear_namespaces_code_tree()
@@ -150,7 +150,7 @@ def _boxed_types_generated_code(self) -> _GeneratedCode | None:
         standalone_options.cpp_indent_size = self.options().cpp_indent_size
         standalone_options.cpp_indent_with_tabs = self.options().cpp_indent_with_tabs
 
-        adapted_unit = code_to_adapted_unit(LitgenContext(standalone_options), boxed_types_cpp_code)
+        adapted_unit = code_to_adapted_unit_in_context(LitgenContext(standalone_options), boxed_types_cpp_code)
 
         stub_code = adapted_unit.str_stub()
         pydef_code = adapted_unit.str_pydef()