From 1282f6066c0612c89ad66822b53c1e2ae70c7cbf Mon Sep 17 00:00:00 2001
From: D4rkr34lm <darth.m.frohn@gmail.com>
Date: Wed, 13 Mar 2024 23:18:57 +0100
Subject: [PATCH] Added count, fuse and time flags

---
 .../features/tracer/rendering/ViewFlag.java   | 35 +++++++++++++++++++
 1 file changed, 35 insertions(+)

diff --git a/BauSystem_Main/src/de/steamwar/bausystem/features/tracer/rendering/ViewFlag.java b/BauSystem_Main/src/de/steamwar/bausystem/features/tracer/rendering/ViewFlag.java
index da6603af..d11bacec 100644
--- a/BauSystem_Main/src/de/steamwar/bausystem/features/tracer/rendering/ViewFlag.java
+++ b/BauSystem_Main/src/de/steamwar/bausystem/features/tracer/rendering/ViewFlag.java
@@ -100,7 +100,42 @@ public abstract class ViewFlag {
                 new RFallingBlockEntity(server, pos, Material.WHITE_STAINED_GLASS);
              }
         }
+    };
 
+    public static ViewFlag COUNT = new ViewFlag(true, false, "count") {
+        @Override
+        public List<TNTRecord> filter(List<TNTRecord> records) { return  records; }
+
+        @Override
+        public void modify(REntityServer server, List<TraceEntity> entities) {
+            for(TraceEntity entity: entities){
+                entity.setDisplayName(String.valueOf(entity.getRecords().size()));
+            }
+        }
+    };
+
+    public static ViewFlag FUSE = new ViewFlag(true, false, "fuse") {
+        @Override
+        public List<TNTRecord> filter(List<TNTRecord> records) { return  records; }
+
+        @Override
+        public void modify(REntityServer server, List<TraceEntity> entities) {
+            for(TraceEntity entity: entities){
+                entity.setDisplayName(String.valueOf(entity.getRecords().get(0).getFuse()));
+            }
+        }
+    };
+
+    public static ViewFlag TIME = new ViewFlag(true, false, "time") {
+        @Override
+        public List<TNTRecord> filter(List<TNTRecord> records) { return  records; }
+
+        @Override
+        public void modify(REntityServer server, List<TraceEntity> entities) {
+            for(TraceEntity entity: entities){
+                entity.setDisplayName(String.valueOf(entity.getRecords().get(0).getTicksSinceStart()));
+            }
+        }
     };
 
     /**