зеркало из https://github.com/microsoft/scenepic.git
61 строка
1.8 KiB
Python
61 строка
1.8 KiB
Python
|
"""Test script for dropdown CSS"""
|
||
|
|
|
||
|
|
import random
|
||
|
|
from typing import List
|
||
|
|
import numpy as np
|
||
|
|
import scenepic as sp
|
||
|
|
|
||
|
|
|
||
|
|
def _create_cubes(scene: sp.Scene, num_cubes: int) -> List[sp.Mesh]:
|
||
|
|
meshes = []
|
||
|
|
for i in range(num_cubes):
|
||
|
|
color = np.random.rand(3)
|
||
|
|
mesh: sp.Mesh = scene.create_mesh(layer_id="cube{}".format(i), shared_color=color)
|
||
|
|
mesh.add_cube()
|
||
|
|
meshes.append(mesh)
|
||
|
|
|
||
|
|
return meshes
|
||
|
|
|
||
|
|
|
||
|
|
def _add_objects(canvas: sp.Canvas3D, cubes: List[sp.Mesh]):
|
||
|
|
frame = canvas.create_frame()
|
||
|
|
num_cubes = np.random.randint(5, 10)
|
||
|
|
choices = np.random.choice(len(cubes), num_cubes, replace=False)
|
||
|
|
for index in choices:
|
||
|
|
pos = np.random.rand(3) * 2 - 1
|
||
|
|
frame.add_mesh(cubes[index], sp.Transforms.Translate(pos))
|
||
|
|
|
||
|
|
|
||
|
|
def _add_circles(canvas: sp.Canvas2D):
|
||
|
|
frame: sp.Frame2D = canvas.create_frame()
|
||
|
|
num_circles = np.random.randint(5, 10)
|
||
|
|
for i in range(num_circles):
|
||
|
|
pos = np.random.rand(2) * 100
|
||
|
|
radius = np.random.random() * 10
|
||
|
|
color = np.random.rand(3)
|
||
|
|
frame.add_circle(pos[0], pos[1], radius, fill_color=color, layer_id="circle{}".format(i))
|
||
|
|
|
||
|
|
|
||
|
|
def _main():
|
||
|
|
scene = sp.Scene()
|
||
|
|
|
||
|
|
cubes = _create_cubes(scene, 20)
|
||
|
|
for row in range(1, 5):
|
||
|
|
for col in range(1, 5):
|
||
|
|
canvasID = "canvas" + str(row) + str(col)
|
||
|
|
if random.random() < 0.5:
|
||
|
|
canvas = scene.create_canvas_3d(canvasID, 100, 100)
|
||
|
|
_add_objects(canvas, cubes)
|
||
|
|
else:
|
||
|
|
canvas = scene.create_canvas_2d(canvasID, 100, 100)
|
||
|
|
_add_circles(canvas)
|
||
|
|
|
||
|
|
scene.place(canvasID, str(row), str(col))
|
||
|
|
|
||
|
|
scene.grid("400px", "auto auto auto auto", "auto auto auto auto")
|
||
|
|
scene.save_as_html("overlap.html")
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
_main()
|