From 0b0415060a483ae5ffa5f8c86667ccfe747192bc Mon Sep 17 00:00:00 2001
From: josch <j.schauer@email.de>
Date: Tue, 15 May 2012 12:39:11 +0200
Subject: [PATCH] selective rotation of articles and pallet through environment
 variables

---
 bruteforce2.py | 51 ++++++++++++++++++++++++++++++++++++++++++++++----
 1 file changed, 47 insertions(+), 4 deletions(-)

diff --git a/bruteforce2.py b/bruteforce2.py
index d3d1fdf..9f4931d 100644
--- a/bruteforce2.py
+++ b/bruteforce2.py
@@ -5,6 +5,7 @@ from arrange_spread2 import arrange_in_layer, spread_articles
 import cPickle
 from binascii import b2a_base64
 import zlib
+import os
 
 def rotate(node):
     if node is None:
@@ -71,14 +72,47 @@ def main():
         else:
             bins[article['Article']['Height']] = [article]
 
-    product_it = product_varlength(4)
+    if os.environ.get("rot_article"):
+        try_rot_article = bool(int(os.environ["rot_article"]))
+    else:
+        try_rot_article = True
+
+    if os.environ.get("rot_pallet"):
+        try_rot_pallet = bool(int(os.environ["rot_pallet"]))
+    else:
+        try_rot_pallet = True
+
+    if try_rot_article and try_rot_pallet:
+        product_it = product_varlength(4)
+    elif try_rot_article or try_rot_pallet:
+        product_it = product_varlength(2)
     while True:
         rests = list()
         layers = list()
         try:
-            rot_article, rot_pallet = get_bitmask(product_it.send(True), 2) # start a new combination
+            if try_rot_article and try_rot_pallet:
+                rot_article, rot_pallet = get_bitmask(product_it.send(True), 2)
+            elif try_rot_article and not try_rot_pallet:
+                rot_article = get_bitmask(product_it.send(True), 1)[0]
+                rot_pallet = False
+            elif not try_rot_article and try_rot_pallet:
+                rot_article = False
+                rot_pallet = get_bitmask(product_it.send(True), 1)[0]
+            else:
+                rot_article = False
+                rot_pallet = False
         except TypeError:
-            rot_article, rot_pallet = get_bitmask(product_it.next(), 2) # can't send to a just-started generator
+            if try_rot_article and try_rot_pallet:
+                rot_article, rot_pallet = get_bitmask(product_it.next(), 2)
+            elif try_rot_article and not try_rot_pallet:
+                rot_article = get_bitmask(product_it.next(), 1)[0]
+                rot_pallet = False
+            elif not try_rot_article and try_rot_pallet:
+                rot_article = False
+                rot_pallet = get_bitmask(product_it.next(), 1)[0]
+            else:
+                rot_article = False
+                rot_pallet = False
         except StopIteration:
             break # generator empty
         it = get_layers(bins, pallet, rot_article, rot_pallet)
@@ -90,7 +124,14 @@ def main():
 
         while True:
             try:
-                layer, rest = it.send(get_bitmask(product_it.send(False), 2))
+                if try_rot_article and try_rot_pallet:
+                    layer, rest = it.send(get_bitmask(product_it.send(False), 2))
+                elif try_rot_article and not try_rot_pallet:
+                    layer, rest = it.send((get_bitmask(product_it.send(False), 1)[0], False))
+                elif not try_rot_article and try_rot_pallet:
+                    layer, rest = it.send((False, get_bitmask(product_it.send(False), 1)[0]))
+                else:
+                    layer, rest = it.send((False, False))
                 if layer:
                     layers.append(layer)
                 if rest:
@@ -98,5 +139,7 @@ def main():
             except StopIteration:
                 break
         print b2a_base64(zlib.compress(cPickle.dumps((layers, rests, pallet)))),
+        if not try_rot_article and not try_rot_pallet:
+            break # only one iteration if both are deactivated
 if __name__ == "__main__":
     main()