Change to data style for some known locations

This commit is contained in:
Rob McMullen 2017-03-20 14:29:08 -07:00
parent 72c4fb430a
commit f152a7dd2d
2 changed files with 5 additions and 1 deletions

View File

@ -577,6 +577,7 @@ class AtariDosDiskImage(DiskImageBase):
addr = 0 addr = 0
start, count = self.get_contiguous_sectors(self.first_vtoc, self.num_vtoc) start, count = self.get_contiguous_sectors(self.first_vtoc, self.num_vtoc)
segment = RawSectorsSegment(r[start:start+count], self.first_vtoc, self.num_vtoc, count, 128, 3, self.header.sector_size, name="VTOC") segment = RawSectorsSegment(r[start:start+count], self.first_vtoc, self.num_vtoc, count, 128, 3, self.header.sector_size, name="VTOC")
segment.style[:] = get_style_bits(data=True)
segments.append(segment) segments.append(segment)
if self.vtoc2 > 0: if self.vtoc2 > 0:
start, count = self.get_contiguous_sectors(self.vtoc2, 1) start, count = self.get_contiguous_sectors(self.vtoc2, 1)
@ -590,6 +591,7 @@ class AtariDosDiskImage(DiskImageBase):
addr = 0 addr = 0
start, count = self.get_contiguous_sectors(361, 8) start, count = self.get_contiguous_sectors(361, 8)
segment = RawSectorsSegment(r[start:start+count], 361, 8, count, 128, 3, self.header.sector_size, name="Directory") segment = RawSectorsSegment(r[start:start+count], 361, 8, count, 128, 3, self.header.sector_size, name="Directory")
segment.style[:] = get_style_bits(data=True)
segments.append(segment) segments.append(segment)
return segments return segments

View File

@ -3,7 +3,7 @@ import numpy as np
from errors import * from errors import *
from diskimages import BaseHeader, DiskImageBase from diskimages import BaseHeader, DiskImageBase
from utils import Directory, VTOC, WriteableSector, BaseSectorList, Dirent from utils import Directory, VTOC, WriteableSector, BaseSectorList, Dirent
from segments import DefaultSegment, EmptySegment, ObjSegment, RawTrackSectorSegment, SegmentSaver from segments import DefaultSegment, EmptySegment, ObjSegment, RawTrackSectorSegment, SegmentSaver, get_style_bits
import logging import logging
log = logging.getLogger(__name__) log = logging.getLogger(__name__)
@ -495,6 +495,7 @@ class Dos33DiskImage(DiskImageBase):
addr = 0 addr = 0
start, count = self.get_contiguous_sectors(self.header.first_vtoc, 1) start, count = self.get_contiguous_sectors(self.header.first_vtoc, 1)
segment = RawTrackSectorSegment(r[start:start+count], self.header.first_vtoc, 1, count, 0, 0, self.header.sector_size, name="VTOC") segment = RawTrackSectorSegment(r[start:start+count], self.header.first_vtoc, 1, count, 0, 0, self.header.sector_size, name="VTOC")
segment.style[:] = get_style_bits(data=True)
segments.append(segment) segments.append(segment)
return segments return segments
@ -511,6 +512,7 @@ class Dos33DiskImage(DiskImageBase):
sector = self.header.sector_from_track(raw[1], raw[2]) sector = self.header.sector_from_track(raw[1], raw[2])
raw = self.rawdata.get_indexed(byte_order) raw = self.rawdata.get_indexed(byte_order)
segment = DefaultSegment(raw, name="Catalog") segment = DefaultSegment(raw, name="Catalog")
segment.style[:] = get_style_bits(data=True)
segments.append(segment) segments.append(segment)
return segments return segments