stupidpythonprojects/cellular_automaton/cellular.py

121 lines
3.7 KiB
Python
Raw Normal View History

2017-05-14 23:10:43 +02:00
#!/usr/bin/env python3
import numpy as np
import argparse
# cellular automaton functions:
def generatestart(startcondition,border,size):
"""
:param startcondition: r for random numers zero or one
:param border: the size of the border one or two
:param size: size of the whole array
:return: the initialized array to work with
"""
#all cells
cells = np.zeros((size),int)
#cells without borders
writableCells = cells[border:size-border]
#startcondition for seed
if(startcondition == "s"):
cells[size // 2] = 1
#condition for setting random values
else:
for i in range(writableCells.shape[0]):
writableCells[i]=np.round(np.random.rand())
#print(np.shape(writableCells))
return cells
def code2FunctionTable(code, r):
result = {}
c = code
if r == 1:
2017-05-15 08:47:26 +02:00
e = 7
# for each bit, loop over {1,0}
for i in range(1,-1, -1):
for j in range(1,-1, -1):
for k in range(1,-1, -1):
2017-05-14 23:10:43 +02:00
if 2**e > c:
result[(i,j,k)] = 0
else:
result[(i,j,k)] = 1
c -= 2 ** e
e -= 1
elif r == 2:
2017-05-15 08:47:26 +02:00
e = 31
for i in range(1,-1, -1):
for j in range(1,-1, -1):
for k in range(1,-1, -1):
for l in range(1,-1, -1):
for m in range(1,-1, -1):
if 2**e > c:
2017-05-14 23:10:43 +02:00
result[(i, j, k, l, m)] = 0
else:
result[(i, j, k, l, m)] = 1
c -= 2 ** e
e -= 1
return result
def getNextCellValue(functionTable, row, pos, r):
if r == 1:
i = row[pos - 1]
j = row[pos]
k = row[pos + 1]
return functionTable[(i,j,k)]
elif r == 2:
i = row[pos - 2]
j = row[pos - 1]
k = row[pos]
l = row[pos + 1]
m = row[pos + 2]
return functionTable[(i,j,k,l,m)]
return None
def calculateNextStep(functionTable, size, row, r):
result = np.zeros((size),int)
for i in range(r, size - r):
result[i] = getNextCellValue(functionTable, row, i, r)
return result
if __name__ == "__main__":
# parsing args:
parser = argparse.ArgumentParser(description="one dimensional cellular automaton for k=2")
#parser.add_argument('--steps', dest='steps', default=20, help='steps per second')
parser.add_argument('--w', dest='w', default=84, help='field width')
#parser.add_argument('--h', dest='h', default=livingSpaceHeight, help='field height')
#parser.add_argument('--fullscreen', dest='fullscreen', action='store_true')
#parser.add_argument('--window_w', dest='win_w', default=window_w, help='window width')
#parser.add_argument('--window_h', dest='win_h', default=window_h, help='window height')
parser.add_argument('--code', dest='code', default='17', help='code for the automaton')
parser.add_argument('--random', dest='random', action='store_true')
parser.add_argument('--r', dest='r', default=1, help='radius. can be 1 or 2')
parser.set_defaults(random=False)
args = parser.parse_args()
r = int(args.r)
c = int(args.code)
size = int(args.w)
startcondition = ""
if args.random:
startcondition = "r"
else:
startcondition = "s"
functionTable = code2FunctionTable(c, r)
print(functionTable)
cells = generatestart(startcondition, r, size)
print(cells)
for i in range(20):
cells = calculateNextStep(functionTable,size,cells, r)
print(cells)