ir_call: Add method to set the function signature being called

diff --git a/src/glsl/ir.cpp b/src/glsl/ir.cpp
index 4257842..f3ee12c 100644 (file)
--- a/src/glsl/ir.cpp
+++ b/src/glsl/ir.cpp
@@ -837,6 +837,14 @@ ir_call::get_error_instruction(void *ctx)
 }
 
 void
+ir_call::set_callee(const ir_function_signature *sig)
+{
+   assert((this->type == NULL) || (this->type == sig->return_type));
+
+   this->callee = sig;
+}
+
+void
 visit_exec_list(exec_list *list, ir_visitor *visitor)
 {
    foreach_iter(exec_list_iterator, iter, *list) {

diff --git a/src/glsl/ir.h b/src/glsl/ir.h
index 0d5bbc2..89922c6 100644 (file)
--- a/src/glsl/ir.h
+++ b/src/glsl/ir.h
@@ -695,6 +695,11 @@ public:
    }
 
    /**
+    * Set the function call target
+    */
+   void set_callee(const ir_function_signature *sig);
+
+   /**
     * Generates an inline version of the function before @ir,
     * returning the return value of the function.
     */