feat(gccjit): Allow set_loccation to various classes

Including:
1. field
2. function
3. rvalue

Author: tempdragon

gcc/jit/jit-recording.h

@@ -1041,6 +1041,7 @@ class array_type : public type
   bool is_signed () const final override { return false; }
 
   void replay_into (replayer *) final override;
+  void set_loc(location * loc) { m_loc = loc; }
 
  private:
   string * make_debug_string () final override;
@@ -1130,6 +1131,7 @@ class field : public memento
 
   compound_type * get_container () const { return m_container; }
   void set_container (compound_type *c) { m_container = c; }
+  void set_loc (location * loc) { m_loc = loc; }
 
   void replay_into (replayer *) override;
 
@@ -1204,6 +1206,7 @@ class compound_type : public type
   bool is_signed () const final override { return false; }
 
   bool has_known_size () const final override { return m_fields != NULL; }
+  void set_loc (location * loc) { m_loc = loc; }
 
   playback::compound_type *
   playback_compound_type ()
@@ -1345,6 +1348,7 @@ class rvalue : public memento
   }
 
   location * get_loc () const { return m_loc; }
+  void set_loc (location * loc) { m_loc = loc; }
 
   /* Get the recording::type of this rvalue.
 
@@ -1536,6 +1540,7 @@ class function : public memento
   new_block (const char *name);
 
   location *get_loc () const { return m_loc; }
+  void set_loc (location * loc) { m_loc = loc; }
   type *get_return_type () const { return m_return_type; }
   string * get_name () const { return m_name; }
   const vec<param *> &get_params () const { return m_params; }
@@ -1673,6 +1678,7 @@ class block : public memento
   bool validate ();
 
   location *get_loc () const;
+  void set_loc (location * loc);
 
   statement *get_first_statement () const;
   statement *get_last_statement () const;
@@ -2480,6 +2486,7 @@ class statement : public memento
 
   block *get_block () const { return m_block; }
   location *get_loc () const { return m_loc; }
+  void set_loc (location * loc) { m_loc = loc; }
 
 protected:
   statement (block *b, location *loc)

gcc/jit/libgccjit.cc

@@ -4745,3 +4745,46 @@ gcc_jit_context_add_top_level_asm (gcc_jit_context *ctxt,
   RETURN_IF_FAIL (asm_stmts, ctxt, NULL, "NULL asm_stmts");
   ctxt->add_top_level_asm (loc, asm_stmts);
 }
+
+/* Public entrypoint.  See description in libgccjit.h.
+
+   After error-checking, this calls the trivial
+   gcc::jit::recording::field::set_loc method, in jit-recording.h.  */
+
+void
+gcc_jit_field_set_location (gcc_jit_field *field,
+                            gcc_jit_location *loc)
+{
+  RETURN_IF_FAIL (field, NULL, NULL, "NULL field");
+
+  field->set_loc (loc);
+}
+
+
+/* Public entrypoint.  See description in libgccjit.h.
+
+   After error-checking, this calls the trivial
+   gcc::jit::recording::rvalue::set_loc method , in jit-recording.h.  */
+
+void
+gcc_jit_rvalue_set_location (gcc_jit_rvalue *rvalue,
+			     gcc_jit_location *loc)
+{
+  RETURN_IF_FAIL (rvalue, NULL, NULL, "NULL rvalue");
+
+  rvalue->set_loc (loc);
+}
+
+/* Public entrypoint.  See description in libgccjit.h.
+
+   After error-checking, this calls the trivial
+   gcc::jit::recording::function::set_loc method, in jit-recording.h.  */
+
+void
+gcc_jit_function_set_location (gcc_jit_function *func,
+                               gcc_jit_location *loc)
+{
+  RETURN_IF_FAIL (func, NULL, NULL, "NULL func");
+
+  func->set_loc (loc);
+}

gcc/jit/libgccjit.h

@@ -2180,6 +2180,16 @@ gcc_jit_target_info_supports_128bit_int (gcc_jit_target_info *info);
 extern void
 gcc_jit_type_set_packed (gcc_jit_type *type);
 
+extern void
+gcc_jit_field_set_location (gcc_jit_field *field,
+			    gcc_jit_location *loc);
+extern void
+gcc_jit_function_set_location (gcc_jit_function *func,
+			       gcc_jit_location *loc);
+extern void
+gcc_jit_rvalue_set_location (gcc_jit_rvalue *rvalue,
+			     gcc_jit_location *loc);
+
 #ifdef __cplusplus
 }
 #endif /* __cplusplus */

gcc/jit/libgccjit.map

@@ -340,3 +340,10 @@ LIBGCCJIT_ABI_36 {
   global:
     gcc_jit_function_new_temp;
 } LIBGCCJIT_ABI_35;
+
+LIBGCCJIT_ABI_37 {
+   global:
+    gcc_jit_field_set_location;
+    gcc_jit_function_set_location;
+    gcc_jit_rvalue_set_location;
+} LIBGCCJIT_ABI_36;