Added test generation tools using franny to create disk images

test/create_binary.py

@@ -0,0 +1,89 @@
+#!/usr/bin/env python
+"""
+
+"""
+
+import os
+import numpy as np
+import subprocess
+import tempfile
+
+def create_binary(filename, num, outfile, options):
+    """Create patterned binary data with the filename interleaved with a byte
+    ramp, e.g.  A128\x00A128\x01A128\x02 etc.
+    """
+    root, _ = outfile.split(".")
+    a = np.fromstring(root + " ", dtype=np.uint8)
+    b = np.tile(a, (num / np.alen(a)) + 1)[0:num]
+    e = np.where(b == 32)[0]
+    b[e] = np.arange(np.alen(e), dtype=np.uint8)
+    with open(filename, "wb") as fh:
+        fh.write(b.tostring())
+
+def num_to_letter(num):
+    text = []
+    while True:
+        num, rem = divmod(num, 26)
+        text[0:0] = chr(ord("A") + rem)
+        if num == 0:
+            break
+    return "".join(text)
+
+def get_filename(num, used):
+    if num not in used:
+        used[num] = 0
+    prefix = num_to_letter(used[num])
+    filename = "%s%d.DAT" % (prefix, num)
+    used[num] += 1
+    return filename
+
+def franny(filename, *options):
+    cmd = ["franny", filename]
+    cmd.extend(options)
+    msg = subprocess.Popen(cmd, stdout=subprocess.PIPE).communicate()[0]
+
+if __name__ == "__main__":
+    import sys
+    import argparse
+    
+    parser = argparse.ArgumentParser(description="Extract images off ATR format disks")
+    parser.add_argument("-v", "--verbose", default=0, action="count")
+    parser.add_argument("-p", "--prefix", default="B", action="store")
+    parser.add_argument("-f", "--filesystem", default="a", action="store")
+    parser.add_argument("-d", "--density", default="s", action="store")
+    parser.add_argument("-s", "--sectors", default=720, type=int, action="store")
+    parser.add_argument("-t", "--template", default="", action="store")
+    parser.add_argument("-o", "--output", default="a.out", action="store")
+    parser.add_argument("files", metavar="NUM", nargs="+", help="list of numbers specifying the size of each binary file")
+    options, extra_args = parser.parse_known_args()
+    
+    out = options.output
+    if options.template:
+        franny(out, "-C", "-f", options.filesystem, "-t", options.template)
+    else:
+        franny(out, "-C", "-f", options.filesystem, "-d", options.density, "-s", str(options.sectors))
+    fh, binfile = tempfile.mkstemp()
+    os.close(fh)
+    used = {}
+    files = []
+    print "Creating image %s" % out
+    for entry in options.files:
+        entry = entry.lower()
+        if entry.startswith("d"):
+            index = int(entry[1:])
+            outfile = files[index]
+            franny(out, "-U", outfile)
+        else:
+            if "*" in entry:
+                repeat, num = entry.split("*", 1)
+                num = int(num)
+                repeat = int(repeat)
+            else:
+                num = int(entry)
+                repeat = 1
+            for i in range(repeat):
+                outfile = get_filename(num, used)
+                create_binary(binfile, num, outfile, options)
+                franny(out, "-A", "-i", binfile, "-o", outfile)
+                files.append(outfile)
+    os.unlink(binfile)

test/rebuild.sh

@@ -0,0 +1,16 @@
+
+
+function create() {
+    ROOT=$1
+    ARGS=$2
+    ./create_binary.py $ARGS -o ${ROOT}test1.atr 128 256 512 1024 4096
+    ./create_binary.py $ARGS -o ${ROOT}test2.atr 50*256 2*512 256 1024 256 d1 d2 d3 4096
+    ./create_binary.py $ARGS -o ${ROOT}test3.atr 10*256 d1 d7 4096 d3 d5 d9 8000
+    ./create_binary.py $ARGS -o ${ROOT}test4.atr 10*4096 d1 d3 d5 d7 d9
+}
+
+create "dos_sd_" "-f a -t s"
+create "dos_ed_" "-f a -t m"
+create "dos_dd_" "-f a -t d"
+create "sd_sd_" "-f s -t s"
+create "sd_dd_" "-f s -t d"