src/compiler/nir/nir_search.h

   1 /*
   2  * Copyright © 2014 Intel Corporation
   3  *
   4  * Permission is hereby granted, free of charge, to any person obtaining a
   5  * copy of this software and associated documentation files (the "Software"),
   6  * to deal in the Software without restriction, including without limitation
   7  * the rights to use, copy, modify, merge, publish, distribute, sublicense,
   8  * and/or sell copies of the Software, and to permit persons to whom the
   9  * Software is furnished to do so, subject to the following conditions:
  10  *
  11  * The above copyright notice and this permission notice (including the next
  12  * paragraph) shall be included in all copies or substantial portions of the
  13  * Software.
  14  *
  15  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  16  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  17  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL
  18  * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  19  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
  20  * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
  21  * IN THE SOFTWARE.
  22  *
  23  * Authors:
  24  *    Jason Ekstrand (jason@jlekstrand.net)
  25  *
  26  */
  27
  28 #ifndef _NIR_SEARCH_
  29 #define _NIR_SEARCH_
  30
  31 #include "nir.h"
  32
  33 #define NIR_SEARCH_MAX_VARIABLES 16
  34
  35 typedef enum {
  36    nir_search_value_expression,
  37    nir_search_value_variable,
  38    nir_search_value_constant,
  39 } nir_search_value_type;
  40
  41 typedef struct {
  42    nir_search_value_type type;
  43
  44    unsigned bit_size;
  45 } nir_search_value;
  46
  47 typedef struct {
  48    nir_search_value value;
  49
  50    /** The variable index;  Must be less than NIR_SEARCH_MAX_VARIABLES */
  51    unsigned variable;
  52
  53    /** Indicates that the given variable must be a constant
  54     *
  55     * This is only alloed in search expressions and indicates that the
  56     * given variable is only allowed to match constant values.
  57     */
  58    bool is_constant;
  59
  60    /** Indicates that the given variable must have a certain type
  61     *
  62     * This is only allowed in search expressions and indicates that the
  63     * given variable is only allowed to match values that come from an ALU
  64     * instruction with the given output type.  A type of nir_type_void
  65     * means it can match any type.
  66     *
  67     * Note: A variable that is both constant and has a non-void type will
  68     * never match anything.
  69     */
  70    nir_alu_type type;
  71 } nir_search_variable;
  72
  73 typedef struct {
  74    nir_search_value value;
  75
  76    nir_alu_type type;
  77
  78    union {
  79       uint64_t u;
  80       int64_t i;
  81       double d;
  82    } data;
  83 } nir_search_constant;
  84
  85 typedef struct {
  86    nir_search_value value;
  87
  88    /* When set on a search expression, the expression will only match an SSA
  89     * value that does *not* have the exact bit set.  If unset, the exact bit
  90     * on the SSA value is ignored.
  91     */
  92    bool inexact;
  93
  94    nir_op opcode;
  95    const nir_search_value *srcs[4];
  96 } nir_search_expression;
  97
  98 NIR_DEFINE_CAST(nir_search_value_as_variable, nir_search_value,
  99                 nir_search_variable, value)
 100 NIR_DEFINE_CAST(nir_search_value_as_constant, nir_search_value,
 101                 nir_search_constant, value)
 102 NIR_DEFINE_CAST(nir_search_value_as_expression, nir_search_value,
 103                 nir_search_expression, value)
 104
 105 nir_alu_instr *
 106 nir_replace_instr(nir_alu_instr *instr, const nir_search_expression *search,
 107                   const nir_search_value *replace, void *mem_ctx);
 108
 109 #endif /* _NIR_SEARCH_ */