Add writing support to Apple II interleaved disk image.

DiscImageChef.DiscImages/AppleDOS.cs

@@ -38,11 +38,15 @@ using DiscImageChef.Filters;
 
 namespace DiscImageChef.DiscImages
 {
-    public class AppleDos : IMediaImage
+    public class AppleDos : IWritableImage
     {
-        readonly int[] interleave = {0, 14, 13, 12, 11, 10, 9, 8, 7, 6, 5, 4, 3, 2, 1, 15};
         readonly int[] deinterleave = {0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15};
+        readonly int[] interleave   = {0, 14, 13, 12, 11, 10, 9, 8, 7, 6, 5, 4, 3, 2, 1, 15};
+
+        byte[]     deinterleaved;
+        string     extension;
         ImageInfo  imageInfo;
+        FileStream writingStream;
 
         public AppleDos()
         {
@@ -76,7 +80,8 @@ namespace DiscImageChef.DiscImages
         public string Name => "Apple ][ Interleaved Disk Image";
         public Guid   Id   => new Guid("A5828AC0-62C9-4304-81D4-EFD4AAE47360");
 
-        public string Format => extension == ".po"
+        public string Format =>
+            extension == ".po"
                 ? "Apple ][ Interleaved Disk Image (ProDOS order)"
                 : "Apple ][ Interleaved Disk Image (DOS order)";
 
@@ -108,7 +113,9 @@ namespace DiscImageChef.DiscImages
 
             extension = Path.GetExtension(imageFilter.GetFilename())?.ToLower();
 
-            int[] offsets = extension == ".do" ? (isDos ? deinterleave : interleave) : (isDos ? interleave : deinterleave);
+            int[] offsets = extension == ".do"
+                                ? (isDos ? deinterleave : interleave)
+                                : (isDos ? interleave : deinterleave);
 
             for(int t = 0; t < 35; t++)
             {
@@ -218,9 +225,6 @@ namespace DiscImageChef.DiscImages
             throw new FeatureUnsupportedImageException("Feature not supported by image format");
         }
 
-        byte[] deinterleaved;
-        string extension;
-
         public byte[] ReadSectorTag(ulong sectorAddress, SectorTagType tag)
         {
             throw new FeatureUnsupportedImageException("Feature not supported by image format");
@@ -258,5 +262,168 @@ namespace DiscImageChef.DiscImages
         {
             throw new FeatureUnsupportedImageException("Feature not supported by image format");
         }
+
+        public IEnumerable<MediaTagType>  SupportedMediaTags  => new MediaTagType[] { };
+        public IEnumerable<SectorTagType> SupportedSectorTags => new SectorTagType[] { };
+        public IEnumerable<MediaType>     SupportedMediaTypes =>
+            new[] {MediaType.Apple33SS};
+        public IEnumerable<(string name, Type type, string description)> SupportedOptions =>
+            new (string name, Type type, string description)[] { };
+        public IEnumerable<string> KnownExtensions => new[] {".do", ".po"};
+        public bool                IsWriting       { get; private set; }
+        public string              ErrorMessage    { get; private set; }
+
+        public bool Create(string path, MediaType mediaType, Dictionary<string, string> options, ulong sectors,
+                           uint   sectorSize)
+        {
+            if(sectorSize != 256)
+            {
+                ErrorMessage = "Unsupported sector size";
+                return false;
+            }
+
+            if(mediaType != MediaType.Apple33SS)
+            {
+                ErrorMessage = $"Unsupport media format {mediaType}";
+                return false;
+            }
+
+            if(sectors > uint.MaxValue)
+            {
+                ErrorMessage = "Too many sectors";
+                return false;
+            }
+
+            imageInfo = new ImageInfo {MediaType = mediaType, SectorSize = sectorSize, Sectors = sectors};
+
+            try { writingStream = new FileStream(path, FileMode.CreateNew, FileAccess.ReadWrite, FileShare.None); }
+            catch(IOException e)
+            {
+                ErrorMessage = $"Could not create new image file, exception {e.Message}";
+                return false;
+            }
+
+            deinterleaved = new byte[35 * 16 * 256];
+            extension     = Path.GetExtension(path);
+
+            IsWriting    = true;
+            ErrorMessage = null;
+            return true;
+        }
+
+        public bool WriteMediaTag(byte[] data, MediaTagType tag)
+        {
+            ErrorMessage = "Unsupported feature";
+            return false;
+        }
+
+        public bool WriteSector(byte[] data, ulong sectorAddress)
+        {
+            return WriteSectors(data, sectorAddress, 1);
+        }
+
+        public bool WriteSectors(byte[] data, ulong sectorAddress, uint length)
+        {
+            if(!IsWriting)
+            {
+                ErrorMessage = "Tried to write on a non-writable image";
+                return false;
+            }
+
+            if(data.Length % imageInfo.SectorSize != 0)
+            {
+                ErrorMessage = "Incorrect data size";
+                return false;
+            }
+
+            if(sectorAddress + length > imageInfo.Sectors)
+            {
+                ErrorMessage = "Tried to write past image size";
+                return false;
+            }
+
+            Array.Copy(data, 0, deinterleaved, (int)(sectorAddress * imageInfo.SectorSize), data.Length);
+
+            ErrorMessage = "";
+            return true;
+        }
+
+        public bool WriteSectorLong(byte[] data, ulong sectorAddress)
+        {
+            ErrorMessage = "Writing sectors with tags is not supported.";
+            return false;
+        }
+
+        public bool WriteSectorsLong(byte[] data, ulong sectorAddress, uint length)
+        {
+            ErrorMessage = "Writing sectors with tags is not supported.";
+            return false;
+        }
+
+        public bool SetTracks(List<Track> tracks)
+        {
+            ErrorMessage = "Unsupported feature";
+            return false;
+        }
+
+        public bool Close()
+        {
+            if(!IsWriting)
+            {
+                ErrorMessage = "Image is not opened for writing";
+                return false;
+            }
+
+            bool isDos = deinterleaved[0x11001] == 17  && deinterleaved[0x11002] < 16 &&
+                         deinterleaved[0x11027] <= 122 &&
+                         deinterleaved[0x11034] == 35  && deinterleaved[0x11035] == 16 &&
+                         deinterleaved[0x11036] == 0   &&
+                         deinterleaved[0x11037] == 1;
+
+            byte[] tmp = new byte[deinterleaved.Length];
+
+            int[] offsets = extension == ".do"
+                                ? (isDos ? deinterleave : interleave)
+                                : (isDos ? interleave : deinterleave);
+
+            for(int t = 0; t < 35; t++)
+            {
+                for(int s = 0; s < 16; s++)
+                    Array.Copy(deinterleaved, t * 16 * 256 + offsets[s] * 256, tmp, t * 16 * 256 + s * 256, 256);
+            }
+
+            writingStream.Seek(0, SeekOrigin.Begin);
+            writingStream.Write(tmp, 0, tmp.Length);
+            writingStream.Flush();
+            writingStream.Close();
+
+            ErrorMessage = "";
+            IsWriting    = false;
+
+            return true;
+        }
+
+        public bool SetMetadata(ImageInfo metadata)
+        {
+            return true;
+        }
+
+        public bool SetGeometry(uint cylinders, uint heads, uint sectorsPerTrack)
+        {
+            // Geometry is not stored in image
+            return true;
+        }
+
+        public bool WriteSectorTag(byte[] data, ulong sectorAddress, SectorTagType tag)
+        {
+            ErrorMessage = "Writing sectors with tags is not supported.";
+            return false;
+        }
+
+        public bool WriteSectorsTag(byte[] data, ulong sectorAddress, uint length, SectorTagType tag)
+        {
+            ErrorMessage = "Writing sectors with tags is not supported.";
+            return false;
+        }
     }
 }