Merge pull request #400 from vil02/add_fortran90

feat: add `to_fortran90`

Author: vil02

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "string-to-code"
-version = "0.3.0"
+version = "0.4.0"
 description = "Generates a piece of messy code in a given language displaying a given string "
 authors = ["piotr.idzik <[email protected]>"]
 readme = "./string_to_code/README.md"

string_to_code/to_fortran90.py

@@ -0,0 +1,68 @@
+"""
+provides string_to_fortran90 utilities
+"""
+
+from . import core
+from . import utils
+
+
+_get_function_name = utils.get_function_name_fun("fun_")
+
+
+def _atom_to_code(in_atom: core.Atom) -> str:
+    if in_atom.atom_char == "\n":
+        return "write (*, *)"
+    res = 'write (*, "(A)", advance="no") '
+    if in_atom.atom_char == '"':
+        return res + "'\"'"
+    if in_atom.atom_char == "\t":
+        return res + "char(9)"
+
+    return res + f'"{in_atom.atom_char}"'
+
+
+_function_call_str = utils.get_function_call_str_fun(_get_function_name, "call ", "()")
+
+_call_function_or_atom = utils.get_call_function_or_atom(
+    _atom_to_code, _function_call_str
+)
+
+
+_body_to_str = utils.get_body_to_str("\n", "        ", _call_function_or_atom, "", "")
+
+
+def _merge_to_full_function(in_function_name: str, in_function_body: str) -> str:
+    body_str = ""
+    if in_function_body:
+        body_str = in_function_body + "\n"
+    return f"""    subroutine {in_function_name}()
+{body_str}    end subroutine {in_function_name}
+"""
+
+
+_function_to_code = utils.get_function_to_code(
+    _get_function_name, _body_to_str, _merge_to_full_function
+)
+
+
+def _main_call_to_code(in_initial_call: core.InitialCall, **kwargs) -> str:
+    if in_initial_call is None:
+        return ""
+    return _call_function_or_atom(in_initial_call, **kwargs)
+
+
+def _join_to_final(main_call: str, function_definitions: list[str], **_kwargs) -> str:
+    main_call_str = ""
+    if main_call:
+        main_call_str = "\n    " + main_call
+    contains_str = "\n"
+    if function_definitions:
+        contains_str = "\ncontains\n" + "\n".join(function_definitions)
+    return f"""program main
+    implicit none{main_call_str}{contains_str}end program main
+"""
+
+
+proc_printer_program, proc = utils.get_all_proc_functions(
+    _main_call_to_code, _function_to_code, _join_to_final
+)

system_setup_scripts/setup_for_fortran90.sh

@@ -0,0 +1,5 @@
+#!/usr/bin/env bash
+
+set -euo pipefail
+
+apt-get install -y --no-install-recommends gfortran

…e_data/fortran90/function_used_twice.f90 …e_data/fortran90/function_used_twice.f90tests/unused_example_data/fortran90/function_used_twice.f90 renamed to tests/example_data/fortran90/function_used_twice.f90 tests/unused_example_data/fortran90/function_used_twice.f90 renamed to tests/example_data/fortran90/function_used_twice.f90

@@ -0,0 +1,53 @@
+"""
+setup for the tests of the module string_to_fortran90
+"""
+
+import general_utilities as gu
+
+from string_to_code import to_fortran90
+
+_GFOTRAN = "gfortran"
+
+_FILE_EXTENSION = "f90"
+
+
+def _compile_code(in_code, tmp_folder):
+    source_filename = gu.get_unique_filename(tmp_folder, _FILE_EXTENSION)
+    gu.save_str_to_file(tmp_folder / source_filename, in_code)
+    executable_filename = gu.get_unique_filename(tmp_folder, "o")
+
+    gu.subprocess_run_with_check(
+        [
+            _GFOTRAN,
+            "-Wall",
+            "-Wextra",
+            "-Wpedantic",
+            "-Waliasing",
+            "-Wconversion-extra",
+            "-Wimplicit-interface",
+            "-Wimplicit-procedure",
+            "-Wsurprising",
+            "-Werror",
+            source_filename,
+            "-o",
+            executable_filename,
+        ],
+        cwd=str(tmp_folder),
+    )
+    return executable_filename
+
+
+def _run_code(in_code, tmp_folder):
+    return gu.run_executable(_compile_code(in_code, tmp_folder), tmp_folder)
+
+
+def get_test_data():
+    """returns test data for the module string_to_fortran90"""
+    return gu.Language(
+        tool_names=[_GFOTRAN],
+        string_to_code=to_fortran90.proc,
+        printer_program_to_code=to_fortran90.proc_printer_program,
+        run_code=_run_code,
+        id="fortran90",
+        source_code_file_extension=_FILE_EXTENSION,
+    )