Browse Source

erase: New command to erase all tracks on a disk.

Keir Fraser 4 years ago
parent
commit
73d129f6fd

+ 62 - 0
scripts/greaseweazle/tools/erase.py

@@ -0,0 +1,62 @@
+# greaseweazle/tools/erase.py
+#
+# Greaseweazle control script: Erase a Disk.
+#
+# Written & released by Keir Fraser <keir.xen@gmail.com>
+#
+# This is free and unencumbered software released into the public domain.
+# See the file COPYING for more details, or visit <http://unlicense.org>.
+
+import sys, argparse
+
+from greaseweazle.tools import util
+from greaseweazle import usb as USB
+
+def erase(usb, args):
+
+    # @drive_ticks is the time in Greaseweazle ticks between index pulses.
+    # We will adjust the flux intervals per track to allow for this.
+    flux = usb.read_track(2)
+    drive_ticks = (flux.index_list[0] + flux.index_list[1]) / 2
+    del flux
+
+    for cyl in range(args.scyl, args.ecyl+1):
+        for side in range(0, args.nr_sides):
+            print("\rErasing Track %u.%u..." % (cyl, side), end="")
+            usb.seek(cyl, side)
+            usb.erase_track(drive_ticks * 1.1)
+
+    print()
+
+
+def main(argv):
+
+    parser = argparse.ArgumentParser(
+        formatter_class=argparse.ArgumentDefaultsHelpFormatter)
+    parser.add_argument("--drive", type=util.drive_letter, default='A',
+                        help="drive to write (A,B,0,1,2)")
+    parser.add_argument("--scyl", type=int, default=0,
+                        help="first cylinder to write")
+    parser.add_argument("--ecyl", type=int, default=81,
+                        help="last cylinder to write")
+    parser.add_argument("--single-sided", action="store_true",
+                        help="single-sided write")
+    parser.add_argument("device", nargs="?", default="auto",
+                        help="serial device")
+    parser.prog += ' ' + argv[1]
+    args = parser.parse_args(argv[2:])
+    args.nr_sides = 1 if args.single_sided else 2
+
+    try:
+        usb = util.usb_open(args.device)
+        util.with_drive_selected(erase, usb, args)
+    except USB.CmdError as error:
+        print("Command Failed: %s" % error)
+
+
+if __name__ == "__main__":
+    main(sys.argv)
+
+# Local variables:
+# python-indent: 4
+# End:

+ 1 - 1
scripts/greaseweazle/tools/reset.py

@@ -1,4 +1,4 @@
-# greaseweazle/tools/rset.py
+# greaseweazle/tools/reset.py
 #
 # Greaseweazle control script: Reset to power-on defaults.
 #

+ 1 - 1
scripts/greaseweazle/tools/write.py

@@ -16,7 +16,7 @@ from greaseweazle import usb as USB
 # Writes the specified image file to floppy disk.
 def write_from_image(usb, args):
 
-    # @drive_ticks is the time in Gresaeweazle ticks between index pulses.
+    # @drive_ticks is the time in Greaseweazle ticks between index pulses.
     # We will adjust the flux intervals per track to allow for this.
     flux = usb.read_track(2)
     drive_ticks = (flux.index_list[0] + flux.index_list[1]) / 2

+ 10 - 2
scripts/gw.py

@@ -38,13 +38,21 @@ For installation instructions please read the wiki:
           % ', '.join(missing_modules))
     sys.exit(1)
 
-actions = [ 'read', 'write', 'delays', 'update', 'pin', 'reset', 'bandwidth' ]
+actions = [ 'read',
+            'write',
+            'erase',
+            'delays',
+            'update',
+            'pin',
+            'reset',
+            'bandwidth' ]
 argv = sys.argv
 
 if len(argv) < 2 or argv[1] not in actions:
-    print("Usage: %s [action] ..." % (argv[0]))
+    print("Usage: %s [action] [-h] ..." % (argv[0]))
     print("Actions: ", end="")
     print(", ".join(str(x) for x in actions))
+    print("  -h, --help  Show help message for specified action")
     sys.exit(1)
 
 mod = importlib.import_module('greaseweazle.tools.' + argv[1])