def make_fov(sweep=90, steps=16, scale=100):
z = 1 + random.uniform(-0.01, 0.01)
data = EggData()
vp = EggVertexPool('fan')
data.addChild(vp)
poly = EggPolygon()
data.addChild(poly)
v = EggVertex()
v.setPos(Point3D(0, 0, z))
poly.addVertex(vp.addVertex(v))
rads = deg2Rad(sweep)
for i in range(steps + 1):
a = rads * i / steps
y = math.sin(a)
x = math.cos(a)
v = EggVertex()
v.setPos(Point3D(x*scale, y*scale, z))
poly.addVertex(vp.addVertex(v))
node = loadEggData(data)
np = NodePath(node)
np.setH(sweep/2)
return np
def createGizmo(self, angleDegrees=360, numSteps=16, axis=0, scale=10):
data = EggData()
vp = EggVertexPool('fan')
data.addChild(vp)
poly = EggPolygon()
data.addChild(poly)
v = EggVertex()
v.setPos(Point3D(0, 0, 0))
poly.addVertex(vp.addVertex(v))
angleRadians = deg2Rad(angleDegrees)
for i in range(numSteps + 1):
a = angleRadians * i / numSteps
y = math.sin(a) * scale
x = math.cos(a) * scale
v = EggVertex()
if axis is 0:
v.setPos(Point3D(x, 0, y))
elif axis is 1:
v.setPos(Point3D(x, y, 0))
else:
v.setPos(Point3D(0, x, y))
poly.addVertex(vp.addVertex(v))
node = loadEggData(data)
return NodePath(node)
def getNodepath(self):
myCurve = EggNurbsCurve()
myCurve.setup(3, len(self.myverts) + 3)
myCurve.setCurveType(1)
for i in self.myverts:
myCurve.addVertex(i)
self.eggGroup.addChild(myCurve)
return NodePath(loadEggData(self.data))
def DisplacementUVSphere(radius,
heightmap,
scale,
rings=5,
sectors=5,
inv_texture_u=False,
inv_texture_v=True):
data = EggData()
pool = EggVertexPool('pool')
vertices = []
data.addChild(pool)
R = 1. / (rings)
S = 1. / (sectors)
for r in range(0, rings + 1):
for s in range(0, sectors + 1):
cos_s = cos(2 * pi * s * S + pi)
sin_s = sin(2 * pi * s * S + pi)
sin_r = sin(pi * r * R)
cos_r = cos(pi * r * R)
x = cos_s * sin_r
y = sin_s * sin_r
z = cos_r
vertex = EggVertex()
u = s * S
v = r * R
height = radius + heightmap.get_height_uv(u, v) * scale
vertex.setPos(LPoint3d(x * height, y * height, z * height))
if inv_texture_v:
v = 1.0 - v
if inv_texture_u:
u = 1.0 - u
vertex.setUv(LPoint2d(u, v))
pool.addVertex(vertex)
vertices.append(vertex)
index = 0
for r in range(0, rings):
for s in range(0, sectors):
poly = EggPolygon()
data.addChild(poly)
poly.addVertex(vertices[index + sectors + 1])
poly.addVertex(vertices[index])
poly.addVertex(vertices[index + sectors])
poly = EggPolygon()
data.addChild(poly)
poly.addVertex(vertices[index + sectors + 1])
poly.addVertex(vertices[index + 1])
poly.addVertex(vertices[index])
index += 1
data.removeUnusedVertices(True)
data.recomputeVertexNormals(45)
data.recomputeTangentBinormal(GlobPattern(""))
node = loadEggData(data)
path = NodePath(node)
path.flattenStrong()
return path
def obj2bam(infile,
outfile,
coordinateSystem='z-up',
recomputeVertexNormals=False,
recomputeTangentBinormal=False,
recomputePolygonNormals=False,
triangulatePolygons=False,
degreeSmoothing=30.0):
if coordinateSystem == 'z-up' or coordinateSystem == 'z-up-right':
coordSys = CSZupRight
elif coordinateSystem == 'z-up-left':
coordSys = CSZupLeft
elif coordinateSystem == 'y-up' or coordinateSystem == 'y-up-right':
coordSys = CSYupRight
elif coordinateSystem == 'y-up-left':
coordSys = CSYupLeft
else:
raise Exception('Unsupported coordinate system: %s' %
(coordinateSystem))
os.chdir(os.path.dirname(infile))
obj = ObjFile(infile)
egg = obj.toEgg()
egg.setCoordinateSystem(coordSys)
egg.removeUnusedVertices(GlobPattern(""))
if recomputeVertexNormals:
egg.recomputeVertexNormals(float(degreeSmoothing))
if recomputeTangentBinormal:
egg.recomputeTangentBinormal(GlobPattern(""))
if recomputePolygonNormals:
egg.recomputePolygonNormals()
if triangulatePolygons:
egg.triangulatePolygons(EggData.TConvex & EggData.TPolygon)
np = NodePath(loadEggData(egg))
np.writeBamFile(outfile)
np.removeNode()
def make_circle(self, num_steps):
data = EggData()
vertex_pool = EggVertexPool('fan')
data.addChild(vertex_pool)
poly = EggPolygon()
data.addChild(poly)
for i in range(num_steps + 1):
angle = 2 * math.pi * i / num_steps
y = math.sin(angle) * config.minimap.size / 2
x = math.cos(angle) * config.minimap.size / 2
vertex = EggVertex()
vertex.setPos(Point3D(x, 0, y))
poly.addVertex(vertex_pool.addVertex(vertex))
node = loadEggData(data)
return node
def createCuttingPlanes(self):
self.planes = self.render.attachNewNode("planes")
pl = self.planes.attachNewNode("pl")
data = EggData()
self.planedata = []
vp = EggVertexPool('plane')
data.addChild(vp)
fac = 1.0
expanse = 10.0
rng = 4
for i in range(-rng, rng):
poly = EggPolygon()
data.addChild(poly)
self.planedata.append(i * fac)
v = EggVertex()
v.setPos(Point3D(i * fac, -expanse, -expanse))
poly.addVertex(vp.addVertex(v))
v = EggVertex()
v.setPos(Point3D(i * fac, -expanse, +expanse))
poly.addVertex(vp.addVertex(v))
v = EggVertex()
v.setPos(Point3D(i * fac, +expanse, +expanse))
poly.addVertex(vp.addVertex(v))
v = EggVertex()
v.setPos(Point3D(i * fac, +expanse, -expanse))
poly.addVertex(vp.addVertex(v))
node = loadEggData(data)
np = NodePath(node)
np.reparentTo(pl)
np.setColor(0, 1, 0, 1)
np.setTwoSided(True)
np.setTransparency(TransparencyAttrib.MAlpha)
np.setAlphaScale(0.1)
np.setCollideMask(BitMask32(0x0))
return self.planes
def getNodePath(self):
return p3d.NodePath(egg.loadEggData(self.getEggData()))
