r300g: Add new debug option for logging vertex/fragment program stats

1 files changed, 62 insertions, 4 deletions

diff --git a/src/mesa/drivers/dri/r300/compiler/radeon_compiler.c b/src/mesa/drivers/dri/r300/compiler/radeon_compiler.c
index b410b2daf4..125d64e649 100644
--- a/src/mesa/drivers/dri/r300/compiler/radeon_compiler.c
+++ b/src/mesa/drivers/dri/r300/compiler/radeon_compiler.c
@@ -26,7 +26,9 @@
 #include <stdio.h>
 #include <stdlib.h>
 
+#include "radeon_dataflow.h"
 #include "radeon_program.h"
+#include "radeon_program_pair.h"
 
 
 void rc_init(struct radeon_compiler * c)
@@ -50,7 +52,7 @@ void rc_debug(struct radeon_compiler * c, const char * fmt, ...)
 {
 	va_list ap;
 
-	if (!c->Debug)
+	if (!(c->Debug & RC_DBG_LOG))
 		return;
 
 	va_start(ap, fmt);
@@ -84,7 +86,7 @@ void rc_error(struct radeon_compiler * c, const char * fmt, ...)
 		}
 	}
 
-	if (c->Debug) {
+	if (c->Debug & RC_DBG_LOG) {
 		fprintf(stderr, "r300compiler error: ");
 
 		va_start(ap, fmt);
@@ -351,11 +353,65 @@ void rc_transform_fragment_face(struct radeon_compiler *c, unsigned face)
 	}
 }
 
+static void reg_count_callback(void * userdata, struct rc_instruction * inst,
+		rc_register_file file, unsigned int index, unsigned int mask)
+{
+	unsigned int * max_reg = userdata;
+	if (file == RC_FILE_TEMPORARY)
+		index > *max_reg ? *max_reg = index : 0;
+}
+
+static void print_stats(struct radeon_compiler * c)
+{
+	struct rc_instruction * tmp;
+	unsigned i, max_reg, insts, fc, tex, alpha, rgb, presub;
+	max_reg = insts = fc = tex = alpha = rgb = presub = 0;
+	for(tmp = c->Program.Instructions.Next; tmp != &c->Program.Instructions;
+							tmp = tmp->Next){
+		const struct rc_opcode_info * info;
+		rc_for_all_reads_mask(tmp, reg_count_callback, &max_reg);
+		if (tmp->Type == RC_INSTRUCTION_NORMAL) {
+			if (tmp->U.I.PreSub.Opcode != RC_PRESUB_NONE)
+				presub++;
+			info = rc_get_opcode_info(tmp->U.I.Opcode);
+		} else {
+			if (tmp->U.P.RGB.Src[RC_PAIR_PRESUB_SRC].Used)
+				presub++;
+			if (tmp->U.P.Alpha.Src[RC_PAIR_PRESUB_SRC].Used)
+				presub++;
+			/* Assuming alpha will never be a flow control or
+			 * a tex instruction. */
+			if (tmp->U.P.Alpha.Opcode != RC_OPCODE_NOP)
+				alpha++;
+			if (tmp->U.P.RGB.Opcode != RC_OPCODE_NOP)
+				rgb++;
+			info = rc_get_opcode_info(tmp->U.P.RGB.Opcode);
+		}
+		if (info->IsFlowControl)
+			fc++;
+		if (info->HasTexture)
+			tex++;
+		insts++;
+	}
+	if (insts < 4)
+		return;
+	fprintf(stderr,"~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~\n"
+		       "~%4u Instructions\n"
+		       "~%4u Vector Instructions (RGB)\n"
+		       "~%4u Scalar Instructions (Alpha)\n"
+		       "~%4u Flow Control Instructions\n"
+		       "~%4u Texture Instructions\n"
+		       "~%4u Presub Operations\n"
+		       "~%4u Temporary Registers\n"
+		       "~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~\n",
+		       insts, rgb, alpha, fc, tex, presub, max_reg + 1);
+}
+
 /* Executes a list of compiler passes given in the parameter 'list'. */
 void rc_run_compiler(struct radeon_compiler *c, struct radeon_compiler_pass *list,
 		     const char *shader_name)
 {
-	if (c->Debug) {
+	if (c->Debug & RC_DBG_LOG) {
 		fprintf(stderr, "%s: before compilation\n", shader_name);
 		rc_print_program(&c->Program);
 	}
@@ -367,12 +423,14 @@ void rc_run_compiler(struct radeon_compiler *c, struct radeon_compiler_pass *lis
 			if (c->Error)
 				return;
 
-			if (c->Debug && list[i].dump) {
+			if ((c->Debug & RC_DBG_LOG) && list[i].dump) {
 				fprintf(stderr, "%s: after '%s'\n", shader_name, list[i].name);
 				rc_print_program(&c->Program);
 			}
 		}
 	}
+	if (c->Debug & RC_DBG_STATS)
+		print_stats(c);
 }
 
 void rc_validate_final_shader(struct radeon_compiler *c, void *user)