Fixed get_comment_locations: was using a copy and returning a subset of the original

This commit is contained in:
Rob McMullen 2017-03-17 23:43:43 -07:00
parent 46409addae
commit d9ced9fbd2
1 changed files with 3 additions and 2 deletions

View File

@ -3,6 +3,7 @@ import cStringIO
import uuid
import numpy as np
from profilehooks import coverage
from errors import *
from utils import to_numpy, to_numpy_list
@ -500,10 +501,10 @@ class DefaultSegment(object):
r = self.rawdata.copy()
print len(r.style)
print len(r.style_base)
base = r.style_base & style_bits
r.style_base[:] &= style_bits
comment_indexes = np.asarray(self.rawdata.extra.comments.keys(), dtype=np.uint32)
print comment_indexes
base[comment_indexes] |= comment_bit_mask
r.style_base[comment_indexes] |= comment_bit_mask
return r.style
def get_entire_style_ranges(self, split_comments=None, **kwargs):