src/mesa/program/prog_parameter.c

   1 /*
   2  * Mesa 3-D graphics library
   3  * Version:  7.3
   4  *
   5  * Copyright (C) 1999-2008  Brian Paul   All Rights Reserved.
   6  *
   7  * Permission is hereby granted, free of charge, to any person obtaining a
   8  * copy of this software and associated documentation files (the "Software"),
   9  * to deal in the Software without restriction, including without limitation
  10  * the rights to use, copy, modify, merge, publish, distribute, sublicense,
  11  * and/or sell copies of the Software, and to permit persons to whom the
  12  * Software is furnished to do so, subject to the following conditions:
  13  *
  14  * The above copyright notice and this permission notice shall be included
  15  * in all copies or substantial portions of the Software.
  16  *
  17  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
  18  * OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  19  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL
  20  * BRIAN PAUL BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN
  21  * AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
  22  * CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
  23  */
  24
  25 /**
  26  * \file prog_parameter.c
  27  * Program parameter lists and functions.
  28  * \author Brian Paul
  29  */
  30
  31
  32 #include "main/glheader.h"
  33 #include "main/imports.h"
  34 #include "main/macros.h"
  35 #include "prog_instruction.h"
  36 #include "prog_parameter.h"
  37 #include "prog_statevars.h"
  38
  39
  40 struct gl_program_parameter_list *
  41 _mesa_new_parameter_list(void)
  42 {
  43    return CALLOC_STRUCT(gl_program_parameter_list);
  44 }
  45
  46
  47 struct gl_program_parameter_list *
  48 _mesa_new_parameter_list_sized(unsigned size)
  49 {
  50    struct gl_program_parameter_list *p = _mesa_new_parameter_list();
  51
  52    if ((p != NULL) && (size != 0)) {
  53       p->Size = size;
  54
  55       /* alloc arrays */
  56       p->Parameters = (struct gl_program_parameter *)
  57          calloc(1, size * sizeof(struct gl_program_parameter));
  58
  59       p->ParameterValues = (GLfloat (*)[4])
  60          _mesa_align_malloc(size * 4 *sizeof(GLfloat), 16);
  61
  62
  63       if ((p->Parameters == NULL) || (p->ParameterValues == NULL)) {
  64          free(p->Parameters);
  65          _mesa_align_free(p->ParameterValues);
  66          free(p);
  67          p = NULL;
  68       }
  69    }
  70
  71    return p;
  72 }
  73
  74
  75 /**
  76  * Free a parameter list and all its parameters
  77  */
  78 void
  79 _mesa_free_parameter_list(struct gl_program_parameter_list *paramList)
  80 {
  81    GLuint i;
  82    for (i = 0; i < paramList->NumParameters; i++) {
  83       if (paramList->Parameters[i].Name)
  84          free((void *) paramList->Parameters[i].Name);
  85    }
  86    free(paramList->Parameters);
  87    if (paramList->ParameterValues)
  88       _mesa_align_free(paramList->ParameterValues);
  89    free(paramList);
  90 }
  91
  92
  93 /**
  94  * Add a new parameter to a parameter list.
  95  * Note that parameter values are usually 4-element GLfloat vectors.
  96  * When size > 4 we'll allocate a sequential block of parameters to
  97  * store all the values (in blocks of 4).
  98  *
  99  * \param paramList  the list to add the parameter to
 100  * \param type  type of parameter, such as
 101  * \param name  the parameter name, will be duplicated/copied!
 102  * \param size  number of elements in 'values' vector (1..4, or more)
 103  * \param datatype  GL_FLOAT, GL_FLOAT_VECx, GL_INT, GL_INT_VECx or GL_NONE.
 104  * \param values  initial parameter value, up to 4 GLfloats, or NULL
 105  * \param state  state indexes, or NULL
 106  * \return  index of new parameter in the list, or -1 if error (out of mem)
 107  */
 108 GLint
 109 _mesa_add_parameter(struct gl_program_parameter_list *paramList,
 110                     gl_register_file type, const char *name,
 111                     GLuint size, GLenum datatype, const GLfloat *values,
 112                     const gl_state_index state[STATE_LENGTH],
 113                     GLbitfield flags)
 114 {
 115    const GLuint oldNum = paramList->NumParameters;
 116    const GLuint sz4 = (size + 3) / 4; /* no. of new param slots needed */
 117
 118    assert(size > 0);
 119
 120    if (oldNum + sz4 > paramList->Size) {
 121       /* Need to grow the parameter list array (alloc some extra) */
 122       paramList->Size = paramList->Size + 4 * sz4;
 123
 124       /* realloc arrays */
 125       paramList->Parameters = (struct gl_program_parameter *)
 126          _mesa_realloc(paramList->Parameters,
 127                        oldNum * sizeof(struct gl_program_parameter),
 128                        paramList->Size * sizeof(struct gl_program_parameter));
 129
 130       paramList->ParameterValues = (GLfloat (*)[4])
 131          _mesa_align_realloc(paramList->ParameterValues,         /* old buf */
 132                              oldNum * 4 * sizeof(GLfloat),      /* old size */
 133                              paramList->Size * 4 *sizeof(GLfloat), /* new sz */
 134                              16);
 135    }
 136
 137    if (!paramList->Parameters ||
 138        !paramList->ParameterValues) {
 139       /* out of memory */
 140       paramList->NumParameters = 0;
 141       paramList->Size = 0;
 142       return -1;
 143    }
 144    else {
 145       GLuint i;
 146
 147       paramList->NumParameters = oldNum + sz4;
 148
 149       memset(&paramList->Parameters[oldNum], 0,
 150              sz4 * sizeof(struct gl_program_parameter));
 151
 152       for (i = 0; i < sz4; i++) {
 153          struct gl_program_parameter *p = paramList->Parameters + oldNum + i;
 154          p->Name = name ? _mesa_strdup(name) : NULL;
 155          p->Type = type;
 156          p->Size = size;
 157          p->DataType = datatype;
 158          p->Flags = flags;
 159          if (values) {
 160             COPY_4V(paramList->ParameterValues[oldNum + i], values);
 161             values += 4;
 162             p->Initialized = GL_TRUE;
 163          }
 164          else {
 165             /* silence valgrind */
 166             ASSIGN_4V(paramList->ParameterValues[oldNum + i], 0, 0, 0, 0);
 167          }
 168          size -= 4;
 169       }
 170
 171       if (state) {
 172          for (i = 0; i < STATE_LENGTH; i++)
 173             paramList->Parameters[oldNum].StateIndexes[i] = state[i];
 174       }
 175
 176       return (GLint) oldNum;
 177    }
 178 }
 179
 180
 181 /**
 182  * Add a new named program parameter (Ex: NV_fragment_program DEFINE statement)
 183  * \return index of the new entry in the parameter list
 184  */
 185 GLint
 186 _mesa_add_named_parameter(struct gl_program_parameter_list *paramList,
 187                           const char *name, const GLfloat values[4])
 188 {
 189    return _mesa_add_parameter(paramList, PROGRAM_NAMED_PARAM, name,
 190                               4, GL_NONE, values, NULL, 0x0);
 191
 192 }
 193
 194
 195 /**
 196  * Add a new named constant to the parameter list.
 197  * This will be used when the program contains something like this:
 198  *    PARAM myVals = { 0, 1, 2, 3 };
 199  *
 200  * \param paramList  the parameter list
 201  * \param name  the name for the constant
 202  * \param values  four float values
 203  * \return index/position of the new parameter in the parameter list
 204  */
 205 GLint
 206 _mesa_add_named_constant(struct gl_program_parameter_list *paramList,
 207                          const char *name, const GLfloat values[4],
 208                          GLuint size)
 209 {
 210    /* first check if this is a duplicate constant */
 211    GLint pos;
 212    for (pos = 0; pos < (GLint)paramList->NumParameters; pos++) {
 213       const GLfloat *pvals = paramList->ParameterValues[pos];
 214       if (pvals[0] == values[0] &&
 215           pvals[1] == values[1] &&
 216           pvals[2] == values[2] &&
 217           pvals[3] == values[3] &&
 218           strcmp(paramList->Parameters[pos].Name, name) == 0) {
 219          /* Same name and value is already in the param list - reuse it */
 220          return pos;
 221       }
 222    }
 223    /* not found, add new parameter */
 224    return _mesa_add_parameter(paramList, PROGRAM_CONSTANT, name,
 225                               size, GL_NONE, values, NULL, 0x0);
 226 }
 227
 228
 229 /**
 230  * Add a new unnamed constant to the parameter list.  This will be used
 231  * when a fragment/vertex program contains something like this:
 232  *    MOV r, { 0, 1, 2, 3 };
 233  * If swizzleOut is non-null we'll search the parameter list for an
 234  * existing instance of the constant which matches with a swizzle.
 235  *
 236  * \param paramList  the parameter list
 237  * \param values  four float values
 238  * \param swizzleOut  returns swizzle mask for accessing the constant
 239  * \return index/position of the new parameter in the parameter list.
 240  */
 241 GLint
 242 _mesa_add_unnamed_constant(struct gl_program_parameter_list *paramList,
 243                            const GLfloat values[4], GLuint size,
 244                            GLuint *swizzleOut)
 245 {
 246    GLint pos;
 247    ASSERT(size >= 1);
 248    ASSERT(size <= 4);
 249
 250    if (swizzleOut &&
 251        _mesa_lookup_parameter_constant(paramList, values,
 252                                        size, &pos, swizzleOut)) {
 253       return pos;
 254    }
 255
 256    /* Look for empty space in an already unnamed constant parameter
 257     * to add this constant.  This will only work for single-element
 258     * constants because we rely on smearing (i.e. .yyyy or .zzzz).
 259     */
 260    if (size == 1 && swizzleOut) {
 261       for (pos = 0; pos < (GLint) paramList->NumParameters; pos++) {
 262          struct gl_program_parameter *p = paramList->Parameters + pos;
 263          if (p->Type == PROGRAM_CONSTANT && p->Size + size <= 4) {
 264             /* ok, found room */
 265             GLfloat *pVal = paramList->ParameterValues[pos];
 266             GLuint swz = p->Size; /* 1, 2 or 3 for Y, Z, W */
 267             pVal[p->Size] = values[0];
 268             p->Size++;
 269             *swizzleOut = MAKE_SWIZZLE4(swz, swz, swz, swz);
 270             return pos;
 271          }
 272       }
 273    }
 274
 275    /* add a new parameter to store this constant */
 276    pos = _mesa_add_parameter(paramList, PROGRAM_CONSTANT, NULL,
 277                              size, GL_NONE, values, NULL, 0x0);
 278    if (pos >= 0 && swizzleOut) {
 279       if (size == 1)
 280          *swizzleOut = SWIZZLE_XXXX;
 281       else
 282          *swizzleOut = SWIZZLE_NOOP;
 283    }
 284    return pos;
 285 }
 286
 287 /**
 288  * Add parameter representing a varying variable.
 289  */
 290 GLint
 291 _mesa_add_varying(struct gl_program_parameter_list *paramList,
 292                   const char *name, GLuint size, GLenum datatype,
 293                   GLbitfield flags)
 294 {
 295    GLint i = _mesa_lookup_parameter_index(paramList, -1, name);
 296    if (i >= 0 && paramList->Parameters[i].Type == PROGRAM_VARYING) {
 297       /* already in list */
 298       return i;
 299    }
 300    else {
 301       /*assert(size == 4);*/
 302       i = _mesa_add_parameter(paramList, PROGRAM_VARYING, name,
 303                               size, datatype, NULL, NULL, flags);
 304       return i;
 305    }
 306 }
 307
 308
 309 /**
 310  * Add parameter representing a vertex program attribute.
 311  * \param size  size of attribute (in floats), may be -1 if unknown
 312  * \param attrib  the attribute index, or -1 if unknown
 313  */
 314 GLint
 315 _mesa_add_attribute(struct gl_program_parameter_list *paramList,
 316                     const char *name, GLint size, GLenum datatype, GLint attrib)
 317 {
 318    GLint i = _mesa_lookup_parameter_index(paramList, -1, name);
 319    if (i >= 0) {
 320       /* replace */
 321       if (attrib < 0)
 322          attrib = i;
 323       paramList->Parameters[i].StateIndexes[0] = attrib;
 324    }
 325    else {
 326       /* add */
 327       gl_state_index state[STATE_LENGTH];
 328       state[0] = (gl_state_index) attrib;
 329       if (size < 0)
 330          size = 4;
 331       i = _mesa_add_parameter(paramList, PROGRAM_INPUT, name,
 332                               size, datatype, NULL, state, 0x0);
 333    }
 334    return i;
 335 }
 336
 337
 338
 339 #if 0 /* not used yet */
 340 /**
 341  * Returns the number of 4-component registers needed to store a piece
 342  * of GL state.  For matrices this may be as many as 4 registers,
 343  * everything else needs
 344  * just 1 register.
 345  */
 346 static GLuint
 347 sizeof_state_reference(const GLint *stateTokens)
 348 {
 349    if (stateTokens[0] == STATE_MATRIX) {
 350       GLuint rows = stateTokens[4] - stateTokens[3] + 1;
 351       assert(rows >= 1);
 352       assert(rows <= 4);
 353       return rows;
 354    }
 355    else {
 356       return 1;
 357    }
 358 }
 359 #endif
 360
 361
 362 /**
 363  * Add a new state reference to the parameter list.
 364  * This will be used when the program contains something like this:
 365  *    PARAM ambient = state.material.front.ambient;
 366  *
 367  * \param paramList  the parameter list
 368  * \param stateTokens  an array of 5 (STATE_LENGTH) state tokens
 369  * \return index of the new parameter.
 370  */
 371 GLint
 372 _mesa_add_state_reference(struct gl_program_parameter_list *paramList,
 373                           const gl_state_index stateTokens[STATE_LENGTH])
 374 {
 375    const GLuint size = 4; /* XXX fix */
 376    char *name;
 377    GLint index;
 378
 379    /* Check if the state reference is already in the list */
 380    for (index = 0; index < (GLint) paramList->NumParameters; index++) {
 381       GLuint i, match = 0;
 382       for (i = 0; i < STATE_LENGTH; i++) {
 383          if (paramList->Parameters[index].StateIndexes[i] == stateTokens[i]) {
 384             match++;
 385          }
 386          else {
 387             break;
 388          }
 389       }
 390       if (match == STATE_LENGTH) {
 391          /* this state reference is already in the parameter list */
 392          return index;
 393       }
 394    }
 395
 396    name = _mesa_program_state_string(stateTokens);
 397    index = _mesa_add_parameter(paramList, PROGRAM_STATE_VAR, name,
 398                                size, GL_NONE,
 399                                NULL, (gl_state_index *) stateTokens, 0x0);
 400    paramList->StateFlags |= _mesa_program_state_flags(stateTokens);
 401
 402    /* free name string here since we duplicated it in add_parameter() */
 403    free(name);
 404
 405    return index;
 406 }
 407
 408
 409 /**
 410  * Lookup a parameter value by name in the given parameter list.
 411  * \return pointer to the float[4] values.
 412  */
 413 GLfloat *
 414 _mesa_lookup_parameter_value(const struct gl_program_parameter_list *paramList,
 415                              GLsizei nameLen, const char *name)
 416 {
 417    GLint i = _mesa_lookup_parameter_index(paramList, nameLen, name);
 418    if (i < 0)
 419       return NULL;
 420    else
 421       return paramList->ParameterValues[i];
 422 }
 423
 424
 425 /**
 426  * Given a program parameter name, find its position in the list of parameters.
 427  * \param paramList  the parameter list to search
 428  * \param nameLen  length of name (in chars).
 429  *                 If length is negative, assume that name is null-terminated.
 430  * \param name  the name to search for
 431  * \return index of parameter in the list.
 432  */
 433 GLint
 434 _mesa_lookup_parameter_index(const struct gl_program_parameter_list *paramList,
 435                              GLsizei nameLen, const char *name)
 436 {
 437    GLint i;
 438
 439    if (!paramList)
 440       return -1;
 441
 442    if (nameLen == -1) {
 443       /* name is null-terminated */
 444       for (i = 0; i < (GLint) paramList->NumParameters; i++) {
 445          if (paramList->Parameters[i].Name &&
 446              strcmp(paramList->Parameters[i].Name, name) == 0)
 447             return i;
 448       }
 449    }
 450    else {
 451       /* name is not null-terminated, use nameLen */
 452       for (i = 0; i < (GLint) paramList->NumParameters; i++) {
 453          if (paramList->Parameters[i].Name &&
 454              strncmp(paramList->Parameters[i].Name, name, nameLen) == 0
 455              && strlen(paramList->Parameters[i].Name) == (size_t)nameLen)
 456             return i;
 457       }
 458    }
 459    return -1;
 460 }
 461
 462
 463 /**
 464  * Look for a float vector in the given parameter list.  The float vector
 465  * may be of length 1, 2, 3 or 4.  If swizzleOut is non-null, we'll try
 466  * swizzling to find a match.
 467  * \param list  the parameter list to search
 468  * \param v  the float vector to search for
 469  * \param vSize  number of element in v
 470  * \param posOut  returns the position of the constant, if found
 471  * \param swizzleOut  returns a swizzle mask describing location of the
 472  *                    vector elements if found.
 473  * \return GL_TRUE if found, GL_FALSE if not found
 474  */
 475 GLboolean
 476 _mesa_lookup_parameter_constant(const struct gl_program_parameter_list *list,
 477                                 const GLfloat v[], GLuint vSize,
 478                                 GLint *posOut, GLuint *swizzleOut)
 479 {
 480    GLuint i;
 481
 482    assert(vSize >= 1);
 483    assert(vSize <= 4);
 484
 485    if (!list) {
 486       *posOut = -1;
 487       return GL_FALSE;
 488    }
 489
 490    for (i = 0; i < list->NumParameters; i++) {
 491       if (list->Parameters[i].Type == PROGRAM_CONSTANT) {
 492          if (!swizzleOut) {
 493             /* swizzle not allowed */
 494             GLuint j, match = 0;
 495             for (j = 0; j < vSize; j++) {
 496                if (v[j] == list->ParameterValues[i][j])
 497                   match++;
 498             }
 499             if (match == vSize) {
 500                *posOut = i;
 501                return GL_TRUE;
 502             }
 503          }
 504          else {
 505             /* try matching w/ swizzle */
 506              if (vSize == 1) {
 507                 /* look for v[0] anywhere within float[4] value */
 508                 GLuint j;
 509                 for (j = 0; j < list->Parameters[i].Size; j++) {
 510                    if (list->ParameterValues[i][j] == v[0]) {
 511                       /* found it */
 512                       *posOut = i;
 513                       *swizzleOut = MAKE_SWIZZLE4(j, j, j, j);
 514                       return GL_TRUE;
 515                    }
 516                 }
 517              }
 518              else if (vSize <= list->Parameters[i].Size) {
 519                 /* see if we can match this constant (with a swizzle) */
 520                 GLuint swz[4];
 521                 GLuint match = 0, j, k;
 522                 for (j = 0; j < vSize; j++) {
 523                    if (v[j] == list->ParameterValues[i][j]) {
 524                       swz[j] = j;
 525                       match++;
 526                    }
 527                    else {
 528                       for (k = 0; k < list->Parameters[i].Size; k++) {
 529                          if (v[j] == list->ParameterValues[i][k]) {
 530                             swz[j] = k;
 531                             match++;
 532                             break;
 533                          }
 534                       }
 535                    }
 536                 }
 537                 /* smear last value to remaining positions */
 538                 for (; j < 4; j++)
 539                    swz[j] = swz[j-1];
 540
 541                 if (match == vSize) {
 542                    *posOut = i;
 543                    *swizzleOut = MAKE_SWIZZLE4(swz[0], swz[1], swz[2], swz[3]);
 544                    return GL_TRUE;
 545                 }
 546              }
 547          }
 548       }
 549    }
 550
 551    *posOut = -1;
 552    return GL_FALSE;
 553 }
 554
 555
 556 struct gl_program_parameter_list *
 557 _mesa_clone_parameter_list(const struct gl_program_parameter_list *list)
 558 {
 559    struct gl_program_parameter_list *clone;
 560    GLuint i;
 561
 562    clone = _mesa_new_parameter_list();
 563    if (!clone)
 564       return NULL;
 565
 566    /** Not too efficient, but correct */
 567    for (i = 0; i < list->NumParameters; i++) {
 568       struct gl_program_parameter *p = list->Parameters + i;
 569       struct gl_program_parameter *pCopy;
 570       GLuint size = MIN2(p->Size, 4);
 571       GLint j = _mesa_add_parameter(clone, p->Type, p->Name, size, p->DataType,
 572                                     list->ParameterValues[i], NULL, 0x0);
 573       ASSERT(j >= 0);
 574       pCopy = clone->Parameters + j;
 575       pCopy->Flags = p->Flags;
 576       /* copy state indexes */
 577       if (p->Type == PROGRAM_STATE_VAR) {
 578          GLint k;
 579          for (k = 0; k < STATE_LENGTH; k++) {
 580             pCopy->StateIndexes[k] = p->StateIndexes[k];
 581          }
 582       }
 583       else {
 584          clone->Parameters[j].Size = p->Size;
 585       }
 586
 587    }
 588
 589    clone->StateFlags = list->StateFlags;
 590
 591    return clone;
 592 }
 593
 594
 595 /**
 596  * Return a new parameter list which is listA + listB.
 597  */
 598 struct gl_program_parameter_list *
 599 _mesa_combine_parameter_lists(const struct gl_program_parameter_list *listA,
 600                               const struct gl_program_parameter_list *listB)
 601 {
 602    struct gl_program_parameter_list *list;
 603
 604    if (listA) {
 605       list = _mesa_clone_parameter_list(listA);
 606       if (list && listB) {
 607          GLuint i;
 608          for (i = 0; i < listB->NumParameters; i++) {
 609             struct gl_program_parameter *param = listB->Parameters + i;
 610             _mesa_add_parameter(list, param->Type, param->Name, param->Size,
 611                                 param->DataType,
 612                                 listB->ParameterValues[i],
 613                                 param->StateIndexes,
 614                                 param->Flags);
 615          }
 616       }
 617    }
 618    else if (listB) {
 619       list = _mesa_clone_parameter_list(listB);
 620    }
 621    else {
 622       list = NULL;
 623    }
 624    return list;
 625 }
 626
 627
 628
 629 /**
 630  * Find longest name of all uniform parameters in list.
 631  */
 632 GLuint
 633 _mesa_longest_parameter_name(const struct gl_program_parameter_list *list,
 634                              gl_register_file type)
 635 {
 636    GLuint i, maxLen = 0;
 637    if (!list)
 638       return 0;
 639    for (i = 0; i < list->NumParameters; i++) {
 640       if (list->Parameters[i].Type == type) {
 641          GLuint len = strlen(list->Parameters[i].Name);
 642          if (len > maxLen)
 643             maxLen = len;
 644       }
 645    }
 646    return maxLen;
 647 }
 648
 649
 650 /**
 651  * Count the number of parameters in the last that match the given type.
 652  */
 653 GLuint
 654 _mesa_num_parameters_of_type(const struct gl_program_parameter_list *list,
 655                              gl_register_file type)
 656 {
 657    GLuint i, count = 0;
 658    if (list) {
 659       for (i = 0; i < list->NumParameters; i++) {
 660          if (list->Parameters[i].Type == type)
 661             count++;
 662       }
 663    }
 664    return count;
 665 }