Commit 3b996389 authored by santiaago's avatar santiaago
Browse files

add hasArmDownExtension func

parent ff749b74
Loading
Loading
Loading
Loading
+25 −14
Original line number Diff line number Diff line
@@ -246,6 +246,29 @@ func hasArmDown(invader invader, armIndex, squares, xQ, yQ int) (armDown bool) {
	return
}

func hasArmDownExtension(invader invader, armIndex, squares, xQ, yQ int) (armDownExtension bool) {
	if yQ != armIndex || invader.armsUp || invader.armSize != 3 {
		return
	}
	if invader.arms <= 0 {
		return
	}

	leftOver := squares - invader.length
	half := leftOver / 2

	if (squares - half - half - 1) >= invader.length {
		if xQ == half-1 || xQ == squares-half {
			armDownExtension = true
		}
	} else {
		if xQ == half-2 || xQ == squares+1-half {
			armDownExtension = true
		}
	}
	return
}

func hasBody(invader invader, squares, xQ int) (body bool) {
	leftOver := squares - invader.length
	half := leftOver / 2
@@ -400,21 +423,9 @@ func SpaceInvaders(w http.ResponseWriter, key string, colors []color.RGBA, size
				fill = draw.FillFromRGBA(colorMap[xQ])
			}

			// arm down extension.
			if yQ == armIndex && !invader.armsUp && invader.armSize == 3 {
				leftOver := squares - invader.length
				if invader.arms > 0 {
					if (squares - leftOver/2 - (leftOver / 2) - 1) >= invader.length {
						if xQ == (leftOver/2)-1 || xQ == squares-leftOver/2 {
			if hasArmDownExtension(invader, armIndex, squares, xQ, yQ) {
				fill = draw.FillFromRGBA(colorMap[xQ])
			}
					} else {
						if xQ == (leftOver/2)-2 || xQ == squares+1-leftOver/2 {
							fill = draw.FillFromRGBA(colorMap[xQ])
						}
					}
				}
			}

			// big arm extension
			if yQ == armIndex+1 && !invader.armsUp && invader.armSize == 3 {