@@ -131,7 +131,8 @@ def generate_random(previous_atoms):
131
131
while True :
132
132
x , y = Atom .generate_random_position (co .H_RADIUS )
133
133
if all (not atom .isColliding (x , y , co .H_RADIUS ) for atom in previous_atoms ):
134
- break
134
+ if math .dist ((x , y ), (co .RESTART_BTN_POS_X , co .RESTART_BTN_POS_X )) > co .H_RADIUS + 20 :
135
+ break
135
136
return Hydrogen (x , y )
136
137
137
138
@@ -144,7 +145,8 @@ def generate_random(previous_atoms):
144
145
while True :
145
146
x , y = Atom .generate_random_position (co .O_RADIUS )
146
147
if all (not atom .isColliding (x , y , co .O_RADIUS ) for atom in previous_atoms ):
147
- break
148
+ if math .dist ((x , y ), (co .RESTART_BTN_POS_X , co .RESTART_BTN_POS_X )) > co .O_RADIUS + 20 :
149
+ break
148
150
return Oxygen (x , y )
149
151
150
152
@@ -157,7 +159,8 @@ def generate_random(previous_atoms):
157
159
while True :
158
160
x , y = Atom .generate_random_position (co .N_RADIUS )
159
161
if all (not atom .isColliding (x , y , co .N_RADIUS ) for atom in previous_atoms ):
160
- break
162
+ if math .dist ((x , y ), (co .RESTART_BTN_POS_X , co .RESTART_BTN_POS_X )) > co .N_RADIUS + 20 :
163
+ break
161
164
return Nitrogen (x , y )
162
165
163
166
@@ -170,7 +173,8 @@ def generate_random(previous_atoms):
170
173
while True :
171
174
x , y = Atom .generate_random_position (co .C_RADIUS )
172
175
if all (not atom .isColliding (x , y , co .C_RADIUS ) for atom in previous_atoms ):
173
- break
176
+ if math .dist ((x , y ), (co .RESTART_BTN_POS_X , co .RESTART_BTN_POS_X )) > co .C_RADIUS + 20 :
177
+ break
174
178
return Carbon (x , y )
175
179
176
180
@@ -183,5 +187,6 @@ def generate_random(previous_atoms):
183
187
while True :
184
188
x , y = Atom .generate_random_position (co .ELECTRON_RADIUS )
185
189
if all (not atom .isColliding (x , y , co .ELECTRON_RADIUS ) for atom in previous_atoms ):
186
- break
190
+ if math .dist ((x , y ), (co .RESTART_BTN_POS_X , co .RESTART_BTN_POS_X )) > co .ELECTRON_RADIUS + 20 :
191
+ break
187
192
return Electron (x , y )
0 commit comments