diff --git a/atrcopy/ataridos.py b/atrcopy/ataridos.py index dce21af..1681def 100644 --- a/atrcopy/ataridos.py +++ b/atrcopy/ataridos.py @@ -577,6 +577,7 @@ class AtariDosDiskImage(DiskImageBase): addr = 0 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.style[:] = get_style_bits(data=True) segments.append(segment) if self.vtoc2 > 0: start, count = self.get_contiguous_sectors(self.vtoc2, 1) @@ -590,6 +591,7 @@ class AtariDosDiskImage(DiskImageBase): addr = 0 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.style[:] = get_style_bits(data=True) segments.append(segment) return segments diff --git a/atrcopy/dos33.py b/atrcopy/dos33.py index 1626e2b..0e48dae 100644 --- a/atrcopy/dos33.py +++ b/atrcopy/dos33.py @@ -3,7 +3,7 @@ import numpy as np from errors import * from diskimages import BaseHeader, DiskImageBase 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 log = logging.getLogger(__name__) @@ -495,6 +495,7 @@ class Dos33DiskImage(DiskImageBase): addr = 0 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.style[:] = get_style_bits(data=True) segments.append(segment) return segments @@ -511,6 +512,7 @@ class Dos33DiskImage(DiskImageBase): sector = self.header.sector_from_track(raw[1], raw[2]) raw = self.rawdata.get_indexed(byte_order) segment = DefaultSegment(raw, name="Catalog") + segment.style[:] = get_style_bits(data=True) segments.append(segment) return segments