Track end operations to align artifacts with pgraph log

Author: abaire

Trace.py

@@ -115,6 +115,7 @@ def __init__(
         self.nv2a_log = NV2ALog(os.path.join(output_dir, "nv2a_log.txt"))
         self.flip_stall_count = 0
         self.command_count = 0
+        self.end_op_count = 0
 
         self.real_dma_pull_addr = dma_pull_addr
         self.real_dma_push_addr = dma_push_addr
@@ -353,8 +354,9 @@ def dump(img_tags, adjusted_offset, layer):
                 layer_name = ""
 
             if self.alpha_mode != self.ALPHA_MODE_KEEP:
-                no_alpha_path = "command%d--tex_%d%scolor.png" % (
+                no_alpha_path = "command%d-draw%d--tex_%d%scolor.png" % (
                     self.command_count,
+                    self.end_op_count,
                     index,
                     layer_name,
                 )
@@ -366,8 +368,9 @@ def dump(img_tags, adjusted_offset, layer):
                 no_alpha_path = None
 
             if self.alpha_mode != self.ALPHA_MODE_DROP:
-                alpha_path = "command%d--tex_%d%scolor-a.png" % (
+                alpha_path = "command%d-draw%d--tex_%d%scolor-a.png" % (
                     self.command_count,
+                    self.end_op_count,
                     index,
                     layer_name,
                 )
@@ -456,7 +459,10 @@ def dump_surfaces(self, _data, *_args):
         # FIXME: Respect anti-aliasing
         img_tags = ""
         if self.alpha_mode != self.ALPHA_MODE_KEEP:
-            no_alpha_path = "command%d--color.png" % (self.command_count)
+            no_alpha_path = "command%d_draw%d--color.png" % (
+                self.command_count,
+                self.end_op_count,
+            )
             img_tags += '<img height="128px" src="%s" alt="%s"/>' % (
                 no_alpha_path,
                 no_alpha_path,
@@ -465,15 +471,18 @@ def dump_surfaces(self, _data, *_args):
             no_alpha_path = None
 
         if self.alpha_mode != self.ALPHA_MODE_DROP:
-            alpha_path = "command%d--color-a.png" % (self.command_count)
+            alpha_path = "command%d_draw%d--color-a.png" % (
+                self.command_count,
+                self.end_op_count,
+            )
             img_tags += '<img height="128px" src="%s" alt="%s"/>' % (
                 alpha_path,
                 alpha_path,
             )
         else:
             alpha_path = None
 
-        path = "command%d--color.png" % (self.command_count)
+        path = "command%d_draw%d--color.png" % (self.command_count, self.end_op_count)
         extra_html = []
 
         extra_html += [img_tags]
@@ -558,7 +567,14 @@ def _handle_begin(self, data, *args):
         if not data:
             return []
 
-        print("BEGIN %d" % self.command_count)
+        print(
+            "BEGIN %d (%d:%d)"
+            % (self.command_count, self.flip_stall_count, self.end_op_count)
+        )
+        self.nv2a_log.log(
+            "BEGIN: command %d frame_draw %d\n"
+            % (self.command_count, self.end_op_count)
+        )
 
         extra_html = []
         extra_html += self.dump_textures(data, *args)
@@ -572,6 +588,7 @@ def _handle_end(self, data, *args):
 
         extra_html = []
         extra_html += self.dump_surfaces(data, *args)
+        self.end_op_count += 1
         return extra_html
 
     def _begin_pgraph_recording(self, _data, *_args):
@@ -701,6 +718,7 @@ def _end_pgraph_recording(self, _data, *_args):
     def _handle_flip_stall(self, _data, *_args):
         print("Flip (Stall)")
         self.flip_stall_count += 1
+        self.end_op_count = 0
 
         self.nv2a_log.log("Flip (stall) %d\n\n" % self.flip_stall_count)
 
@@ -900,7 +918,11 @@ def process_push_buffer_command(self, pull_addr):
     def _write(self, suffix, contents):
         """Writes a raw byte dump."""
         out_path = (
-            os.path.join(self.output_dir, "command%d_" % self.command_count) + suffix
+            os.path.join(
+                self.output_dir,
+                "command%d_draw%d_" % (self.command_count, self.end_op_count),
+            )
+            + suffix
         )
         with open(out_path, "wb") as dumpfile:
             dumpfile.write(contents)