From caee2554f8c3a182dcadebeaed9819388ca15a61 Mon Sep 17 00:00:00 2001
From: Rob McMullen <feedback@playermissile.com>
Date: Sat, 26 May 2018 20:14:57 -0700
Subject: [PATCH] Added segment saver for Apple 2 binary format

---
 atrcopy/segments.py | 16 +++++++++++++++-
 1 file changed, 15 insertions(+), 1 deletion(-)

diff --git a/atrcopy/segments.py b/atrcopy/segments.py
index 77f2587..9dbd3ae 100644
--- a/atrcopy/segments.py
+++ b/atrcopy/segments.py
@@ -74,6 +74,20 @@ class SegmentSaver(object):
         return segment.tobytes()
 
 
+class BSAVESaver(object):
+    export_data_name = "Apple ][ Binary"
+    export_extensions = [".bsave"]
+
+    @classmethod
+    def encode_data(cls, segment, ui_control):
+        data = segment.tobytes()
+        header = np.empty(2, dtype="<u2")
+        header[0] = segment.start_addr
+        header[1] = len(data)
+        print("binary data: %x bytes at %x" % (header[1], header[0]))
+        return header.tobytes() + segment.tobytes()
+
+
 class OrderWrapper(object):
     """Wrapper for numpy data so that manipulations can use normal numpy syntax
     and still affect the data according to the byte ordering.
@@ -380,7 +394,7 @@ class SegmentData(object):
 
 
 class DefaultSegment(object):
-    savers = [SegmentSaver]
+    savers = [SegmentSaver, BSAVESaver]
     can_resize_default = False
 
     base_serializable_attributes = ['start_addr', 'error', 'name', 'verbose_name', 'page_size', 'map_width', 'uuid', 'can_resize']