Add functions to retrieve a reference to the element type of a java array

Author: svorenova

jbmc/src/java_bytecode/java_types.cpp

@@ -131,6 +131,26 @@ typet java_array_element_type(const symbol_typet &array_type)
   return array_type.find_type(ID_C_element_type);
 }
 
+/// Return a const reference to the element type of a given java array type
+/// \param array_type The java array type
+const typet &java_array_element_type_ref(const symbol_typet &array_type)
+{
+  INVARIANT(
+    is_java_array_tag(array_type.get_identifier()),
+    "Symbol should have array tag");
+  return array_type.find_type(ID_C_element_type);
+}
+
+/// Return a non-const reference to the element type of a given java array type
+/// \param array_type The java array type
+typet &java_array_element_type_ref(symbol_typet &array_type)
+{
+  INVARIANT(
+    is_java_array_tag(array_type.get_identifier()),
+    "Symbol should have array tag");
+  return const_cast<typet &>(array_type.find_type(ID_C_element_type));
+}
+
 /// See above
 /// \par parameters: Struct tag 'tag'
 /// \return True if the given struct is a Java array

jbmc/src/java_bytecode/java_types.h

@@ -236,6 +236,8 @@ symbol_typet java_classname(const std::string &);
 
 reference_typet java_array_type(const char subtype);
 typet java_array_element_type(const symbol_typet &array_type);
+const typet &java_array_element_type_ref(const symbol_typet &array_type);
+typet &java_array_element_type_ref(symbol_typet &array_type);
 
 bool is_reference_type(char t);