class Plant(EntityBase):
type_id = 'plant'
def __init__(self, pos, angle=0, text='plant'):
super().__init__(pos, angle)
self.text = text
self.words = []
def branch(self):
if self.world != None:
if len(self.words) == 0:
if self.text == '':
self.text = self.world.markov.chain(self.text, self.text)
length = (len(self.text) + 1) * 5
angle = super().rot - math.pi / 2
x, y = super().pos
newpos = (x + math.cos(angle) * (length + 15),
y + math.sin(angle) * (length + 15))
w = Word(newpos, angle, self.text, self)
self.text = ''
self.words.append(w)
self.world.add_entity(w)
elif len(self.words) < 25:
# Structural determination. x^y A tree may be best represented by y of around 0.5; a shrub by y = 2
# index = random.randint(0, len(self.words)-1)
index = int(
math.pow(random.random(), 0.7) *
(len(self.words) -
0.001)) # converts to int via truncation
self.words[index].branch()
self.words[-1].parent_word = self.words[index].text
text = dataprop("text")
words = []
class TerrBlock(EntityBase):
type_id = 'terrblock'
def __init__(self, pos, scale, angle=0):
super().__init__(pos, angle)
self.scale = scale
scale = dataprop('scale')
class Box(EntityBase):
type_id = 'box'
def __init__(self, pos, angle=0, text=''):
super().__init__(pos, angle)
self.text = text
text = dataprop('text')
class Word(EntityBase):
type_id = 'word'
def __init__(self,
pos,
angle=0,
text='word',
parent_plant=None,
parent_branch=None):
super().__init__(pos, angle)
self.text = text
self.parent_plant = parent_plant
self.parent_branch = parent_branch
self.branches = []
self.alive = parent_plant != None
def kill(self):
self.alive = False
if self.parent_plant != None:
self.parent_plant.words.remove(self)
self.parent_plant = None
for b in self.branches:
b.kill()
def on_removed(self):
super().on_removed()
if (self.parent_branch != None):
self.parent_branch.branches.remove(self)
self.kill()
def branch(self):
# 2 branch limit for trees.
if len(self.branches) < 2 and self.alive:
# First, calculate offset based on current word
x, y = super().pos
x = x + math.cos(super().rot) * (len(self.text) + 1) * 5
y = y + math.sin(super().rot) * (len(self.text) + 1) * 5
# next, calculate offset based on future word.
next_word = self.world.markov.chain(self.text, self.parent_word)
# This should be tree-only code.
new_angle = 0
if len(self.branches) == 1:
new_angle = super().rot + (super().rot - self.branches[0].rot
) * 0.3 * random.random()
else:
new_angle = super().rot + (random.random() - 0.5) * 1.2
if abs(new_angle - self.parent_plant.rot -
math.pi / 2) > math.pi / 3:
new_angle = (new_angle + self.parent_plant.rot -
math.pi / 2) / 2
good = True
for b in self.branches:
if abs(b.rot - new_angle) < 0.3 or next_word == b.text:
good = False
if good:
w = Word((x + math.cos(new_angle) * (len(next_word) + 1) * 5,
y + math.sin(new_angle) * (len(next_word) + 1) * 5),
new_angle, next_word, self.parent_plant, self)
self.branches.append(w)
self.parent_plant.words.append(w)
self.world.add_entity(w)
text = dataprop('text')
branches = []
parent_plant = None
parent_word = ''
class Ant(EntityBase):
type_id = 'ant'
def __init__(self, client, pos, angle=0):
super().__init__(
pos,
angle,
id=client.face,
)
self.client = client
self.correcting = False
self.speech = None
self.gut = CounterNetProp(self, 'gut')
speech = dataprop('speech')
bell_type = dataprop('bellType')
def add_text_act(self, act):
t = act['type']
gut = self.data.setdefault('gut', {})
if t == 'eat':
char = act['char']
gut[char] = gut.get(char, 0) + 1
elif t == 'speak':
char = act['char']
gut[char] -= 1
if gut[char] <= 0:
del gut[char]
self.data['speech'] = self.data['speech'] + char
elif t == 'untracked':
text = act['text']
if text not in ' ':
text = f'\0{text}\0'
self.data['speech'] = self.data['speech'] + text
elif t == 'unspeak':
speech = self.data['speech']
if speech[-1] == '\0':
self.data['speech'] = speech[:speech.rfind('\0', 0, -1)]
else:
self.data['speech'] = speech[:-1]
if speech[-1] != ' ':
char = speech[-1]
gut[char] = gut.get(char, 0) + 1
elif t == 'clear':
untracked = False
for chunk in self.data['speech'].split('\0'):
if untracked:
continue
for char in chunk:
if char == ' ':
continue
gut[char] = gut.get(char, 0) + 1
untracked = not untracked
self.data['speech'] = None
elif t == 'destroy':
self.data['speech'] = None
else:
raise ValueError(f'Invalid text act type {t}')
self.diff.setdefault('textActs', []).append(act)