changeset 33:c9cc6f9f3bad

Progress on the lug list part of key list generation.
author Brian Neal <bgneal@gmail.com>
date Fri, 21 Jun 2013 20:20:43 -0500
parents 70673126c158
children 502b6bbcfefb
files m209/keylist/generate.py m209/keylist/tests/test_generate.py m209/main.py
diffstat 3 files changed, 140 insertions(+), 12 deletions(-) [+]
line wrap: on
line diff
--- a/m209/keylist/generate.py	Fri Jun 21 16:33:06 2013 -0500
+++ b/m209/keylist/generate.py	Fri Jun 21 20:20:43 2013 -0500
@@ -87,22 +87,24 @@
     group = GROUP_A if rn > 10 else GROUP_B
     selection = random.choice(group)
 
+    logger.debug("Selecting from group %s", 'A' if group is GROUP_A else 'B')
+    logger.debug("Selection: %s", selection)
+
     # 2b: Rearrange the numbers so they appear in a random order
     random.shuffle(selection)
+    logger.debug("Shuffled selection: %s", selection)
 
     # 2c: Distribution of Overlaps
     overlap = sum(selection) - 27
+    logger.debug("Overlap: %d", overlap)
 
-    # TODO:
-    # (1) Most of the six number should be involved
-    # (2) Overlaps should include numbers which are separated, and numbers which
-    # are side by side.
-    # (3) Several small overlaps should be used in preference to one large
-    # overlap
-    # (4) There must not be more than four overlaps between any two numbers.
-
-    # 2d: Checking placement of overlaps
-    # TODO
+    for n in range(MAX_ATTEMPTS):
+        overlaps = distribute_overlaps(selection, overlap)
+        if overlaps and check_overlaps(overlaps):
+            break
+    else:
+        raise M209Error("generate_lugs: too many attempts")
+    logger.info("Lugs generated in %s iteration(s)", n + 1)
 
     # Build lug string
     # TODO
@@ -110,6 +112,105 @@
     return ''
 
 
+def distribute_overlaps(selection, overlap):
+    """Distributes the overlaps over the selection and returns an overlap list.
+
+    If successful, return a list consisting of 3-tuples of the form (position1,
+    position2, overlap). If unable to come up with a solution, an empty list is
+    returned.
+
+    """
+
+    # The remaining list keeps track of how many lugs are remaining in each
+    # position:
+    remaining = list(selection)
+
+    # Generate all combinations of ways to pick 2 positions in random order
+    combs = list(itertools.combinations(range(0, 6), 2))
+    random.shuffle(combs)
+
+    # chunk_limit enforces the rules:
+    # (3) Several small overlaps should be used in preference to one large
+    # overlap.
+    # (4) There must not be more than four overlaps between any two numbers.
+    if 1 <= overlap <= 3:
+        divisor = 1
+    elif 4 <= overlap <= 8:
+        divisor = 2
+    else:
+        divisor = 3
+    chunk_limit = max(1, min(4, overlap // divisor))
+    logger.debug("chunk_limit: %d", chunk_limit)
+
+    overlaps = []
+    for c in combs:
+        if overlap <= 0:
+            break
+        x, y = c
+        # Figure out the largest 'chunk' we can remove from overlap
+        max_chunk = min(remaining[x], remaining[y], overlap, chunk_limit)
+
+        if max_chunk <= 0:
+            continue
+
+        chunk = random.randint(1, max_chunk)
+
+        overlap -= chunk
+        remaining[x] -= chunk
+        remaining[y] -= chunk
+
+        overlaps.append((x, y, chunk))
+
+    overlaps.sort()
+
+    logger.debug("Overlap: %d", overlap)
+    logger.debug('Overlaps: %s', overlaps)
+
+    return overlaps if overlap == 0 else []
+
+
+def check_overlaps(overlaps):
+    """Checks the overlap list for the desired qualities.
+
+    Returns True if all checks pass and False otherwise.
+
+    """
+    # (1) Most of the six numbers should be involved
+    # We only check this if we have 3 or more overlaps.
+
+    if len(overlaps) >= 3:
+        numbers = set()
+        for x, y, n in overlaps:
+            numbers.add(x)
+            numbers.add(y)
+        if len(numbers) <= 3:
+            logger.debug("check_overlaps: failed most of 6 check")
+            return False
+
+    # (2) Overlaps should include numbers which are separated, and numbers which
+    # are side by side.
+    # We only check this if we have 2 or more overlaps:
+
+    if len(overlaps) >= 2:
+        for x, y, n in overlaps:
+            if y - x > 1:
+                break
+        else:
+            logger.debug("check_overlaps: failed separated check")
+            return False
+
+        for x, y, n in overlaps:
+            if y - x == 1:
+                break
+        else:
+            logger.debug("check_overlaps: failed side by side check")
+            return False
+
+    # 2d: Checking placement of overlaps
+
+    return True
+
+
 def generate_pin_list():
     """Return a random pin list based on Army procedure."""
 
--- a/m209/keylist/tests/test_generate.py	Fri Jun 21 16:33:06 2013 -0500
+++ b/m209/keylist/tests/test_generate.py	Fri Jun 21 20:20:43 2013 -0500
@@ -8,7 +8,7 @@
 import random
 import unittest
 
-from ..generate import generate_key_list, pin_list_check
+from ..generate import generate_key_list, pin_list_check, check_overlaps
 from m209.converter import M209
 from m209.data import KEY_WHEEL_DATA
 
@@ -177,3 +177,30 @@
         pin_list[5] = 'DEFHIKLM'
         self.assertFalse(pin_list_check(pin_list))
 
+
+class CheckOverlapsTestCase(unittest.TestCase):
+
+    def test_most_of_six(self):
+
+        self.assertTrue(check_overlaps([(0, 1, 2), (0, 2, 2)]))
+        self.assertTrue(check_overlaps([(0, 1, 2), (0, 2, 1), (0, 3, 1)]))
+        self.assertTrue(check_overlaps([(0, 1, 2), (1, 2, 1), (1, 3, 1), (4, 5, 1)]))
+        self.assertFalse(check_overlaps([(0, 1, 2), (0, 2, 1), (1, 2, 1)]))
+        self.assertFalse(check_overlaps([(2, 4, 2), (2, 5, 1), (4, 5, 1)]))
+
+    def test_separated(self):
+
+        self.assertTrue(check_overlaps([(0, 1, 1)]))
+        self.assertTrue(check_overlaps([(0, 2, 1)]))
+        self.assertFalse(check_overlaps([(0, 1, 1), (1, 2, 1)]))
+        self.assertFalse(check_overlaps([(0, 1, 1), (1, 2, 1), (3, 4, 1)]))
+        self.assertFalse(check_overlaps([(0, 1, 1), (1, 2, 1), (3, 4, 1), (4, 5, 1)]))
+
+    def test_side_by_side(self):
+
+        self.assertTrue(check_overlaps([(0, 1, 1)]))
+        self.assertTrue(check_overlaps([(0, 2, 1)]))
+        self.assertFalse(check_overlaps([(0, 2, 1), (1, 3, 1)]))
+        self.assertFalse(check_overlaps([(0, 2, 1), (1, 3, 1), (2, 4, 1)]))
+        self.assertFalse(check_overlaps([(0, 2, 1), (1, 3, 1), (2, 4, 1), (2, 5, 1)]))
+
--- a/m209/main.py	Fri Jun 21 16:33:06 2013 -0500
+++ b/m209/main.py	Fri Jun 21 20:20:43 2013 -0500
@@ -33,7 +33,7 @@
     """Key list generation subcommand processor"""
     print('Creating key list!', args)
     if not valid_indicator(args.keylist_indicator):
-        sys.exit("Invalid key list indicator\n")
+        sys.exit("Invalid key list indicator %s\n" % args.keylist_indicator)
 
     print(generate_key_list(args.keylist_indicator))