Added the ability to change READ access to TYPE_INFO

Author: LonelyCat124

src/psyclone/core/access_sequence.py

@@ -90,6 +90,16 @@ def change_read_to_write(self):
                                 "which does not have 'READ' access.")
         self._access_type = AccessType.WRITE
 
+    def change_read_to_type_info(self):
+        '''This changes the access mode from READ to TYPE_INFO.
+
+        :raises InternalError: if the variable does not have READ acccess.
+        '''
+        if self._access_type != AccessType.READ:
+            raise InternalError("Trying to change variable to 'TYPE_INFO' "
+                                "which does not have 'READ' access.")
+        self._access_type = AccessType.TYPE_INFO
+
     @property
     def component_indices(self):
         '''
@@ -310,6 +320,35 @@ def add_access(
         '''
         self.append(AccessInfo(access_type, node, component_indices))
 
+    def change_read_to_type_info(self):
+        '''This function is used to change a READ into a TYPEINFO.
+
+        :raises InternalError: if there is an access that is not READ or
+                               INQUIRY or there is > 1 READ accesses.
+        '''
+        read_access = None
+        for acc in self:
+
+            if acc.access_type == AccessType.READ:
+                if read_access:
+                    raise InternalError(
+                        f"Trying to change variable '{self._signature}' to "
+                        f"'TYPE_INFO' but it has more than one 'READ' access."
+                    )
+                read_access = acc
+
+            elif acc.access_type not in AccessType.non_data_accesses():
+                raise InternalError(
+                    f"Variable '{self._signature}' has a '{acc.access_type}' "
+                    f"access. change_read_to_type_info() expects only "
+                    f"inquiry accesses and a single 'READ' access.")
+
+        if not read_access:
+            raise InternalError(
+                f"Trying to change variable '{self._signature}' to "
+                f"'TYPE_INFO' but it does not have a 'READ' access.")
+        read_access.change_read_to_type_info()
+
     def change_read_to_write(self):
         '''This function is only used when analysing an assignment statement.
         The LHS has first all variables identified, which will be READ.

src/psyclone/tests/core/access_sequence_test.py

@@ -62,6 +62,15 @@ def test_access_info():
         access_info.change_read_to_write()
     assert "Trying to change variable to 'WRITE' which does not have "\
         "'READ' access." in str(err.value)
+    access_info2 = AccessInfo(AccessType.READ, Node())
+    assert str(access_info2) == "READ"
+    access_info2.change_read_to_type_info()
+    assert str(access_info2) == "TYPE_INFO"
+    assert access_info2.access_type == AccessType.TYPE_INFO
+    with pytest.raises(InternalError) as err:
+        access_info2.change_read_to_type_info()
+    assert ("Trying to change variable to 'TYPE_INFO' which does not have "
+        "'READ' access." in str(err.value))
 
     # Test setter and getter:
     component_indices = ComponentIndices([["i"]])
@@ -200,6 +209,42 @@ def test_variable_access_sequence():
     assert not accesses.has_data_access()
 
 
+def test_variable_access_sequence_read_to_type_info():
+    '''
+    Test the read_to_type_info functionality of AccessSequence
+    '''
+    accesses = AccessSequence(Signature("var_name"))
+    accesses.add_access(AccessType.INQUIRY, Node(), component_indices=None)
+    accesses.add_access(AccessType.READ, Node(), component_indices=None)
+    accesses.change_read_to_type_info()
+    assert not accesses.is_read()
+    assert not accesses.is_written()
+    assert not accesses.has_data_access()
+
+    with pytest.raises(InternalError) as err:
+        accesses.change_read_to_type_info()
+    assert ("Trying to change variable 'var_name' to "
+            "'TYPE_INFO' but it does not have a 'READ' access."
+            in str(err.value))
+
+    accesses.add_access(AccessType.WRITE, Node(), component_indices=None)
+    with pytest.raises(InternalError) as err:
+        accesses.change_read_to_type_info()
+    assert ("Variable 'var_name' has a 'WRITE' access. "
+            "change_read_to_type_info() "
+            "expects only inquiry accesses and a single 'READ' access."
+            in str(err.value))
+
+    accesses = AccessSequence(Signature("var_name"))
+    accesses.add_access(AccessType.READ, Node(), component_indices=None)
+    accesses.add_access(AccessType.READ, Node(), component_indices=None)
+
+    with pytest.raises(InternalError) as err:
+        accesses.change_read_to_type_info()
+    assert ("Trying to change variable 'var_name' to "
+            "'TYPE_INFO' but it has more than one 'READ' access."
+            in str(err.value))
+
 def test_variable_access_sequence_has_indices(fortran_reader):
     '''Test that the AccessSequence class handles indices as expected.