2460 DynamoPSy split & rename

doc/user_guide/dynamo0p3_topclasses.dot

@@ -5,7 +5,7 @@ rankdir=BT
 "20" [label="{LFRicInvoke|\l|arg_for_funcspace()\lfield_on_space()\lgen_code()\lis_coloured()\lunique_fss()\lunique_proxy_declarations()\l}", shape="record"];
 "21" [label="{DynInvokeSchedule|\l|view()\l}", shape="record"];
 "35" [label="{DynamoInvokes|\l|}", shape="record"];
-"36" [label="{DynamoPSy|\l|}", shape="record"];
+"36" [label="{LFRicPSy|\l|}", shape="record"];
 "45" [label="{Invoke|\l|first_access()\lgen()\lgen_code()\lunique_declarations()\lunique_declns_by_intent()\l}", shape="record", style=filled, fillcolor="antiquewhite"];
 "46" [label="{InvokeSchedule|\l|gen_code()\lview()\l}", shape="record"];
 "47" [label="{Invokes|\l|gen_code()\lgen_ocl_init()\lget()\l}", shape="record", style=filled, fillcolor="antiquewhite"];

examples/lfric/eg14/acc_parallel.py

@@ -52,10 +52,10 @@ def trans(psy):
     transformed through the addition of a routine directive.
 
     :param psy: the PSy object containing the invokes to transform.
-    :type psy: :py:class:`psyclone.dynamo0p3.DynamoPSy`
+    :type psy: :py:class:`psyclone.domain.lfric.LFRicPSy`
 
     :returns: the transformed PSy object.
-    :rtype: :py:class:`psyclone.dynamo0p3.DynamoPSy`
+    :rtype: :py:class:`psyclone.domain.lfric.LFRicPSy`
 
     '''
     const = LFRicConstants()

examples/lfric/eg2/loop_fuse_trans.py

@@ -50,10 +50,10 @@ def trans(psy):
 
     :param psy: the PSy object that PSyclone has constructed for the \
                 'invoke'(s) found in the Algorithm file.
-    :type psy: :py:class:`psyclone.dynamo0p3.DynamoPSy`
+    :type psy: :py:class:`psyclone.domain.lfric.LFRicPSy`
 
     :returns: the transformed PSy object.
-    :rtype: :py:class:`psyclone.dynamo0p3.DynamoPSy`
+    :rtype: :py:class:`psyclone.domain.lfric.LFRicPSy`
 
     '''
     print(psy.gen)

examples/lfric/eg2/module_inline_trans.py

@@ -48,10 +48,10 @@ def trans(psy):
 
     :param psy: the PSy object that PSyclone has constructed for the \
                 'invoke'(s) found in the Algorithm file.
-    :type psy: :py:class:`psyclone.dynamo0p3.DynamoPSy`
+    :type psy: :py:class:`psyclone.domain.lfric.LFRicPSy`
 
     :returns: the transformed PSy object.
-    :rtype: :py:class:`psyclone.dynamo0p3.DynamoPSy`
+    :rtype: :py:class:`psyclone.domain.lfric.LFRicPSy`
 
     '''
     invokes = psy.invokes

examples/lfric/eg2/print_psyir_trans.py

@@ -46,10 +46,10 @@ def trans(psy):
 
     :param psy: the PSy object that PSyclone has constructed for the \
                 'invoke'(s) found in the Algorithm file.
-    :type psy: :py:class:`psyclone.dynamo0p3.DynamoPSy`
+    :type psy: :py:class:`psyclone.domain.lfric.LFRicPSy`
 
     :returns: the supplied PSy object unmodified.
-    :rtype: :py:class:`psyclone.dynamo0p3.DynamoPSy`
+    :rtype: :py:class:`psyclone.domain.lfric.LFRicPSy`
 
     '''
     print("Supplied code has Invokes: ", psy.invokes.names)

src/psyclone/domain/lfric/__init__.py

@@ -71,6 +71,7 @@
 from psyclone.domain.lfric.lfric_scalar_args import LFRicScalarArgs
 from psyclone.domain.lfric.lfric_loop_bounds import LFRicLoopBounds
 from psyclone.domain.lfric.lfric_kern_metadata import LFRicKernMetadata
+from psyclone.domain.lfric.lfric_psy import LFRicPSy
 from psyclone.domain.lfric.lfric_dofmaps import LFRicDofmaps
 from psyclone.domain.lfric.lfric_stencils import LFRicStencils
 
@@ -95,6 +96,7 @@
     'LFRicKernMetadata',
     'LFRicLoop',
     'LFRicLoopBounds',
+    'LFRicPSy',
     'LFRicRunTimeChecks',
     'LFRicScalarArgs',
     'LFRicStencils',

src/psyclone/domain/lfric/lfric_invokes.py

@@ -53,7 +53,7 @@ class LFRicInvokes(Invokes):
                       information.
     :type alg_calls: List[:py:class:`psyclone.parse.algorithm.InvokeCall`]
     :param psy: The PSy object containing this LFRicInvokes object.
-    :type psy: :py:class:`psyclone.dynamo0p3.DynamoPSy`
+    :type psy: :py:class:`psyclone.domain.lfric.LFRicPSy`
 
     '''
     def __init__(self, alg_calls, psy):

src/psyclone/domain/lfric/lfric_psy.py

@@ -0,0 +1,183 @@
+# -----------------------------------------------------------------------------
+# BSD 3-Clause License
+#
+# Copyright (c) 2017-2023, Science and Technology Facilities Council.
+# All rights reserved.
+#
+# Redistribution and use in source and binary forms, with or without
+# modification, are permitted provided that the following conditions are met:
+#
+# * Redistributions of source code must retain the above copyright notice, this
+#   list of conditions and the following disclaimer.
+#
+# * Redistributions in binary form must reproduce the above copyright notice,
+#   this list of conditions and the following disclaimer in the documentation
+#   and/or other materials provided with the distribution.
+#
+# * Neither the name of the copyright holder nor the names of its
+#   contributors may be used to endorse or promote products derived from
+#   this software without specific prior written permission.
+#
+# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
+# "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
+# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
+# FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
+# COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
+# INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
+# BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
+# LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
+# CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
+# LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
+# ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
+# POSSIBILITY OF SUCH DAMAGE.
+# -----------------------------------------------------------------------------
+# Authors R. W. Ford, A. R. Porter and S. Siso, STFC Daresbury Lab
+# Modified I. Kavcic, A. Coughtrie, L. Turner and O. Brunt, Met Office
+# Modified J. Henrichs, Bureau of Meteorology
+# Modified A. B. G. Chalk and N. Nobre, STFC Daresbury Lab
+
+''' This module creates an LFRic-specific Invokes object which controls all
+    the required invocation calls. It also overrides the PSy gen method so
+    that LFRic-specific PSy module code is generated.
+    '''
+
+from collections import OrderedDict
+
+from psyclone.configuration import Config
+from psyclone.domain.lfric import (LFRicConstants, LFRicSymbolTable,
+                                   LFRicInvokes)
+from psyclone.f2pygen import ModuleGen, UseGen, PSyIRGen
+from psyclone.psyGen import PSy, InvokeSchedule
+from psyclone.psyir.nodes import ScopingNode
+
+
+class LFRicPSy(PSy):
+    '''
+    The LFRic-specific PSy class. This creates an LFRic-specific
+    Invokes object (which controls all the required invocation calls).
+    It also overrides the PSy gen method so that we generate
+    LFRic-specific PSy module code.
+
+    :param invoke_info: object containing the required invocation information \
+                        for code optimisation and generation.
+    :type invoke_info: :py:class:`psyclone.parse.algorithm.FileInfo`
+
+    '''
+    def __init__(self, invoke_info):
+        # Make sure the scoping node creates LFRicSymbolTables
+        # TODO #1954: Remove the protected access using a factory
+        ScopingNode._symbol_table_class = LFRicSymbolTable
+        PSy.__init__(self, invoke_info)
+        self._invokes = LFRicInvokes(invoke_info.calls, self)
+        # Initialise the dictionary that holds the names of the required
+        # LFRic constants, data structures and data structure proxies for
+        # the "use" statements in modules that contain PSy-layer routines.
+        const = LFRicConstants()
+        const_mod = const.UTILITIES_MOD_MAP["constants"]["module"]
+        infmod_list = [const_mod]
+        # Add all field and operator modules that might be used in the
+        # algorithm layer. These do not appear in the code unless a
+        # variable is added to the "only" part of the
+        # '_infrastructure_modules' map.
+        for data_type_info in const.DATA_TYPE_MAP.values():
+            infmod_list.append(data_type_info["module"])
+
+        # This also removes any duplicates from infmod_list
+        self._infrastructure_modules = OrderedDict(
+            (k, set()) for k in infmod_list)
+
+        kind_names = set()
+
+        # The infrastructure declares integer types with default
+        # precision so always add this.
+        api_config = Config.get().api_conf("dynamo0.3")
+        kind_names.add(api_config.default_kind["integer"])
+
+        # Datatypes declare precision information themselves. However,
+        # that is not the case for literals. Therefore deal
+        # with these separately here.
+        for invoke in self.invokes.invoke_list:
+            schedule = invoke.schedule
+            for kernel in schedule.kernels():
+                for arg in kernel.args:
+                    if arg.is_literal:
+                        kind_names.add(arg.precision)
+        # Add precision names to the dictionary storing the required
+        # LFRic constants.
+        self._infrastructure_modules[const_mod] = kind_names
+
+    @property
+    def name(self):
+        '''
+        :returns: a name for the PSy layer. This is used as the PSy module \
+                  name. We override the default value as the Met Office \
+                  prefer "_psy" to be appended, rather than prepended.
+        :rtype: str
+
+        '''
+        return self._name + "_psy"
+
+    @property
+    def orig_name(self):
+        '''
+        :returns: the unmodified PSy-layer name.
+        :rtype: str
+
+        '''
+        return self._name
+
+    @property
+    def infrastructure_modules(self):
+        '''
+        :returns: the dictionary that holds the names of the required \
+                  LFRic infrastructure modules to create "use" \
+                  statements in the PSy-layer modules.
+        :rtype: dict of set
+
+        '''
+        return self._infrastructure_modules
+
+    @property
+    def gen(self):
+        '''
+        Generate PSy code for the LFRic (Dynamo0.3) API.
+
+        :returns: root node of generated Fortran AST.
+        :rtype: :py:class:`psyir.nodes.Node`
+
+        '''
+        # Create an empty PSy layer module
+        psy_module = ModuleGen(self.name)
+
+        # If the container has a Routine that is not an InvokeSchedule
+        # it should also be added to the generated module.
+        for routine in self.container.children:
+            if not isinstance(routine, InvokeSchedule):
+                psy_module.add(PSyIRGen(psy_module, routine))
+
+        # Add all invoke-specific information
+        self.invokes.gen_code(psy_module)
+
+        # Include required constants and infrastructure modules. The sets of
+        # required LFRic data structures and their proxies are updated in
+        # the relevant field and operator subclasses of LFRicCollection.
+        # Here we sort the inputs in reverse order to have "_type" before
+        # "_proxy_type" and "operator_" before "columnwise_operator_".
+        # We also iterate through the dictionary in reverse order so the
+        # "use" statements for field types are before the "use" statements
+        # for operator types.
+        for infmod in reversed(self._infrastructure_modules):
+            if self._infrastructure_modules[infmod]:
+                infmod_types = sorted(
+                    list(self._infrastructure_modules[infmod]), reverse=True)
+                psy_module.add(UseGen(psy_module, name=infmod,
+                                      only=True, funcnames=infmod_types))
+
+        # Return the root node of the generated code
+        return psy_module.root
+
+
+# ---------- Documentation utils -------------------------------------------- #
+# The list of module members that we wish AutoAPI to generate
+# documentation for. (See https://psyclone-ref.readthedocs.io)
+__all__ = ['LFRicPSy']