ID

73f404d

date

2021-03-27 23:56:08+00:00

author

Alex Chan <alex@alexwlchan.net>

parent

e4dcb98

message

Add a parameter for setting the maximum rotation

changed files

1 file, 6 additions, 1 deletion

• happybackgrounds.py

Changed files

happybackgrounds.py (9707 → 9891)

diff --git a/happybackgrounds.py b/happybackgrounds.py
index b14f894..5a177b9 100755
--- a/happybackgrounds.py
+++ b/happybackgrounds.py
@@ -136,6 +136,7 @@ def create_svg(
     max_icon_count,
     min_scale,
     max_scale,
+    max_rotation,
     avoid_center=False,
     out_path=None,
 ):
@@ -193,10 +194,12 @@ def create_svg(
     ):
         scale = random.uniform(min_scale, max_scale)
 
-        rotation_angle = random.randint(0, 360)
+        rotation_angle = random.randint(-max_rotation, max_rotation)
         rotation_center_x = scale * icon_width // 2
         rotation_center_y = scale * icon_height // 2
 
+        scale = random.uniform(min_scale, max_scale)
+
         # icon_blur = random.randint(0, 20)
         icon_blur = 0
 
@@ -243,6 +246,8 @@ def parse_args():
     parser.add_argument("--max_icon_count", type=int, default=30, help="(default: 30)")
     parser.add_argument("--min_scale", type=float, default=0.15, help="(default: 0.15)")
     parser.add_argument("--max_scale", type=float, default=0.3, help="(default: 0.3)")
+    parser.add_argument("--max_rotation", type=float, default=180, help="(default: 180)")
+
     parser.add_argument("--avoid_center", action="store_true")
 
     # Make a consideration for British people ;-)