Python光学仿真wxpython透镜演示系统计算与绘图

def getRay(self):
 self.rays,self.abcs,self.dots = [[],[],[]]
 sDot = self.source #光源为第一个点
 sRay = rp.getABC(self.sourceDict['theta'],sDot)
 inPoint,outPoint,flec,frac = self.opti.singleReflect(sRay,sDot,1)
 if inPoint == []: return [] #无交点返回空list
 self.dots.append(inPoint)
 self.rays.append([sDot,inPoint])
 crossflec = self.crossRagion(flec,inPoint)
 if crossflec != []:
  self.dots.append(crossflec)
  self.rays.append([inPoint,crossflec])
  self.abcs.append(flec)
 if outPoint == []: return []
 self.dots.append(outPoint)
 self.rays.append([inPoint,outPoint])
 if frac == []: return []
 crossfrac = self.crossRagion(frac,outPoint)
 if crossflec != []:
  self.dots.append(crossfrac)
  self.rays.append([outPoint,crossfrac])
  self.abcs.append(frac)
##求光线与界面边缘的交点
def crossRagion(self,ray,point):
 w,h = self.drawPanel.GetSize()
 edges = [[(0,0),(0,w)],[(0,h/2),(0,-h/2)],[(0,-h/2),(w,-h/2)],
[(w,-h/2),(w,h/2)],[(w,h/2),(0,h/2)]]
 for dots in edges:
  cross=rp.getCross(ray,dots,point)
  if cross!=[]:
return cross
 return []

def DrawPath(self):
 w,h = self.drawPanel.GetSize()#获取画布尺寸
 dc = wx.ClientDC(self.drawPanel)
 dc.SetPen(wx.Pen('#666666'))
 dc.DrawRectangle(0,0,w,h)
 dc.SetDeviceOrigin(0,h/2)
 dc.SetAxisOrientation(True,True) #坐标系翻转 
 dc.SetPen(wx.Pen('#0000FF'))
 dc.DrawLine(0,0,w,0)
 dc.SetPen(wx.Pen('#00FF00'))
 ##绘制透镜
 for edge in self.opti.edges:
  dots = edge['dots']
  if len(dots)==2: #此时为平面
dc.DrawLine(dots[0][0],dots[0][1],
dots[1][0],dots[1][1])
  elif len(dots)==3:  #此时为曲面
x3,y3,_=rp.arc2cir(dots)
if dots[1][0]>dots[2][0]:#画劣弧
 dc.DrawArc(dots[0][0],dots[0][1],dots[1][0],dots[1][1],x3,y3)
else:
 dc.DrawArc(dots[1][0],dots[1][1],dots[0][0],dots[0][1],x3,y3)
 dc.SetPen(wx.Pen('#FF0000'))
 ##绘制光源
 dc.DrawCircle(self.source[0],self.source[1],10)
 ##绘制光线
 for ray in self.rays:
  dc.DrawLine(ray[0][0],ray[0][1],
  ray[1][0],ray[1][1])
 ##绘制光线与物体表面的交点
 dc.SetPen(wx.Pen('#FF00FF'))
 for dot in self.dots:
  dc.DrawCircle(dot[0],dot[1],5)