def _get_cylinder(self, start, end, radius): texture = vp.Texture( vp.Pigment('color', [1 / 255. * e for e in (0, 125, 255)])) c1 = vp.Sphere(start, radius, texture) c2 = vp.Sphere(end, radius, texture) c3 = vp.Cylinder(start, end, radius, texture) return vp.Union(c1, c2, c3)
def _get_cylinder(self, start, end, radius): color = (40 / 255., 40 / 255., 40 / 255.) texture = vp.Texture(vp.Pigment('color', color), vp.Finish("ambient", color, "diffuse", 0.0) ) c1 = vp.Sphere(start, radius, texture) c2 = vp.Sphere(end, radius, texture) c3 = vp.Cylinder(start, end, radius, texture) return vp.Union(c1, c2, c3)
def __init__(self, filename=None, ratio=2., lift=0, simple=False): texture = vp.Texture( vp.Pigment('color', [0.1 * e for e in [0.20, 0.20, 0.20]]), vp.Finish("ambient", 1.0, "diffuse", 0.0)) cylinder = vp.Cylinder([0, 0, 0], [0, 2.5, 0], 0.02, texture) texture = vp.Texture( "" if simple else "uv_mapping", vp.Pigment(vp.ImageMap('png', filename, 'once')), vp.Finish("ambient", 1.0)) if filename is not None else vp.Texture( vp.Pigment('color', [0.9 * e for e in [1.0, 1.00, 1.00]]), vp.Finish("ambient", 1.0, "diffuse", 0.0)) sheet = vp.Polygon(5, [0, 0, 0], [0, 1, 0], [1, 1, 0], [1, 0, 0], [0, 0, 0], texture, "scale", [ratio * 0.5, 0.5, 0], "translate", [0, 2, 0]) if simple else Parametric( self._get_parametric_body(), texture, "scale", [ratio * 0.5, 0.5, 0], "translate", [0, 2, 0]) # self.flag = vp.Union(cylinder, sheet) self.flag = self.flag.add_args( ["rotate", [0, 90, 0], "translate", [0, lift, 0]])
def geometry_to_povray(appearance, geometry, object, transform, properties): if get_property_yes(properties, [geometry, object], "hide"): return [] #analyze appearance tex = get_property(properties, [geometry, object], 'texture') if tex is None: tex_params = [] #pigment pigment = get_property(properties, [geometry, object], 'pigment') if pigment is None: transmit = 1. - appearance.getColor()[3] pigment = vp.Pigment(*[ 'color', list(appearance.getColor())[0:3], 'transmit', get_property(properties, [geometry, object], 'ambient', transmit) ]) tex_params.append(pigment) #finish finish = get_property(properties, [geometry, object], 'finish') if finish is None: finish=vp.Finish(*['ambient',get_property(properties,[geometry,object],'ambient',0.2), \ 'diffuse',get_property(properties,[geometry,object],'diffuse',0.7), \ 'phong',get_property(properties,[geometry,object],'phong',1.), \ 'phong_size',get_property(properties,[geometry,object],'phong_size',50)]) tex_params.append(finish) #normal normal = get_property(properties, [geometry, object], 'normal') if normal is not None: tex_params.append(normal) #texture tex = vp.Texture(*tex_params) #create geometry ret = [] if transform is None: transform = geometry.getCurrentTransform() else: transform = se3.mul(transform, geometry.getCurrentTransform()) if geometry.type() == "GeometricPrimitive": prim = geometry.getGeometricPrimitive() if get_property_yes(properties, [prim, geometry, object], "hide"): return ret if prim.type == "Point": rad = get_property(properties, [prim, geometry, object], "radius") if rad is not None: mesh_param = [se3.apply(transform, prim.properties[0:3]), rad] mesh_param.append(tex) mesh = vp.Sphere(*mesh_param) ret.append(mesh) elif prim.type == "Sphere": mesh_param = [ se3.apply(transform, prim.properties[0:3]), prim.properties[3] ] mesh_param.append(tex) mesh = vp.Sphere(*mesh_param) ret.append(mesh) elif prim.type == "Segment": rad = get_property(properties, [prim, geometry, object], "radius") if rad is not None: mesh_param = [ se3.apply(transform, prim.properties[0:3]), se3.apply(transform, prim.properties[3:6]), rad ] mesh_param.append(tex) mesh = vp.Cylinder(*mesh_param) ret.append(mesh) elif prim.type == "AABB": mesh_param = [ se3.apply(transform, prim.properties[0:3]), se3.apply(transform, prim.properties[3:6]) ] mesh_param.append(tex) mesh = vp.Box(*mesh_param) ret.append(mesh) elif geometry.type() == "Group": for idElem in range(geometry.numElements()): elem = geometry.getElement(idElem) elem.getCurrentTransform() ret += geometry_to_povray(appearance=appearance, geometry=elem, object=object, transform=transform, properties=properties) elif geometry.type() == "TriangleMesh": tm = geometry.getTriangleMesh() if get_property_yes(properties, [geometry, object], "smooth"): vss = [ se3.apply(transform, tuple(tm.vertices[i * 3:i * 3 + 3])) for i in range(len(tm.vertices) // 3) ] iss = [ tuple(tm.indices[i * 3:i * 3 + 3]) for i in range(len(tm.indices) // 3) ] mesh_param = [ vp.VertexVectors(*([len(vss)] + vss)), vp.FaceIndices(*([len(iss)] + iss)) ] mesh_param.append(tex) mesh = vp.Mesh2(*mesh_param) else: vss = [ se3.apply(transform, tuple(tm.vertices[i * 3:i * 3 + 3])) for i in range(len(tm.vertices) // 3) ] iss = [ tuple(tm.indices[i * 3:i * 3 + 3]) for i in range(len(tm.indices) // 3) ] mesh_param = [ vp.Triangle(vss[it[0]], vss[it[1]], vss[it[2]]) for it in iss ] mesh_param.append(tex) mesh = vp.Mesh(*mesh_param) ret.append(mesh) elif geometry.type() == "VolumeGrid": from skimage import measure import numpy as np grid = geometry.getVolumeGrid() volume = np.reshape(np.array(list(grid.values)), tuple(grid.dims)) spacing = [ (b - a) / d for a, b, d in zip(grid.bbox[0:3], grid.bbox[3:6], grid.dims[0:3]) ] vss, iss, nss, _ = measure.marching_cubes_lewiner(volume, level=0., spacing=spacing) vss += np.expand_dims(np.array(grid.bbox[0:3]).T, 0) vss = [vss[it, :].tolist() for it in range(vss.shape[0])] iss = [iss[it, :].tolist() for it in range(iss.shape[0])] nss = [nss[it, :].tolist() for it in range(nss.shape[0])] mesh_param = [ vp.VertexVectors(*([len(vss)] + vss)), vp.NormalVectors(*([len(nss)] + nss)), vp.FaceIndices(*([len(iss)] + iss)) ] mesh_param.append(tex) mesh = vp.Mesh2(*mesh_param) ret.append(mesh) elif geometry.type() == "PointCloud": cloud_param = [] cloud = geometry.getPointCloud() rad = get_property(properties, [cloud, geometry, object], "radius") for id in range(len(cloud.vertices) // 3): cloud_param.append( vp.Sphere(cloud.vertices[id * 3:id * 3 + 3], rad)) cloud_param.append(tex) mesh = vp.Union(*cloud_param) ret.append(mesh) else: print("Geometry (name=%s) type: %s not supported!" % (object.getName(), geometry.type())) return ret
def _get_cylinder(self, start, end, radius, texture): c1 = vp.Sphere(start, radius, texture) c2 = vp.Sphere(end, radius, texture) c3 = vp.Cylinder(start, end, radius, texture) return vp.Union(c1, c2, c3)
def _get_body(self, texture): return vp.Cylinder([0, 0, 0], [0, .2, 0], .1, texture, "scale", [1, 2, 2])
def trav(start, branch, r, objects, t, iter): end = branch[0] # print(end[3]) # print(start, end) cylinder = vp.Cone( [start[0], start[2], start[1]], end[3], [end[0], end[2], end[1]], end[3], vp.Texture(vp.Pigment('color', [0.7 / 255. * e for e in [98, 78, 44]])), vp.Finish('ambient', [0.7 / 255. * e for e in [98, 78, 44]], "diffuse", 0.1)) top = vp.Sphere( [end[0], end[2], end[1]], end[3], vp.Texture(vp.Pigment('color', [.7 / 255. * e for e in [98, 78, 44]])), vp.Finish('ambient', [0.7 / 255. * e for e in [98, 78, 44]], "diffuse", 0.1)) for i in range(0, int(t * 450)): (dx, dy, dz) = Vector3D.fromAngles( random.randint(0, 180), random.randint(0, 360), t * random.randint(0, 5) / (5. * (iter + 2) if iter != 0 else 1000)) # dx = random.randint(-5, 5) / 30. # dy = random.randint(-5, 5) / 30. # dz = random.randint(-5, 5) / 30. dr = random.randint(-5, 5) / 30. dg = random.randint(-5, 5) / 30. db = random.randint(-5, 5) / 30. ddx = random.randint(1, 10) / 10000. ddy = random.randint(0, 10) / 10000. ddz = random.randint(0, 10) / 10000. color = (random.randint(40, 180), random.randint(120, 250), random.randint(0, 50)) leave = vp.Cylinder( [end[0] + dx, end[2] + dy, end[1] + dz], [end[0] + dx + ddx, end[2] + dy + ddy, end[1] + dz + ddz], .01 + random.randint(0, 10) / 1000., vp.Texture(vp.Pigment('color', [2 / 255. * e for e in color])), vp.Finish('phong', 1)) objects.append( leave.add_args(["scale", [2, 2, 2], "translate", [0, 0.58, 0]])) for i in range(0, int(random.randint(0, 10) / 5)): (dx, dy, dz) = Vector3D.fromAngles( random.randint(90, 180), random.randint(0, 360), t * random.randint(4, 4) / (5 * (iter + 2) if iter != 0 else 1000)) color = (random.randint(190, 250), random.randint(40, 80), random.randint(40, 80)) fruit = vp.Sphere([end[0] + dx, end[2] + dz, end[1] + dy], 0.04, vp.Texture( vp.Pigment('color', [2 / 255. * e for e in color]))) #objects.append(fruit.add_args(["scale", [2, 2, 2], "translate", [0, 0.58, 0]])) objects.append( cylinder.add_args(["scale", [2, 2, 2], "translate", [0, 0.58, 0]])) objects.append( top.add_args(["scale", [2, 2, 2], "translate", [0, 0.58, 0]])) for branch in branch[1:]: trav(end, branch, r - (0.005 if r != 0.05 else 0.02), objects, t, iter + 1)
def get_rounded_cylinder(self, start, end, radius, texture): cylinder = vapory.Cylinder(start, end, radius, texture) tip_1 = vapory.Sphere(start, radius, texture) tip_2 = vapory.Sphere(end, radius, texture) return vapory.Union(cylinder, tip_1, tip_2)
def _get_cylinder(self, start, end, radius): c1 = vp.Sphere(start, radius, vp.Texture("Brass_Metal")) c2 = vp.Sphere(end, radius, vp.Texture("Brass_Metal")) c3 = vp.Cylinder(start, end, radius, vp.Texture("Brass_Metal")) return vp.Union(c1, c2, c3)
def _get_cylinder(self, p1, p2, radius, color) -> vp.Cylinder: return vp.Cylinder( p1, p2, radius, vp.Texture(vp.Pigment('color', [1. / 255. * e for e in color])), vp.Finish("ambient", 1.0, "diffuse", 0.0))