Write out compressed and uncompressed packet sizes

Author: per-mathisen-arm

src/filewriter.cpp

@@ -258,6 +258,8 @@ buffer file_writer::compress_chunk(buffer& uncompressed)
 	uint64_t header[2] = { was_written, was_read }; // store compressed and uncompressed sizes
 	memcpy(compressed.data(), header, header_size); // use memcpy to avoid aliasing issues
 	compressed.shrink(was_written + header_size);
+	compressed_sizes.push_back(was_written);
+	uncompressed_sizes.push_back(was_read);
 	DLOG3("Filewriter thread %d handing over compressed buffer of %lu bytes, was %lu bytes uncompressed", mTid, (unsigned long)(was_written + header_size), (unsigned long)was_read);
 	return compressed;
 }

src/filewriter.h

@@ -196,6 +196,11 @@ class file_writer
 		chunk_mutex.unlock();
 	}
 
+protected:
+	// these only written to by worker thread until the end when they are read out
+	std::vector<uint64_t> compressed_sizes;
+	std::vector<uint64_t> uncompressed_sizes;
+
 public:
 	/// Stop sending packets to compression, useful if you want to keep pointers to written memory to keep working
 	inline void freeze() { if (!holding) holding = true; }

src/write.cpp

@@ -90,6 +90,10 @@ lava_file_writer::~lava_file_writer()
 		v["frames"].append(k);
 		highest = std::max(highest, frame.global_frame);
 	}
+	v["uncompressed_sizes"] = Json::arrayValue;
+	v["compressed_sizes"] = Json::arrayValue;
+	for (const auto i : uncompressed_sizes) v["uncompressed_sizes"].append((Json::Value::UInt64)i);
+	for (const auto i : compressed_sizes) v["compressed_sizes"].append((Json::Value::UInt64)i);
 	DLOG("Wrapping up thread %u with %d frames", current.thread, highest);
 	v["highest_global_frame"] = highest;
 	const std::string path = mPath + "/frames_" + _to_string(current.thread) + ".json";