netgen/py_tutorials/opengl.py

from netgen.csg import *
import netgen.meshing as meshing

from OpenGL.GL import *
from OpenGL.GLU import *
from OpenGL.GLUT import *



sp1 = Sphere (Pnt(0,0,0), 0.2)
sp2 = Sphere (Pnt(0.5,0,0), 0.2)
sp3 = Sphere (Pnt(0,0,0.5), 0.2)
sp4 = Sphere (Pnt(0,0.2,0.7), 0.2)
sp5 = Sphere (Pnt(0,0,0.3), 0.2)


all = sp1+sp2+sp3+sp4+sp5


geom = CSGeometry()
geom.Add (all)


vis = VS(geom)
# vis.Draw()


window = 0                                             # glut window number
width, height = 500, 500



def mydraw():
    vis.Draw()
    glutSwapBuffers()


xold = -1;
yold = -1;
mode = 'r'

def myMotionHandler( x, y ):
    global xold, yold
    MouseMove(vis,xold,yold, x,y, mode)  # 'm','z'
    xold = x
    yold = y
    mydraw()

def myPassiveMotionHandler( x, y ):
    global xold, yold
    xold = x
    yold = y

def myMouseHandler( button, state, x, y ):
    print(button,state,x,y)
    modes = {0:'r', 1:'m', 2:'z'}
    global mode
    if button<3:
        if state==0:
            mode = modes[button]
        else:
            mode = 'r'



glutInit("mainwin")                                   # initialize glut
glutInitDisplayMode(GLUT_RGBA | GLUT_DOUBLE | GLUT_ALPHA | GLUT_DEPTH)
glutInitWindowSize(width, height)                      # set window size
glutInitWindowPosition(0, 0)                           # set window position
window = glutCreateWindow(b"ngs")                      # create window with title

glutMotionFunc(myMotionHandler)
glutMouseFunc(myMouseHandler)
glutPassiveMotionFunc(myPassiveMotionHandler)
glutDisplayFunc(mydraw)                                 # set draw function callback
# glutIdleFunc(mydraw)                                     # draw all the time



glMatrixMode(GL_PROJECTION);
glLoadIdentity();
pnear = 0.1;
pfar = 10;
gluPerspective(20.0, 1.0*width / height, pnear, pfar);
glViewport(0, 0, width, height);
glMatrixMode(GL_MODELVIEW);



glutMainLoop()        



# param = meshing.MeshingParameters(maxh=0.2)
# mesh = GenerateMesh (geom, param)
# mesh.Save ("test.vol")