Changed to labeled bit mask constants rather than hard-coded bit values

* changed to one user bit mask value rather than separate bits
This commit is contained in:
Rob McMullen 2016-04-18 11:54:29 -07:00
parent e9640269f9
commit 02ba76f358
2 changed files with 14 additions and 14 deletions

View File

@ -11,7 +11,7 @@ from errors import *
from ataridos import AtariDosDiskImage, AtariDosFile
from diskimages import AtrHeader, BootDiskImage
from kboot import KBootImage
from segments import SegmentData, SegmentSaver, DefaultSegment, EmptySegment, ObjSegment, RawSectorsSegment
from segments import SegmentData, SegmentSaver, DefaultSegment, EmptySegment, ObjSegment, RawSectorsSegment, user_bit_mask, match_bit_mask, comment_bit_mask, data_bit_mask, selected_bit_mask
from spartados import SpartaDosDiskImage
from utils import to_numpy

View File

@ -3,6 +3,12 @@ import numpy as np
from errors import *
from utils import to_numpy, to_numpy_list
user_bit_mask = 0x0f
match_bit_mask = 0x10
comment_bit_mask = 0x20
data_bit_mask = 0x40
selected_bit_mask = 0x80
class SegmentSaver(object):
name = "Raw Data"
@ -265,7 +271,7 @@ class DefaultSegment(object):
def tostring(self):
return self.data.tostring()
def get_style_bits(self, match=False, comment=False, selected=False, data=False, user1=False, user2=False, user3=False, user4=False):
def get_style_bits(self, match=False, comment=False, selected=False, data=False, user=0):
""" Return an int value that contains the specified style bits set.
Available styles for each byte are:
@ -276,22 +282,16 @@ class DefaultSegment(object):
data: labeled in the disassembler as a data region (i.e. not disassembled)
"""
style_bits = 0
if user:
style_bits |= (user & user_bit_mask)
if match:
style_bits |= 1
style_bits |= match_bit_mask
if comment:
style_bits |= 2
style_bits |= comment_bit_mask
if data:
style_bits |= 4
if user1:
style_bits |= 8
if user2:
style_bits |= 0x10
if user3:
style_bits |= 0x20
if user4:
style_bits |= 0x40
style_bits |= data_bit_mask
if selected:
style_bits |= 0x80
style_bits |= selected_bit_mask
return style_bits
def get_style_mask(self, **kwargs):