glsl2: Add pass for supporting variable vector indexing in rvalues.

[mesa.git] / src / mesa / shader / ir_to_mesa.cpp

/*
 * Copyright (C) 2005-2007  Brian Paul   All Rights Reserved.
 * Copyright (C) 2008  VMware, Inc.   All Rights Reserved.
 * Copyright © 2010 Intel Corporation
 *
 * Permission is hereby granted, free of charge, to any person obtaining a
 * copy of this software and associated documentation files (the "Software"),
 * to deal in the Software without restriction, including without limitation
 * the rights to use, copy, modify, merge, publish, distribute, sublicense,
 * and/or sell copies of the Software, and to permit persons to whom the
 * Software is furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice (including the next
 * paragraph) shall be included in all copies or substantial portions of the
 * Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL
 * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
 * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
 * DEALINGS IN THE SOFTWARE.
 */

/**
 * \file ir_to_mesa.cpp
 *
 * Translates the IR to ARB_fragment_program text if possible,
 * printing the result
 */

#include <stdio.h>
#include "ir.h"
#include "ir_visitor.h"
#include "ir_print_visitor.h"
#include "ir_expression_flattening.h"
#include "glsl_types.h"
#include "glsl_parser_extras.h"
#include "../glsl/program.h"
#include "ir_optimization.h"
#include "ast.h"

extern "C" {
#include "main/mtypes.h"
#include "shader/prog_instruction.h"
#include "shader/prog_optimize.h"
#include "shader/prog_print.h"
#include "shader/program.h"
#include "shader/prog_uniform.h"
#include "shader/prog_parameter.h"
#include "shader/shader_api.h"
}

/**
 * This struct is a corresponding struct to Mesa prog_src_register, with
 * wider fields.
 */
typedef struct ir_to_mesa_src_reg {
   int file; /**< PROGRAM_* from Mesa */
   int index; /**< temporary index, VERT_ATTRIB_*, FRAG_ATTRIB_*, etc. */
   GLuint swizzle; /**< SWIZZLE_XYZWONEZERO swizzles from Mesa. */
   int negate; /**< NEGATE_XYZW mask from mesa */
   bool reladdr; /**< Register index should be offset by address reg. */
} ir_to_mesa_src_reg;

typedef struct ir_to_mesa_dst_reg {
   int file; /**< PROGRAM_* from Mesa */
   int index; /**< temporary index, VERT_ATTRIB_*, FRAG_ATTRIB_*, etc. */
   int writemask; /**< Bitfield of WRITEMASK_[XYZW] */
   GLuint cond_mask:4;
} ir_to_mesa_dst_reg;

extern ir_to_mesa_src_reg ir_to_mesa_undef;

class ir_to_mesa_instruction : public exec_node {
public:
   enum prog_opcode op;
   ir_to_mesa_dst_reg dst_reg;
   ir_to_mesa_src_reg src_reg[3];
   /** Pointer to the ir source this tree came from for debugging */
   ir_instruction *ir;
   GLboolean cond_update;
   int sampler; /**< sampler index */
   int tex_target; /**< One of TEXTURE_*_INDEX */
   GLboolean tex_shadow;
};

class temp_entry : public exec_node {
public:
   temp_entry(ir_variable *var, int file, int index)
      : file(file), index(index), var(var)
   {
      /* empty */
   }

   int file;
   int index;
   ir_variable *var; /* variable that maps to this, if any */
};

class ir_to_mesa_visitor : public ir_visitor {
public:
   ir_to_mesa_visitor();

   GLcontext *ctx;
   struct gl_program *prog;

   int next_temp;

   temp_entry *find_variable_storage(ir_variable *var);

   ir_to_mesa_src_reg get_temp(const glsl_type *type);

   struct ir_to_mesa_src_reg src_reg_for_float(float val);

   /**
    * \name Visit methods
    *
    * As typical for the visitor pattern, there must be one \c visit method for
    * each concrete subclass of \c ir_instruction.  Virtual base classes within
    * the hierarchy should not have \c visit methods.
    */
   /*@{*/
   virtual void visit(ir_variable *);
   virtual void visit(ir_loop *);
   virtual void visit(ir_loop_jump *);
   virtual void visit(ir_function_signature *);
   virtual void visit(ir_function *);
   virtual void visit(ir_expression *);
   virtual void visit(ir_swizzle *);
   virtual void visit(ir_dereference_variable  *);
   virtual void visit(ir_dereference_array *);
   virtual void visit(ir_dereference_record *);
   virtual void visit(ir_assignment *);
   virtual void visit(ir_constant *);
   virtual void visit(ir_call *);
   virtual void visit(ir_return *);
   virtual void visit(ir_discard *);
   virtual void visit(ir_texture *);
   virtual void visit(ir_if *);
   /*@}*/

   struct ir_to_mesa_src_reg result;

   /** List of temp_entry */
   exec_list variable_storage;

   /** List of ir_to_mesa_instruction */
   exec_list instructions;

   ir_to_mesa_instruction *ir_to_mesa_emit_op1(ir_instruction *ir,
                                               enum prog_opcode op,
                                               ir_to_mesa_dst_reg dst,
                                               ir_to_mesa_src_reg src0);

   ir_to_mesa_instruction *ir_to_mesa_emit_op2(ir_instruction *ir,
                                               enum prog_opcode op,
                                               ir_to_mesa_dst_reg dst,
                                               ir_to_mesa_src_reg src0,
                                               ir_to_mesa_src_reg src1);

   ir_to_mesa_instruction *ir_to_mesa_emit_op3(ir_instruction *ir,
                                               enum prog_opcode op,
                                               ir_to_mesa_dst_reg dst,
                                               ir_to_mesa_src_reg src0,
                                               ir_to_mesa_src_reg src1,
                                               ir_to_mesa_src_reg src2);

   void ir_to_mesa_emit_scalar_op1(ir_instruction *ir,
                                   enum prog_opcode op,
                                   ir_to_mesa_dst_reg dst,
                                   ir_to_mesa_src_reg src0);

   void ir_to_mesa_emit_scalar_op2(ir_instruction *ir,
                                   enum prog_opcode op,
                                   ir_to_mesa_dst_reg dst,
                                   ir_to_mesa_src_reg src0,
                                   ir_to_mesa_src_reg src1);

   int *sampler_map;
   int sampler_map_size;

   void map_sampler(int location, int sampler);
   int get_sampler_number(int location);

   void *mem_ctx;
};

ir_to_mesa_src_reg ir_to_mesa_undef = {
   PROGRAM_UNDEFINED, 0, SWIZZLE_NOOP, NEGATE_NONE, false,
};

ir_to_mesa_dst_reg ir_to_mesa_undef_dst = {
   PROGRAM_UNDEFINED, 0, SWIZZLE_NOOP
};

ir_to_mesa_dst_reg ir_to_mesa_address_reg = {
   PROGRAM_ADDRESS, 0, WRITEMASK_X
};

static int swizzle_for_size(int size)
{
   int size_swizzles[4] = {
      MAKE_SWIZZLE4(SWIZZLE_X, SWIZZLE_X, SWIZZLE_X, SWIZZLE_X),
      MAKE_SWIZZLE4(SWIZZLE_X, SWIZZLE_Y, SWIZZLE_Y, SWIZZLE_Y),
      MAKE_SWIZZLE4(SWIZZLE_X, SWIZZLE_Y, SWIZZLE_Z, SWIZZLE_Z),
      MAKE_SWIZZLE4(SWIZZLE_X, SWIZZLE_Y, SWIZZLE_Z, SWIZZLE_W),
   };

   return size_swizzles[size - 1];
}

ir_to_mesa_instruction *
ir_to_mesa_visitor::ir_to_mesa_emit_op3(ir_instruction *ir,
                                        enum prog_opcode op,
                                        ir_to_mesa_dst_reg dst,
                                        ir_to_mesa_src_reg src0,
                                        ir_to_mesa_src_reg src1,
                                        ir_to_mesa_src_reg src2)
{
   ir_to_mesa_instruction *inst = new(mem_ctx) ir_to_mesa_instruction();

   inst->op = op;
   inst->dst_reg = dst;
   inst->src_reg[0] = src0;
   inst->src_reg[1] = src1;
   inst->src_reg[2] = src2;
   inst->ir = ir;

   this->instructions.push_tail(inst);

   return inst;
}


ir_to_mesa_instruction *
ir_to_mesa_visitor::ir_to_mesa_emit_op2(ir_instruction *ir,
                                        enum prog_opcode op,
                                        ir_to_mesa_dst_reg dst,
                                        ir_to_mesa_src_reg src0,
                                        ir_to_mesa_src_reg src1)
{
   return ir_to_mesa_emit_op3(ir, op, dst, src0, src1, ir_to_mesa_undef);
}

ir_to_mesa_instruction *
ir_to_mesa_visitor::ir_to_mesa_emit_op1(ir_instruction *ir,
                                        enum prog_opcode op,
                                        ir_to_mesa_dst_reg dst,
                                        ir_to_mesa_src_reg src0)
{
   return ir_to_mesa_emit_op3(ir, op, dst,
                              src0, ir_to_mesa_undef, ir_to_mesa_undef);
}

void
ir_to_mesa_visitor::map_sampler(int location, int sampler)
{
   if (this->sampler_map_size <= location) {
      this->sampler_map = talloc_realloc(this->mem_ctx, this->sampler_map,
                                         int, location + 1);
      this->sampler_map_size = location + 1;
   }

   this->sampler_map[location] = sampler;
}

int
ir_to_mesa_visitor::get_sampler_number(int location)
{
   assert(location < this->sampler_map_size);
   return this->sampler_map[location];
}

inline ir_to_mesa_dst_reg
ir_to_mesa_dst_reg_from_src(ir_to_mesa_src_reg reg)
{
   ir_to_mesa_dst_reg dst_reg;

   dst_reg.file = reg.file;
   dst_reg.index = reg.index;
   dst_reg.writemask = WRITEMASK_XYZW;
   dst_reg.cond_mask = COND_TR;

   return dst_reg;
}

inline ir_to_mesa_src_reg
ir_to_mesa_src_reg_from_dst(ir_to_mesa_dst_reg reg)
{
   ir_to_mesa_src_reg src_reg;

   src_reg.file = reg.file;
   src_reg.index = reg.index;
   src_reg.swizzle = SWIZZLE_XYZW;
   src_reg.negate = 0;
   src_reg.reladdr = 0;

   return src_reg;
}

/**
 * Emits Mesa scalar opcodes to produce unique answers across channels.
 *
 * Some Mesa opcodes are scalar-only, like ARB_fp/vp.  The src X
 * channel determines the result across all channels.  So to do a vec4
 * of this operation, we want to emit a scalar per source channel used
 * to produce dest channels.
 */
void
ir_to_mesa_visitor::ir_to_mesa_emit_scalar_op2(ir_instruction *ir,
                                               enum prog_opcode op,
                                               ir_to_mesa_dst_reg dst,
                                               ir_to_mesa_src_reg orig_src0,
                                               ir_to_mesa_src_reg orig_src1)
{
   int i, j;
   int done_mask = ~dst.writemask;

   /* Mesa RCP is a scalar operation splatting results to all channels,
    * like ARB_fp/vp.  So emit as many RCPs as necessary to cover our
    * dst channels.
    */
   for (i = 0; i < 4; i++) {
      GLuint this_mask = (1 << i);
      ir_to_mesa_instruction *inst;
      ir_to_mesa_src_reg src0 = orig_src0;
      ir_to_mesa_src_reg src1 = orig_src1;

      if (done_mask & this_mask)
         continue;

      GLuint src0_swiz = GET_SWZ(src0.swizzle, i);
      GLuint src1_swiz = GET_SWZ(src1.swizzle, i);
      for (j = i + 1; j < 4; j++) {
         if (!(done_mask & (1 << j)) &&
             GET_SWZ(src0.swizzle, j) == src0_swiz &&
             GET_SWZ(src1.swizzle, j) == src1_swiz) {
            this_mask |= (1 << j);
         }
      }
      src0.swizzle = MAKE_SWIZZLE4(src0_swiz, src0_swiz,
                                   src0_swiz, src0_swiz);
      src1.swizzle = MAKE_SWIZZLE4(src1_swiz, src1_swiz,
                                  src1_swiz, src1_swiz);

      inst = ir_to_mesa_emit_op2(ir, op,
                                 dst,
                                 src0,
                                 src1);
      inst->dst_reg.writemask = this_mask;
      done_mask |= this_mask;
   }
}

void
ir_to_mesa_visitor::ir_to_mesa_emit_scalar_op1(ir_instruction *ir,
                                               enum prog_opcode op,
                                               ir_to_mesa_dst_reg dst,
                                               ir_to_mesa_src_reg src0)
{
   ir_to_mesa_src_reg undef = ir_to_mesa_undef;

   undef.swizzle = SWIZZLE_XXXX;

   ir_to_mesa_emit_scalar_op2(ir, op, dst, src0, undef);
}

struct ir_to_mesa_src_reg
ir_to_mesa_visitor::src_reg_for_float(float val)
{
   ir_to_mesa_src_reg src_reg;

   src_reg.file = PROGRAM_CONSTANT;
   src_reg.index = _mesa_add_unnamed_constant(this->prog->Parameters,
                                              &val, 1, &src_reg.swizzle);

   return src_reg;
}

static int
type_size(const struct glsl_type *type)
{
   unsigned int i;
   int size;

   switch (type->base_type) {
   case GLSL_TYPE_UINT:
   case GLSL_TYPE_INT:
   case GLSL_TYPE_FLOAT:
   case GLSL_TYPE_BOOL:
      if (type->is_matrix()) {
         return 4; /* FINISHME: Not all matrices are 4x4. */
      } else {
         /* Regardless of size of vector, it gets a vec4. This is bad
          * packing for things like floats, but otherwise arrays become a
          * mess.  Hopefully a later pass over the code can pack scalars
          * down if appropriate.
          */
         return 1;
      }
   case GLSL_TYPE_ARRAY:
      return type_size(type->fields.array) * type->length;
   case GLSL_TYPE_STRUCT:
      size = 0;
      for (i = 0; i < type->length; i++) {
         size += type_size(type->fields.structure[i].type);
      }
      return size;
   default:
      assert(0);
   }
}

/**
 * In the initial pass of codegen, we assign temporary numbers to
 * intermediate results.  (not SSA -- variable assignments will reuse
 * storage).  Actual register allocation for the Mesa VM occurs in a
 * pass over the Mesa IR later.
 */
ir_to_mesa_src_reg
ir_to_mesa_visitor::get_temp(const glsl_type *type)
{
   ir_to_mesa_src_reg src_reg;
   int swizzle[4];
   int i;

   assert(!type->is_array());

   src_reg.file = PROGRAM_TEMPORARY;
   src_reg.index = next_temp;
   src_reg.reladdr = false;
   next_temp += type_size(type);

   for (i = 0; i < type->vector_elements; i++)
      swizzle[i] = i;
   for (; i < 4; i++)
      swizzle[i] = type->vector_elements - 1;
   src_reg.swizzle = MAKE_SWIZZLE4(swizzle[0], swizzle[1],
                                   swizzle[2], swizzle[3]);
   src_reg.negate = 0;

   return src_reg;
}

temp_entry *
ir_to_mesa_visitor::find_variable_storage(ir_variable *var)
{
   
   temp_entry *entry;

   foreach_iter(exec_list_iterator, iter, this->variable_storage) {
      entry = (temp_entry *)iter.get();

      if (entry->var == var)
         return entry;
   }

   return NULL;
}

void
ir_to_mesa_visitor::visit(ir_variable *ir)
{
   (void)ir;
}

void
ir_to_mesa_visitor::visit(ir_loop *ir)
{
   assert(!ir->from);
   assert(!ir->to);
   assert(!ir->increment);
   assert(!ir->counter);

   ir_to_mesa_emit_op1(NULL, OPCODE_BGNLOOP,
                       ir_to_mesa_undef_dst, ir_to_mesa_undef);

   visit_exec_list(&ir->body_instructions, this);

   ir_to_mesa_emit_op1(NULL, OPCODE_ENDLOOP,
                       ir_to_mesa_undef_dst, ir_to_mesa_undef);
}

void
ir_to_mesa_visitor::visit(ir_loop_jump *ir)
{
   switch (ir->mode) {
   case ir_loop_jump::jump_break:
      ir_to_mesa_emit_op1(NULL, OPCODE_BRK,
                          ir_to_mesa_undef_dst, ir_to_mesa_undef);
      break;
   case ir_loop_jump::jump_continue:
      ir_to_mesa_emit_op1(NULL, OPCODE_CONT,
                          ir_to_mesa_undef_dst, ir_to_mesa_undef);
      break;
   }
}


void
ir_to_mesa_visitor::visit(ir_function_signature *ir)
{
   assert(0);
   (void)ir;
}

void
ir_to_mesa_visitor::visit(ir_function *ir)
{
   /* Ignore function bodies other than main() -- we shouldn't see calls to
    * them since they should all be inlined before we get to ir_to_mesa.
    */
   if (strcmp(ir->name, "main") == 0) {
      const ir_function_signature *sig;
      exec_list empty;

      sig = ir->matching_signature(&empty);

      assert(sig);

      foreach_iter(exec_list_iterator, iter, sig->body) {
         ir_instruction *ir = (ir_instruction *)iter.get();

         ir->accept(this);
      }
   }
}

void
ir_to_mesa_visitor::visit(ir_expression *ir)
{
   unsigned int operand;
   struct ir_to_mesa_src_reg op[2];
   struct ir_to_mesa_src_reg result_src;
   struct ir_to_mesa_dst_reg result_dst;
   const glsl_type *vec4_type = glsl_type::get_instance(GLSL_TYPE_FLOAT, 4, 1);
   const glsl_type *vec3_type = glsl_type::get_instance(GLSL_TYPE_FLOAT, 3, 1);
   const glsl_type *vec2_type = glsl_type::get_instance(GLSL_TYPE_FLOAT, 2, 1);

   for (operand = 0; operand < ir->get_num_operands(); operand++) {
      this->result.file = PROGRAM_UNDEFINED;
      ir->operands[operand]->accept(this);
      if (this->result.file == PROGRAM_UNDEFINED) {
         ir_print_visitor v;
         printf("Failed to get tree for expression operand:\n");
         ir->operands[operand]->accept(&v);
         exit(1);
      }
      op[operand] = this->result;

      /* Only expression implemented for matrices yet */
      assert(!ir->operands[operand]->type->is_matrix() ||
             ir->operation == ir_binop_mul);
   }

   this->result.file = PROGRAM_UNDEFINED;

   /* Storage for our result.  Ideally for an assignment we'd be using
    * the actual storage for the result here, instead.
    */
   result_src = get_temp(ir->type);
   /* convenience for the emit functions below. */
   result_dst = ir_to_mesa_dst_reg_from_src(result_src);
   /* Limit writes to the channels that will be used by result_src later.
    * This does limit this temp's use as a temporary for multi-instruction
    * sequences.
    */
   result_dst.writemask = (1 << ir->type->vector_elements) - 1;

   switch (ir->operation) {
   case ir_unop_logic_not:
      ir_to_mesa_emit_op2(ir, OPCODE_SEQ, result_dst,
                          op[0], src_reg_for_float(0.0));
      break;
   case ir_unop_neg:
      op[0].negate = ~op[0].negate;
      result_src = op[0];
      break;
   case ir_unop_abs:
      ir_to_mesa_emit_op1(ir, OPCODE_ABS, result_dst, op[0]);
      break;
   case ir_unop_sign:
      ir_to_mesa_emit_op1(ir, OPCODE_SSG, result_dst, op[0]);
      break;
   case ir_unop_rcp:
      ir_to_mesa_emit_scalar_op1(ir, OPCODE_RCP, result_dst, op[0]);
      break;

   case ir_unop_exp:
      ir_to_mesa_emit_scalar_op2(ir, OPCODE_POW, result_dst,
                                 src_reg_for_float(M_E), op[0]);
      break;
   case ir_unop_exp2:
      ir_to_mesa_emit_scalar_op1(ir, OPCODE_EX2, result_dst, op[0]);
      break;
   case ir_unop_log:
      ir_to_mesa_emit_scalar_op1(ir, OPCODE_LOG, result_dst, op[0]);
      break;
   case ir_unop_log2:
      ir_to_mesa_emit_scalar_op1(ir, OPCODE_LG2, result_dst, op[0]);
      break;
   case ir_unop_sin:
      ir_to_mesa_emit_scalar_op1(ir, OPCODE_SIN, result_dst, op[0]);
      break;
   case ir_unop_cos:
      ir_to_mesa_emit_scalar_op1(ir, OPCODE_COS, result_dst, op[0]);
      break;

   case ir_unop_dFdx:
      ir_to_mesa_emit_op1(ir, OPCODE_DDX, result_dst, op[0]);
      break;
   case ir_unop_dFdy:
      ir_to_mesa_emit_op1(ir, OPCODE_DDY, result_dst, op[0]);
      break;

   case ir_binop_add:
      ir_to_mesa_emit_op2(ir, OPCODE_ADD, result_dst, op[0], op[1]);
      break;
   case ir_binop_sub:
      ir_to_mesa_emit_op2(ir, OPCODE_SUB, result_dst, op[0], op[1]);
      break;
   case ir_binop_mul:
      if (ir->operands[0]->type->is_matrix() &&
          !ir->operands[1]->type->is_matrix()) {
         if (ir->operands[1]->type->is_scalar()) {
            ir_to_mesa_dst_reg dst_column = result_dst;
            ir_to_mesa_src_reg src_column = op[0];
            for (int i = 0; i < ir->operands[0]->type->matrix_columns; i++) {
               ir_to_mesa_emit_op2(ir, OPCODE_MUL,
                                   dst_column, src_column, op[1]);
               dst_column.index++;
               src_column.index++;
            }
         } else {
            ir_to_mesa_src_reg src_column = op[0];
            ir_to_mesa_src_reg src_chan = op[1];
            assert(!ir->operands[1]->type->is_matrix() ||
                    !"FINISHME: matrix * matrix");
             for (int i = 0; i < ir->operands[0]->type->matrix_columns; i++) {
                src_chan.swizzle = MAKE_SWIZZLE4(i, i, i, i);
                if (i == 0) {
                   ir_to_mesa_emit_op2(ir, OPCODE_MUL,
                                       result_dst, src_column, src_chan);
                } else {
                   ir_to_mesa_emit_op3(ir, OPCODE_MAD,
                                       result_dst, src_column, src_chan,
                                       result_src);
                }
                src_column.index++;
            }
         }
      } else if (ir->operands[1]->type->is_matrix()) {
         if (ir->operands[0]->type->is_scalar()) {
            ir_to_mesa_dst_reg dst_column = result_dst;
            ir_to_mesa_src_reg src_column = op[1];
            for (int i = 0; i < ir->operands[1]->type->matrix_columns; i++) {
               ir_to_mesa_emit_op2(ir, OPCODE_MUL,
                                   dst_column, src_column, op[0]);
               dst_column.index++;
               src_column.index++;
            }
         } else {
            ir_to_mesa_src_reg src_column = op[1];
            ir_to_mesa_dst_reg dst_chan = result_dst;

            /* FINISHME here and above: non-square matrices */
            assert(ir->operands[1]->type->vector_elements ==
                   ir->operands[1]->type->matrix_columns);

            for (int i = 0; i < ir->operands[0]->type->vector_elements; i++) {
               dst_chan.writemask = (1 << i);
               switch (ir->operands[0]->type->vector_elements) {
               case 2:
                  ir_to_mesa_emit_op2(ir, OPCODE_DP2, dst_chan, op[0], src_column);
                  break;
               case 3:
                  ir_to_mesa_emit_op2(ir, OPCODE_DP3, dst_chan, op[0], src_column);
                  break;
               case 4:
                  ir_to_mesa_emit_op2(ir, OPCODE_DP4, dst_chan, op[0], src_column);
                  break;
               default:
                  assert(0);
               }
               src_column.index++;
            }
         }
      } else {
         assert(!ir->operands[0]->type->is_matrix());
         assert(!ir->operands[1]->type->is_matrix());
         ir_to_mesa_emit_op2(ir, OPCODE_MUL, result_dst, op[0], op[1]);
      }
      break;
   case ir_binop_div:
      assert(!"not reached: should be handled by ir_div_to_mul_rcp");
   case ir_binop_mod:
      assert(!"ir_binop_mod should have been converted to b * fract(a/b)");
      break;

   case ir_binop_less:
      ir_to_mesa_emit_op2(ir, OPCODE_SLT, result_dst, op[0], op[1]);
      break;
   case ir_binop_greater:
      ir_to_mesa_emit_op2(ir, OPCODE_SGT, result_dst, op[0], op[1]);
      break;
   case ir_binop_lequal:
      ir_to_mesa_emit_op2(ir, OPCODE_SLE, result_dst, op[0], op[1]);
      break;
   case ir_binop_gequal:
      ir_to_mesa_emit_op2(ir, OPCODE_SGE, result_dst, op[0], op[1]);
      break;
   case ir_binop_equal:
      ir_to_mesa_emit_op2(ir, OPCODE_SEQ, result_dst, op[0], op[1]);
      break;
   case ir_binop_logic_xor:
   case ir_binop_nequal:
      ir_to_mesa_emit_op2(ir, OPCODE_SNE, result_dst, op[0], op[1]);
      break;

   case ir_binop_logic_or:
      /* This could be a saturated add and skip the SNE. */
      ir_to_mesa_emit_op2(ir, OPCODE_ADD,
                          result_dst,
                          op[0], op[1]);

      ir_to_mesa_emit_op2(ir, OPCODE_SNE,
                          result_dst,
                          result_src, src_reg_for_float(0.0));
      break;

   case ir_binop_logic_and:
      /* the bool args are stored as float 0.0 or 1.0, so "mul" gives us "and". */
      ir_to_mesa_emit_op2(ir, OPCODE_MUL,
                          result_dst,
                          op[0], op[1]);
      break;

   case ir_binop_dot:
      if (ir->operands[0]->type == vec4_type) {
         assert(ir->operands[1]->type == vec4_type);
         ir_to_mesa_emit_op2(ir, OPCODE_DP4,
                             result_dst,
                             op[0], op[1]);
      } else if (ir->operands[0]->type == vec3_type) {
         assert(ir->operands[1]->type == vec3_type);
         ir_to_mesa_emit_op2(ir, OPCODE_DP3,
                             result_dst,
                             op[0], op[1]);
      } else if (ir->operands[0]->type == vec2_type) {
         assert(ir->operands[1]->type == vec2_type);
         ir_to_mesa_emit_op2(ir, OPCODE_DP2,
                             result_dst,
                             op[0], op[1]);
      }
      break;
   case ir_unop_sqrt:
      ir_to_mesa_emit_scalar_op1(ir, OPCODE_RSQ, result_dst, op[0]);
      ir_to_mesa_emit_scalar_op1(ir, OPCODE_RCP, result_dst, result_src);
      /* For incoming channels < 0, set the result to 0. */
      ir_to_mesa_emit_op3(ir, OPCODE_CMP, result_dst,
                          op[0], src_reg_for_float(0.0), result_src);
      break;
   case ir_unop_rsq:
      ir_to_mesa_emit_scalar_op1(ir, OPCODE_RSQ, result_dst, op[0]);
      break;
   case ir_unop_i2f:
   case ir_unop_b2f:
   case ir_unop_b2i:
      /* Mesa IR lacks types, ints are stored as truncated floats. */
      result_src = op[0];
      break;
   case ir_unop_f2i:
      ir_to_mesa_emit_op1(ir, OPCODE_TRUNC, result_dst, op[0]);
      break;
   case ir_unop_f2b:
   case ir_unop_i2b:
      ir_to_mesa_emit_op2(ir, OPCODE_SNE, result_dst,
                          result_src, src_reg_for_float(0.0));
      break;
   case ir_unop_trunc:
      ir_to_mesa_emit_op1(ir, OPCODE_TRUNC, result_dst, op[0]);
      break;
   case ir_unop_ceil:
      op[0].negate = ~op[0].negate;
      ir_to_mesa_emit_op1(ir, OPCODE_FLR, result_dst, op[0]);
      result_src.negate = ~result_src.negate;
      break;
   case ir_unop_floor:
      ir_to_mesa_emit_op1(ir, OPCODE_FLR, result_dst, op[0]);
      break;
   case ir_unop_fract:
      ir_to_mesa_emit_op1(ir, OPCODE_FRC, result_dst, op[0]);
      break;

   case ir_binop_min:
      ir_to_mesa_emit_op2(ir, OPCODE_MIN, result_dst, op[0], op[1]);
      break;
   case ir_binop_max:
      ir_to_mesa_emit_op2(ir, OPCODE_MAX, result_dst, op[0], op[1]);
      break;
   case ir_binop_pow:
      ir_to_mesa_emit_scalar_op2(ir, OPCODE_POW, result_dst, op[0], op[1]);
      break;

   case ir_unop_bit_not:
   case ir_unop_u2f:
   case ir_binop_lshift:
   case ir_binop_rshift:
   case ir_binop_bit_and:
   case ir_binop_bit_xor:
   case ir_binop_bit_or:
      assert(!"GLSL 1.30 features unsupported");
      break;
   }

   this->result = result_src;
}


void
ir_to_mesa_visitor::visit(ir_swizzle *ir)
{
   ir_to_mesa_src_reg src_reg;
   int i;
   int swizzle[4];

   /* Note that this is only swizzles in expressions, not those on the left
    * hand side of an assignment, which do write masking.  See ir_assignment
    * for that.
    */

   ir->val->accept(this);
   src_reg = this->result;
   assert(src_reg.file != PROGRAM_UNDEFINED);

   for (i = 0; i < 4; i++) {
      if (i < ir->type->vector_elements) {
         switch (i) {
         case 0:
            swizzle[i] = GET_SWZ(src_reg.swizzle, ir->mask.x);
            break;
         case 1:
            swizzle[i] = GET_SWZ(src_reg.swizzle, ir->mask.y);
            break;
         case 2:
            swizzle[i] = GET_SWZ(src_reg.swizzle, ir->mask.z);
            break;
         case 3:
            swizzle[i] = GET_SWZ(src_reg.swizzle, ir->mask.w);
            break;
         }
      } else {
         /* If the type is smaller than a vec4, replicate the last
          * channel out.
          */
         swizzle[i] = swizzle[ir->type->vector_elements - 1];
      }
   }

   src_reg.swizzle = MAKE_SWIZZLE4(swizzle[0],
                                   swizzle[1],
                                   swizzle[2],
                                   swizzle[3]);

   this->result = src_reg;
}

static temp_entry *
get_builtin_matrix_ref(void *mem_ctx, struct gl_program *prog, ir_variable *var)
{
   /*
    * NOTE: The ARB_vertex_program extension specified that matrices get
    * loaded in registers in row-major order.  With GLSL, we want column-
    * major order.  So, we need to transpose all matrices here...
    */
   static const struct {
      const char *name;
      int matrix;
      int modifier;
   } matrices[] = {
      { "gl_ModelViewMatrix", STATE_MODELVIEW_MATRIX, STATE_MATRIX_TRANSPOSE },
      { "gl_ModelViewMatrixInverse", STATE_MODELVIEW_MATRIX, STATE_MATRIX_INVTRANS },
      { "gl_ModelViewMatrixTranspose", STATE_MODELVIEW_MATRIX, 0 },
      { "gl_ModelViewMatrixInverseTranspose", STATE_MODELVIEW_MATRIX, STATE_MATRIX_INVERSE },

      { "gl_ProjectionMatrix", STATE_PROJECTION_MATRIX, STATE_MATRIX_TRANSPOSE },
      { "gl_ProjectionMatrixInverse", STATE_PROJECTION_MATRIX, STATE_MATRIX_INVTRANS },
      { "gl_ProjectionMatrixTranspose", STATE_PROJECTION_MATRIX, 0 },
      { "gl_ProjectionMatrixInverseTranspose", STATE_PROJECTION_MATRIX, STATE_MATRIX_INVERSE },

      { "gl_ModelViewProjectionMatrix", STATE_MVP_MATRIX, STATE_MATRIX_TRANSPOSE },
      { "gl_ModelViewProjectionMatrixInverse", STATE_MVP_MATRIX, STATE_MATRIX_INVTRANS },
      { "gl_ModelViewProjectionMatrixTranspose", STATE_MVP_MATRIX, 0 },
      { "gl_ModelViewProjectionMatrixInverseTranspose", STATE_MVP_MATRIX, STATE_MATRIX_INVERSE },

      { "gl_TextureMatrix", STATE_TEXTURE_MATRIX, STATE_MATRIX_TRANSPOSE },
      { "gl_TextureMatrixInverse", STATE_TEXTURE_MATRIX, STATE_MATRIX_INVTRANS },
      { "gl_TextureMatrixTranspose", STATE_TEXTURE_MATRIX, 0 },
      { "gl_TextureMatrixInverseTranspose", STATE_TEXTURE_MATRIX, STATE_MATRIX_INVERSE },

      { "gl_NormalMatrix", STATE_MODELVIEW_MATRIX, STATE_MATRIX_INVERSE },

   };
   unsigned int i;
   temp_entry *entry;

   /* C++ gets angry when we try to use an int as a gl_state_index, so we use
    * ints for gl_state_index.  Make sure they're compatible.
    */
   assert(sizeof(gl_state_index) == sizeof(int));

   for (i = 0; i < Elements(matrices); i++) {
      if (strcmp(var->name, matrices[i].name) == 0) {
         int j;
         int last_pos = -1, base_pos = -1;
         int tokens[STATE_LENGTH];

         tokens[0] = matrices[i].matrix;
         tokens[1] = 0; /* array index! */
         tokens[4] = matrices[i].modifier;

         /* Add a ref for each column.  It looks like the reason we do
          * it this way is that _mesa_add_state_reference doesn't work
          * for things that aren't vec4s, so the tokens[2]/tokens[3]
          * range has to be equal.
          */
         for (j = 0; j < 4; j++) {
            tokens[2] = j;
            tokens[3] = j;
            int pos = _mesa_add_state_reference(prog->Parameters,
                                                (gl_state_index *)tokens);
            assert(last_pos == -1 || last_pos == base_pos + j);
            if (base_pos == -1)
               base_pos = pos;
         }

         entry = new(mem_ctx) temp_entry(var,
                                         PROGRAM_STATE_VAR,
                                         base_pos);

         return entry;
      }
   }

   return NULL;
}

void
ir_to_mesa_visitor::visit(ir_dereference_variable *ir)
{
   ir_to_mesa_src_reg src_reg;
   temp_entry *entry = find_variable_storage(ir->var);
   unsigned int loc;

   if (!entry) {
      switch (ir->var->mode) {
      case ir_var_uniform:
         entry = get_builtin_matrix_ref(this->mem_ctx, this->prog, ir->var);
         if (entry)
            break;

         /* FINISHME: Fix up uniform name for arrays and things */
         if (ir->var->type->base_type == GLSL_TYPE_SAMPLER) {
            /* FINISHME: we whack the location of the var here, which
             * is probably not expected.  But we need to communicate
             * mesa's sampler number to the tex instruction.
             */
            int sampler = _mesa_add_sampler(this->prog->Parameters,
                                            ir->var->name,
                                            ir->var->type->gl_type);
            map_sampler(ir->var->location, sampler);

            entry = new(mem_ctx) temp_entry(ir->var, PROGRAM_SAMPLER, sampler);
            this->variable_storage.push_tail(entry);
            break;
         }

         assert(ir->var->type->gl_type != 0 &&
                ir->var->type->gl_type != GL_INVALID_ENUM);
         loc = _mesa_add_uniform(this->prog->Parameters,
                                 ir->var->name,
                                 type_size(ir->var->type) * 4,
                                 ir->var->type->gl_type,
                                 NULL);

         /* Always mark the uniform used at this point.  If it isn't
          * used, dead code elimination should have nuked the decl already.
          */
         this->prog->Parameters->Parameters[loc].Used = GL_TRUE;

         entry = new(mem_ctx) temp_entry(ir->var, PROGRAM_UNIFORM, loc);
         this->variable_storage.push_tail(entry);
         break;
      case ir_var_in:
      case ir_var_out:
      case ir_var_inout:
         /* The linker assigns locations for varyings and attributes,
          * including deprecated builtins (like gl_Color), user-assign
          * generic attributes (glBindVertexLocation), and
          * user-defined varyings.
          *
          * FINISHME: We would hit this path for function arguments.  Fix!
          */
         assert(ir->var->location != -1);
         if (ir->var->mode == ir_var_in ||
             ir->var->mode == ir_var_inout) {
            entry = new(mem_ctx) temp_entry(ir->var,
                                            PROGRAM_INPUT,
                                            ir->var->location);

            if (this->prog->Target == GL_VERTEX_PROGRAM_ARB &&
                ir->var->location >= VERT_ATTRIB_GENERIC0) {
               _mesa_add_attribute(prog->Attributes,
                                   ir->var->name,
                                   type_size(ir->var->type) * 4,
                                   ir->var->type->gl_type,
                                   ir->var->location - VERT_ATTRIB_GENERIC0);
            }
         } else {
            entry = new(mem_ctx) temp_entry(ir->var,
                                            PROGRAM_OUTPUT,
                                            ir->var->location);
         }

         break;
      case ir_var_auto:
         entry = new(mem_ctx) temp_entry(ir->var, PROGRAM_TEMPORARY,
                                         this->next_temp);
         this->variable_storage.push_tail(entry);

         next_temp += type_size(ir->var->type);
         break;
      }

      if (!entry) {
         printf("Failed to make storage for %s\n", ir->var->name);
         exit(1);
      }
   }

   src_reg.file = entry->file;
   src_reg.index = entry->index;
   /* If the type is smaller than a vec4, replicate the last channel out. */
   src_reg.swizzle = swizzle_for_size(ir->var->type->vector_elements);
   src_reg.reladdr = false;
   src_reg.negate = 0;

   this->result = src_reg;
}

void
ir_to_mesa_visitor::visit(ir_dereference_array *ir)
{
   ir_constant *index;
   ir_to_mesa_src_reg src_reg;

   index = ir->array_index->constant_expression_value();

   /* By the time we make it to this stage, matrices should be broken down
    * to vectors.
    */
   assert(!ir->type->is_matrix());

   ir->array->accept(this);
   src_reg = this->result;

   if (src_reg.file == PROGRAM_INPUT ||
       src_reg.file == PROGRAM_OUTPUT) {
      assert(index); /* FINISHME: Handle variable indexing of builtins. */

      src_reg.index += index->value.i[0];
   } else {
      if (index) {
         src_reg.index += index->value.i[0];
      } else {
         ir_to_mesa_src_reg array_base = this->result;
         /* Variable index array dereference.  It eats the "vec4" of the
          * base of the array and an index that offsets the Mesa register
          * index.
          */
         ir->array_index->accept(this);

         /* FINISHME: This doesn't work when we're trying to do the LHS
          * of an assignment.
          */
         src_reg.reladdr = true;
         ir_to_mesa_emit_op1(ir, OPCODE_ARL, ir_to_mesa_address_reg,
                             this->result);

         this->result = get_temp(ir->type);
         ir_to_mesa_emit_op1(ir, OPCODE_MOV,
                             ir_to_mesa_dst_reg_from_src(this->result),
                             src_reg);
      }
   }

   /* If the type is smaller than a vec4, replicate the last channel out. */
   src_reg.swizzle = swizzle_for_size(ir->type->vector_elements);

   this->result = src_reg;
}

void
ir_to_mesa_visitor::visit(ir_dereference_record *ir)
{
   unsigned int i;
   const glsl_type *struct_type = ir->record->type;
   int offset = 0;

   ir->record->accept(this);

   for (i = 0; i < struct_type->length; i++) {
      if (strcmp(struct_type->fields.structure[i].name, ir->field) == 0)
         break;
      offset += type_size(struct_type->fields.structure[i].type);
   }
   this->result.index += offset;
}

/**
 * We want to be careful in assignment setup to hit the actual storage
 * instead of potentially using a temporary like we might with the
 * ir_dereference handler.
 *
 * Thanks to ir_swizzle_swizzle, and ir_vec_index_to_swizzle, we
 * should only see potentially one variable array index of a vector,
 * and one swizzle, before getting to actual vec4 storage.  So handle
 * those, then go use ir_dereference to handle the rest.
 */
static struct ir_to_mesa_dst_reg
get_assignment_lhs(ir_instruction *ir, ir_to_mesa_visitor *v)
{
   struct ir_to_mesa_dst_reg dst_reg;
   ir_dereference *deref;
   ir_swizzle *swiz;

   /* Use the rvalue deref handler for the most part.  We'll ignore
    * swizzles in it and write swizzles using writemask, though.
    */
   ir->accept(v);
   dst_reg = ir_to_mesa_dst_reg_from_src(v->result);

   if ((deref = ir->as_dereference())) {
      ir_dereference_array *deref_array = ir->as_dereference_array();
      assert(!deref_array || deref_array->array->type->is_array());

      ir->accept(v);
   } else if ((swiz = ir->as_swizzle())) {
      dst_reg.writemask = 0;
      if (swiz->mask.num_components >= 1)
         dst_reg.writemask |= (1 << swiz->mask.x);
      if (swiz->mask.num_components >= 2)
         dst_reg.writemask |= (1 << swiz->mask.y);
      if (swiz->mask.num_components >= 3)
         dst_reg.writemask |= (1 << swiz->mask.z);
      if (swiz->mask.num_components >= 4)
         dst_reg.writemask |= (1 << swiz->mask.w);
   }

   return dst_reg;
}

static GLuint
reswizzle_for_writemask(GLuint writemask, GLuint swizzle)
{
   int new_swizzle[4], pos = 0;
   int i;

   /* reswizzle the rhs so the components are in place for the
    * components we'll assign to the lhs.
    */
   for (i = 0; i < 4; i++) {
      if (writemask & (1 << i)) {
         new_swizzle[i] = GET_SWZ(swizzle, pos++);
      } else {
         new_swizzle[i] = GET_SWZ(swizzle, 0);
      }
   }

   return MAKE_SWIZZLE4(new_swizzle[0],
                        new_swizzle[1],
                        new_swizzle[2],
                        new_swizzle[3]);
}

void
ir_to_mesa_visitor::visit(ir_assignment *ir)
{
   struct ir_to_mesa_dst_reg l;
   struct ir_to_mesa_src_reg r;

   assert(!ir->lhs->type->is_matrix());
   assert(!ir->lhs->type->is_array());
   assert(ir->lhs->type->base_type != GLSL_TYPE_STRUCT);

   l = get_assignment_lhs(ir->lhs, this);

   ir->rhs->accept(this);
   r = this->result;

   r.swizzle = reswizzle_for_writemask(l.writemask, r.swizzle);

   assert(l.file != PROGRAM_UNDEFINED);
   assert(r.file != PROGRAM_UNDEFINED);

   if (ir->condition) {
      ir_to_mesa_src_reg condition;

      ir->condition->accept(this);
      condition = this->result;

      /* We use the OPCODE_CMP (a < 0 ? b : c) for conditional moves,
       * and the condition we produced is 0.0 or 1.0.  By flipping the
       * sign, we can choose which value OPCODE_CMP produces without
       * an extra computing the condition.
       */
      condition.negate = ~condition.negate;
      ir_to_mesa_emit_op3(ir, OPCODE_CMP, l,
                          condition, r, ir_to_mesa_src_reg_from_dst(l));
   } else {
      ir_to_mesa_emit_op1(ir, OPCODE_MOV, l, r);
   }
}


void
ir_to_mesa_visitor::visit(ir_constant *ir)
{
   ir_to_mesa_src_reg src_reg;
   GLfloat stack_vals[4];
   GLfloat *values = stack_vals;
   unsigned int i;

   if (ir->type->is_matrix() || ir->type->is_array()) {
      assert(!"FINISHME: array/matrix constants");
   }

   src_reg.file = PROGRAM_CONSTANT;
   switch (ir->type->base_type) {
   case GLSL_TYPE_FLOAT:
      values = &ir->value.f[0];
      break;
   case GLSL_TYPE_UINT:
      for (i = 0; i < ir->type->vector_elements; i++) {
         values[i] = ir->value.u[i];
      }
      break;
   case GLSL_TYPE_INT:
      for (i = 0; i < ir->type->vector_elements; i++) {
         values[i] = ir->value.i[i];
      }
      break;
   case GLSL_TYPE_BOOL:
      for (i = 0; i < ir->type->vector_elements; i++) {
         values[i] = ir->value.b[i];
      }
      break;
   default:
      assert(!"Non-float/uint/int/bool constant");
   }

   src_reg.index = _mesa_add_unnamed_constant(this->prog->Parameters,
                                              values, ir->type->vector_elements,
                                              &src_reg.swizzle);
   src_reg.reladdr = false;
   src_reg.negate = 0;

   this->result = src_reg;
}


void
ir_to_mesa_visitor::visit(ir_call *ir)
{
   printf("Can't support call to %s\n", ir->callee_name());
   exit(1);
}


void
ir_to_mesa_visitor::visit(ir_texture *ir)
{
   ir_to_mesa_src_reg result_src, coord, lod_info, projector;
   ir_to_mesa_dst_reg result_dst, coord_dst;
   ir_to_mesa_instruction *inst = NULL;
   prog_opcode opcode = OPCODE_NOP;

   ir->coordinate->accept(this);

   /* Put our coords in a temp.  We'll need to modify them for shadow,
    * projection, or LOD, so the only case we'd use it as is is if
    * we're doing plain old texturing.  Mesa IR optimization should
    * handle cleaning up our mess in that case.
    */
   coord = get_temp(glsl_type::vec4_type);
   coord_dst = ir_to_mesa_dst_reg_from_src(coord);
   ir_to_mesa_emit_op1(ir, OPCODE_MOV, coord_dst,
                       this->result);

   if (ir->projector) {
      ir->projector->accept(this);
      projector = this->result;
   }

   /* Storage for our result.  Ideally for an assignment we'd be using
    * the actual storage for the result here, instead.
    */
   result_src = get_temp(glsl_type::vec4_type);
   result_dst = ir_to_mesa_dst_reg_from_src(result_src);

   switch (ir->op) {
   case ir_tex:
      opcode = OPCODE_TEX;
      break;
   case ir_txb:
      opcode = OPCODE_TXB;
      ir->lod_info.bias->accept(this);
      lod_info = this->result;
      break;
   case ir_txl:
      opcode = OPCODE_TXL;
      ir->lod_info.lod->accept(this);
      lod_info = this->result;
      break;
   case ir_txd:
   case ir_txf:
      assert(!"GLSL 1.30 features unsupported");
      break;
   }

   if (ir->projector) {
      if (opcode == OPCODE_TEX) {
         /* Slot the projector in as the last component of the coord. */
         coord_dst.writemask = WRITEMASK_W;
         ir_to_mesa_emit_op1(ir, OPCODE_MOV, coord_dst, projector);
         coord_dst.writemask = WRITEMASK_XYZW;
         opcode = OPCODE_TXP;
      } else {
         ir_to_mesa_src_reg coord_w = coord;
         coord_w.swizzle = SWIZZLE_WWWW;

         /* For the other TEX opcodes there's no projective version
          * since the last slot is taken up by lod info.  Do the
          * projective divide now.
          */
         coord_dst.writemask = WRITEMASK_W;
         ir_to_mesa_emit_op1(ir, OPCODE_RCP, coord_dst, projector);

         coord_dst.writemask = WRITEMASK_XYZ;
         ir_to_mesa_emit_op2(ir, OPCODE_MUL, coord_dst, coord, coord_w);

         coord_dst.writemask = WRITEMASK_XYZW;
         coord.swizzle = SWIZZLE_XYZW;
      }
   }

   if (ir->shadow_comparitor) {
      /* Slot the shadow value in as the second to last component of the
       * coord.
       */
      ir->shadow_comparitor->accept(this);
      coord_dst.writemask = WRITEMASK_Z;
      ir_to_mesa_emit_op1(ir, OPCODE_MOV, coord_dst, this->result);
      coord_dst.writemask = WRITEMASK_XYZW;
   }

   if (opcode == OPCODE_TXL || opcode == OPCODE_TXB) {
      /* Mesa IR stores lod or lod bias in the last channel of the coords. */
      coord_dst.writemask = WRITEMASK_W;
      ir_to_mesa_emit_op1(ir, OPCODE_MOV, coord_dst, lod_info);
      coord_dst.writemask = WRITEMASK_XYZW;
   }

   inst = ir_to_mesa_emit_op1(ir, opcode, result_dst, coord);

   if (ir->shadow_comparitor)
      inst->tex_shadow = GL_TRUE;

   ir_dereference_variable *sampler = ir->sampler->as_dereference_variable();
   assert(sampler); /* FINISHME: sampler arrays */
   /* generate the mapping, remove when we generate storage at
    * declaration time
    */
   sampler->accept(this);

   inst->sampler = get_sampler_number(sampler->var->location);

   switch (sampler->type->sampler_dimensionality) {
   case GLSL_SAMPLER_DIM_1D:
      inst->tex_target = TEXTURE_1D_INDEX;
      break;
   case GLSL_SAMPLER_DIM_2D:
      inst->tex_target = TEXTURE_2D_INDEX;
      break;
   case GLSL_SAMPLER_DIM_3D:
      inst->tex_target = TEXTURE_3D_INDEX;
      break;
   case GLSL_SAMPLER_DIM_CUBE:
      inst->tex_target = TEXTURE_CUBE_INDEX;
      break;
   default:
      assert(!"FINISHME: other texture targets");
   }

   this->result = result_src;
}

void
ir_to_mesa_visitor::visit(ir_return *ir)
{
   assert(0);

   ir->get_value()->accept(this);
}

void
ir_to_mesa_visitor::visit(ir_discard *ir)
{
   assert(ir->condition == NULL); /* FINISHME */

   ir_to_mesa_emit_op1(ir, OPCODE_KIL_NV,
                       ir_to_mesa_undef_dst, ir_to_mesa_undef);
}

void
ir_to_mesa_visitor::visit(ir_if *ir)
{
   ir_to_mesa_instruction *cond_inst, *if_inst, *else_inst = NULL;
   ir_to_mesa_instruction *prev_inst;

   prev_inst = (ir_to_mesa_instruction *)this->instructions.get_tail();

   ir->condition->accept(this);
   assert(this->result.file != PROGRAM_UNDEFINED);

   if (ctx->Shader.EmitCondCodes) {
      cond_inst = (ir_to_mesa_instruction *)this->instructions.get_tail();

      /* See if we actually generated any instruction for generating
       * the condition.  If not, then cook up a move to a temp so we
       * have something to set cond_update on.
       */
      if (cond_inst == prev_inst) {
         ir_to_mesa_src_reg temp = get_temp(glsl_type::bool_type);
         cond_inst = ir_to_mesa_emit_op1(ir->condition, OPCODE_MOV,
                                         ir_to_mesa_dst_reg_from_src(temp),
                                         result);
      }
      cond_inst->cond_update = GL_TRUE;

      if_inst = ir_to_mesa_emit_op1(ir->condition,
                                    OPCODE_IF, ir_to_mesa_undef_dst,
                                    ir_to_mesa_undef);
      if_inst->dst_reg.cond_mask = COND_NE;
   } else {
      if_inst = ir_to_mesa_emit_op1(ir->condition,
                                    OPCODE_IF, ir_to_mesa_undef_dst,
                                    this->result);
   }

   this->instructions.push_tail(if_inst);

   visit_exec_list(&ir->then_instructions, this);

   if (!ir->else_instructions.is_empty()) {
      else_inst = ir_to_mesa_emit_op1(ir->condition, OPCODE_ELSE,
                                      ir_to_mesa_undef_dst,
                                      ir_to_mesa_undef);
      visit_exec_list(&ir->else_instructions, this);
   }

   if_inst = ir_to_mesa_emit_op1(ir->condition, OPCODE_ENDIF,
                                 ir_to_mesa_undef_dst, ir_to_mesa_undef);
}

ir_to_mesa_visitor::ir_to_mesa_visitor()
{
   result.file = PROGRAM_UNDEFINED;
   next_temp = 1;
   sampler_map = NULL;
   sampler_map_size = 0;
}

static struct prog_src_register
mesa_src_reg_from_ir_src_reg(ir_to_mesa_src_reg reg)
{
   struct prog_src_register mesa_reg;

   mesa_reg.File = reg.file;
   assert(reg.index < (1 << INST_INDEX_BITS) - 1);
   mesa_reg.Index = reg.index;
   mesa_reg.Swizzle = reg.swizzle;
   mesa_reg.RelAddr = reg.reladdr;
   mesa_reg.Negate = reg.negate;
   mesa_reg.Abs = 0;

   return mesa_reg;
}

static void
set_branchtargets(struct prog_instruction *mesa_instructions,
                  int num_instructions)
{
   int if_count = 0, loop_count = 0;
   int *if_stack, *loop_stack;
   int if_stack_pos = 0, loop_stack_pos = 0;
   int i, j;

   for (i = 0; i < num_instructions; i++) {
      switch (mesa_instructions[i].Opcode) {
      case OPCODE_IF:
         if_count++;
         break;
      case OPCODE_BGNLOOP:
         loop_count++;
         break;
      case OPCODE_BRK:
      case OPCODE_CONT:
         mesa_instructions[i].BranchTarget = -1;
         break;
      default:
         break;
      }
   }

   if_stack = (int *)calloc(if_count, sizeof(*if_stack));
   loop_stack = (int *)calloc(loop_count, sizeof(*loop_stack));

   for (i = 0; i < num_instructions; i++) {
      switch (mesa_instructions[i].Opcode) {
      case OPCODE_IF:
         if_stack[if_stack_pos] = i;
         if_stack_pos++;
         break;
      case OPCODE_ELSE:
         mesa_instructions[if_stack[if_stack_pos - 1]].BranchTarget = i;
         if_stack[if_stack_pos - 1] = i;
         break;
      case OPCODE_ENDIF:
         mesa_instructions[if_stack[if_stack_pos - 1]].BranchTarget = i;
         if_stack_pos--;
         break;
      case OPCODE_BGNLOOP:
         loop_stack[loop_stack_pos] = i;
         loop_stack_pos++;
         break;
      case OPCODE_ENDLOOP:
         loop_stack_pos--;
         /* Rewrite any breaks/conts at this nesting level (haven't
          * already had a BranchTarget assigned) to point to the end
          * of the loop.
          */
         for (j = loop_stack[loop_stack_pos]; j < i; j++) {
            if (mesa_instructions[j].Opcode == OPCODE_BRK ||
                mesa_instructions[j].Opcode == OPCODE_CONT) {
               if (mesa_instructions[j].BranchTarget == -1) {
                  mesa_instructions[j].BranchTarget = i;
               }
            }
         }
         /* The loop ends point at each other. */
         mesa_instructions[i].BranchTarget = loop_stack[loop_stack_pos];
         mesa_instructions[loop_stack[loop_stack_pos]].BranchTarget = i;
      default:
         break;
      }
   }

   free(if_stack);
}

static void
print_program(struct prog_instruction *mesa_instructions,
              ir_instruction **mesa_instruction_annotation,
              int num_instructions)
{
   ir_instruction *last_ir = NULL;
   int i;

   for (i = 0; i < num_instructions; i++) {
      struct prog_instruction *mesa_inst = mesa_instructions + i;
      ir_instruction *ir = mesa_instruction_annotation[i];

      if (last_ir != ir && ir) {
         ir_print_visitor print;
         ir->accept(&print);
         printf("\n");
         last_ir = ir;
      }

      _mesa_print_instruction(mesa_inst);
   }
}

static void
count_resources(struct gl_program *prog)
{
   unsigned int i;

   prog->InputsRead = 0;
   prog->OutputsWritten = 0;
   prog->SamplersUsed = 0;

   for (i = 0; i < prog->NumInstructions; i++) {
      struct prog_instruction *inst = &prog->Instructions[i];
      unsigned int reg;

      switch (inst->DstReg.File) {
      case PROGRAM_OUTPUT:
         prog->OutputsWritten |= BITFIELD64_BIT(inst->DstReg.Index);
         break;
      case PROGRAM_INPUT:
         prog->InputsRead |= BITFIELD64_BIT(inst->DstReg.Index);
         break;
      default:
         break;
      }

      for (reg = 0; reg < _mesa_num_inst_src_regs(inst->Opcode); reg++) {
         switch (inst->SrcReg[reg].File) {
         case PROGRAM_OUTPUT:
            prog->OutputsWritten |= BITFIELD64_BIT(inst->SrcReg[reg].Index);
            break;
         case PROGRAM_INPUT:
            prog->InputsRead |= BITFIELD64_BIT(inst->SrcReg[reg].Index);
            break;
         default:
            break;
         }
      }

      /* Instead of just using the uniform's value to map to a
       * sampler, Mesa first allocates a separate number for the
       * sampler (_mesa_add_sampler), then we reindex it down to a
       * small integer (sampler_map[], SamplersUsed), then that gets
       * mapped to the uniform's value, and we get an actual sampler.
       */
      if (_mesa_is_tex_instruction(inst->Opcode)) {
         prog->SamplerTargets[inst->TexSrcUnit] =
            (gl_texture_index)inst->TexSrcTarget;
         prog->SamplersUsed |= 1 << inst->TexSrcUnit;
         if (inst->TexShadow) {
            prog->ShadowSamplers |= 1 << inst->TexSrcUnit;
         }
      }
   }

   _mesa_update_shader_textures_used(prog);
}

/* Each stage has some uniforms in its Parameters list.  The Uniforms
 * list for the linked shader program has a pointer to these uniforms
 * in each of the stage's Parameters list, so that their values can be
 * updated when a uniform is set.
 */
static void
link_uniforms_to_shared_uniform_list(struct gl_uniform_list *uniforms,
                                     struct gl_program *prog)
{
   unsigned int i;

   for (i = 0; i < prog->Parameters->NumParameters; i++) {
      const struct gl_program_parameter *p = prog->Parameters->Parameters + i;

      if (p->Type == PROGRAM_UNIFORM || p->Type == PROGRAM_SAMPLER) {
         struct gl_uniform *uniform =
            _mesa_append_uniform(uniforms, p->Name, prog->Target, i);
         if (uniform)
            uniform->Initialized = p->Initialized;
      }
   }
}

struct gl_program *
get_mesa_program(GLcontext *ctx, void *mem_ctx, struct gl_shader *shader)
{
   ir_to_mesa_visitor v;
   struct prog_instruction *mesa_instructions, *mesa_inst;
   ir_instruction **mesa_instruction_annotation;
   int i;
   struct gl_program *prog;
   GLenum target;

   switch (shader->Type) {
   case GL_VERTEX_SHADER:   target = GL_VERTEX_PROGRAM_ARB; break;
   case GL_FRAGMENT_SHADER: target = GL_FRAGMENT_PROGRAM_ARB; break;
   default: assert(!"should not be reached"); break;
   }

   prog = ctx->Driver.NewProgram(ctx, target, 1);
   if (!prog)
      return NULL;
   prog->Parameters = _mesa_new_parameter_list();
   prog->Varying = _mesa_new_parameter_list();
   prog->Attributes = _mesa_new_parameter_list();
   v.ctx = ctx;
   v.prog = prog;

   v.mem_ctx = talloc_new(NULL);
   visit_exec_list(shader->ir, &v);
   v.ir_to_mesa_emit_op1(NULL, OPCODE_END,
                         ir_to_mesa_undef_dst, ir_to_mesa_undef);

   prog->NumTemporaries = v.next_temp;

   int num_instructions = 0;
   foreach_iter(exec_list_iterator, iter, v.instructions) {
      num_instructions++;
   }

   mesa_instructions =
      (struct prog_instruction *)calloc(num_instructions,
                                        sizeof(*mesa_instructions));
   mesa_instruction_annotation = talloc_array(mem_ctx, ir_instruction *,
                                              num_instructions);

   mesa_inst = mesa_instructions;
   i = 0;
   foreach_iter(exec_list_iterator, iter, v.instructions) {
      ir_to_mesa_instruction *inst = (ir_to_mesa_instruction *)iter.get();

      mesa_inst->Opcode = inst->op;
      mesa_inst->CondUpdate = inst->cond_update;
      mesa_inst->DstReg.File = inst->dst_reg.file;
      mesa_inst->DstReg.Index = inst->dst_reg.index;
      mesa_inst->DstReg.CondMask = inst->dst_reg.cond_mask;
      mesa_inst->DstReg.WriteMask = inst->dst_reg.writemask;
      mesa_inst->SrcReg[0] = mesa_src_reg_from_ir_src_reg(inst->src_reg[0]);
      mesa_inst->SrcReg[1] = mesa_src_reg_from_ir_src_reg(inst->src_reg[1]);
      mesa_inst->SrcReg[2] = mesa_src_reg_from_ir_src_reg(inst->src_reg[2]);
      mesa_inst->TexSrcUnit = inst->sampler;
      mesa_inst->TexSrcTarget = inst->tex_target;
      mesa_inst->TexShadow = inst->tex_shadow;
      mesa_instruction_annotation[i] = inst->ir;

      mesa_inst++;
      i++;
   }

   set_branchtargets(mesa_instructions, num_instructions);
   if (0) {
      print_program(mesa_instructions, mesa_instruction_annotation,
                    num_instructions);
   }

   prog->Instructions = mesa_instructions;
   prog->NumInstructions = num_instructions;

   _mesa_reference_program(ctx, &shader->Program, prog);

   if ((ctx->Shader.Flags & GLSL_NO_OPT) == 0) {
      _mesa_optimize_program(ctx, prog);
   }

   return prog;
}

extern "C" {

static void
steal_memory(ir_instruction *ir, void *new_ctx)
{
   talloc_steal(new_ctx, ir);
}

void
_mesa_glsl_compile_shader(GLcontext *ctx, struct gl_shader *shader)
{
   struct _mesa_glsl_parse_state *state;

   state = talloc_zero(shader, struct _mesa_glsl_parse_state);
   switch (shader->Type) {
   case GL_VERTEX_SHADER:   state->target = vertex_shader; break;
   case GL_FRAGMENT_SHADER: state->target = fragment_shader; break;
   case GL_GEOMETRY_SHADER: state->target = geometry_shader; break;
   }

   state->scanner = NULL;
   state->translation_unit.make_empty();
   state->symbols = new(shader) glsl_symbol_table;
   state->info_log = talloc_strdup(shader, "");
   state->error = false;
   state->temp_index = 0;
   state->loop_or_switch_nesting = NULL;
   state->ARB_texture_rectangle_enable = true;

   state->extensions = &ctx->Extensions;
   state->Const.MaxDrawBuffers = ctx->Const.MaxDrawBuffers;
   state->Const.MaxTextureCoords = ctx->Const.MaxTextureCoordUnits;

   const char *source = shader->Source;
   state->error = preprocess(state, &source, &state->info_log,
                             &ctx->Extensions);

   if (!state->error) {
     _mesa_glsl_lexer_ctor(state, source);
     _mesa_glsl_parse(state);
     _mesa_glsl_lexer_dtor(state);
   }

   shader->ir = new(shader) exec_list;
   if (!state->error && !state->translation_unit.is_empty())
      _mesa_ast_to_hir(shader->ir, state);

   /* Lowering */
   do_mod_to_fract(shader->ir);
   do_div_to_mul_rcp(shader->ir);

   /* Optimization passes */
   if (!state->error && !shader->ir->is_empty()) {
      bool progress;
      do {
         progress = false;

         progress = do_function_inlining(shader->ir) || progress;
         progress = do_if_simplification(shader->ir) || progress;
         progress = do_copy_propagation(shader->ir) || progress;
         progress = do_dead_code_local(shader->ir) || progress;
         progress = do_dead_code_unlinked(state, shader->ir) || progress;
         progress = do_constant_variable_unlinked(shader->ir) || progress;
         progress = do_constant_folding(shader->ir) || progress;

         progress = do_vec_index_to_swizzle(shader->ir) || progress;
         /* Do this one after the previous to let the easier pass handle
          * constant vector indexing.
          */
         progress = do_vec_index_to_cond_assign(shader->ir) || progress;

         progress = do_swizzle_swizzle(shader->ir) || progress;
      } while (progress);
   }

   shader->symbols = state->symbols;

   shader->CompileStatus = !state->error;
   shader->InfoLog = state->info_log;

   /* Retain any live IR, but trash the rest. */
   foreach_list(node, shader->ir) {
      visit_tree((ir_instruction *) node, steal_memory, shader);
   }

   talloc_free(state);
 }

void
_mesa_glsl_link_shader(GLcontext *ctx, struct gl_shader_program *prog)
{
   unsigned int i;

   _mesa_clear_shader_program_data(ctx, prog);

   prog->LinkStatus = GL_TRUE;

   for (i = 0; i < prog->NumShaders; i++) {
      if (!prog->Shaders[i]->CompileStatus) {
         prog->InfoLog =
            talloc_asprintf_append(prog->InfoLog,
                                   "linking with uncompiled shader");
         prog->LinkStatus = GL_FALSE;
      }
   }

   prog->Varying = _mesa_new_parameter_list();
   _mesa_reference_vertprog(ctx, &prog->VertexProgram, NULL);
   _mesa_reference_fragprog(ctx, &prog->FragmentProgram, NULL);

   if (prog->LinkStatus) {
      link_shaders(prog);

      /* We don't use the linker's uniforms list, and cook up our own at
       * generate time.
       */
      free(prog->Uniforms);
      prog->Uniforms = _mesa_new_uniform_list();
   }

   prog->LinkStatus = prog->LinkStatus;

   /* FINISHME: This should use the linker-generated code */
   if (prog->LinkStatus) {
      for (i = 0; i < prog->NumShaders; i++) {
         struct gl_program *linked_prog;

         linked_prog = get_mesa_program(ctx, prog,
                                        prog->Shaders[i]);
         count_resources(linked_prog);

         link_uniforms_to_shared_uniform_list(prog->Uniforms, linked_prog);

         switch (prog->Shaders[i]->Type) {
         case GL_VERTEX_SHADER:
            _mesa_reference_vertprog(ctx, &prog->VertexProgram,
                                     (struct gl_vertex_program *)linked_prog);
            ctx->Driver.ProgramStringNotify(ctx, GL_VERTEX_PROGRAM_ARB,
                                            linked_prog);
            break;
         case GL_FRAGMENT_SHADER:
            _mesa_reference_fragprog(ctx, &prog->FragmentProgram,
                                     (struct gl_fragment_program *)linked_prog);
            ctx->Driver.ProgramStringNotify(ctx, GL_FRAGMENT_PROGRAM_ARB,
                                            linked_prog);
            break;
         }
      }
   }
}

} /* extern "C" */