ArbInst.hpp

// Sh: A GPU metaprogramming language.
//
// Copyright 2003-2005 Serious Hack Inc.
// 
// This library is free software; you can redistribute it and/or
// modify it under the terms of the GNU Lesser General Public
// License as published by the Free Software Foundation; either
// version 2.1 of the License, or (at your option) any later version.
//
// This library is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
// Lesser General Public License for more details.
//
// You should have received a copy of the GNU Lesser General Public
// License along with this library; if not, write to the Free Software
// Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, 
// MA  02110-1301, USA
#ifndef ARBINST_HPP
#define ARBINST_HPP

#include "ShVariable.hpp"

namespace shgl {

enum ArbOp {
  // VERTEX AND FRAGMENT
  
  // Vector
  SH_ARB_ABS,
  SH_ARB_FLR,
  SH_ARB_FRC,
  SH_ARB_LIT,
  SH_ARB_MOV,

  // Scalar
  SH_ARB_EX2,
  SH_ARB_LG2,
  SH_ARB_RCP,
  SH_ARB_RSQ,

  // Binary scalar
  SH_ARB_POW,

  // Binary vector
  SH_ARB_ADD,
  SH_ARB_DP3,
  SH_ARB_DP4,
  SH_ARB_DPH,
  SH_ARB_DST,
  SH_ARB_MAX,
  SH_ARB_MIN,
  SH_ARB_MUL,
  SH_ARB_SGE,
  SH_ARB_SLT,
  SH_ARB_SUB,
  SH_ARB_XPD,

  // Trinary
  SH_ARB_MAD,

  // Swizzling
  SH_ARB_SWZ,

  // VERTEX ONLY
  // Scalar
  SH_ARB_EXP,
  SH_ARB_LOG,
  
  // Weird,
  SH_ARB_ARL,
  SH_ARB_ARRAYMOV, // Kind of hacky but works.
  
  // FRAGMENT ONLY
  // Scalar
  SH_ARB_COS,
  SH_ARB_SIN,
  SH_ARB_SCS,

  // Trinary
  SH_ARB_CMP,
  SH_ARB_LRP,

  // Sampling
  SH_ARB_TEX,
  SH_ARB_TXP,
  SH_ARB_TXB,

  // KIL
  SH_ARB_KIL,

  // NV_vertex_program2/NV_fragment_program
  SH_ARB_SEQ,
  SH_ARB_SGT,
  SH_ARB_SLE,
  SH_ARB_SNE,
  SH_ARB_SFL,
  SH_ARB_STR,

  // NV_fragment_program
  SH_ARB_DDX,
  SH_ARB_DDY,
  SH_ARB_RFL,
  SH_ARB_TXD,

  // NV_vertex_program2
  SH_ARB_SSG,
  SH_ARB_BRA,
  SH_ARB_LABEL, // Special label "instruction"
  
  // NV_fragment_program2
  SH_ARB_DIV,
  SH_ARB_DP2,
  SH_ARB_NRM,
  SH_ARB_IF,
  SH_ARB_ELSE,
  SH_ARB_ENDIF,
  SH_ARB_REP,
  SH_ARB_ENDREP,
  SH_ARB_BRK,
  SH_ARB_RET,
  
  // Special "operations" for emit
  SH_ARB_FUN,

  // Just comments
  SH_ARB_COMMENT,
};

struct ArbOpInfo {
  char* name;
  int arity;
  bool collectingOp;
};

extern ArbOpInfo arbOpInfo[];

struct ArbInst {
  static const int max_num_sources = 4;
  
  explicit ArbInst(ArbOp op)
    : op(op), invert(false), update_cc(false), ccode(NOCC)
  {
  }

  ArbInst(ArbOp op, int label)
    : op(op), label(label), invert(false),
      update_cc(false), ccode(NOCC)
  {
  }

  ArbInst(ArbOp op, const std::string& comment)
    : op(op), invert(false), update_cc(false), ccode(NOCC),
    comment(comment)
  {
  }

  ArbInst(ArbOp op, int label, const SH::ShVariable& condition)
    : op(op), label(label), invert(false),
      update_cc(false), ccode(NOCC)
  {
    src[0] = condition;
  }
  
  ArbInst(ArbOp op, const SH::ShVariable& dest)
    : op(op), dest(dest), invert(false),
      update_cc(false), ccode(NOCC)
  {
  }

  ArbInst(ArbOp op, const SH::ShVariable& dest, const SH::ShVariable& src0)
    : op(op), dest(dest), invert(false),
      update_cc(false), ccode(NOCC)
  {
    src[0] = src0;
  }

  ArbInst(ArbOp op, const SH::ShVariable& dest, const SH::ShVariable& src0,
          const SH::ShVariable& src1)
    : op(op), dest(dest), invert(false),
      update_cc(false), ccode(NOCC)
  {
    src[0] = src0;
    src[1] = src1;
  }
  ArbInst(ArbOp op, const SH::ShVariable& dest, const SH::ShVariable& src0,
          const SH::ShVariable& src1, const SH::ShVariable& src2)
    : op(op), dest(dest), invert(false),
      update_cc(false), ccode(NOCC)
  {
    src[0] = src0;
    src[1] = src1;
    src[2] = src2;
  }
  ArbInst(ArbOp op, const SH::ShVariable& dest, const SH::ShVariable& src0,
          const SH::ShVariable& src1, const SH::ShVariable& src2,
          const SH::ShVariable& src3)
    : op(op), dest(dest), invert(false),
      update_cc(false), ccode(NOCC)
  {
    src[0] = src0;
    src[1] = src1;
    src[2] = src2;
    src[3] = src3;
  }
  
  ArbOp op;
  SH::ShVariable dest;

  SH::ShVariable src[max_num_sources];

  int label; // For branching instructions and labels
  bool invert; // Invert the sense of a break condition.

  // NV specific stuff for condition codes
  bool update_cc; 
  
  enum CCode {
    NOCC,
    EQ,
    GE,
    GT,
    LE,
    LT,
    NE,
    TR,
    FL
  } ccode; 

  SH::ShSwizzle ccswiz;
  
  // Only used for COMMENTs
  std::string comment;
};

extern char* arbCCnames[];

}

#endif

---