class Moon(Planet):
def __init__(self, parent, control="RANDOM"):
self.parent = parent
if (control == "LUNA"):
self.initluna()
return
if (control == "PHOBOS"):
self.initphobos()
return
if (control == "DEIMOS"):
self.initdeimos()
return
if (control == "IO"):
self.initio()
return
if (control == "EUROPA"):
self.initeuropa()
return
if (control == "GANYMEDE"):
self.initganymede()
return
if (control == "CALLISTO"):
self.initcallisto()
return
if (control == "TITAN"):
self.inittitan()
return
if (control == "TRITON"):
self.inittriton()
return
if (control == "CHARON"):
self.initcharon()
return
self.index = random.randint(
0, 2147483647
) #I want these to be indicies but here I can assure they're unique which is a start
self.resources = [
random.randint(0, 255),
random.randint(0, 255),
random.randint(0, 255)
]
self.size = random.lognormvariate(math.log(1.5), 0.3)
self.culture = OnlineMarkov()
for i in range(-1, int(self.size * 2 / 2)):
cultureString = self.culture.randomString(int(self.size * 2 / 2))
#print(cultureString)
self.culture.contribute(cultureString)
self.radius = random.lognormvariate(math.log(parent.size * 2), 0.125)
#print(self.radius)
self.inclination = random.normalvariate(0, 15)
self.longitudeAscendingNode = random.uniform(0, 360)
self.trueAnomaly = random.uniform(0, 360)
self.period = Planet.getPeriodFromRadius(self.radius, self.parent.size)
self.rotationMatrix = Planet.eulerAngles2Matrix(
self.inclination, self.longitudeAscendingNode)
self.position = self.getPosition()
def getPosition(self):
#compute position in the plane of the orbit
x = self.radius * math.sin(self.trueAnomaly * math.pi / 180.0)
y = self.radius * math.cos(self.trueAnomaly * math.pi / 180.0)
z = 0
#ignore inclination and longitude of ascending node for now
return add(self.parent.getPosition(),
np.matmul(self.rotationMatrix, [x, y, z]))
def initluna(self):
self.size = 3
self.radius = 20
self.resources = [64.0, 64.0, 64.0]
self.culture = OnlineMarkov()
for i in range(-1, int(self.size * 2 / 2)):
cultureString = self.culture.randomString(int(self.size * 2 / 2))
#print(cultureString)
self.culture.contribute(cultureString)
self.trueAnomaly = 25
self.period = Planet.getPeriodFromRadius(self.radius, self.parent.size)
self.inclination = 15
self.longitudeAscendingNode = 150
self.rotationMatrix = Planet.eulerAngles2Matrix(
self.inclination, self.longitudeAscendingNode)
self.position = self.getPosition()
def initphobos(self):
self.size = 2
self.radius = 10
self.resources = [64.0, 64.0, 64.0]
self.culture = OnlineMarkov()
for i in range(-1, int(self.size * 2 / 2)):
cultureString = self.culture.randomString(int(self.size * 2 / 2))
#print(cultureString)
self.culture.contribute(cultureString)
self.trueAnomaly = 151
self.period = Planet.getPeriodFromRadius(self.radius, self.parent.size)
self.inclination = 5
self.longitudeAscendingNode = 110
self.rotationMatrix = Planet.eulerAngles2Matrix(
self.inclination, self.longitudeAscendingNode)
self.position = self.getPosition()
def initdeimos(self):
self.size = 1
self.radius = 12
self.resources = [64.0, 64.0, 64.0]
self.culture = OnlineMarkov()
for i in range(-1, int(self.size * 2 / 2)):
cultureString = self.culture.randomString(int(self.size * 2 / 2))
#print(cultureString)
self.culture.contribute(cultureString)
self.trueAnomaly = 274
self.period = Planet.getPeriodFromRadius(self.radius, self.parent.size)
self.inclination = 16
self.longitudeAscendingNode = 280
self.rotationMatrix = Planet.eulerAngles2Matrix(
self.inclination, self.longitudeAscendingNode)
self.position = self.getPosition()
def initio(self):
self.resources = [
random.randint(0, 255),
random.randint(0, 255),
random.randint(0, 255)
]
self.size = 4
self.culture = OnlineMarkov()
for i in range(-1, int(self.size * 2 / 2)):
cultureString = self.culture.randomString(int(self.size * 2 / 2))
#print(cultureString)
self.culture.contribute(cultureString)
self.radius = 36
#print(self.radius)
self.inclination = random.normalvariate(0, 15)
self.longitudeAscendingNode = random.uniform(0, 360)
self.trueAnomaly = random.uniform(0, 360)
self.period = Planet.getPeriodFromRadius(self.radius, self.parent.size)
self.rotationMatrix = Planet.eulerAngles2Matrix(
self.inclination, self.longitudeAscendingNode)
self.position = self.getPosition()
def initeuropa(self):
self.resources = [
random.randint(0, 255),
random.randint(0, 255),
random.randint(0, 255)
]
self.size = 3
self.culture = OnlineMarkov()
for i in range(-1, int(self.size * 2 / 2)):
cultureString = self.culture.randomString(int(self.size * 2 / 2))
#print(cultureString)
self.culture.contribute(cultureString)
self.radius = 49
#print(self.radius)
self.inclination = random.normalvariate(0, 15)
self.longitudeAscendingNode = random.uniform(0, 360)
self.trueAnomaly = random.uniform(0, 360)
self.period = Planet.getPeriodFromRadius(self.radius, self.parent.size)
self.rotationMatrix = Planet.eulerAngles2Matrix(
self.inclination, self.longitudeAscendingNode)
self.position = self.getPosition()
def initganymede(self):
self.resources = [
random.randint(0, 255),
random.randint(0, 255),
random.randint(0, 255)
]
self.size = 4
self.culture = OnlineMarkov()
for i in range(-1, int(self.size * 2 / 2)):
cultureString = self.culture.randomString(int(self.size * 2 / 2))
#print(cultureString)
self.culture.contribute(cultureString)
self.radius = 64
#print(self.radius)
self.inclination = random.normalvariate(0, 15)
self.longitudeAscendingNode = random.uniform(0, 360)
self.trueAnomaly = random.uniform(0, 360)
self.period = Planet.getPeriodFromRadius(self.radius, self.parent.size)
self.rotationMatrix = Planet.eulerAngles2Matrix(
self.inclination, self.longitudeAscendingNode)
self.position = self.getPosition()
def initcallisto(self):
self.resources = [
random.randint(0, 255),
random.randint(0, 255),
random.randint(0, 255)
]
self.size = 4
self.culture = OnlineMarkov()
for i in range(-1, int(self.size * 2 / 2)):
cultureString = self.culture.randomString(int(self.size * 2 / 2))
#print(cultureString)
self.culture.contribute(cultureString)
self.radius = 81
#print(self.radius)
self.inclination = random.normalvariate(0, 15)
self.longitudeAscendingNode = random.uniform(0, 360)
self.trueAnomaly = random.uniform(0, 360)
self.period = Planet.getPeriodFromRadius(self.radius, self.parent.size)
self.rotationMatrix = Planet.eulerAngles2Matrix(
self.inclination, self.longitudeAscendingNode)
self.position = self.getPosition()
def generateMoonList(planetList):
[earth, mars, jupiter, saturn, uranus, neptune, pluto] = planetList
luna = Moon(earth, "LUNA")
phobos = Moon(mars, "PHOBOS")
deimos = Moon(mars, "DEIMOS")
io = Moon(jupiter, "IO")
europa = Moon(jupiter, "EUROPA")
ganymede = Moon(jupiter, "GANYMEDE")
callisto = Moon(jupiter, "CALLISTO")
titan = Moon(saturn, "RANDOM")
saturni = []
for i in range(7):
saturni += [Moon(saturn, "RANDOM")]
uranii = []
for i in range(6):
uranii += [Moon(uranus, "RANDOM")]
triton = Moon(neptune, "RANDOM")
neptunii = []
for i in range(2):
neptunii += [Moon(neptune, "RANDOM")]
charon = Moon(pluto, "RANDOM")
return [luna, phobos, deimos, io, europa, ganymede, callisto, titan
] + saturni + uranii + [triton] + neptunii + [charon]
class Planet:
def __init__(self, control = "RANDOM"):
if(control=="SUN"):
self.initsun()
return
if(control=="MERCURY"):
self.initmercury()
return
if(control=="VENUS"):
self.initvenus()
return
if(control=="EARTH"):
self.initearth()
return
if(control=="MARS"):
self.initmars()
return
if(control=="CERES"):
self.initceres()
return
if(control=="JUPITER"):
self.initjupiter()
return
if(control=="SATURN"):
self.initsaturn()
return
if(control=="URANUS"):
self.inituranus()
return
if(control=="NEPTUNE"):
self.initneptune()
return
if(control=="PLUTO"):
self.initpluto()
return
if(control=="HAUMEA"):
self.inithaumea()
return
self.index = random.randint(0,2147483647)#I want these to be indicies but here I can assure they're unique which is a start
self.resources = [random.randint(0,255),random.randint(0,255),random.randint(0,255)]
self.size = random.lognormvariate(math.log(3),0.3)
self.culture = OnlineMarkov()
for i in range(-1,int(self.size*2/2)):
cultureString = self.culture.randomString(int(self.size*2/2))
#print(cultureString)
self.culture.contribute(cultureString)
self.radius = random.lognormvariate(math.log(300),0.125)
#print(self.radius)
self.inclination = random.normalvariate(0,15)
self.longitudeAscendingNode = random.uniform(0,360)
self.trueAnomaly = random.uniform(0,360)
self.period = Planet.getPeriodFromRadius(self.radius)
self.rotationMatrix = Planet.eulerAngles2Matrix(self.inclination,self.longitudeAscendingNode)
self.position = self.getPosition()
def getPeriodFromRadius(radius,parentSize=40.0):
if(radius)==0:
print("tried to get period for planet with 0 radius orbit")
return 22.0
return math.sqrt(radius/parentSize*40.0)**3*25
def eulerAngles2Matrix(x,z):
thetaZ = z*math.pi/180.0
Rz = [ [math.cos(thetaZ), -math.sin(thetaZ), 0],
[math.sin(thetaZ), math.cos(thetaZ), 0],
[0, 0, 1]]
thetaX = x*math.pi/180.0
Rx = [ [1, 0, 0],
[0, math.cos(thetaX), -math.sin(thetaX)],
[0, math.sin(thetaX), math.cos(thetaX)]]
return np.matmul(Rz,Rx)
def initsun(self):
self.size = 40
self.radius = 0
self.position = [0.0,0.0,0.0]
self.resources = [225.0,225.0,225.0]
self.culture = OnlineMarkov()
self.culture.erase()#clear out the dictionary on sun for a suntutorial
for tutorialLine in suntutorial:
self.culture.contribute(tutorialLine)
#the suntutorial dictionary is committed twice to help it to speak more cogently.
self.culture.contribute(tutorialLine)
#self.culture.print()
self.trueAnomaly = 0
self.period = 22
self.rotationMatrix = [[1,0,0],[0,1,0],[0,0,1]]
def initmercury(self):
self.size = 8
self.radius = 60
self.resources = [225.0,0.0,225.0]
self.culture = OnlineMarkov()
for i in range(-1,int(self.size*2/2)):
cultureString = self.culture.randomString(int(self.size*2/2))
#print(cultureString)
self.culture.contribute(cultureString)
self.trueAnomaly = 120
self.period = Planet.getPeriodFromRadius(self.radius)
self.inclination = 20
self.longitudeAscendingNode = 120
self.rotationMatrix = Planet.eulerAngles2Matrix(self.inclination,self.longitudeAscendingNode)
self.position = self.getPosition()
def initvenus(self):
self.size = 12
self.radius = 100
self.resources = [0.0,225.0,225.0]
self.culture = OnlineMarkov()
for i in range(-1,int(self.size*2/2)):
cultureString = self.culture.randomString(int(self.size*2/2))
#print(cultureString)
self.culture.contribute(cultureString)
self.trueAnomaly = 20
self.period = Planet.getPeriodFromRadius(self.radius)
self.inclination = 17
self.longitudeAscendingNode = 100
self.rotationMatrix = Planet.eulerAngles2Matrix(self.inclination,self.longitudeAscendingNode)
self.position = self.getPosition()
def initearth(self):
self.size = 14
self.radius = 150
self.resources = [0.0,225.0,0.0]
self.culture = OnlineMarkov()
for i in range(-1,int(self.size*2/2)):
cultureString = self.culture.randomString(int(self.size*2/2))
#print(cultureString)
self.culture.contribute(cultureString)
self.trueAnomaly = 25
self.period = Planet.getPeriodFromRadius(self.radius)
self.inclination = 15
self.longitudeAscendingNode = 150
self.rotationMatrix = Planet.eulerAngles2Matrix(self.inclination,self.longitudeAscendingNode)
self.position = self.getPosition()
def initmars(self):
self.size = 8
self.radius = 200
self.resources = [225.0,0.0,0.0]
self.culture = OnlineMarkov()
for i in range(-1,int(self.size*2/2)):
cultureString = self.culture.randomString(int(self.size*2/2))
#print(cultureString)
self.culture.contribute(cultureString)
self.trueAnomaly = 49
self.period = Planet.getPeriodFromRadius(self.radius)
self.inclination = 10
self.longitudeAscendingNode = 135
self.rotationMatrix = Planet.eulerAngles2Matrix(self.inclination,self.longitudeAscendingNode)
self.position = self.getPosition()
def initjupiter(self):
self.size = 30
self.radius = 420
self.resources = [255.0,225.0,0.0]
self.culture = OnlineMarkov()
for i in range(-1,int(self.size*2/2)):
cultureString = self.culture.randomString(int(self.size*2/2))
#print(cultureString)
self.culture.contribute(cultureString)
self.trueAnomaly = 180
self.period = Planet.getPeriodFromRadius(self.radius)
self.inclination = 8
self.longitudeAscendingNode = 170
self.rotationMatrix = Planet.eulerAngles2Matrix(self.inclination,self.longitudeAscendingNode)
self.position = self.getPosition()
def initsaturn(self):
self.size = 26
self.radius = 600
self.resources = [120.0,120.0,100.0]
self.culture = OnlineMarkov()
for i in range(-1,int(self.size*2/2)):
cultureString = self.culture.randomString(int(self.size*2/2))
#print(cultureString)
self.culture.contribute(cultureString)
self.trueAnomaly = 249
self.period = Planet.getPeriodFromRadius(self.radius)
self.inclination = 12
self.longitudeAscendingNode = 230
self.rotationMatrix = Planet.eulerAngles2Matrix(self.inclination,self.longitudeAscendingNode)
self.position = self.getPosition()
def inituranus(self):
self.size = 25
self.radius = 700
self.resources = [50.0,70.0,50.0]
self.culture = OnlineMarkov()
for i in range(-1,int(self.size*2/2)):
cultureString = self.culture.randomString(int(self.size*2/2))
#print(cultureString)
self.culture.contribute(cultureString)
self.trueAnomaly = 25
self.period = Planet.getPeriodFromRadius(self.radius)
self.inclination = 15
self.longitudeAscendingNode = 150
self.rotationMatrix = Planet.eulerAngles2Matrix(self.inclination,self.longitudeAscendingNode)
self.position = self.getPosition()
def initneptune(self):
self.size = 20
self.radius = 800
self.resources = [0.0,0.0,225.0]
self.culture = OnlineMarkov()
for i in range(-1,int(self.size*2/2)):
cultureString = self.culture.randomString(int(self.size*2/2))
#print(cultureString)
self.culture.contribute(cultureString)
self.trueAnomaly = 25
self.period = Planet.getPeriodFromRadius(self.radius)
self.inclination = 15
self.longitudeAscendingNode = 150
self.rotationMatrix = Planet.eulerAngles2Matrix(self.inclination,self.longitudeAscendingNode)
self.position = self.getPosition()
def initpluto(self):
self.size = 5
self.radius = 1200
self.resources = [40.0,30.0,40.0]
self.culture = OnlineMarkov()
for i in range(-1,int(self.size*2/2)):
cultureString = self.culture.randomString(int(self.size*2/2))
#print(cultureString)
self.culture.contribute(cultureString)
self.trueAnomaly = 59
self.period = Planet.getPeriodFromRadius(self.radius)
self.inclination = 30
self.longitudeAscendingNode = 60
self.rotationMatrix = Planet.eulerAngles2Matrix(self.inclination,self.longitudeAscendingNode)
self.position = self.getPosition()
def getPosition(self):
#compute position in the plane of the orbit
x = self.radius*math.sin(self.trueAnomaly*math.pi/180.0)
y = self.radius*math.cos(self.trueAnomaly*math.pi/180.0)
z = 0
#ignore inclination and longitude of ascending node for now
return add(np.matmul(self.rotationMatrix,[x,y,z]),[0,0,200])
def step(self,tick):
#position update
self.trueAnomaly += 360.0*tick/self.period
if self.trueAnomaly>=360:
self.trueAnomaly-=360
self.position = self.getPosition()
#resource update
#if self.resources[1]<1:
# if self.resources[0]>1:
# self.resources[0] -=
def generatePlanetList():
#initialize sun, the tutorial planet
sun = Planet("SUN")
mercury = Planet("MERCURY")
venus = Planet("VENUS")
earth = Planet("EARTH")
mars = Planet("MARS")
jupiter = Planet("JUPITER")
saturn = Planet("SATURN")
neptune = Planet("NEPTUNE")
uranus = Planet("URANUS")
pluto = Planet("PLUTO")
planetList = [mercury,venus,earth,mars,jupiter,saturn,uranus,neptune,pluto]
#set sun as resting on the "table"
minZ = 10
for i in range(NUMBEROFRANDOMPLANETS):
planetList = planetList+[Planet()]
for eachPlanet in planetList:
if(eachPlanet.position[2]<minZ):
minZ = eachPlanet.position[2]
for eachPlanet in planetList:
eachPlanet.position[2] -= minZ-eachPlanet.size
#planetList.sort(key = getY)
return [sun,planetList]
def drawReflections(self,win,screenCenter,yscaling,zscaling):
if(self.position[2]>0):
dimColor = (self.resources[0]*0.25,self.resources[1]*0.25,self.resources[2]*0.25)
proj = (screenCenter[0]+self.position[0],screenCenter[1]+yscaling*self.position[1])
self.npos = (screenCenter[0]+self.position[0],screenCenter[1]+yscaling*self.position[1]-zscaling*self.position[2])
pygame.draw.aaline(win,(16,16,16),self.npos,proj)
aafilledcircle(win,self.npos,self.size,dimColor)
def drawImages(self,win,screenCenter,yscaling,zscaling):
planetColor = (self.resources[0],self.resources[1],self.resources[2])
self.pos = (screenCenter[0]+self.position[0],screenCenter[1]+yscaling*self.position[1]+zscaling*self.position[2])
proj = (screenCenter[0]+self.position[0],screenCenter[1]+yscaling*self.position[1])
pygame.draw.aaline(win,(64,64,64),self.pos,proj)
aafilledcircle(win,self.pos,self.size,planetColor)
def collidepoint(self,mousePos):
if(magnitude(sub(self.pos,mousePos))<self.size+5):
#self.print()
return True
return False
def print(self):
print(self.resources)
print(self.size)
print(self.culture)
print(self.position)
def listen(self):
return self.culture.generate()
def talk(self,string):
self.culture.contribute(string)