print total PNG files size

Author: arnaudroques

src/main/java/com/pdiff/RunCommand.java

@@ -10,6 +10,7 @@
 import java.util.concurrent.Executors;
 import java.util.concurrent.TimeUnit;
 import java.util.concurrent.atomic.AtomicInteger;
+import java.util.concurrent.atomic.AtomicLong;
 import java.util.function.Predicate;
 import java.util.stream.Collectors;
 
@@ -38,13 +39,13 @@ public class RunCommand {
 	@Parameter(names = { "-p", "--perf" }, description = "Do not generate output files", required = false)
 	private boolean onlyPerf = false;
 
-
 	private AtomicInteger done = new AtomicInteger();
 
 	private int minimalPrefix;
 
 	private MagicOutput magicOutput;
 	private RemainingTime remainingTime;
+	private final AtomicLong totalSize = new AtomicLong();
 
 	private Predicate<? super DbFileBeforeRun> getFilter() {
 		if (filter == null)
@@ -53,7 +54,7 @@ private Predicate<? super DbFileBeforeRun> getFilter() {
 	}
 
 	public void doit() throws IOException, InterruptedException {
-		
+
 		final DbCollection dbCollection = new DbCollection();
 		this.minimalPrefix = dbCollection.getMinimalPrefix();
 
@@ -104,17 +105,31 @@ public void doit() throws IOException, InterruptedException {
 		}
 		System.out.print(Ansi.ansi().eraseLine());
 		System.out.println("Done!");
+	}
 
+	public static String formatSize(long bytes) {
+		if (bytes < 1024) {
+			return bytes + " bytes";
+		} else if (bytes < 1024 * 1024) {
+			return String.format("%.1f K", bytes / 1024.0);
+		} else {
+			return String.format("%.1f M", bytes / (1024.0 * 1024));
+		}
 	}
 
 	private void processFile(DbFileBeforeRun dbFile) throws Exception {
 		final String threadName = Thread.currentThread().getName();
 		final int workerId = Integer.parseInt(threadName.split("-")[1]);
 
 		magicOutput.updateLivingPart(workerId, workerId + " " + dbFile.getFileName(minimalPrefix));
-		magicOutput.updateLivingPart(slot, remainingTime.updateCountAndGetStatus(done.incrementAndGet()));
+		final String status = remainingTime.updateCountAndGetStatus(done.incrementAndGet());
+
+		magicOutput.updateLivingPart(slot, "(" + formatSize(totalSize.get()) + ") " + status);
+
+		final long sizeInBytes = dbFile.convertMe(this.minimalPrefix, onlyPerf);
+		final long newSize = totalSize.addAndGet(sizeInBytes);
+		magicOutput.updateLivingPart(slot, "(" + formatSize(newSize) + ") " + status);
 
-		dbFile.convertMe(this.minimalPrefix, onlyPerf);
 		magicOutput.updateLivingPart(workerId, "");
 
 	}

src/main/java/com/pdiff/core/DbFileBeforeRun.java

@@ -43,7 +43,7 @@ private DbFileAfterRun getDbFileAfterRun() {
 		return res.get();
 	}
 
-	public void convertMe(int minimalPrefix, boolean onlyPerf) throws Exception {
+	public long convertMe(int minimalPrefix, boolean onlyPerf) throws Exception {
 		List<String> all = Files.readAllLines(getPumlPath());
 		all = all.subList(DbCollection.getStartingLine(all), all.size());
 
@@ -55,6 +55,8 @@ public void convertMe(int minimalPrefix, boolean onlyPerf) throws Exception {
 		final long start = System.currentTimeMillis();
 		final OutputRun outputRun = Introspection.outputImage(outputPathPng, text, onlyPerf);
 
+		final long sizeInBytes = Files.size(outputPathPng);
+
 		final long duration = System.currentTimeMillis() - start;
 
 		final BufferedImage bufferedImage = ImageIO.read(outputPathPng.toFile());
@@ -69,11 +71,14 @@ public void convertMe(int minimalPrefix, boolean onlyPerf) throws Exception {
 		png.addProperty("width", bufferedImage.getWidth());
 		png.addProperty("height", bufferedImage.getHeight());
 		png.addProperty("crc", imageSignature.getCrc());
+		png.addProperty("sizeInBytes", sizeInBytes);
 
 		jsonObject.add("png", png);
 
 		final Gson gson = new GsonBuilder().setPrettyPrinting().create();
 		Files.write(outputPathJson, List.of(gson.toJson(jsonObject)));
+		
+		return sizeInBytes;
 
 	}
 
@@ -82,8 +87,10 @@ private static String getOtherLink(DbFileBeforeRun other) {
 		return "../" + name.substring(0, 2) + "/" + name;
 	}
 
-	public void createStandaloneHtml(int minimalPrefix, DbFileBeforeRun prev, DbFileBeforeRun next/*,
-			List<DbFileAfterRun> allAfterRuns*/) throws IOException {
+	public void createStandaloneHtml(int minimalPrefix, DbFileBeforeRun prev,
+			DbFileBeforeRun next/*
+								 * , List<DbFileAfterRun> allAfterRuns
+								 */) throws IOException {
 		final Path outputPathHtml = transformPath(getPumlPath(), ".html");
 
 		List<String> all = Files.readAllLines(getPumlPath());