From 40a5a3207abf768c5553d7ef32344539df012b2e Mon Sep 17 00:00:00 2001 From: Rob McMullen Date: Mon, 29 Oct 2018 13:32:59 -0700 Subject: [PATCH] Added SegmentParser.reconstruct_segments to fill segments with actual data * used when restoring segment parser from json --- atrcopy/parsers.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/atrcopy/parsers.py b/atrcopy/parsers.py index b51fb3d..08257d2 100644 --- a/atrcopy/parsers.py +++ b/atrcopy/parsers.py @@ -78,6 +78,12 @@ class SegmentParser: raise errors.InvalidSegmentParser(e) self.segments.extend(self.image.segments) + def reconstruct_segments(self, new_rawdata): + self.image = self.get_image(new_rawdata) + self.segment_data = new_rawdata + for s in self.segments: + s.reconstruct_raw(new_rawdata) + def get_image(self, r): return self.image_type(r)