Merge commit 'origin/master' into gallium-0.2
[mesa.git] / src / mesa / shader / 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 /**
48 * Free a parameter list and all its parameters
49 */
50 void
51 _mesa_free_parameter_list(struct gl_program_parameter_list *paramList)
52 {
53 GLuint i;
54 for (i = 0; i < paramList->NumParameters; i++) {
55 if (paramList->Parameters[i].Name)
56 _mesa_free((void *) paramList->Parameters[i].Name);
57 }
58 _mesa_free(paramList->Parameters);
59 if (paramList->ParameterValues)
60 _mesa_align_free(paramList->ParameterValues);
61 _mesa_free(paramList);
62 }
63
64
65 /**
66 * Add a new parameter to a parameter list.
67 * Note that parameter values are usually 4-element GLfloat vectors.
68 * When size > 4 we'll allocate a sequential block of parameters to
69 * store all the values (in blocks of 4).
70 *
71 * \param paramList the list to add the parameter to
72 * \param type type of parameter, such as
73 * \param name the parameter name, will be duplicated/copied!
74 * \param size number of elements in 'values' vector (1..4, or more)
75 * \param values initial parameter value, up to 4 GLfloats, or NULL
76 * \param state state indexes, or NULL
77 * \return index of new parameter in the list, or -1 if error (out of mem)
78 */
79 GLint
80 _mesa_add_parameter(struct gl_program_parameter_list *paramList,
81 enum register_file type, const char *name,
82 GLuint size, GLenum datatype, const GLfloat *values,
83 const gl_state_index state[STATE_LENGTH],
84 GLbitfield flags)
85 {
86 const GLuint oldNum = paramList->NumParameters;
87 const GLuint sz4 = (size + 3) / 4; /* no. of new param slots needed */
88
89 assert(size > 0);
90
91 if (oldNum + sz4 > paramList->Size) {
92 /* Need to grow the parameter list array (alloc some extra) */
93 paramList->Size = paramList->Size + 4 * sz4;
94
95 /* realloc arrays */
96 paramList->Parameters = (struct gl_program_parameter *)
97 _mesa_realloc(paramList->Parameters,
98 oldNum * sizeof(struct gl_program_parameter),
99 paramList->Size * sizeof(struct gl_program_parameter));
100
101 paramList->ParameterValues = (GLfloat (*)[4])
102 _mesa_align_realloc(paramList->ParameterValues, /* old buf */
103 oldNum * 4 * sizeof(GLfloat), /* old size */
104 paramList->Size * 4 *sizeof(GLfloat), /* new sz */
105 16);
106 }
107
108 if (!paramList->Parameters ||
109 !paramList->ParameterValues) {
110 /* out of memory */
111 paramList->NumParameters = 0;
112 paramList->Size = 0;
113 return -1;
114 }
115 else {
116 GLuint i;
117
118 paramList->NumParameters = oldNum + sz4;
119
120 _mesa_memset(&paramList->Parameters[oldNum], 0,
121 sz4 * sizeof(struct gl_program_parameter));
122
123 for (i = 0; i < sz4; i++) {
124 struct gl_program_parameter *p = paramList->Parameters + oldNum + i;
125 p->Name = name ? _mesa_strdup(name) : NULL;
126 p->Type = type;
127 p->Size = size;
128 p->DataType = datatype;
129 p->Flags = flags;
130 if (values) {
131 COPY_4V(paramList->ParameterValues[oldNum + i], values);
132 values += 4;
133 p->Initialized = GL_TRUE;
134 }
135 else {
136 /* silence valgrind */
137 ASSIGN_4V(paramList->ParameterValues[oldNum + i], 0, 0, 0, 0);
138 }
139 size -= 4;
140 }
141
142 if (state) {
143 for (i = 0; i < STATE_LENGTH; i++)
144 paramList->Parameters[oldNum].StateIndexes[i] = state[i];
145 }
146
147 return (GLint) oldNum;
148 }
149 }
150
151
152 /**
153 * Add a new named program parameter (Ex: NV_fragment_program DEFINE statement)
154 * \return index of the new entry in the parameter list
155 */
156 GLint
157 _mesa_add_named_parameter(struct gl_program_parameter_list *paramList,
158 const char *name, const GLfloat values[4])
159 {
160 return _mesa_add_parameter(paramList, PROGRAM_NAMED_PARAM, name,
161 4, GL_NONE, values, NULL, 0x0);
162
163 }
164
165
166 /**
167 * Add a new named constant to the parameter list.
168 * This will be used when the program contains something like this:
169 * PARAM myVals = { 0, 1, 2, 3 };
170 *
171 * \param paramList the parameter list
172 * \param name the name for the constant
173 * \param values four float values
174 * \return index/position of the new parameter in the parameter list
175 */
176 GLint
177 _mesa_add_named_constant(struct gl_program_parameter_list *paramList,
178 const char *name, const GLfloat values[4],
179 GLuint size)
180 {
181 #if 0 /* disable this for now -- we need to save the name! */
182 GLint pos;
183 GLuint swizzle;
184 ASSERT(size == 4); /* XXX future feature */
185 /* check if we already have this constant */
186 if (_mesa_lookup_parameter_constant(paramList, values, 4, &pos, &swizzle)) {
187 return pos;
188 }
189 #endif
190 return _mesa_add_parameter(paramList, PROGRAM_CONSTANT, name,
191 size, GL_NONE, values, NULL, 0x0);
192 }
193
194
195 /**
196 * Add a new unnamed constant to the parameter list. This will be used
197 * when a fragment/vertex program contains something like this:
198 * MOV r, { 0, 1, 2, 3 };
199 * We'll search the parameter list for an existing instance of the
200 * constant. If swizzleOut is non-null, we'll try swizzling when
201 * looking for a match.
202 *
203 * \param paramList the parameter list
204 * \param values four float values
205 * \param swizzleOut returns swizzle mask for accessing the constant
206 * \return index/position of the new parameter in the parameter list.
207 */
208 GLint
209 _mesa_add_unnamed_constant(struct gl_program_parameter_list *paramList,
210 const GLfloat values[4], GLuint size,
211 GLuint *swizzleOut)
212 {
213 GLint pos;
214 ASSERT(size >= 1);
215 ASSERT(size <= 4);
216
217 if (_mesa_lookup_parameter_constant(paramList, values,
218 size, &pos, swizzleOut)) {
219 return pos;
220 }
221
222 /* Look for empty space in an already unnamed constant parameter
223 * to add this constant. This will only work for single-element
224 * constants because we rely on smearing (i.e. .yyyy or .zzzz).
225 */
226 if (size == 1 && swizzleOut) {
227 for (pos = 0; pos < (GLint) paramList->NumParameters; pos++) {
228 struct gl_program_parameter *p = paramList->Parameters + pos;
229 if (p->Type == PROGRAM_CONSTANT && p->Size + size <= 4) {
230 /* ok, found room */
231 GLfloat *pVal = paramList->ParameterValues[pos];
232 GLuint swz = p->Size; /* 1, 2 or 3 for Y, Z, W */
233 pVal[p->Size] = values[0];
234 p->Size++;
235 *swizzleOut = MAKE_SWIZZLE4(swz, swz, swz, swz);
236 return pos;
237 }
238 }
239 }
240
241 /* add a new parameter to store this constant */
242 pos = _mesa_add_parameter(paramList, PROGRAM_CONSTANT, NULL,
243 size, GL_NONE, values, NULL, 0x0);
244 if (pos >= 0 && swizzleOut) {
245 if (size == 1)
246 *swizzleOut = SWIZZLE_XXXX;
247 else
248 *swizzleOut = SWIZZLE_NOOP;
249 }
250 return pos;
251 }
252
253
254 /**
255 * Add a uniform to the parameter list.
256 * Note that if the uniform is an array, size may be greater than
257 * what's implied by the datatype.
258 * \param name uniform's name
259 * \param size number of floats to allocate
260 * \param datatype GL_FLOAT_VEC3, GL_FLOAT_MAT4, etc.
261 */
262 GLint
263 _mesa_add_uniform(struct gl_program_parameter_list *paramList,
264 const char *name, GLuint size, GLenum datatype,
265 const GLfloat *values)
266 {
267 GLint i = _mesa_lookup_parameter_index(paramList, -1, name);
268 ASSERT(datatype != GL_NONE);
269 if (i >= 0 && paramList->Parameters[i].Type == PROGRAM_UNIFORM) {
270 ASSERT(paramList->Parameters[i].Size == size);
271 ASSERT(paramList->Parameters[i].DataType == datatype);
272 /* already in list */
273 return i;
274 }
275 else {
276 i = _mesa_add_parameter(paramList, PROGRAM_UNIFORM, name,
277 size, datatype, values, NULL, 0x0);
278 return i;
279 }
280 }
281
282
283 /**
284 * Mark the named uniform as 'used'.
285 */
286 void
287 _mesa_use_uniform(struct gl_program_parameter_list *paramList,
288 const char *name)
289 {
290 GLuint i;
291 for (i = 0; i < paramList->NumParameters; i++) {
292 struct gl_program_parameter *p = paramList->Parameters + i;
293 if (p->Type == PROGRAM_UNIFORM && _mesa_strcmp(p->Name, name) == 0) {
294 p->Used = GL_TRUE;
295 /* Note that large uniforms may occupy several slots so we're
296 * not done searching yet.
297 */
298 }
299 }
300 }
301
302
303 /**
304 * Add a sampler to the parameter list.
305 * \param name uniform's name
306 * \param datatype GL_SAMPLER_2D, GL_SAMPLER_2D_RECT_ARB, etc.
307 * \param index the sampler number (as seen in TEX instructions)
308 * \return sampler index (starting at zero) or -1 if error
309 */
310 GLint
311 _mesa_add_sampler(struct gl_program_parameter_list *paramList,
312 const char *name, GLenum datatype)
313 {
314 GLint i = _mesa_lookup_parameter_index(paramList, -1, name);
315 if (i >= 0 && paramList->Parameters[i].Type == PROGRAM_SAMPLER) {
316 ASSERT(paramList->Parameters[i].Size == 1);
317 ASSERT(paramList->Parameters[i].DataType == datatype);
318 /* already in list */
319 return (GLint) paramList->ParameterValues[i][0];
320 }
321 else {
322 GLuint i;
323 const GLint size = 1; /* a sampler is basically a texture unit number */
324 GLfloat value;
325 GLint numSamplers = 0;
326 for (i = 0; i < paramList->NumParameters; i++) {
327 if (paramList->Parameters[i].Type == PROGRAM_SAMPLER)
328 numSamplers++;
329 }
330 value = (GLfloat) numSamplers;
331 (void) _mesa_add_parameter(paramList, PROGRAM_SAMPLER, name,
332 size, datatype, &value, NULL, 0x0);
333 return numSamplers;
334 }
335 }
336
337
338 /**
339 * Add parameter representing a varying variable.
340 */
341 GLint
342 _mesa_add_varying(struct gl_program_parameter_list *paramList,
343 const char *name, GLuint size, GLbitfield flags)
344 {
345 GLint i = _mesa_lookup_parameter_index(paramList, -1, name);
346 if (i >= 0 && paramList->Parameters[i].Type == PROGRAM_VARYING) {
347 /* already in list */
348 return i;
349 }
350 else {
351 /*assert(size == 4);*/
352 i = _mesa_add_parameter(paramList, PROGRAM_VARYING, name,
353 size, GL_NONE, NULL, NULL, flags);
354 return i;
355 }
356 }
357
358
359 /**
360 * Add parameter representing a vertex program attribute.
361 * \param size size of attribute (in floats), may be -1 if unknown
362 * \param attrib the attribute index, or -1 if unknown
363 */
364 GLint
365 _mesa_add_attribute(struct gl_program_parameter_list *paramList,
366 const char *name, GLint size, GLenum datatype, GLint attrib)
367 {
368 GLint i = _mesa_lookup_parameter_index(paramList, -1, name);
369 if (i >= 0) {
370 /* replace */
371 if (attrib < 0)
372 attrib = i;
373 paramList->Parameters[i].StateIndexes[0] = attrib;
374 }
375 else {
376 /* add */
377 gl_state_index state[STATE_LENGTH];
378 state[0] = (gl_state_index) attrib;
379 if (size < 0)
380 size = 4;
381 i = _mesa_add_parameter(paramList, PROGRAM_INPUT, name,
382 size, datatype, NULL, state, 0x0);
383 }
384 return i;
385 }
386
387
388
389 #if 0 /* not used yet */
390 /**
391 * Returns the number of 4-component registers needed to store a piece
392 * of GL state. For matrices this may be as many as 4 registers,
393 * everything else needs
394 * just 1 register.
395 */
396 static GLuint
397 sizeof_state_reference(const GLint *stateTokens)
398 {
399 if (stateTokens[0] == STATE_MATRIX) {
400 GLuint rows = stateTokens[4] - stateTokens[3] + 1;
401 assert(rows >= 1);
402 assert(rows <= 4);
403 return rows;
404 }
405 else {
406 return 1;
407 }
408 }
409 #endif
410
411
412 /**
413 * Add a new state reference to the parameter list.
414 * This will be used when the program contains something like this:
415 * PARAM ambient = state.material.front.ambient;
416 *
417 * \param paramList the parameter list
418 * \param stateTokens an array of 5 (STATE_LENGTH) state tokens
419 * \return index of the new parameter.
420 */
421 GLint
422 _mesa_add_state_reference(struct gl_program_parameter_list *paramList,
423 const gl_state_index stateTokens[STATE_LENGTH])
424 {
425 const GLuint size = 4; /* XXX fix */
426 char *name;
427 GLint index;
428
429 /* Check if the state reference is already in the list */
430 for (index = 0; index < (GLint) paramList->NumParameters; index++) {
431 GLuint i, match = 0;
432 for (i = 0; i < STATE_LENGTH; i++) {
433 if (paramList->Parameters[index].StateIndexes[i] == stateTokens[i]) {
434 match++;
435 }
436 else {
437 break;
438 }
439 }
440 if (match == STATE_LENGTH) {
441 /* this state reference is already in the parameter list */
442 return index;
443 }
444 }
445
446 name = _mesa_program_state_string(stateTokens);
447 index = _mesa_add_parameter(paramList, PROGRAM_STATE_VAR, name,
448 size, GL_NONE,
449 NULL, (gl_state_index *) stateTokens, 0x0);
450 paramList->StateFlags |= _mesa_program_state_flags(stateTokens);
451
452 /* free name string here since we duplicated it in add_parameter() */
453 _mesa_free(name);
454
455 return index;
456 }
457
458
459 /**
460 * Lookup a parameter value by name in the given parameter list.
461 * \return pointer to the float[4] values.
462 */
463 GLfloat *
464 _mesa_lookup_parameter_value(const struct gl_program_parameter_list *paramList,
465 GLsizei nameLen, const char *name)
466 {
467 GLuint i = _mesa_lookup_parameter_index(paramList, nameLen, name);
468 if (i < 0)
469 return NULL;
470 else
471 return paramList->ParameterValues[i];
472 }
473
474
475 /**
476 * Given a program parameter name, find its position in the list of parameters.
477 * \param paramList the parameter list to search
478 * \param nameLen length of name (in chars).
479 * If length is negative, assume that name is null-terminated.
480 * \param name the name to search for
481 * \return index of parameter in the list.
482 */
483 GLint
484 _mesa_lookup_parameter_index(const struct gl_program_parameter_list *paramList,
485 GLsizei nameLen, const char *name)
486 {
487 GLint i;
488
489 if (!paramList)
490 return -1;
491
492 if (nameLen == -1) {
493 /* name is null-terminated */
494 for (i = 0; i < (GLint) paramList->NumParameters; i++) {
495 if (paramList->Parameters[i].Name &&
496 _mesa_strcmp(paramList->Parameters[i].Name, name) == 0)
497 return i;
498 }
499 }
500 else {
501 /* name is not null-terminated, use nameLen */
502 for (i = 0; i < (GLint) paramList->NumParameters; i++) {
503 if (paramList->Parameters[i].Name &&
504 _mesa_strncmp(paramList->Parameters[i].Name, name, nameLen) == 0
505 && _mesa_strlen(paramList->Parameters[i].Name) == (size_t)nameLen)
506 return i;
507 }
508 }
509 return -1;
510 }
511
512
513 /**
514 * Look for a float vector in the given parameter list. The float vector
515 * may be of length 1, 2, 3 or 4. If swizzleOut is non-null, we'll try
516 * swizzling to find a match.
517 * \param list the parameter list to search
518 * \param v the float vector to search for
519 * \param size number of element in v
520 * \param posOut returns the position of the constant, if found
521 * \param swizzleOut returns a swizzle mask describing location of the
522 * vector elements if found.
523 * \return GL_TRUE if found, GL_FALSE if not found
524 */
525 GLboolean
526 _mesa_lookup_parameter_constant(const struct gl_program_parameter_list *list,
527 const GLfloat v[], GLuint vSize,
528 GLint *posOut, GLuint *swizzleOut)
529 {
530 GLuint i;
531
532 assert(vSize >= 1);
533 assert(vSize <= 4);
534
535 if (!list)
536 return -1;
537
538 for (i = 0; i < list->NumParameters; i++) {
539 if (list->Parameters[i].Type == PROGRAM_CONSTANT) {
540 if (!swizzleOut) {
541 /* swizzle not allowed */
542 GLuint j, match = 0;
543 for (j = 0; j < vSize; j++) {
544 if (v[j] == list->ParameterValues[i][j])
545 match++;
546 }
547 if (match == vSize) {
548 *posOut = i;
549 return GL_TRUE;
550 }
551 }
552 else {
553 /* try matching w/ swizzle */
554 if (vSize == 1) {
555 /* look for v[0] anywhere within float[4] value */
556 GLuint j;
557 for (j = 0; j < 4; j++) {
558 if (list->ParameterValues[i][j] == v[0]) {
559 /* found it */
560 *posOut = i;
561 *swizzleOut = MAKE_SWIZZLE4(j, j, j, j);
562 return GL_TRUE;
563 }
564 }
565 }
566 else if (vSize <= list->Parameters[i].Size) {
567 /* see if we can match this constant (with a swizzle) */
568 GLuint swz[4];
569 GLuint match = 0, j, k;
570 for (j = 0; j < vSize; j++) {
571 if (v[j] == list->ParameterValues[i][j]) {
572 swz[j] = j;
573 match++;
574 }
575 else {
576 for (k = 0; k < list->Parameters[i].Size; k++) {
577 if (v[j] == list->ParameterValues[i][k]) {
578 swz[j] = k;
579 match++;
580 break;
581 }
582 }
583 }
584 }
585 /* smear last value to remaining positions */
586 for (; j < 4; j++)
587 swz[j] = swz[j-1];
588
589 if (match == vSize) {
590 *posOut = i;
591 *swizzleOut = MAKE_SWIZZLE4(swz[0], swz[1], swz[2], swz[3]);
592 return GL_TRUE;
593 }
594 }
595 }
596 }
597 }
598
599 *posOut = -1;
600 return GL_FALSE;
601 }
602
603
604 struct gl_program_parameter_list *
605 _mesa_clone_parameter_list(const struct gl_program_parameter_list *list)
606 {
607 struct gl_program_parameter_list *clone;
608 GLuint i;
609
610 clone = _mesa_new_parameter_list();
611 if (!clone)
612 return NULL;
613
614 /** Not too efficient, but correct */
615 for (i = 0; i < list->NumParameters; i++) {
616 struct gl_program_parameter *p = list->Parameters + i;
617 struct gl_program_parameter *pCopy;
618 GLuint size = MIN2(p->Size, 4);
619 GLint j = _mesa_add_parameter(clone, p->Type, p->Name, size, p->DataType,
620 list->ParameterValues[i], NULL, 0x0);
621 ASSERT(j >= 0);
622 pCopy = clone->Parameters + j;
623 pCopy->Used = p->Used;
624 pCopy->Flags = p->Flags;
625 /* copy state indexes */
626 if (p->Type == PROGRAM_STATE_VAR) {
627 GLint k;
628 for (k = 0; k < STATE_LENGTH; k++) {
629 pCopy->StateIndexes[k] = p->StateIndexes[k];
630 }
631 }
632 else {
633 clone->Parameters[j].Size = p->Size;
634 }
635
636 }
637
638 clone->StateFlags = list->StateFlags;
639
640 return clone;
641 }
642
643
644 /**
645 * Return a new parameter list which is listA + listB.
646 */
647 struct gl_program_parameter_list *
648 _mesa_combine_parameter_lists(const struct gl_program_parameter_list *listA,
649 const struct gl_program_parameter_list *listB)
650 {
651 struct gl_program_parameter_list *list;
652
653 if (listA) {
654 list = _mesa_clone_parameter_list(listA);
655 if (list && listB) {
656 GLuint i;
657 for (i = 0; i < listB->NumParameters; i++) {
658 struct gl_program_parameter *param = listB->Parameters + i;
659 _mesa_add_parameter(list, param->Type, param->Name, param->Size,
660 param->DataType,
661 listB->ParameterValues[i],
662 param->StateIndexes,
663 param->Flags);
664 }
665 }
666 }
667 else if (listB) {
668 list = _mesa_clone_parameter_list(listB);
669 }
670 else {
671 list = NULL;
672 }
673 return list;
674 }
675
676
677
678 /**
679 * Find longest name of all uniform parameters in list.
680 */
681 GLuint
682 _mesa_longest_parameter_name(const struct gl_program_parameter_list *list,
683 enum register_file type)
684 {
685 GLuint i, maxLen = 0;
686 if (!list)
687 return 0;
688 for (i = 0; i < list->NumParameters; i++) {
689 if (list->Parameters[i].Type == type) {
690 GLuint len = _mesa_strlen(list->Parameters[i].Name);
691 if (len > maxLen)
692 maxLen = len;
693 }
694 }
695 return maxLen;
696 }
697
698
699 /**
700 * Count the number of parameters in the last that match the given type.
701 */
702 GLuint
703 _mesa_num_parameters_of_type(const struct gl_program_parameter_list *list,
704 enum register_file type)
705 {
706 GLuint i, count = 0;
707 if (list) {
708 for (i = 0; i < list->NumParameters; i++) {
709 if (list->Parameters[i].Type == type)
710 count++;
711 }
712 }
713 return count;
714 }