/**** 
* Plugins
****/ 
var tween = LK.import("@upit/tween.v1");

/**** 
* Classes
****/ 
// Enemy class
var Enemy = Container.expand(function () {
	var self = Container.call(this);
	var enemySprite = self.attachAsset('enemy', {
		anchorX: 0.5,
		anchorY: 0.5
	});
	self.width = enemySprite.width;
	self.height = enemySprite.height;
	self.speed = 12; // Downwards, can be increased by game speed
	self.alive = true;
	self.update = function () {
		self.y += self.speed * gameSpeed;
		if (self.y > GAMEPLAY_AREA_BOTTOM) {
			self.alpha = 0;
		} else {
			self.alpha = 1;
		}
	};
	return self;
});
// FireRateBoost drop
var FireRateBoost = Container.expand(function () {
	var self = Container.call(this);
	var sprite = self.attachAsset('speedBoost', {
		anchorX: 0.5,
		anchorY: 0.5,
		color: 0x00bfff // blue
	});
	self.width = sprite.width;
	self.height = sprite.height;
	self.speed = 10;
	self.type = 'firerate';
	self.update = function () {
		self.y += self.speed * gameSpeed;
	};
	return self;
});
// FireStyleBoost drop
var FireStyleBoost = Container.expand(function () {
	var self = Container.call(this);
	var sprite = self.attachAsset('speedBoost', {
		anchorX: 0.5,
		anchorY: 0.5,
		color: 0xff8800 // orange
	});
	self.width = sprite.width;
	self.height = sprite.height;
	self.speed = 10;
	self.type = 'firestyle';
	self.update = function () {
		self.y += self.speed * gameSpeed;
	};
	return self;
});
// Hero class
var Hero = Container.expand(function () {
	var self = Container.call(this);
	var heroSprite = self.attachAsset('hero', {
		anchorX: 0.5,
		anchorY: 0.5
	});
	self.width = heroSprite.width;
	self.height = heroSprite.height;
	self.shootCooldown = 0; // frames until next allowed shot
	// Shoot method
	self.shoot = function () {
		if (self.shootCooldown > 0) return;
		// Fire style: 1=single, 2=double, 3=triple
		if (fireStyle === 1) {
			var bullet = new HeroBullet();
			bullet.x = self.x;
			bullet.y = self.y - self.height / 2 - bullet.height / 2;
			heroBullets.push(bullet);
			game.addChild(bullet);
		} else if (fireStyle === 2) {
			for (var i = -1; i <= 1; i += 2) {
				var bullet = new HeroBullet();
				bullet.x = self.x + i * 40;
				bullet.y = self.y - self.height / 2 - bullet.height / 2;
				heroBullets.push(bullet);
				game.addChild(bullet);
			}
		} else if (fireStyle === 3) {
			for (var i = -1; i <= 1; i++) {
				var bullet = new HeroBullet();
				bullet.x = self.x + i * 40;
				bullet.y = self.y - self.height / 2 - bullet.height / 2;
				heroBullets.push(bullet);
				game.addChild(bullet);
			}
		}
		if (fireRateActive) {
			self.shootCooldown = 4; // much faster
		} else {
			self.shootCooldown = 12; // normal
		}
		LK.getSound('shoot').play();
	};
	// Called every tick
	self.update = function () {
		if (self.shootCooldown > 0) self.shootCooldown--;
	};
	return self;
});
// Hero bullet class
var HeroBullet = Container.expand(function () {
	var self = Container.call(this);
	var bulletSprite = self.attachAsset('heroBullet', {
		anchorX: 0.5,
		anchorY: 0.5
	});
	self.width = bulletSprite.width;
	self.height = bulletSprite.height;
	self.speed = -36; // Upwards
	self.update = function () {
		self.y += self.speed;
	};
	return self;
});
// ShieldBoost drop
var ShieldBoost = Container.expand(function () {
	var self = Container.call(this);
	var sprite = self.attachAsset('speedBoost', {
		anchorX: 0.5,
		anchorY: 0.5,
		color: 0x00ffea // cyan
	});
	self.width = sprite.width;
	self.height = sprite.height;
	self.speed = 10;
	self.type = 'shield';
	self.update = function () {
		self.y += self.speed * gameSpeed;
	};
	return self;
});
// Speed boost drop class
var SpeedBoost = Container.expand(function () {
	var self = Container.call(this);
	var boostSprite = self.attachAsset('speedBoost', {
		anchorX: 0.5,
		anchorY: 0.5
	});
	self.width = boostSprite.width;
	self.height = boostSprite.height;
	self.speed = 10;
	self.type = 'speed'; // default type
	self.update = function () {
		self.y += self.speed * gameSpeed;
	};
	return self;
});

/**** 
* Initialize Game
****/ 
var game = new LK.Game({
	backgroundColor: 0x111111
});

/**** 
* Game Code
****/ 
// --- Global variables ---
// Hero (player)
// Hero bullet
// Enemy
// Speed boost drop
// Road (background)
// Sound effects
// Music
// soldier: gray box
// tank: green box
// jungle green
var hero;
var heroBullets = [];
var enemies = [];
var boosts = [];
var lives = 3;
var gameTime = 0; // ms
var gameDuration = 20000; // 20 seconds
var gameSpeed = 1; // Multiplier, affected by boosts
var boostActive = false;
var boostTimer = 0;
// Power-up states
var fireRateActive = false;
var fireRateTimer = 0;
var fireStyle = 1; // 1: single, 2: double, 3: triple
var fireStyleTimer = 0;
var shieldActive = false;
var shieldTimer = 0;
var lastEnemySpawnTick = 0;
var enemySpawnInterval = 48; // frames (0.8s at 60fps)
var score = 0;
var finished = false;
var level = 1;
var killTarget = 10; // Level 1: 10, Level 2: 15, Level 3: 20, etc.
function updateKillTarget() {
	if (level === 1) {
		killTarget = 10;
	} else if (level === 2) {
		killTarget = 15;
	} else {
		killTarget = 15 + (level - 2) * 5;
	}
}
updateKillTarget();
// --- Define gameplay and control areas ---
var CONTROL_AREA_HEIGHT = 400; // px reserved for controller at bottom
var GAMEPLAY_AREA_TOP = 0;
var GAMEPLAY_AREA_BOTTOM = 2732 - CONTROL_AREA_HEIGHT;
// --- Define 5 lanes ---
var LANE_COUNT = 5;
var ROAD_WIDTH = 900;
var LANE_WIDTH = ROAD_WIDTH / LANE_COUNT; // 180px per lane
var ROAD_LEFT = 2048 / 2 - ROAD_WIDTH / 2;
var currentLane = 2; // Start hero in middle lane (0-4)
// Get lane center X position
function getLaneX(laneIndex) {
	return ROAD_LEFT + (laneIndex + 0.5) * LANE_WIDTH;
}
// --- Controller area background (army green) ---
var controllerBg = LK.getAsset('centerCircle', {
	anchorX: 0,
	anchorY: 0,
	width: 2048,
	height: CONTROL_AREA_HEIGHT,
	color: 0x4B5320,
	// army green
	x: 0,
	y: 2732 - CONTROL_AREA_HEIGHT
});
game.addChild(controllerBg);
// --- Road background (restricted to end at controller line) ---
var road = LK.getAsset('road', {
	anchorX: 0.5,
	anchorY: 0,
	width: 900,
	height: 2732 - CONTROL_AREA_HEIGHT // End at controller line
});
road.x = 2048 / 2;
road.y = 0;
game.addChild(road);
// --- Hero ---
hero = new Hero();
hero.x = getLaneX(currentLane);
hero.y = GAMEPLAY_AREA_BOTTOM - 150;
game.addChild(hero);
// --- GUI: Lives ---
var livesTxt = new Text2('♥♥♥', {
	size: 90,
	fill: 0xFF4D4D
});
livesTxt.anchor.set(0.5, 0);
LK.gui.top.addChild(livesTxt);
// --- GUI: Level/Target ---
var levelTxt = new Text2('Level 1', {
	size: 90,
	fill: 0xFFFFFF
});
levelTxt.anchor.set(0.5, 0);
LK.gui.topRight.addChild(levelTxt);
var targetTxt = new Text2('Target: 10', {
	size: 90,
	fill: 0xFFFFFF
});
targetTxt.anchor.set(0.5, 0);
LK.gui.bottom.addChild(targetTxt);
function updateLevelDisplay() {
	levelTxt.setText('Level ' + level);
	targetTxt.setText('Target: ' + killTarget);
}
// --- GUI: Score ---
var scoreTxt = new Text2('0', {
	size: 90,
	fill: 0xFFE066
});
scoreTxt.anchor.set(0.5, 0);
LK.gui.top.addChild(scoreTxt);
// --- Music ---
LK.playMusic('bgmusic');
// --- Helper: Update GUI ---
function updateLivesDisplay() {
	var s = '';
	for (var i = 0; i < lives; i++) s += '♥';
	livesTxt.setText(s);
}
function updateTimerDisplay() {
	// Timer GUI removed, nothing to update
}
function updateScoreDisplay() {
	scoreTxt.setText(score);
}
// --- Touch controls: Move hero, shoot ---
var dragHero = false;
var leftArrowBtn, rightArrowBtn, fireBtn;
var leftBtnPressed = false;
var rightBtnPressed = false;
var fireBtnPressed = false;
var controllerBtnSize = 200; // Size for arrow buttons
var fireBtnSize = 250; // Size for fire button
// Move hero to specific lane
function moveHeroToLane(newLane) {
	currentLane = Math.max(0, Math.min(LANE_COUNT - 1, newLane));
	hero.x = getLaneX(currentLane);
}
// Controller buttons (left arrow, right arrow, fire)
leftArrowBtn = LK.getAsset('leftArrow', {
	anchorX: 0.5,
	anchorY: 0.5
});
rightArrowBtn = LK.getAsset('rightArrow', {
	anchorX: 0.5,
	anchorY: 0.5
});
fireBtn = LK.getAsset('fireButton', {
	anchorX: 0.5,
	anchorY: 0.5
});
leftArrowBtn.alpha = 0.4;
rightArrowBtn.alpha = 0.4;
fireBtn.alpha = 0.4;
// Place buttons side-by-side in the control area
leftArrowBtn.x = 350;
leftArrowBtn.y = 2732 - CONTROL_AREA_HEIGHT / 2;
rightArrowBtn.x = 550;
rightArrowBtn.y = 2732 - CONTROL_AREA_HEIGHT / 2;
fireBtn.x = 2048 - 400;
fireBtn.y = 2732 - CONTROL_AREA_HEIGHT / 2;
// Make sure controllerBg is below buttons (already added above)
// Add controller buttons
game.addChild(leftArrowBtn);
game.addChild(rightArrowBtn);
game.addChild(fireBtn);
// Helper: check if point is inside button
function isInsideBtn(btn, x, y) {
	return x >= btn.x - btn.width / 2 && x <= btn.x + btn.width / 2 && y >= btn.y - btn.height / 2 && y <= btn.y + btn.height / 2;
}
game.down = function (x, y, obj) {
	// If touch is on left arrow, set pressed
	if (isInsideBtn(leftArrowBtn, x, y)) {
		leftBtnPressed = true;
		leftArrowBtn.alpha = 0.8;
		return;
	}
	// If touch is on right arrow, set pressed
	if (isInsideBtn(rightArrowBtn, x, y)) {
		rightBtnPressed = true;
		rightArrowBtn.alpha = 0.8;
		return;
	}
	// If touch is on fire button, set pressed and shoot
	if (isInsideBtn(fireBtn, x, y)) {
		fireBtnPressed = true;
		fireBtn.alpha = 0.8;
		hero.shoot();
		return;
	}
};
game.move = function (x, y, obj) {
	// Drag control disabled
	// If finger moves off controller, release
	if (!isInsideBtn(leftArrowBtn, x, y)) {
		leftBtnPressed = false;
		leftArrowBtn.alpha = 0.4;
	}
	if (!isInsideBtn(rightArrowBtn, x, y)) {
		rightBtnPressed = false;
		rightArrowBtn.alpha = 0.4;
	}
	if (!isInsideBtn(fireBtn, x, y)) {
		fireBtnPressed = false;
		fireBtn.alpha = 0.4;
	}
};
game.up = function (x, y, obj) {
	leftBtnPressed = false;
	rightBtnPressed = false;
	fireBtnPressed = false;
	leftArrowBtn.alpha = 0.4;
	rightArrowBtn.alpha = 0.4;
	fireBtn.alpha = 0.4;
};
// --- Enemy spawn logic ---
function spawnEnemy() {
	// Enemies spawn randomly in one of the 5 lanes
	var enemy = new Enemy();
	var enemyLane = Math.floor(Math.random() * LANE_COUNT);
	enemy.x = getLaneX(enemyLane);
	// Spawn at the top of the gameplay area, not above the road
	enemy.y = GAMEPLAY_AREA_TOP - enemy.height / 2 - 10;
	// Randomize speed a bit
	enemy.speed = 10 + Math.random() * 6;
	enemies.push(enemy);
	game.addChild(enemy);
}
// --- Boost logic ---
function spawnBoost(x, y) {
	// Randomly choose power-up type
	var r = Math.random();
	var boost;
	if (r < 0.25) {
		boost = new SpeedBoost();
	} else if (r < 0.5) {
		boost = new FireRateBoost();
	} else if (r < 0.75) {
		boost = new FireStyleBoost();
	} else {
		boost = new ShieldBoost();
	}
	boost.x = x;
	boost.y = y;
	boosts.push(boost);
	game.addChild(boost);
}
// --- Game update ---
game.update = function () {
	if (finished) return;
	// --- Controller movement ---
	if (leftBtnPressed) {
		moveHeroToLane(currentLane - 1);
		leftBtnPressed = false; // Single lane move per press
		leftArrowBtn.alpha = 0.4;
	}
	if (rightBtnPressed) {
		moveHeroToLane(currentLane + 1);
		rightBtnPressed = false; // Single lane move per press
		rightArrowBtn.alpha = 0.4;
	}
	// --- Timer ---
	// (Timer removed, no time-based win/lose condition)
	updateTimerDisplay();
	// --- Win condition ---
	// (Handled by kill target below)
	// --- Boost logic ---
	if (boostActive) {
		boostTimer -= 1000 / 60;
		if (boostTimer <= 0) {
			boostActive = false;
			gameSpeed = 1;
		}
	}
	if (fireRateActive) {
		fireRateTimer -= 1000 / 60;
		if (fireRateTimer <= 0) {
			fireRateActive = false;
		}
	}
	if (fireStyle > 1) {
		fireStyleTimer -= 1000 / 60;
		if (fireStyleTimer <= 0) {
			fireStyle = 1;
		}
	}
	if (shieldActive) {
		shieldTimer -= 1000 / 60;
		if (shieldTimer <= 0) {
			shieldActive = false;
		}
	}
	// --- Enemy spawn ---
	if (LK.ticks - lastEnemySpawnTick >= enemySpawnInterval) {
		spawnEnemy();
		lastEnemySpawnTick = LK.ticks;
	}
	// --- Update hero ---
	hero.update();
	// --- Update hero bullets ---
	for (var i = heroBullets.length - 1; i >= 0; i--) {
		var b = heroBullets[i];
		b.update();
		// Remove if off screen
		if (b.y < -b.height) {
			b.destroy();
			heroBullets.splice(i, 1);
		}
	}
	// --- Update enemies ---
	for (var i = enemies.length - 1; i >= 0; i--) {
		var e = enemies[i];
		e.update();
		// Remove if off screen
		if (e.y > 2732 + e.height) {
			e.destroy();
			enemies.splice(i, 1);
			continue;
		}
		// Check collision with hero
		if (e.alive && e.intersects(hero)) {
			if (shieldActive) {
				// Ignore hit, just destroy enemy
				LK.effects.flashObject(hero, 0x00ffea, 200);
				e.alive = false;
				e.destroy();
				enemies.splice(i, 1);
				continue;
			}
			// Lose a life, destroy enemy
			lives--;
			updateLivesDisplay();
			LK.getSound('hit').play();
			LK.effects.flashObject(hero, 0xff0000, 400);
			e.alive = false;
			e.destroy();
			enemies.splice(i, 1);
			// Game over?
			if (lives <= 0) {
				finished = true;
				LK.showGameOver();
				return;
			}
			continue;
		}
		// Check collision with hero bullets
		for (var j = heroBullets.length - 1; j >= 0; j--) {
			var b = heroBullets[j];
			if (e.alive && b.intersects(e)) {
				// Enemy down
				e.alive = false;
				e.destroy();
				enemies.splice(i, 1);
				b.destroy();
				heroBullets.splice(j, 1);
				score++;
				updateScoreDisplay();
				LK.getSound('enemyDown').play();
				// Win if enough enemies killed for this level
				if (score >= killTarget) {
					finished = true;
					LK.showYouWin();
					// Prepare for next level
					level++;
					updateKillTarget();
					updateLevelDisplay();
					return;
				}
				// Chance to drop boost (30%)
				if (Math.random() < 0.3) {
					spawnBoost(e.x, e.y);
				}
				break;
			}
		}
	}
	// --- Update boosts ---
	for (var i = boosts.length - 1; i >= 0; i--) {
		var boost = boosts[i];
		boost.update();
		// Remove if off screen
		if (boost.y > 2732 + boost.height) {
			boost.destroy();
			boosts.splice(i, 1);
			continue;
		}
		// Check collision with hero
		if (boost.intersects(hero)) {
			if (boost.type === 'speed') {
				boostActive = true;
				boostTimer = 2000; // 2 seconds
				gameSpeed = 2.2;
				LK.getSound('boost').play();
				LK.effects.flashObject(hero, 0x44e07b, 400);
			} else if (boost.type === 'firerate') {
				fireRateActive = true;
				fireRateTimer = 4000; // 4 seconds
				LK.effects.flashObject(hero, 0x00bfff, 400);
			} else if (boost.type === 'firestyle') {
				fireStyle = Math.min(3, fireStyle + 1); // double, then triple
				fireStyleTimer = 4000; // 4 seconds
				LK.effects.flashObject(hero, 0xff8800, 400);
			} else if (boost.type === 'shield') {
				shieldActive = true;
				shieldTimer = 4000; // 4 seconds
				LK.effects.flashObject(hero, 0x00ffea, 400);
			}
			boost.destroy();
			boosts.splice(i, 1);
		}
	}
};
// --- Initial GUI update ---
updateLivesDisplay();
updateScoreDisplay();
updateLevelDisplay();

===================================================================
--- original.js
+++ change.js
@@ -313,28 +313,19 @@
 	currentLane = Math.max(0, Math.min(LANE_COUNT - 1, newLane));
 	hero.x = getLaneX(currentLane);
 }
 // Controller buttons (left arrow, right arrow, fire)
-leftArrowBtn = LK.getAsset('centerCircle', {
+leftArrowBtn = LK.getAsset('leftArrow', {
 	anchorX: 0.5,
-	anchorY: 0.5,
-	width: controllerBtnSize,
-	height: controllerBtnSize,
-	color: 0x666666
+	anchorY: 0.5
 });
-rightArrowBtn = LK.getAsset('centerCircle', {
+rightArrowBtn = LK.getAsset('rightArrow', {
 	anchorX: 0.5,
-	anchorY: 0.5,
-	width: controllerBtnSize,
-	height: controllerBtnSize,
-	color: 0x666666
+	anchorY: 0.5
 });
-fireBtn = LK.getAsset('heroBullet', {
+fireBtn = LK.getAsset('fireButton', {
 	anchorX: 0.5,
-	anchorY: 0.5,
-	width: fireBtnSize,
-	height: fireBtnSize,
-	color: 0xff4444
+	anchorY: 0.5
 });
 leftArrowBtn.alpha = 0.4;
 rightArrowBtn.alpha = 0.4;
 fireBtn.alpha = 0.4;