src/glsl/ast.h

   1 /* -*- c++ -*- */
   2 /*
   3  * Copyright © 2009 Intel Corporation
   4  *
   5  * Permission is hereby granted, free of charge, to any person obtaining a
   6  * copy of this software and associated documentation files (the "Software"),
   7  * to deal in the Software without restriction, including without limitation
   8  * the rights to use, copy, modify, merge, publish, distribute, sublicense,
   9  * and/or sell copies of the Software, and to permit persons to whom the
  10  * Software is furnished to do so, subject to the following conditions:
  11  *
  12  * The above copyright notice and this permission notice (including the next
  13  * paragraph) shall be included in all copies or substantial portions of the
  14  * Software.
  15  *
  16  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  17  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  18  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL
  19  * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  20  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
  21  * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
  22  * DEALINGS IN THE SOFTWARE.
  23  */
  24
  25 #pragma once
  26 #ifndef AST_H
  27 #define AST_H
  28
  29 #include "list.h"
  30 #include "glsl_parser_extras.h"
  31
  32 struct _mesa_glsl_parse_state;
  33
  34 struct YYLTYPE;
  35
  36 /**
  37  * \defgroup AST Abstract syntax tree node definitions
  38  *
  39  * An abstract syntax tree is generated by the parser.  This is a fairly
  40  * direct representation of the gramma derivation for the source program.
  41  * No symantic checking is done during the generation of the AST.  Only
  42  * syntactic checking is done.  Symantic checking is performed by a later
  43  * stage that converts the AST to a more generic intermediate representation.
  44  *
  45  *@{
  46  */
  47 /**
  48  * Base class of all abstract syntax tree nodes
  49  */
  50 class ast_node {
  51 public:
  52    DECLARE_RALLOC_CXX_OPERATORS(ast_node);
  53
  54    /**
  55     * Print an AST node in something approximating the original GLSL code
  56     */
  57    virtual void print(void) const;
  58
  59    /**
  60     * Convert the AST node to the high-level intermediate representation
  61     */
  62    virtual ir_rvalue *hir(exec_list *instructions,
  63                           struct _mesa_glsl_parse_state *state);
  64
  65    /**
  66     * Retrieve the source location of an AST node
  67     *
  68     * This function is primarily used to get the source position of an AST node
  69     * into a form that can be passed to \c _mesa_glsl_error.
  70     *
  71     * \sa _mesa_glsl_error, ast_node::set_location
  72     */
  73    struct YYLTYPE get_location(void) const
  74    {
  75       struct YYLTYPE locp;
  76
  77       locp.source = this->location.source;
  78       locp.first_line = this->location.line;
  79       locp.first_column = this->location.column;
  80       locp.last_line = locp.first_line;
  81       locp.last_column = locp.first_column;
  82
  83       return locp;
  84    }
  85
  86    /**
  87     * Set the source location of an AST node from a parser location
  88     *
  89     * \sa ast_node::get_location
  90     */
  91    void set_location(const struct YYLTYPE &locp)
  92    {
  93       this->location.source = locp.source;
  94       this->location.line = locp.first_line;
  95       this->location.column = locp.first_column;
  96    }
  97
  98    /**
  99     * Source location of the AST node.
 100     */
 101    struct {
 102       unsigned source;    /**< GLSL source number. */
 103       unsigned line;      /**< Line number within the source string. */
 104       unsigned column;    /**< Column in the line. */
 105    } location;
 106
 107    exec_node link;
 108
 109 protected:
 110    /**
 111     * The only constructor is protected so that only derived class objects can
 112     * be created.
 113     */
 114    ast_node(void);
 115 };
 116
 117
 118 /**
 119  * Operators for AST expression nodes.
 120  */
 121 enum ast_operators {
 122    ast_assign,
 123    ast_plus,        /**< Unary + operator. */
 124    ast_neg,
 125    ast_add,
 126    ast_sub,
 127    ast_mul,
 128    ast_div,
 129    ast_mod,
 130    ast_lshift,
 131    ast_rshift,
 132    ast_less,
 133    ast_greater,
 134    ast_lequal,
 135    ast_gequal,
 136    ast_equal,
 137    ast_nequal,
 138    ast_bit_and,
 139    ast_bit_xor,
 140    ast_bit_or,
 141    ast_bit_not,
 142    ast_logic_and,
 143    ast_logic_xor,
 144    ast_logic_or,
 145    ast_logic_not,
 146
 147    ast_mul_assign,
 148    ast_div_assign,
 149    ast_mod_assign,
 150    ast_add_assign,
 151    ast_sub_assign,
 152    ast_ls_assign,
 153    ast_rs_assign,
 154    ast_and_assign,
 155    ast_xor_assign,
 156    ast_or_assign,
 157
 158    ast_conditional,
 159
 160    ast_pre_inc,
 161    ast_pre_dec,
 162    ast_post_inc,
 163    ast_post_dec,
 164    ast_field_selection,
 165    ast_array_index,
 166
 167    ast_function_call,
 168
 169    ast_identifier,
 170    ast_int_constant,
 171    ast_uint_constant,
 172    ast_float_constant,
 173    ast_bool_constant,
 174
 175    ast_sequence,
 176    ast_aggregate
 177 };
 178
 179 /**
 180  * Representation of any sort of expression.
 181  */
 182 class ast_expression : public ast_node {
 183 public:
 184    ast_expression(int oper, ast_expression *,
 185                   ast_expression *, ast_expression *);
 186
 187    ast_expression(const char *identifier) :
 188       oper(ast_identifier)
 189    {
 190       subexpressions[0] = NULL;
 191       subexpressions[1] = NULL;
 192       subexpressions[2] = NULL;
 193       primary_expression.identifier = identifier;
 194       this->non_lvalue_description = NULL;
 195    }
 196
 197    static const char *operator_string(enum ast_operators op);
 198
 199    virtual ir_rvalue *hir(exec_list *instructions,
 200                           struct _mesa_glsl_parse_state *state);
 201
 202    virtual void print(void) const;
 203
 204    enum ast_operators oper;
 205
 206    ast_expression *subexpressions[3];
 207
 208    union {
 209       const char *identifier;
 210       int int_constant;
 211       float float_constant;
 212       unsigned uint_constant;
 213       int bool_constant;
 214    } primary_expression;
 215
 216
 217    /**
 218     * List of expressions for an \c ast_sequence or parameters for an
 219     * \c ast_function_call
 220     */
 221    exec_list expressions;
 222
 223    /**
 224     * For things that can't be l-values, this describes what it is.
 225     *
 226     * This text is used by the code that generates IR for assignments to
 227     * detect and emit useful messages for assignments to some things that
 228     * can't be l-values.  For example, pre- or post-incerement expressions.
 229     *
 230     * \note
 231     * This pointer may be \c NULL.
 232     */
 233    const char *non_lvalue_description;
 234 };
 235
 236 class ast_expression_bin : public ast_expression {
 237 public:
 238    ast_expression_bin(int oper, ast_expression *, ast_expression *);
 239
 240    virtual void print(void) const;
 241 };
 242
 243 /**
 244  * Subclass of expressions for function calls
 245  */
 246 class ast_function_expression : public ast_expression {
 247 public:
 248    ast_function_expression(ast_expression *callee)
 249       : ast_expression(ast_function_call, callee,
 250                        NULL, NULL),
 251         cons(false)
 252    {
 253       /* empty */
 254    }
 255
 256    ast_function_expression(class ast_type_specifier *type)
 257       : ast_expression(ast_function_call, (ast_expression *) type,
 258                        NULL, NULL),
 259         cons(true)
 260    {
 261       /* empty */
 262    }
 263
 264    bool is_constructor() const
 265    {
 266       return cons;
 267    }
 268
 269    virtual ir_rvalue *hir(exec_list *instructions,
 270                           struct _mesa_glsl_parse_state *state);
 271
 272 private:
 273    /**
 274     * Is this function call actually a constructor?
 275     */
 276    bool cons;
 277 };
 278
 279 class ast_array_specifier : public ast_node {
 280 public:
 281    /** Unsized array specifier ([]) */
 282    explicit ast_array_specifier(const struct YYLTYPE &locp)
 283      : dimension_count(1), is_unsized_array(true)
 284    {
 285       set_location(locp);
 286    }
 287
 288    /** Sized array specifier ([dim]) */
 289    ast_array_specifier(const struct YYLTYPE &locp, ast_expression *dim)
 290      : dimension_count(1), is_unsized_array(false)
 291    {
 292       set_location(locp);
 293       array_dimensions.push_tail(&dim->link);
 294    }
 295
 296    void add_dimension(ast_expression *dim)
 297    {
 298       array_dimensions.push_tail(&dim->link);
 299       dimension_count++;
 300    }
 301
 302    virtual void print(void) const;
 303
 304    /* Count including sized and unsized dimensions */
 305    unsigned dimension_count;
 306
 307    /* If true, this means that the array has an unsized outermost dimension. */
 308    bool is_unsized_array;
 309
 310    /* This list contains objects of type ast_node containing the
 311     * sized dimensions only, in outermost-to-innermost order.
 312     */
 313    exec_list array_dimensions;
 314 };
 315
 316 /**
 317  * C-style aggregate initialization class
 318  *
 319  * Represents C-style initializers of vectors, matrices, arrays, and
 320  * structures. E.g., vec3 pos = {1.0, 0.0, -1.0} is equivalent to
 321  * vec3 pos = vec3(1.0, 0.0, -1.0).
 322  *
 323  * Specified in GLSL 4.20 and GL_ARB_shading_language_420pack.
 324  *
 325  * \sa _mesa_ast_set_aggregate_type
 326  */
 327 class ast_aggregate_initializer : public ast_expression {
 328 public:
 329    ast_aggregate_initializer()
 330       : ast_expression(ast_aggregate, NULL, NULL, NULL),
 331         constructor_type(NULL)
 332    {
 333       /* empty */
 334    }
 335
 336    /**
 337     * glsl_type of the aggregate, which is inferred from the LHS of whatever
 338     * the aggregate is being used to initialize.  This can't be inferred at
 339     * parse time (since the parser deals with ast_type_specifiers, not
 340     * glsl_types), so the parser leaves it NULL.  However, the ast-to-hir
 341     * conversion code makes sure to fill it in with the appropriate type
 342     * before hir() is called.
 343     */
 344    const glsl_type *constructor_type;
 345
 346    virtual ir_rvalue *hir(exec_list *instructions,
 347                           struct _mesa_glsl_parse_state *state);
 348 };
 349
 350 /**
 351  * Number of possible operators for an ast_expression
 352  *
 353  * This is done as a define instead of as an additional value in the enum so
 354  * that the compiler won't generate spurious messages like "warning:
 355  * enumeration value ‘ast_num_operators’ not handled in switch"
 356  */
 357 #define AST_NUM_OPERATORS (ast_sequence + 1)
 358
 359
 360 class ast_compound_statement : public ast_node {
 361 public:
 362    ast_compound_statement(int new_scope, ast_node *statements);
 363    virtual void print(void) const;
 364
 365    virtual ir_rvalue *hir(exec_list *instructions,
 366                           struct _mesa_glsl_parse_state *state);
 367
 368    int new_scope;
 369    exec_list statements;
 370 };
 371
 372 class ast_declaration : public ast_node {
 373 public:
 374    ast_declaration(const char *identifier,
 375                    ast_array_specifier *array_specifier,
 376                    ast_expression *initializer);
 377    virtual void print(void) const;
 378
 379    const char *identifier;
 380
 381    ast_array_specifier *array_specifier;
 382
 383    ast_expression *initializer;
 384 };
 385
 386
 387 enum {
 388    ast_precision_none = 0, /**< Absence of precision qualifier. */
 389    ast_precision_high,
 390    ast_precision_medium,
 391    ast_precision_low
 392 };
 393
 394 struct ast_type_qualifier {
 395    DECLARE_RALLOC_CXX_OPERATORS(ast_type_qualifier);
 396
 397    union {
 398       struct {
 399          unsigned invariant:1;
 400          unsigned constant:1;
 401          unsigned attribute:1;
 402          unsigned varying:1;
 403          unsigned in:1;
 404          unsigned out:1;
 405          unsigned centroid:1;
 406          unsigned sample:1;
 407          unsigned uniform:1;
 408          unsigned smooth:1;
 409          unsigned flat:1;
 410          unsigned noperspective:1;
 411
 412          /** \name Layout qualifiers for GL_ARB_fragment_coord_conventions */
 413          /*@{*/
 414          unsigned origin_upper_left:1;
 415          unsigned pixel_center_integer:1;
 416          /*@}*/
 417
 418          /**
 419           * Flag set if GL_ARB_explicit_attrib_location "location" layout
 420           * qualifier is used.
 421           */
 422          unsigned explicit_location:1;
 423          /**
 424           * Flag set if GL_ARB_explicit_attrib_location "index" layout
 425           * qualifier is used.
 426           */
 427          unsigned explicit_index:1;
 428
 429          /**
 430           * Flag set if GL_ARB_shading_language_420pack "binding" layout
 431           * qualifier is used.
 432           */
 433          unsigned explicit_binding:1;
 434
 435          /**
 436           * Flag set if GL_ARB_shader_atomic counter "offset" layout
 437           * qualifier is used.
 438           */
 439          unsigned explicit_offset:1;
 440
 441          /** \name Layout qualifiers for GL_AMD_conservative_depth */
 442          /** \{ */
 443          unsigned depth_any:1;
 444          unsigned depth_greater:1;
 445          unsigned depth_less:1;
 446          unsigned depth_unchanged:1;
 447          /** \} */
 448
 449          /** \name Layout qualifiers for GL_ARB_uniform_buffer_object */
 450          /** \{ */
 451          unsigned std140:1;
 452          unsigned shared:1;
 453          unsigned packed:1;
 454          unsigned column_major:1;
 455          unsigned row_major:1;
 456          /** \} */
 457
 458          /** \name Layout qualifiers for GLSL 1.50 geometry shaders */
 459          /** \{ */
 460          unsigned prim_type:1;
 461          unsigned max_vertices:1;
 462          /** \} */
 463
 464          /**
 465           * local_size_{x,y,z} flags for compute shaders.  Bit 0 represents
 466           * local_size_x, and so on.
 467           */
 468          unsigned local_size:3;
 469       }
 470       /** \brief Set of flags, accessed by name. */
 471       q;
 472
 473       /** \brief Set of flags, accessed as a bitmask. */
 474       unsigned i;
 475    } flags;
 476
 477    /** Precision of the type (highp/medium/lowp). */
 478    unsigned precision:2;
 479
 480    /**
 481     * Location specified via GL_ARB_explicit_attrib_location layout
 482     *
 483     * \note
 484     * This field is only valid if \c explicit_location is set.
 485     */
 486    int location;
 487    /**
 488     * Index specified via GL_ARB_explicit_attrib_location layout
 489     *
 490     * \note
 491     * This field is only valid if \c explicit_index is set.
 492     */
 493    int index;
 494
 495    /** Maximum output vertices in GLSL 1.50 geometry shaders. */
 496    int max_vertices;
 497
 498    /** Input or output primitive type in GLSL 1.50 geometry shaders */
 499    GLenum prim_type;
 500
 501    /**
 502     * Binding specified via GL_ARB_shading_language_420pack's "binding" keyword.
 503     *
 504     * \note
 505     * This field is only valid if \c explicit_binding is set.
 506     */
 507    int binding;
 508
 509    /**
 510     * Offset specified via GL_ARB_shader_atomic_counter's "offset"
 511     * keyword.
 512     *
 513     * \note
 514     * This field is only valid if \c explicit_offset is set.
 515     */
 516    int offset;
 517
 518    /**
 519     * Local size specified via GL_ARB_compute_shader's "local_size_{x,y,z}"
 520     * layout qualifier.  Element i of this array is only valid if
 521     * flags.q.local_size & (1 << i) is set.
 522     */
 523    int local_size[3];
 524
 525    /**
 526     * Return true if and only if an interpolation qualifier is present.
 527     */
 528    bool has_interpolation() const;
 529
 530    /**
 531     * Return whether a layout qualifier is present.
 532     */
 533    bool has_layout() const;
 534
 535    /**
 536     * Return whether a storage qualifier is present.
 537     */
 538    bool has_storage() const;
 539
 540    /**
 541     * Return whether an auxiliary storage qualifier is present.
 542     */
 543    bool has_auxiliary_storage() const;
 544
 545    /**
 546     * \brief Return string representation of interpolation qualifier.
 547     *
 548     * If an interpolation qualifier is present, then return that qualifier's
 549     * string representation. Otherwise, return null. For example, if the
 550     * noperspective bit is set, then this returns "noperspective".
 551     *
 552     * If multiple interpolation qualifiers are somehow present, then the
 553     * returned string is undefined but not null.
 554     */
 555    const char *interpolation_string() const;
 556
 557    bool merge_qualifier(YYLTYPE *loc,
 558                         _mesa_glsl_parse_state *state,
 559                         ast_type_qualifier q);
 560 };
 561
 562 class ast_declarator_list;
 563
 564 class ast_struct_specifier : public ast_node {
 565 public:
 566    /**
 567     * \brief Make a shallow copy of an ast_struct_specifier.
 568     *
 569     * Use only if the objects are allocated from the same context and will not
 570     * be modified. Zeros the inherited ast_node's fields.
 571     */
 572    ast_struct_specifier(const ast_struct_specifier& that):
 573       ast_node(), name(that.name), declarations(that.declarations),
 574       is_declaration(that.is_declaration)
 575    {
 576       /* empty */
 577    }
 578
 579    ast_struct_specifier(const char *identifier,
 580                         ast_declarator_list *declarator_list);
 581    virtual void print(void) const;
 582
 583    virtual ir_rvalue *hir(exec_list *instructions,
 584                           struct _mesa_glsl_parse_state *state);
 585
 586    const char *name;
 587    /* List of ast_declarator_list * */
 588    exec_list declarations;
 589    bool is_declaration;
 590 };
 591
 592
 593
 594 class ast_type_specifier : public ast_node {
 595 public:
 596    /**
 597     * \brief Make a shallow copy of an ast_type_specifier, specifying array
 598     *        fields.
 599     *
 600     * Use only if the objects are allocated from the same context and will not
 601     * be modified. Zeros the inherited ast_node's fields.
 602     */
 603    ast_type_specifier(const ast_type_specifier *that,
 604                       ast_array_specifier *array_specifier)
 605       : ast_node(), type_name(that->type_name), structure(that->structure),
 606         array_specifier(array_specifier),
 607         default_precision(that->default_precision)
 608    {
 609       /* empty */
 610    }
 611
 612    /** Construct a type specifier from a type name */
 613    ast_type_specifier(const char *name)
 614       : type_name(name), structure(NULL), array_specifier(NULL),
 615         default_precision(ast_precision_none)
 616    {
 617       /* empty */
 618    }
 619
 620    /** Construct a type specifier from a structure definition */
 621    ast_type_specifier(ast_struct_specifier *s)
 622       : type_name(s->name), structure(s), array_specifier(NULL),
 623         default_precision(ast_precision_none)
 624    {
 625       /* empty */
 626    }
 627
 628    const struct glsl_type *glsl_type(const char **name,
 629                                      struct _mesa_glsl_parse_state *state)
 630       const;
 631
 632    virtual void print(void) const;
 633
 634    ir_rvalue *hir(exec_list *, struct _mesa_glsl_parse_state *);
 635
 636    const char *type_name;
 637    ast_struct_specifier *structure;
 638
 639    ast_array_specifier *array_specifier;
 640
 641    /** For precision statements, this is the given precision; otherwise none. */
 642    unsigned default_precision:2;
 643 };
 644
 645
 646 class ast_fully_specified_type : public ast_node {
 647 public:
 648    virtual void print(void) const;
 649    bool has_qualifiers() const;
 650
 651    ast_fully_specified_type() : qualifier(), specifier(NULL)
 652    {
 653    }
 654
 655    const struct glsl_type *glsl_type(const char **name,
 656                                      struct _mesa_glsl_parse_state *state)
 657       const;
 658
 659    ast_type_qualifier qualifier;
 660    ast_type_specifier *specifier;
 661 };
 662
 663
 664 class ast_declarator_list : public ast_node {
 665 public:
 666    ast_declarator_list(ast_fully_specified_type *);
 667    virtual void print(void) const;
 668
 669    virtual ir_rvalue *hir(exec_list *instructions,
 670                           struct _mesa_glsl_parse_state *state);
 671
 672    ast_fully_specified_type *type;
 673    /** List of 'ast_declaration *' */
 674    exec_list declarations;
 675
 676    /**
 677     * Special flag for vertex shader "invariant" declarations.
 678     *
 679     * Vertex shaders can contain "invariant" variable redeclarations that do
 680     * not include a type.  For example, "invariant gl_Position;".  This flag
 681     * is used to note these cases when no type is specified.
 682     */
 683    int invariant;
 684 };
 685
 686
 687 class ast_parameter_declarator : public ast_node {
 688 public:
 689    ast_parameter_declarator() :
 690       type(NULL),
 691       identifier(NULL),
 692       array_specifier(NULL),
 693       formal_parameter(false),
 694       is_void(false)
 695    {
 696       /* empty */
 697    }
 698
 699    virtual void print(void) const;
 700
 701    virtual ir_rvalue *hir(exec_list *instructions,
 702                           struct _mesa_glsl_parse_state *state);
 703
 704    ast_fully_specified_type *type;
 705    const char *identifier;
 706    ast_array_specifier *array_specifier;
 707
 708    static void parameters_to_hir(exec_list *ast_parameters,
 709                                  bool formal, exec_list *ir_parameters,
 710                                  struct _mesa_glsl_parse_state *state);
 711
 712 private:
 713    /** Is this parameter declaration part of a formal parameter list? */
 714    bool formal_parameter;
 715
 716    /**
 717     * Is this parameter 'void' type?
 718     *
 719     * This field is set by \c ::hir.
 720     */
 721    bool is_void;
 722 };
 723
 724
 725 class ast_function : public ast_node {
 726 public:
 727    ast_function(void);
 728
 729    virtual void print(void) const;
 730
 731    virtual ir_rvalue *hir(exec_list *instructions,
 732                           struct _mesa_glsl_parse_state *state);
 733
 734    ast_fully_specified_type *return_type;
 735    const char *identifier;
 736
 737    exec_list parameters;
 738
 739 private:
 740    /**
 741     * Is this prototype part of the function definition?
 742     *
 743     * Used by ast_function_definition::hir to process the parameters, etc.
 744     * of the function.
 745     *
 746     * \sa ::hir
 747     */
 748    bool is_definition;
 749
 750    /**
 751     * Function signature corresponding to this function prototype instance
 752     *
 753     * Used by ast_function_definition::hir to process the parameters, etc.
 754     * of the function.
 755     *
 756     * \sa ::hir
 757     */
 758    class ir_function_signature *signature;
 759
 760    friend class ast_function_definition;
 761 };
 762
 763
 764 class ast_expression_statement : public ast_node {
 765 public:
 766    ast_expression_statement(ast_expression *);
 767    virtual void print(void) const;
 768
 769    virtual ir_rvalue *hir(exec_list *instructions,
 770                           struct _mesa_glsl_parse_state *state);
 771
 772    ast_expression *expression;
 773 };
 774
 775
 776 class ast_case_label : public ast_node {
 777 public:
 778    ast_case_label(ast_expression *test_value);
 779    virtual void print(void) const;
 780
 781    virtual ir_rvalue *hir(exec_list *instructions,
 782                           struct _mesa_glsl_parse_state *state);
 783
 784    /**
 785     * An test value of NULL means 'default'.
 786     */
 787    ast_expression *test_value;
 788 };
 789
 790
 791 class ast_case_label_list : public ast_node {
 792 public:
 793    ast_case_label_list(void);
 794    virtual void print(void) const;
 795
 796    virtual ir_rvalue *hir(exec_list *instructions,
 797                           struct _mesa_glsl_parse_state *state);
 798
 799    /**
 800     * A list of case labels.
 801     */
 802    exec_list labels;
 803 };
 804
 805
 806 class ast_case_statement : public ast_node {
 807 public:
 808    ast_case_statement(ast_case_label_list *labels);
 809    virtual void print(void) const;
 810
 811    virtual ir_rvalue *hir(exec_list *instructions,
 812                           struct _mesa_glsl_parse_state *state);
 813
 814    ast_case_label_list *labels;
 815
 816    /**
 817     * A list of statements.
 818     */
 819    exec_list stmts;
 820 };
 821
 822
 823 class ast_case_statement_list : public ast_node {
 824 public:
 825    ast_case_statement_list(void);
 826    virtual void print(void) const;
 827
 828    virtual ir_rvalue *hir(exec_list *instructions,
 829                           struct _mesa_glsl_parse_state *state);
 830
 831    /**
 832     * A list of cases.
 833     */
 834    exec_list cases;
 835 };
 836
 837
 838 class ast_switch_body : public ast_node {
 839 public:
 840    ast_switch_body(ast_case_statement_list *stmts);
 841    virtual void print(void) const;
 842
 843    virtual ir_rvalue *hir(exec_list *instructions,
 844                           struct _mesa_glsl_parse_state *state);
 845
 846    ast_case_statement_list *stmts;
 847 };
 848
 849
 850 class ast_selection_statement : public ast_node {
 851 public:
 852    ast_selection_statement(ast_expression *condition,
 853                            ast_node *then_statement,
 854                            ast_node *else_statement);
 855    virtual void print(void) const;
 856
 857    virtual ir_rvalue *hir(exec_list *instructions,
 858                           struct _mesa_glsl_parse_state *state);
 859
 860    ast_expression *condition;
 861    ast_node *then_statement;
 862    ast_node *else_statement;
 863 };
 864
 865
 866 class ast_switch_statement : public ast_node {
 867 public:
 868    ast_switch_statement(ast_expression *test_expression,
 869                         ast_node *body);
 870    virtual void print(void) const;
 871
 872    virtual ir_rvalue *hir(exec_list *instructions,
 873                           struct _mesa_glsl_parse_state *state);
 874
 875    ast_expression *test_expression;
 876    ast_node *body;
 877
 878 protected:
 879    void test_to_hir(exec_list *, struct _mesa_glsl_parse_state *);
 880 };
 881
 882 class ast_iteration_statement : public ast_node {
 883 public:
 884    ast_iteration_statement(int mode, ast_node *init, ast_node *condition,
 885                            ast_expression *rest_expression, ast_node *body);
 886
 887    virtual void print(void) const;
 888
 889    virtual ir_rvalue *hir(exec_list *, struct _mesa_glsl_parse_state *);
 890
 891    enum ast_iteration_modes {
 892       ast_for,
 893       ast_while,
 894       ast_do_while
 895    } mode;
 896
 897
 898    ast_node *init_statement;
 899    ast_node *condition;
 900    ast_expression *rest_expression;
 901
 902    ast_node *body;
 903
 904    /**
 905     * Generate IR from the condition of a loop
 906     *
 907     * This is factored out of ::hir because some loops have the condition
 908     * test at the top (for and while), and others have it at the end (do-while).
 909     */
 910    void condition_to_hir(exec_list *, struct _mesa_glsl_parse_state *);
 911 };
 912
 913
 914 class ast_jump_statement : public ast_node {
 915 public:
 916    ast_jump_statement(int mode, ast_expression *return_value);
 917    virtual void print(void) const;
 918
 919    virtual ir_rvalue *hir(exec_list *instructions,
 920                           struct _mesa_glsl_parse_state *state);
 921
 922    enum ast_jump_modes {
 923       ast_continue,
 924       ast_break,
 925       ast_return,
 926       ast_discard
 927    } mode;
 928
 929    ast_expression *opt_return_value;
 930 };
 931
 932
 933 class ast_function_definition : public ast_node {
 934 public:
 935    ast_function_definition() : prototype(NULL), body(NULL)
 936    {
 937    }
 938
 939    virtual void print(void) const;
 940
 941    virtual ir_rvalue *hir(exec_list *instructions,
 942                           struct _mesa_glsl_parse_state *state);
 943
 944    ast_function *prototype;
 945    ast_compound_statement *body;
 946 };
 947
 948 class ast_interface_block : public ast_node {
 949 public:
 950    ast_interface_block(ast_type_qualifier layout,
 951                        const char *instance_name,
 952                        ast_array_specifier *array_specifier)
 953    : layout(layout), block_name(NULL), instance_name(instance_name),
 954      array_specifier(array_specifier)
 955    {
 956    }
 957
 958    virtual ir_rvalue *hir(exec_list *instructions,
 959                           struct _mesa_glsl_parse_state *state);
 960
 961    ast_type_qualifier layout;
 962    const char *block_name;
 963
 964    /**
 965     * Declared name of the block instance, if specified.
 966     *
 967     * If the block does not have an instance name, this field will be
 968     * \c NULL.
 969     */
 970    const char *instance_name;
 971
 972    /** List of ast_declarator_list * */
 973    exec_list declarations;
 974
 975    /**
 976     * Declared array size of the block instance
 977     *
 978     * If the block is not declared as an array or if the block instance array
 979     * is unsized, this field will be \c NULL.
 980     */
 981    ast_array_specifier *array_specifier;
 982 };
 983
 984
 985 /**
 986  * AST node representing a declaration of the input layout for geometry
 987  * shaders.
 988  */
 989 class ast_gs_input_layout : public ast_node
 990 {
 991 public:
 992    ast_gs_input_layout(const struct YYLTYPE &locp, GLenum prim_type)
 993       : prim_type(prim_type)
 994    {
 995       set_location(locp);
 996    }
 997
 998    virtual ir_rvalue *hir(exec_list *instructions,
 999                           struct _mesa_glsl_parse_state *state);
1000
1001 private:
1002    const GLenum prim_type;
1003 };
1004
1005
1006 /**
1007  * AST node representing a decalaration of the input layout for compute
1008  * shaders.
1009  */
1010 class ast_cs_input_layout : public ast_node
1011 {
1012 public:
1013    ast_cs_input_layout(const struct YYLTYPE &locp, const unsigned *local_size)
1014    {
1015       memcpy(this->local_size, local_size, sizeof(this->local_size));
1016       set_location(locp);
1017    }
1018
1019    virtual ir_rvalue *hir(exec_list *instructions,
1020                           struct _mesa_glsl_parse_state *state);
1021
1022 private:
1023    unsigned local_size[3];
1024 };
1025
1026 /*@}*/
1027
1028 extern void
1029 _mesa_ast_to_hir(exec_list *instructions, struct _mesa_glsl_parse_state *state);
1030
1031 extern ir_rvalue *
1032 _mesa_ast_field_selection_to_hir(const ast_expression *expr,
1033                                  exec_list *instructions,
1034                                  struct _mesa_glsl_parse_state *state);
1035
1036 extern ir_rvalue *
1037 _mesa_ast_array_index_to_hir(void *mem_ctx,
1038                              struct _mesa_glsl_parse_state *state,
1039                              ir_rvalue *array, ir_rvalue *idx,
1040                              YYLTYPE &loc, YYLTYPE &idx_loc);
1041
1042 extern void
1043 _mesa_ast_set_aggregate_type(const glsl_type *type,
1044                              ast_expression *expr);
1045
1046 void
1047 emit_function(_mesa_glsl_parse_state *state, ir_function *f);
1048
1049 extern void
1050 check_builtin_array_max_size(const char *name, unsigned size,
1051                              YYLTYPE loc, struct _mesa_glsl_parse_state *state);
1052
1053 #endif /* AST_H */