Fixed get_comment_locations to return unindexed style so those segments using OrderedWrapper will work

This commit is contained in:
Rob McMullen 2017-03-18 20:54:36 -07:00
parent b7aa965d47
commit 592abfa0ee

View File

@ -173,10 +173,16 @@ class SegmentData(object):
return self.style return self.style
@property @property
def unindexed_view(self): def unindexed_data(self):
if self.is_indexed: if self.is_indexed:
return self.data.unindexed return self.data.unindexed
return self.data return self.data
@property
def unindexed_style(self):
if self.is_indexed:
return self.style.unindexed
return self.style
def byte_bounds_offset(self): def byte_bounds_offset(self):
"""Return start and end offsets of this segment's data into the """Return start and end offsets of this segment's data into the
@ -505,7 +511,7 @@ class DefaultSegment(object):
comment_indexes = np.asarray(self.rawdata.extra.comments.keys(), dtype=np.uint32) comment_indexes = np.asarray(self.rawdata.extra.comments.keys(), dtype=np.uint32)
#print comment_indexes #print comment_indexes
r.style_base[comment_indexes] |= comment_bit_mask r.style_base[comment_indexes] |= comment_bit_mask
return r.style return r.unindexed_style[:]
def get_entire_style_ranges(self, split_comments=None, **kwargs): def get_entire_style_ranges(self, split_comments=None, **kwargs):
"""Find sections of the segment that have the same style value. """Find sections of the segment that have the same style value.