class TransmissionDemo(object): def __init__(self, theta=0.0, phi=0.0, etaIn=1.0, etaOut=1.0): self.etaIn = etaIn self.etaOut = etaOut self.incoming = SphericalCoords(theta * math.pi, phi * 2 * math.pi) self.reflected = self.incoming.reflect() self.refracted = self.incoming.refract(etaIn, etaOut) self.initScene() def update(self, theta=None, phi=None, etaIn=None, etaOut=None): if theta != None: self.incoming.theta = theta * math.pi if phi != None: self.incoming.phi = phi * 2 * math.pi if etaIn != None: self.etaIn = (abs(etaIn) > 0) and etaIn or sys.float_info.epsilon if etaOut != None: self.etaOut = (abs(etaOut) > 0) and etaOut or sys.float_info.epsilon self.reflected = self.incoming.reflect() self.refracted = self.incoming.refract(self.etaIn, self.etaOut) # set incoming arrow so that it point towards the transmission point incomingDir = vector(self.incoming.toCartesian()) self.incomingArrow.pos = incomingDir self.incomingArrow.axis = -vector(incomingDir) self.reflectedArrow.axis = self.reflected.toCartesian() self.refractedArrow.axis = self.refracted.toCartesian() def initScene(self): w = 500 display(x=w, y=0, width=w, height=w, range=1.5, forward=-vector(0,1,1), newzoom=1) self.controls = controls(0, 0, width=w, height=w) self.incomingThetaSlider = slider(pos=(-10,-50), width=7, length=100, axis=(0,1,0), action=lambda: self.update(theta=(self.incomingThetaSlider.value / 100.0)-0.5)) self.incomingPhiSlider = slider(pos=(10,-50), width=7, length=100, axis=(0,1,0), action=lambda: self.update(phi=self.incomingPhiSlider.value / 100.0)) self.etaInSlider = slider(pos=(50,-50), width=7, length=100, axis=(0,1,0), action=lambda: self.update(etaIn=(self.etaInSlider.value / 100.0))) drawWorldCoordinates() box(pos=(0,0,0), axis=(0,0,1), length=0.01, height=1.0, width=1.0, opacity=0.25) box(pos=(0,0,-0.5), axis=(0,0,1), length=1, height=1.0, width=1.0, opacity=0.1) self.incomingArrow = drawRayDirection() self.reflectedArrow = drawRayDirection(color=visual.color.green) self.refractedArrow = drawRayDirection(color=visual.color.orange) self.incomingThetaSlider.value = 100.0 * ((self.incoming.theta / math.pi) + 0.5) self.incomingPhiSlider.value = 100.0 * self.incoming.phi / (2 * math.pi) self.etaInSlider.value = 100.0 * self.etaIn def run(self): while True: rate(100) self.controls.interact() # check for events, drive actions; must be executed repeatedly in a loop
def __init__(self, theta=0.0, phi=0.0, etaIn=1.0, etaOut=1.0): self.etaIn = etaIn self.etaOut = etaOut self.incoming = SphericalCoords(theta * math.pi, phi * 2 * math.pi) self.reflected = self.incoming.reflect() self.refracted = self.incoming.refract(etaIn, etaOut) self.initScene()
def visualizeSphereParametrization(): sphere(radius=1, opacity=0.25, material=materials.show_mat_pos) thetaSampleCount = 20 phiSampleCount = 40 thetaStep = 1.0 / float(thetaSampleCount) phiStep = 1.0 / float(phiSampleCount) for i in range(0, thetaSampleCount + 1): for j in range(0, phiSampleCount + 1): spherical = SphericalCoords( theta = (i * thetaStep - 0.5) * math.pi, phi = (j * phiStep) * 2 * math.pi) color = visual.color.hsv_to_rgb((j * phiStep, i * thetaStep, 1.0)) points(pos=[spherical.toCartesian()], color=color)
class TransmissionDemo(object): def __init__(self, theta=0.0, phi=0.0, etaIn=1.0, etaOut=1.0): self.etaIn = etaIn self.etaOut = etaOut self.incoming = SphericalCoords(theta * math.pi, phi * 2 * math.pi) self.reflected = self.incoming.reflect() self.refracted = self.incoming.refract(etaIn, etaOut) self.initScene() def update(self, theta=None, phi=None, etaIn=None, etaOut=None): if theta != None: self.incoming.theta = theta * math.pi if phi != None: self.incoming.phi = phi * 2 * math.pi if etaIn != None: self.etaIn = (abs(etaIn) > 0) and etaIn or sys.float_info.epsilon if etaOut != None: self.etaOut = (abs(etaOut) > 0) and etaOut or sys.float_info.epsilon self.reflected = self.incoming.reflect() self.refracted = self.incoming.refract(self.etaIn, self.etaOut) # set incoming arrow so that it point towards the transmission point incomingDir = vector(self.incoming.toCartesian()) self.incomingArrow.pos = incomingDir self.incomingArrow.axis = -vector(incomingDir) self.reflectedArrow.axis = self.reflected.toCartesian() self.refractedArrow.axis = self.refracted.toCartesian() def initScene(self): w = 500 display(x=w, y=0, width=w, height=w, range=1.5, forward=-vector(0, 1, 1), newzoom=1) self.controls = controls(0, 0, width=w, height=w) self.incomingThetaSlider = slider( pos=(-10, -50), width=7, length=100, axis=(0, 1, 0), action=lambda: self.update(theta=(self.incomingThetaSlider.value / 100.0) - 0.5)) self.incomingPhiSlider = slider( pos=(10, -50), width=7, length=100, axis=(0, 1, 0), action=lambda: self.update(phi=self.incomingPhiSlider.value / 100.0 )) self.etaInSlider = slider( pos=(50, -50), width=7, length=100, axis=(0, 1, 0), action=lambda: self.update(etaIn=(self.etaInSlider.value / 100.0))) drawWorldCoordinates() box(pos=(0, 0, 0), axis=(0, 0, 1), length=0.01, height=1.0, width=1.0, opacity=0.25) box(pos=(0, 0, -0.5), axis=(0, 0, 1), length=1, height=1.0, width=1.0, opacity=0.1) self.incomingArrow = drawRayDirection() self.reflectedArrow = drawRayDirection(color=visual.color.green) self.refractedArrow = drawRayDirection(color=visual.color.orange) self.incomingThetaSlider.value = 100.0 * ( (self.incoming.theta / math.pi) + 0.5) self.incomingPhiSlider.value = 100.0 * self.incoming.phi / (2 * math.pi) self.etaInSlider.value = 100.0 * self.etaIn def run(self): while True: rate(100) self.controls.interact( ) # check for events, drive actions; must be executed repeatedly in a loop