# lorenz liner module

from math import *
from visual import *

def curve_calc(R,b,mus):
	# Draw axes
	curve( pos = [ (-25,0,0), (25,0,0) ], color = (0,1,1), radius = 0.1 )
	curve( pos = [ (0,-25,0), (0,25,0) ], color = (0,1,1), radius = 0.1 )
	curve( pos = [ (0,0,0), (0,0,50) ], color = (0,1,1), radius = 0.1 )
	# Draw interval lines
	fpoint1 = sqrt(b*(R-1))
	fpoint2 = -1 * fpoint1
	if (mus==1):
		x_interval = fpoint1 / 6
		x = 0
		for i in range(13):
			curve( pos = [ ((fpoint2*3+x_interval*i),-2,0),\
					((fpoint2*3+x_interval*i),2,0) ],\
					color = (1,0,0), radius = 0.3 )
		for i in range(13):
			curve( pos = [ ((fpoint2+x_interval*i),-2,0),\
				((fpoint2+x_interval*i),2,0) ],\
				color = (0,1,0), radius = 0.3 )
		for i in range(13):
			curve( pos = [ ((fpoint1+x_interval*i),-2,0),\
					((fpoint1+x_interval*i),2,0) ],\
					color = (0,0,1),radius = 0.3 )
	else:
		x_interval = fpoint1 / 3.5
		x = 0
		for i in range(8):
			curve( pos = [ ((fpoint2*3+x_interval*i),-2,0),\
					((fpoint2*3+x_interval*i),2,0) ],\
					color = (1,0,0), radius = 0.3 )
		for i in range(8):
			curve( pos = [ ((fpoint2+x_interval*i),-2,0),\
				((fpoint2+x_interval*i),2,0) ],\
				color = (0,1,0), radius = 0.3 )
		for i in range(8):
			curve( pos = [ ((fpoint1+x_interval*i),-2,0),\
					((fpoint1+x_interval*i),2,0) ],\
					color = (0,0,1),radius = 0.3 )