Add simple benchmark

.github/workflows/ci.yaml

@@ -107,3 +107,7 @@ jobs:
     # example
     - name: Run example/example.py
       run: python example/example.py
+
+    # benchmark
+    - name: Run example/benchmark.py
+      run: python example/benchmark.py

README.md

@@ -30,7 +30,6 @@ print("problem:", problem)
 
 # Get a solver
 solver = rps.Solver()
-print("solving...")
 
 # Find a solution
 solution = solver.solve(problem)
@@ -40,7 +39,6 @@ print("solution:", solution)
 Output:
 ```plaintext
 problem: Problem({'n': 4, 'rectangles': [{'id': 0, 'width': 4, 'height': 6, 'rotatable': False}, {'id': 1, 'width': 4, 'height': 4, 'rotatable': False}, {'id': 2, 'width': 2.1, 'height': 3.2, 'rotatable': False}, {'id': 3, 'width': 1, 'height': 5, 'rotatable': True}]})
-solving...
 solution: Solution({'sequence_pair': SequencePair(([0, 1, 3, 2], [3, 0, 2, 1])), 'floorplan': Floorplan({'positions': [{'id': 0, 'x': 0, 'y': 1}, {'id': 1, 'x': 4, 'y': 3.2}, {'id': 2, 'x': 5.0, 'y': 0.0}, {'id': 3, 'x': 0, 'y': 0}], 'bounding_box': (8, 7.2), 'area': 57.6})})
 ```
 

example/benchmark.py

@@ -0,0 +1,54 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+
+# Copyright 2021 Kotaro Terada
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+import random
+import time
+
+import rectangle_packing_solver as rps
+
+
+def generate_rectangles(n=10, range_width=(1, 100), range_height=(1, 100)):
+    rectangles = []
+    for i in range(n):
+        rectangles.append((random.randint(range_width[0], range_width[1]), random.randint(range_height[0], range_height[1])))
+    return rectangles
+
+
+def main():
+    random.seed(12345)
+
+    n_rectangles = list(range(4, 10)) + list(range(10, 50, 10))
+    for n in n_rectangles:
+        print("\n================")
+        print("N =", n)
+
+        rectangles = generate_rectangles(n=n)
+        problem = rps.Problem(rectangles=rectangles)
+        print("problem:", problem)
+
+        solver = rps.Solver()
+
+        start = time.time()
+        solution = solver.solve(problem, simanneal_minutes=0.1, simanneal_steps=100)
+        elapsed_time = time.time() - start
+        print("solution:", solution)
+        print(solution.floorplan.bounding_box, solution.floorplan.area)
+        print("elapsed_time:", elapsed_time)
+
+
+if __name__ == "__main__":
+    main()

example/example.py

@@ -32,7 +32,6 @@ def main():
 
     # Get a solver
     solver = rps.Solver()
-    print("solving...")
 
     # Find a solution
     solution = solver.solve(problem)