nojhan/sho
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

fix: check if bitstring neighborhood does not erase sensors

Browse source
This commit is contained in:
johann dreo 2019-12-16 14:43:52 +01:00
commit 6e83a121fb
1 changed files with 5 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

7
sho/bit.py
View file

@ -63,13 +63,16 @@ def neighb_square(sol, scale, domain_width):
# Indices order is (y,x) in order to match
# coordinates of images (row,col).
if sol[py][px] == 1:
new[py][px] = 0 # Remove original position.
# Add a one somewhere around.
w = scale//2 * domain_width
ny = np.random.randint(py-w,py+w)
nx = np.random.randint(px-w,px+w)
ny = min(max(0,ny),domain_width-1)
nx = min(max(0,nx),domain_width-1)
new[ny][nx] = 1
if new[nx][ny] != 1:
new[py][px] = 0 # Remove original position.
new[ny][nx] = 1
# else pass
return new