mesa/program: Add _mesa_symbol_table_replace_symbol()

This function allows to modify an existing symbol.

v2:
- Remove namespace usage now that it was deleted.

Signed-off-by: Samuel Iglesias Gonsálvez <[email protected]>
Reviewed-by: Timothy Arceri <[email protected]>

2 files changed, 18 insertions, 0 deletions

diff --git a/src/mesa/program/symbol_table.c b/src/mesa/program/symbol_table.c
index 23cb7ec26b1..37066c904cc 100644
--- a/src/mesa/program/symbol_table.c
+++ b/src/mesa/program/symbol_table.c
@@ -211,6 +211,20 @@ _mesa_symbol_table_add_symbol(struct _mesa_symbol_table *table,
    return 0;
 }
 
+int
+_mesa_symbol_table_replace_symbol(struct _mesa_symbol_table *table,
+                                  const char *name,
+                                  void *declaration)
+{
+    struct symbol *sym = find_symbol(table, name);
+
+    /* If the symbol doesn't exist, it cannot be replaced. */
+    if (sym == NULL)
+       return -1;
+
+    sym->data = declaration;
+    return 0;
+}
 
 int
 _mesa_symbol_table_add_global_symbol(struct _mesa_symbol_table *table,
diff --git a/src/mesa/program/symbol_table.h b/src/mesa/program/symbol_table.h
index ff1e6f2065a..cba47143ef2 100644
--- a/src/mesa/program/symbol_table.h
+++ b/src/mesa/program/symbol_table.h
@@ -32,6 +32,10 @@ extern void _mesa_symbol_table_pop_scope(struct _mesa_symbol_table *table);
 extern int _mesa_symbol_table_add_symbol(struct _mesa_symbol_table *symtab,
                                          const char *name, void *declaration);
 
+extern int _mesa_symbol_table_replace_symbol(struct _mesa_symbol_table *table,
+                                             const char *name,
+                                             void *declaration);
+
 extern int
 _mesa_symbol_table_add_global_symbol(struct _mesa_symbol_table *symtab,
                                      const char *name,