FilesExpand file tree

 master

/

boxes.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

51 lines (46 loc) · 2.02 KB

 master

/

boxes.py

Copy path

Top

File metadata and controls

• Code
• Blame

51 lines (46 loc) · 2.02 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

import random

class BoxGenerator:

"""

A class to generate boxes with specified dimensions and constraints.

"""

def __init__(self, min_dim=11, split_dim_min=5):

self.min_dim = min_dim

self.split_dim_min = split_dim_min

def generate_boxes(self, container, num):

"""

Generates a list of boxes within a container based on the specified conditions.

Parameters:

- container (list): A list of cuboids where each cuboid is represented by six integers.

- num (int): The number of boxes to generate.

Returns:

- A list of generated boxes with their dimensions.

"""

retry = 500 * num

while num > 1:

cuboid = random.choice(container)

while cuboid[3] <= self.min_dim or cuboid[4] <= self.min_dim or cuboid[5] <= self.min_dim:

retry -= 1

if retry == 0:

print("Cannot partition into packages. Please try again")

return

cuboid = random.choice(container)

container.remove(cuboid)

prob = random.uniform(0, 1)

x1, y1, z1, x2, y2, z2 = cuboid

# Splitting the cuboid based on the random probability generated

if prob < 0.35:

t = random.randint(self.split_dim_min, int(x2 / 2))

package1 = [x1 + t, y1, z1, x2 - t, y2, z2]

package2 = [x1, y1, z1, t, y2, z2]

elif prob < 0.65:

t = random.randint(self.split_dim_min, int(y2 / 2))

package1 = [x1, y1 + t, z1, x2, y2 - t, z2]

package2 = [x1, y1, z1, x2, t, z2]

else:

t = random.randint(self.split_dim_min, int(z2 / 2))

package1 = [x1, y1, z1 + t, x2, y2, z2 - t]

package2 = [x1, y1, z1, x2, y2, t]

container.append(package1)

container.append(package2)

num -= 1

return container