Home | History | Annotate | Download | only in tgsi
      1 /**************************************************************************
      2  *
      3  * Copyright 2008 Tungsten Graphics, Inc., Cedar Park, Texas.
      4  * All Rights Reserved.
      5  *
      6  * Permission is hereby granted, free of charge, to any person obtaining a
      7  * copy of this software and associated documentation files (the
      8  * "Software"), to deal in the Software without restriction, including
      9  * without limitation the rights to use, copy, modify, merge, publish,
     10  * distribute, sub license, and/or sell copies of the Software, and to
     11  * permit persons to whom the Software is furnished to do so, subject to
     12  * the following conditions:
     13  *
     14  * The above copyright notice and this permission notice (including the
     15  * next paragraph) shall be included in all copies or substantial portions
     16  * of the Software.
     17  *
     18  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
     19  * OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
     20  * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NON-INFRINGEMENT.
     21  * IN NO EVENT SHALL TUNGSTEN GRAPHICS AND/OR ITS SUPPLIERS BE LIABLE FOR
     22  * ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
     23  * TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
     24  * SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
     25  *
     26  **************************************************************************/
     27 
     28 /**
     29  * TGSI program transformation utility.
     30  *
     31  * Authors:  Brian Paul
     32  */
     33 
     34 #include "util/u_debug.h"
     35 
     36 #include "tgsi_transform.h"
     37 
     38 
     39 
     40 static void
     41 emit_instruction(struct tgsi_transform_context *ctx,
     42                  const struct tgsi_full_instruction *inst)
     43 {
     44    uint ti = ctx->ti;
     45 
     46    ti += tgsi_build_full_instruction(inst,
     47                                      ctx->tokens_out + ti,
     48                                      ctx->header,
     49                                      ctx->max_tokens_out - ti);
     50    ctx->ti = ti;
     51 }
     52 
     53 
     54 static void
     55 emit_declaration(struct tgsi_transform_context *ctx,
     56                  const struct tgsi_full_declaration *decl)
     57 {
     58    uint ti = ctx->ti;
     59 
     60    ti += tgsi_build_full_declaration(decl,
     61                                      ctx->tokens_out + ti,
     62                                      ctx->header,
     63                                      ctx->max_tokens_out - ti);
     64    ctx->ti = ti;
     65 }
     66 
     67 
     68 static void
     69 emit_immediate(struct tgsi_transform_context *ctx,
     70                const struct tgsi_full_immediate *imm)
     71 {
     72    uint ti = ctx->ti;
     73 
     74    ti += tgsi_build_full_immediate(imm,
     75                                    ctx->tokens_out + ti,
     76                                    ctx->header,
     77                                    ctx->max_tokens_out - ti);
     78    ctx->ti = ti;
     79 }
     80 
     81 
     82 static void
     83 emit_property(struct tgsi_transform_context *ctx,
     84               const struct tgsi_full_property *prop)
     85 {
     86    uint ti = ctx->ti;
     87 
     88    ti += tgsi_build_full_property(prop,
     89                                   ctx->tokens_out + ti,
     90                                   ctx->header,
     91                                   ctx->max_tokens_out - ti);
     92    ctx->ti = ti;
     93 }
     94 
     95 
     96 /**
     97  * Apply user-defined transformations to the input shader to produce
     98  * the output shader.
     99  * For example, a register search-and-replace operation could be applied
    100  * by defining a transform_instruction() callback that examined and changed
    101  * the instruction src/dest regs.
    102  *
    103  * \return number of tokens emitted
    104  */
    105 int
    106 tgsi_transform_shader(const struct tgsi_token *tokens_in,
    107                       struct tgsi_token *tokens_out,
    108                       uint max_tokens_out,
    109                       struct tgsi_transform_context *ctx)
    110 {
    111    uint procType;
    112 
    113    /* input shader */
    114    struct tgsi_parse_context parse;
    115 
    116    /* output shader */
    117    struct tgsi_processor *processor;
    118 
    119 
    120    /**
    121     ** callback context init
    122     **/
    123    ctx->emit_instruction = emit_instruction;
    124    ctx->emit_declaration = emit_declaration;
    125    ctx->emit_immediate = emit_immediate;
    126    ctx->emit_property = emit_property;
    127    ctx->tokens_out = tokens_out;
    128    ctx->max_tokens_out = max_tokens_out;
    129 
    130 
    131    /**
    132     ** Setup to begin parsing input shader
    133     **/
    134    if (tgsi_parse_init( &parse, tokens_in ) != TGSI_PARSE_OK) {
    135       debug_printf("tgsi_parse_init() failed in tgsi_transform_shader()!\n");
    136       return -1;
    137    }
    138    procType = parse.FullHeader.Processor.Processor;
    139    assert(procType == TGSI_PROCESSOR_FRAGMENT ||
    140           procType == TGSI_PROCESSOR_VERTEX ||
    141           procType == TGSI_PROCESSOR_GEOMETRY);
    142 
    143 
    144    /**
    145     **  Setup output shader
    146     **/
    147    ctx->header = (struct tgsi_header *)tokens_out;
    148    *ctx->header = tgsi_build_header();
    149 
    150    processor = (struct tgsi_processor *) (tokens_out + 1);
    151    *processor = tgsi_build_processor( procType, ctx->header );
    152 
    153    ctx->ti = 2;
    154 
    155 
    156    /**
    157     ** Loop over incoming program tokens/instructions
    158     */
    159    while( !tgsi_parse_end_of_tokens( &parse ) ) {
    160 
    161       tgsi_parse_token( &parse );
    162 
    163       switch( parse.FullToken.Token.Type ) {
    164       case TGSI_TOKEN_TYPE_INSTRUCTION:
    165          {
    166             struct tgsi_full_instruction *fullinst
    167                = &parse.FullToken.FullInstruction;
    168 
    169             if (ctx->transform_instruction)
    170                ctx->transform_instruction(ctx, fullinst);
    171             else
    172                ctx->emit_instruction(ctx, fullinst);
    173          }
    174          break;
    175 
    176       case TGSI_TOKEN_TYPE_DECLARATION:
    177          {
    178             struct tgsi_full_declaration *fulldecl
    179                = &parse.FullToken.FullDeclaration;
    180 
    181             if (ctx->transform_declaration)
    182                ctx->transform_declaration(ctx, fulldecl);
    183             else
    184                ctx->emit_declaration(ctx, fulldecl);
    185          }
    186          break;
    187 
    188       case TGSI_TOKEN_TYPE_IMMEDIATE:
    189          {
    190             struct tgsi_full_immediate *fullimm
    191                = &parse.FullToken.FullImmediate;
    192 
    193             if (ctx->transform_immediate)
    194                ctx->transform_immediate(ctx, fullimm);
    195             else
    196                ctx->emit_immediate(ctx, fullimm);
    197          }
    198          break;
    199       case TGSI_TOKEN_TYPE_PROPERTY:
    200          {
    201             struct tgsi_full_property *fullprop
    202                = &parse.FullToken.FullProperty;
    203 
    204             if (ctx->transform_property)
    205                ctx->transform_property(ctx, fullprop);
    206             else
    207                ctx->emit_property(ctx, fullprop);
    208          }
    209          break;
    210 
    211       default:
    212          assert( 0 );
    213       }
    214    }
    215 
    216    if (ctx->epilog) {
    217       ctx->epilog(ctx);
    218    }
    219 
    220    tgsi_parse_free (&parse);
    221 
    222    return ctx->ti;
    223 }
    224 
    225 
    226 #include "tgsi_text.h"
    227 
    228 extern int tgsi_transform_foo( struct tgsi_token *tokens_out,
    229                                uint max_tokens_out );
    230 
    231 /* This function exists only so that tgsi_text_translate() doesn't get
    232  * magic-ed out of the libtgsi.a archive by the build system.  Don't
    233  * remove unless you know this has been fixed - check on mingw/scons
    234  * builds as well.
    235  */
    236 int
    237 tgsi_transform_foo( struct tgsi_token *tokens_out,
    238                     uint max_tokens_out )
    239 {
    240    const char *text =
    241       "FRAG\n"
    242       "DCL IN[0], COLOR, CONSTANT\n"
    243       "DCL OUT[0], COLOR\n"
    244       "  0: MOV OUT[0], IN[0]\n"
    245       "  1: END";
    246 
    247    return tgsi_text_translate( text,
    248                                tokens_out,
    249                                max_tokens_out );
    250 }
    251