Fix ir_dead_code for function refactoring.

1 files changed, 6 insertions, 0 deletions

diff --git a/ir.h b/ir.h
index fa9a132..c93c043 100644
--- a/ir.h
+++ b/ir.h
@@ -52,6 +52,7 @@ public:
     */
    /*@{*/
    virtual class ir_variable *          as_variable()         { return NULL; }
+   virtual class ir_function *          as_function()         { return NULL; }
    virtual class ir_dereference *       as_dereference()      { return NULL; }
    virtual class ir_rvalue *            as_rvalue()           { return NULL; }
    virtual class ir_label *             as_label()            { return NULL; }
@@ -243,6 +244,11 @@ class ir_function : public ir_instruction {
 public:
    ir_function(const char *name);
 
+   virtual ir_function *as_function()
+   {
+      return this;
+   }
+
    virtual void accept(ir_visitor *v)
    {
       v->visit(this);