def testSelectableFlag(self):
r = IECoreGL.Renderer()
r.setOption("gl:mode", IECore.StringData("deferred"))
r.setOption("gl:searchPath:shader",
IECore.StringData(os.path.dirname(__file__) + "/shaders"))
with IECoreScene.WorldBlock(r):
r.concatTransform(imath.M44f().translate(imath.V3f(0, 0, -5)))
r.setAttribute("name", IECore.StringData("selectableObj"))
r.shader(
"surface", "color",
{"colorValue": IECore.Color3fData(imath.Color3f(1, 0, 0))})
r.sphere(1, -1, 1, 360, {})
r.concatTransform(imath.M44f().translate(imath.V3f(-1, 0, 0)))
r.setAttribute("name", IECore.StringData("unselectableObj"))
r.setAttribute("gl:primitive:selectable", IECore.BoolData(False))
r.sphere(1, -1, 1, 360, {})
s = r.scene()
s.setCamera(IECoreGL.Camera(imath.M44f(), False))
for mode in (IECoreGL.Selector.Mode.GLSelect,
IECoreGL.Selector.Mode.OcclusionQuery,
IECoreGL.Selector.Mode.IDRender):
ss = s.select(mode, imath.Box2f(imath.V2f(0), imath.V2f(1)))
names = [
IECoreGL.NameStateComponent.nameFromGLName(x.name) for x in ss
]
self.assertEqual(names, ["selectableObj"])
def testPointsPrimitiveSelect(self):
r = IECoreGL.Renderer()
r.setOption("gl:mode", IECore.StringData("deferred"))
r.setOption("gl:searchPath:shader",
IECore.StringData(os.path.dirname(__file__) + "/shaders"))
with IECoreScene.WorldBlock(r):
r.concatTransform(imath.M44f().translate(imath.V3f(0, 0, -5)))
r.setAttribute("name", IECore.StringData("pointsNeedSelectingToo"))
r.points(
1, {
"P":
IECoreScene.PrimitiveVariable(
IECoreScene.PrimitiveVariable.Interpolation.Vertex,
IECore.V3fVectorData([imath.V3f(0)]))
})
s = r.scene()
s.setCamera(IECoreGL.Camera(imath.M44f(), False))
for mode in (IECoreGL.Selector.Mode.GLSelect,
IECoreGL.Selector.Mode.OcclusionQuery,
IECoreGL.Selector.Mode.IDRender):
ss = s.select(mode, imath.Box2f(imath.V2f(0), imath.V2f(1)))
names = [
IECoreGL.NameStateComponent.nameFromGLName(x.name) for x in ss
]
self.assertEqual(len(names), 1)
self.assertEqual(names[0], "pointsNeedSelectingToo")
def testIDSelectDepths(self):
r = IECoreGL.Renderer()
r.setOption("gl:mode", IECore.StringData("deferred"))
with IECoreScene.WorldBlock(r):
r.concatTransform(imath.M44f().translate(imath.V3f(0, 0, -5)))
r.setAttribute("name", IECore.StringData("ball"))
r.sphere(1, -1, 1, 360, {})
scene = r.scene()
scene.setCamera(IECoreGL.Camera())
s1 = scene.select(IECoreGL.Selector.Mode.GLSelect,
imath.Box2f(imath.V2f(0), imath.V2f(1)))
self.assertEqual(len(s1), 1)
self.assertEqual(
IECoreGL.NameStateComponent.nameFromGLName(s1[0].name), "ball")
s2 = scene.select(IECoreGL.Selector.Mode.IDRender,
imath.Box2f(imath.V2f(0), imath.V2f(1)))
self.assertEqual(len(s2), 1)
self.assertEqual(
IECoreGL.NameStateComponent.nameFromGLName(s2[0].name), "ball")
self.assertAlmostEqual(s1[0].depthMin, s2[0].depthMin, 5)
def testIDSelectWithCustomVertexShader(self):
r = IECoreGL.Renderer()
r.setOption("gl:mode", IECore.StringData("deferred"))
with IECoreScene.WorldBlock(r):
r.concatTransform(imath.M44f().translate(imath.V3f(0, 0, -5)))
# translate to the left with the transform stack
r.concatTransform(imath.M44f().translate(imath.V3f(-1, 0, 0)))
# but translate back to the right using a vertex shader
r.shader(
"gl:surface", "test", {
"gl:vertexSource":
inspect.cleandoc("""
#version 120
#if __VERSION__ <= 120
#define in attribute
#endif
uniform vec3 offset;
in vec3 vertexP;
void main()
{
vec3 translatedVertexP = vertexP + offset;
vec4 pCam = gl_ModelViewMatrix * vec4( translatedVertexP, 1 );
gl_Position = gl_ProjectionMatrix * pCam;
}
"""),
"offset":
imath.V3f(2, 0, 0),
})
r.setAttribute("name", IECore.StringData("sphere"))
r.sphere(1, -1, 1, 360, {})
s = r.scene()
s.setCamera(IECoreGL.Camera())
# on the left should be nothing, because the vertex shader moved it over
ss = s.select(IECoreGL.Selector.Mode.IDRender,
imath.Box2f(imath.V2f(0.25, 0.5), imath.V2f(0.26, 0.51)))
self.assertEqual(len(ss), 0)
# and on the right we should find the sphere
ss = s.select(IECoreGL.Selector.Mode.IDRender,
imath.Box2f(imath.V2f(0.75, 0.5), imath.V2f(0.76, 0.51)))
self.assertEqual(len(ss), 1)
self.assertEqual(
IECoreGL.NameStateComponent.nameFromGLName(ss[0].name), "sphere")
def testIDSelectWithAdditionalDisplayStyles(self):
r = IECoreGL.Renderer()
r.setOption("gl:mode", IECore.StringData("deferred"))
with IECoreScene.WorldBlock(r):
r.setAttribute("gl:primitive:wireframe", IECore.BoolData(True))
r.setAttribute("gl:primitive:bound", IECore.BoolData(True))
r.setAttribute("gl:primitive:outline", IECore.BoolData(True))
r.setAttribute("gl:primitive:points", IECore.BoolData(True))
r.concatTransform(imath.M44f().translate(imath.V3f(0, 0, -5)))
r.concatTransform(imath.M44f().translate(imath.V3f(-1, 0, 0)))
r.setAttribute("name", IECore.StringData("frontLeft"))
r.sphere(1, -1, 1, 360, {})
r.concatTransform(imath.M44f().translate(imath.V3f(0, 0, -1)))
r.setAttribute("name", IECore.StringData("backLeft"))
r.sphere(1, -1, 1, 360, {})
r.concatTransform(imath.M44f().translate(imath.V3f(2, 0, 1)))
r.setAttribute("name", IECore.StringData("frontRight"))
r.sphere(1, -1, 1, 360, {})
r.concatTransform(imath.M44f().translate(imath.V3f(0, 0, -1)))
r.setAttribute("name", IECore.StringData("backRight"))
r.sphere(1, -1, 1, 360, {})
s = r.scene()
s.setCamera(IECoreGL.Camera())
ss = s.select(IECoreGL.Selector.Mode.IDRender,
imath.Box2f(imath.V2f(0.25, 0.5), imath.V2f(0.26, 0.51)))
self.assertEqual(len(ss), 1)
self.assertEqual(
IECoreGL.NameStateComponent.nameFromGLName(ss[0].name),
"frontLeft")
ss = s.select(IECoreGL.Selector.Mode.IDRender,
imath.Box2f(imath.V2f(0.75, 0.5), imath.V2f(0.76, 0.51)))
self.assertEqual(len(ss), 1)
self.assertEqual(
IECoreGL.NameStateComponent.nameFromGLName(ss[0].name),
"frontRight")
def testCurvesPrimitiveSelectUsingWireframeLines(self):
r = IECoreGL.Renderer()
r.setOption("gl:mode", IECore.StringData("deferred"))
r.setOption("gl:searchPath:shader",
IECore.StringData(os.path.dirname(__file__) + "/shaders"))
with IECoreScene.WorldBlock(r):
r.concatTransform(imath.M44f().translate(imath.V3f(0, 0, -5)))
r.setAttribute("name", IECore.StringData("curvesNeedSelectingToo"))
r.setAttribute("gl:curvesPrimitive:useGLLines",
IECore.BoolData(True))
r.setAttribute("gl:primitive:wireframe", IECore.BoolData(True))
r.setAttribute("gl:primitive:solid", IECore.BoolData(False))
r.curves(
IECore.CubicBasisf.linear(), False, IECore.IntVectorData([2]),
{
"P":
IECoreScene.PrimitiveVariable(
IECoreScene.PrimitiveVariable.Interpolation.Vertex,
IECore.V3fVectorData(
[imath.V3f(
-1,
-1,
0,
), imath.V3f(1, 1, 0)]))
})
s = r.scene()
s.setCamera(IECoreGL.Camera(imath.M44f(), False))
for mode in (IECoreGL.Selector.Mode.GLSelect,
IECoreGL.Selector.Mode.OcclusionQuery,
IECoreGL.Selector.Mode.IDRender):
ss = s.select(mode, imath.Box2f(imath.V2f(0), imath.V2f(1)))
names = [
IECoreGL.NameStateComponent.nameFromGLName(x.name) for x in ss
]
self.assertEqual(len(names), 1)
self.assertEqual(names[0], "curvesNeedSelectingToo")
def testOcclusionQuerySelect(self):
r = IECoreGL.Renderer()
r.setOption("gl:mode", IECore.StringData("deferred"))
with IECoreScene.WorldBlock(r):
r.concatTransform(imath.M44f().translate(imath.V3f(0, 0, -5)))
r.concatTransform(imath.M44f().translate(imath.V3f(-1, 0, 0)))
r.setAttribute("name", IECore.StringData("frontLeft"))
r.sphere(1, -1, 1, 360, {})
r.concatTransform(imath.M44f().translate(imath.V3f(0, 0, -1)))
r.setAttribute("name", IECore.StringData("backLeft"))
r.sphere(1, -1, 1, 360, {})
r.concatTransform(imath.M44f().translate(imath.V3f(2, 0, 1)))
r.setAttribute("name", IECore.StringData("frontRight"))
r.sphere(1, -1, 1, 360, {})
r.concatTransform(imath.M44f().translate(imath.V3f(0, 0, -1)))
r.setAttribute("name", IECore.StringData("backRight"))
r.sphere(1, -1, 1, 360, {})
s = r.scene()
s.setCamera(IECoreGL.Camera())
ss = s.select(IECoreGL.Selector.Mode.OcclusionQuery,
imath.Box2f(imath.V2f(0, 0), imath.V2f(0.25, 1)))
self.assertEqual(len(ss), 2)
self.assertEqual(
set([
IECoreGL.NameStateComponent.nameFromGLName(x.name) for x in ss
]), set(("frontLeft", "backLeft")))
ss = s.select(IECoreGL.Selector.Mode.OcclusionQuery,
imath.Box2f(imath.V2f(0.75, 0), imath.V2f(1, 1)))
self.assertEqual(len(ss), 2)
self.assertEqual(
set([
IECoreGL.NameStateComponent.nameFromGLName(x.name) for x in ss
]), set(("frontRight", "backRight")))
def testContextManager(self):
r = IECoreGL.Renderer()
r.setOption("gl:mode", IECore.StringData("deferred"))
r.setOption("gl:searchPath:shader",
IECore.StringData(os.path.dirname(__file__) + "/shaders"))
with IECoreScene.WorldBlock(r):
r.concatTransform(imath.M44f().translate(imath.V3f(0, 0, -5)))
r.setAttribute("name", IECore.StringData("one"))
r.shader(
"surface", "color",
{"colorValue": IECore.Color3fData(imath.Color3f(1, 0, 0))})
r.sphere(1, -1, 1, 360, {})
r.concatTransform(imath.M44f().translate(imath.V3f(-1, 0, 0)))
r.setAttribute("name", IECore.StringData("two"))
r.sphere(1, -1, 1, 360, {})
scene = r.scene()
scene.setCamera(None)
IECoreGL.Camera(imath.M44f(),
False).render(IECoreGL.State.defaultState())
hits = []
with IECoreGL.Selector(imath.Box2f(imath.V2f(0), imath.V2f(1)),
IECoreGL.Selector.Mode.IDRender,
hits) as selector:
IECoreGL.State.bindBaseState()
selector.baseState().bind()
scene.root().render(selector.baseState())
names = [
IECoreGL.NameStateComponent.nameFromGLName(x.name) for x in hits
]
self.assertEqual(len(names), 2)
self.assert_("one" in names)
self.assert_("two" in names)
def testSelect(self):
r = IECoreGL.Renderer()
r.setOption("gl:mode", IECore.StringData("deferred"))
r.setOption("gl:searchPath:shader",
IECore.StringData(os.path.dirname(__file__) + "/shaders"))
with IECoreScene.WorldBlock(r):
r.concatTransform(imath.M44f().translate(imath.V3f(0, 0, -5)))
r.setAttribute("name", IECore.StringData("one"))
r.shader(
"surface", "color",
{"colorValue": IECore.Color3fData(imath.Color3f(1, 0, 0))})
r.sphere(1, -1, 1, 360, {})
r.concatTransform(imath.M44f().translate(imath.V3f(-1, 0, 0)))
r.setAttribute("name", IECore.StringData("two"))
r.sphere(1, -1, 1, 360, {})
r.concatTransform(imath.M44f().translate(imath.V3f(2, 0, 0)))
r.setAttribute("name", IECore.StringData("three"))
r.sphere(1, -1, 1, 360, {})
s = r.scene()
s.setCamera(IECoreGL.Camera(imath.M44f(), False))
ss = s.select(IECoreGL.Selector.Mode.GLSelect,
imath.Box2f(imath.V2f(0), imath.V2f(1)))
names = [
IECoreGL.NameStateComponent.nameFromGLName(x.name) for x in ss
]
self.assertEqual(len(names), 3)
self.assert_("one" in names)
self.assert_("two" in names)
self.assert_("three" in names)
def testRegionSelect(self):
r = IECoreGL.Renderer()
r.setOption("gl:mode", IECore.StringData("deferred"))
r.setOption("gl:searchPath:shader",
IECore.StringData(os.path.dirname(__file__) + "/shaders"))
with IECoreScene.WorldBlock(r):
r.concatTransform(imath.M44f().translate(imath.V3f(0, 0, -5)))
r.shader(
"surface", "color",
{"colorValue": IECore.Color3fData(imath.Color3f(1, 0, 0))})
r.concatTransform(imath.M44f().translate(imath.V3f(-2, -2, 0)))
r.setAttribute("name", IECore.StringData("red"))
r.sphere(1, -1, 1, 360, {})
r.shader(
"surface", "color",
{"colorValue": IECore.Color3fData(imath.Color3f(0, 1, 0))})
r.concatTransform(imath.M44f().translate(imath.V3f(0, 4, 0)))
r.setAttribute("name", IECore.StringData("green"))
r.sphere(1, -1, 1, 360, {})
r.shader(
"surface", "color",
{"colorValue": IECore.Color3fData(imath.Color3f(0, 0, 1))})
r.concatTransform(imath.M44f().translate(imath.V3f(4, 0, 0)))
r.setAttribute("name", IECore.StringData("blue"))
r.sphere(1, -1, 1, 360, {})
r.shader(
"surface", "color",
{"colorValue": IECore.Color3fData(imath.Color3f(1, 1, 1))})
r.concatTransform(imath.M44f().translate(imath.V3f(0, -4, 0)))
r.setAttribute("name", IECore.StringData("white"))
r.sphere(1, -1, 1, 360, {})
s = r.scene()
s.setCamera(IECoreGL.Camera(imath.M44f(), False))
ss = s.select(IECoreGL.Selector.Mode.GLSelect,
imath.Box2f(imath.V2f(0, 0.5), imath.V2f(0.5, 1)))
self.assertEqual(len(ss), 1)
self.assertEqual(
IECoreGL.NameStateComponent.nameFromGLName(ss[0].name), "red")
ss = s.select(IECoreGL.Selector.Mode.GLSelect,
imath.Box2f(imath.V2f(0), imath.V2f(0.5)))
self.assertEqual(len(ss), 1)
self.assertEqual(
IECoreGL.NameStateComponent.nameFromGLName(ss[0].name), "green")
ss = s.select(IECoreGL.Selector.Mode.GLSelect,
imath.Box2f(imath.V2f(0.5, 0), imath.V2f(1, 0.5)))
self.assertEqual(len(ss), 1)
self.assertEqual(
IECoreGL.NameStateComponent.nameFromGLName(ss[0].name), "blue")
ss = s.select(IECoreGL.Selector.Mode.GLSelect,
imath.Box2f(imath.V2f(0.5), imath.V2f(1)))
self.assertEqual(len(ss), 1)
self.assertEqual(
IECoreGL.NameStateComponent.nameFromGLName(ss[0].name), "white")
