From f09fe30af5316081c87b7531a1f6367c4e7c63c0 Mon Sep 17 00:00:00 2001
From: Thomas Harte <thomas.harte@gmail.com>
Date: Wed, 12 Jul 2017 21:23:59 -0400
Subject: [PATCH] Attempted a full implementation of CSW. All in memory for
 now.

---
 Storage/Tape/Formats/CSW.cpp | 59 +++++++++++++++++++++++++-----------
 Storage/Tape/Formats/CSW.hpp | 12 ++++++--
 2 files changed, 51 insertions(+), 20 deletions(-)

diff --git a/Storage/Tape/Formats/CSW.cpp b/Storage/Tape/Formats/CSW.cpp
index 9ce4b090f..52ab856da 100644
--- a/Storage/Tape/Formats/CSW.cpp
+++ b/Storage/Tape/Formats/CSW.cpp
@@ -11,7 +11,8 @@
 using namespace Storage::Tape;
 
 CSW::CSW(const char *file_name) :
-	Storage::FileHolder(file_name) {
+	Storage::FileHolder(file_name),
+	source_data_pointer_(0) {
 	if(file_stats_.st_size < 0x20) throw ErrorNotCSW;
 
 	// Check signature.
@@ -31,6 +32,7 @@ CSW::CSW(const char *file_name) :
 	if(major_version > 2 || !major_version || minor_version > 1) throw ErrorNotCSW;
 
 	// The header now diverges based on version.
+	uint32_t number_of_waves = 0;
 	if(major_version == 1) {
 		pulse_.length.clock_rate = fgetc16le();
 
@@ -42,7 +44,7 @@ CSW::CSW(const char *file_name) :
 		fseek(file_, 0x20, SEEK_SET);
 	} else {
 		pulse_.length.clock_rate = fgetc32le();
-		number_of_waves_ = fgetc32le();
+		number_of_waves = fgetc32le();
 		switch(fgetc(file_)) {
 			case 1: compression_type_ = RLE;	break;
 			case 2: compression_type_ = ZRLE;	break;
@@ -57,31 +59,54 @@ CSW::CSW(const char *file_name) :
 	}
 
 	if(compression_type_ == ZRLE) {
-		inflation_stream_.zalloc = Z_NULL;
-		inflation_stream_.zfree = Z_NULL;
-		inflation_stream_.opaque = Z_NULL;
-		inflation_stream_.avail_in = 0;
-		inflation_stream_.next_in = Z_NULL;
-		int result = inflateInit(&inflation_stream_);
-		if(result != Z_OK) throw ErrorNotCSW;
+		source_data_.resize((size_t)number_of_waves);
+
+		std::vector<uint8_t> file_data;
+		size_t remaining_data = (size_t)file_stats_.st_size - (size_t)ftell(file_);
+		file_data.resize(remaining_data);
+		fread(file_data.data(), sizeof(uint8_t), remaining_data, file_);
+
+		uLongf output_length = (uLongf)remaining_data;
+		uncompress(source_data_.data(), &output_length, file_data.data(), file_data.size());
+	} else {
+		rle_start_ = ftell(file_);
+	}
+
+	invert_pulse();
+}
+
+uint8_t CSW::get_next_byte() {
+	switch(compression_type_) {
+		case RLE: return (uint8_t)fgetc(file_);
+		case ZRLE: {
+			if(source_data_pointer_ == source_data_.size()) return 0xff;
+			uint8_t result = source_data_[source_data_pointer_];
+			source_data_pointer_++;
+			return result;
+		}
 	}
 }
 
-CSW::~CSW() {
-	if(compression_type_ == ZRLE) {
-		inflateEnd(&inflation_stream_);
-	}
+void CSW::invert_pulse() {
+	pulse_.type = (pulse_.type == Pulse::High) ? Pulse::Low : Pulse::High;
 }
 
-
 bool CSW::is_at_end() {
-	return true;
+	switch(compression_type_) {
+		case RLE: return (bool)feof(file_);
+		case ZRLE: return source_data_pointer_ == source_data_.size();
+	}
 }
 
 void CSW::virtual_reset() {
+	switch(compression_type_) {
+		case RLE:	fseek(file_, rle_start_, SEEK_SET);	break;
+		case ZRLE:	source_data_pointer_ = 0;			break;
+	}
 }
 
 Tape::Pulse CSW::virtual_get_next_pulse() {
-	Tape::Pulse pulse;
-	return pulse;
+	invert_pulse();
+	pulse_.length.length = get_next_byte();
+	return pulse_;
 }
diff --git a/Storage/Tape/Formats/CSW.hpp b/Storage/Tape/Formats/CSW.hpp
index 4063e530b..426310de1 100644
--- a/Storage/Tape/Formats/CSW.hpp
+++ b/Storage/Tape/Formats/CSW.hpp
@@ -13,6 +13,7 @@
 #include "../../FileHolder.hpp"
 
 #include <zlib.h>
+#include <vector>
 
 namespace Storage {
 namespace Tape {
@@ -28,7 +29,6 @@ class CSW: public Tape, public Storage::FileHolder {
 			@throws ErrorNotCSW if this file could not be opened and recognised as a valid CSW file.
 		*/
 		CSW(const char *file_name);
-		~CSW();
 
 		enum {
 			ErrorNotCSW
@@ -46,8 +46,14 @@ class CSW: public Tape, public Storage::FileHolder {
 			RLE,
 			ZRLE
 		} compression_type_;
-		uint32_t number_of_waves_;
-		z_stream inflation_stream_;
+
+		uint8_t get_next_byte();
+		void invert_pulse();
+
+		std::vector<uint8_t> source_data_;
+		size_t source_data_pointer_;
+
+		long rle_start_;
 };
 
 }