From a7b24e705e4acdf3bffefe6fb885d10e3e29d956 Mon Sep 17 00:00:00 2001
From: Rob McMullen <feedback@playermissile.com>
Date: Sun, 19 Mar 2017 13:07:19 -0700
Subject: [PATCH] SegmentParser now stores ref to segment data so it can
 reparse without having to pass that in again

---
 atrcopy/parsers.py | 10 ++++++----
 1 file changed, 6 insertions(+), 4 deletions(-)

diff --git a/atrcopy/parsers.py b/atrcopy/parsers.py
index 3f5bd82..f7310db 100644
--- a/atrcopy/parsers.py
+++ b/atrcopy/parsers.py
@@ -21,9 +21,11 @@ class SegmentParser(object):
         self.image = None
         self.segments = []
         self.strict = strict
-        self.parse(segment_data)
+        self.segment_data = segment_data
+        self.parse()
 
-    def parse(self, r):
+    def parse(self):
+        r = self.segment_data
         self.segments.append(DefaultSegment(r, 0))
         try:
             self.image = self.get_image(r)
@@ -51,8 +53,8 @@ class SegmentParser(object):
 class DefaultSegmentParser(SegmentParser):
     menu_name = "Raw Data"
     
-    def parse(self, r):
-        self.segments = [DefaultSegment(r, 0)]
+    def parse(self):
+        self.segments = [DefaultSegment(self.segment_data, 0)]
 
 
 class KBootSegmentParser(SegmentParser):