correct rotation (2)

diff --git a/function.c b/function.c
index d3439201148a328c71015ba1d7b27eb9d31551e7..4cff58e8921f91b5848acd973234cfebbca19ae2 100644 (file)
--- a/function.c
+++ b/function.c
@@ -238,12 +238,8 @@ void rotateelement (board_t *element, int rot)
 
     rot = (rot > 0) ? rot % 4 : ((1 - rot / 4) * 4 + rot) % 4;
 
-    board_t *newelement = NULL;
-    if (rot % 2 == 0) {
-        newelement = initboard (element->xsize, element->ysize);
-    } else {
-        newelement = initboard (element->ysize, element->xsize);
-    }
+    int maxsize = (element->xsize > element->ysize) ? element->xsize : element->ysize;
+    board_t *newelement = initboard (maxsize, maxsize);
 
     switch (rot) {
     case 0: