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

/**** 
* Classes
****/ 
var DropShape = Container.expand(function (shapeType, color, shapeForm) {
	var self = Container.call(this);
	self.shapeType = shapeType;
	self.color = color;
	self.shapeForm = shapeForm;
	self.speed = 0;
	self.targetY = 0;
	self.isMoving = false;
	self.lane = 0;
	self.row = -1;
	var assetId = color + shapeForm;
	var shapeGraphics = self.attachAsset(assetId, {
		anchorX: 0.5,
		anchorY: 0.5
	});
	self.startDrop = function (lane, startY, endY) {
		self.lane = lane;
		self.y = startY;
		self.targetY = endY;
		self.speed = 8;
		self.isMoving = true;
		// Add slot machine spinning effect
		tween(self, {
			rotation: Math.PI * 4
		}, {
			duration: 2000,
			easing: tween.easeOut
		});
	};
	self.update = function () {
		if (self.isMoving) {
			self.y += self.speed;
			// Realistic slot machine deceleration
			var distanceToTarget = Math.abs(self.targetY - self.y);
			if (distanceToTarget < 150) {
				self.speed = Math.max(0.5, self.speed * 0.85);
			}
			// Stop at target with bounce effect
			if (self.y >= self.targetY) {
				self.y = self.targetY;
				self.isMoving = false;
				self.speed = 0;
				// Add settling bounce animation
				tween(self, {
					scaleY: 0.9
				}, {
					duration: 100,
					onFinish: function onFinish() {
						tween(self, {
							scaleY: 1.0
						}, {
							duration: 150,
							easing: tween.bounceOut
						});
					}
				});
				// Calculate which row this shape is in
				self.row = Math.floor((self.y - gridStartY - cellMargin / 2) / cellHeight);
			}
		}
	};
	return self;
});
var Lever = Container.expand(function () {
	var self = Container.call(this);
	self.isPulled = false;
	self.canPull = true;
	var leverBase = self.attachAsset('lever', {
		anchorX: 0.5,
		anchorY: 1.0
	});
	var leverHandle = self.attachAsset('leverHandle', {
		anchorX: 0.5,
		anchorY: 0.5
	});
	leverHandle.y = -250;
	self.pull = function () {
		if (!self.canPull) return;
		self.canPull = false;
		self.isPulled = true;
		LK.getSound('leverPull').play();
		// Add lever base animation for more realistic feel
		tween(leverBase, {
			rotation: 0.1
		}, {
			duration: 100
		});
		// Animate lever pull with bounce effect
		tween(leverHandle, {
			y: -150,
			scaleX: 1.1,
			scaleY: 0.9
		}, {
			duration: 200,
			easing: tween.easeIn,
			onFinish: function onFinish() {
				tween(leverHandle, {
					y: -250,
					scaleX: 1.0,
					scaleY: 1.0
				}, {
					duration: 300,
					easing: tween.bounceOut,
					onFinish: function onFinish() {
						tween(leverBase, {
							rotation: 0
						}, {
							duration: 200
						});
						self.isPulled = false;
						self.canPull = true;
					}
				});
			}
		});
	};
	self.down = function (x, y, obj) {
		self.pull();
	};
	return self;
});

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

/**** 
* Game Code
****/ 
var lanes = [];
var dropShapes = [];
var gridStartX = 400; // Center the 4x5 grid better
var gridStartY = 600;
var cellMargin = 5; // 5 pixel margin between cells
var laneWidth = 300 + cellMargin; // Adjust for 4 columns with gaps and margins
var laneBackgroundWidth = 250; // Actual lane background width (smaller for visible gaps)
var cellHeight = 180 + cellMargin; // Adjust for 5 rows with margins
var shapeTypes = ['red', 'blue', 'green', 'yellow'];
var shapeForms = ['Shape', 'Circle'];
var lever;
var scoreTxt;
// Create lane backgrounds
for (var i = 0; i < 4; i++) {
	var laneBack = LK.getAsset('laneBackground', {
		width: laneBackgroundWidth,
		anchorX: 0.5,
		anchorY: 0,
		alpha: 0.3
	});
	laneBack.x = gridStartX + i * laneWidth + cellMargin / 2;
	laneBack.y = gridStartY + cellMargin / 2;
	game.addChild(laneBack);
	lanes.push(laneBack);
}
// Add horizontal grid lines for 4x5 grid structure
for (var row = 0; row <= 5; row++) {
	var gridLine = LK.getAsset('laneBackground', {
		width: 1200,
		// Width for 4 columns
		height: 4,
		anchorX: 0.5,
		anchorY: 0.5,
		alpha: 0.5
	});
	gridLine.x = gridStartX + 1.5 * laneWidth + cellMargin / 2; // Center on 4 columns
	gridLine.y = gridStartY + row * cellHeight + cellMargin / 2;
	game.addChild(gridLine);
}
// Add vertical grid lines between lanes for visual separation
for (var col = 1; col < 4; col++) {
	var verticalLine = LK.getAsset('laneBackground', {
		width: 4,
		height: 1000,
		anchorX: 0.5,
		anchorY: 0,
		alpha: 0.5
	});
	verticalLine.x = gridStartX + col * laneWidth - laneWidth / 2 + cellMargin / 2;
	verticalLine.y = gridStartY + cellMargin / 2;
	game.addChild(verticalLine);
}
// Highlight middle row (winning line) - row 2 in 4x5 grid
var winLine = LK.getAsset('laneBackground', {
	width: 1200,
	// Width for 4 columns
	height: 8,
	color: 0xffd700,
	anchorX: 0.5,
	anchorY: 0.5,
	alpha: 0.7
});
winLine.x = gridStartX + 1.5 * laneWidth + cellMargin / 2; // Center on 4 columns
winLine.y = gridStartY + 2 * cellHeight + cellMargin / 2; // Middle row (row 2 of 5)
game.addChild(winLine);
// Create lever
lever = game.addChild(new Lever());
lever.x = 1900; // Move further right
lever.y = 1400;
// Create score display
scoreTxt = new Text2('Score: 0', {
	size: 80,
	fill: 0xFFFFFF
});
scoreTxt.anchor.set(0.5, 0);
LK.gui.top.addChild(scoreTxt);
function updateScore() {
	scoreTxt.setText('Score: ' + LK.getScore());
}
function getRandomShape() {
	var color = shapeTypes[Math.floor(Math.random() * shapeTypes.length)];
	var form = shapeForms[Math.floor(Math.random() * shapeForms.length)];
	return {
		color: color,
		form: form
	};
}
function spawnShapes() {
	// Staggered spawning for authentic slot machine feel
	for (var i = 0; i < 4; i++) {
		(function (laneIndex) {
			LK.setTimeout(function () {
				var shapeData = getRandomShape();
				var shape = new DropShape(shapeData.color + shapeData.form, shapeData.color, shapeData.form);
				shape.x = gridStartX + laneIndex * laneWidth + cellMargin / 2;
				// Find the lowest available position in this lane
				var targetRow = 0;
				for (var row = 4; row >= 0; row--) {
					var occupied = false;
					for (var j = 0; j < dropShapes.length; j++) {
						if (dropShapes[j].lane === laneIndex && dropShapes[j].row === row) {
							occupied = true;
							break;
						}
					}
					if (!occupied) {
						targetRow = row;
						break;
					}
				}
				var targetY = gridStartY + targetRow * cellHeight + cellHeight / 2 + cellMargin / 2;
				shape.startDrop(laneIndex, gridStartY - 100, targetY);
				// Add initial scale animation
				shape.scaleX = 0.5;
				shape.scaleY = 0.5;
				tween(shape, {
					scaleX: 1.0,
					scaleY: 1.0
				}, {
					duration: 300,
					easing: tween.bounceOut
				});
				game.addChild(shape);
				dropShapes.push(shape);
			}, i * 150); // Stagger by 150ms per lane
		})(i);
	}
}
function checkMatches() {
	// Check middle row (row 2) for matches
	var middleRowShapes = [];
	for (var i = 0; i < dropShapes.length; i++) {
		if (dropShapes[i].row === 2) {
			middleRowShapes.push(dropShapes[i]);
		}
	}
	if (middleRowShapes.length < 3) return;
	// Count shapes by type and color
	var shapeCount = {};
	var colorCount = {};
	var shapeColorCount = {};
	for (var i = 0; i < middleRowShapes.length; i++) {
		var shape = middleRowShapes[i];
		shapeCount[shape.shapeForm] = (shapeCount[shape.shapeForm] || 0) + 1;
		colorCount[shape.color] = (colorCount[shape.color] || 0) + 1;
		shapeColorCount[shape.color + shape.shapeForm] = (shapeColorCount[shape.color + shape.shapeForm] || 0) + 1;
	}
	var points = 0;
	var matched = false;
	// Check for 4 same shapes + colors = 1000 points
	for (var key in shapeColorCount) {
		if (shapeColorCount[key] === 4) {
			points = 1000;
			matched = true;
			break;
		}
	}
	// Check for 4 same shapes = 100 points
	if (!matched) {
		for (var key in shapeCount) {
			if (shapeCount[key] === 4) {
				points = 100;
				matched = true;
				break;
			}
		}
	}
	// Check for 3 same shapes + colors = 300 points
	if (!matched) {
		for (var key in shapeColorCount) {
			if (shapeColorCount[key] === 3) {
				points = 300;
				matched = true;
				break;
			}
		}
	}
	// Check for 3 same shapes = 50 points
	if (!matched) {
		for (var key in shapeCount) {
			if (shapeCount[key] === 3) {
				points = 50;
				matched = true;
				break;
			}
		}
	}
	if (matched) {
		LK.setScore(LK.getScore() + points);
		updateScore();
		LK.getSound('scoreMatch').play();
		// Enhanced visual feedback for wins
		var flashColor = points >= 1000 ? 0xffd700 : points >= 300 ? 0xff6600 : 0xffffff;
		for (var i = 0; i < middleRowShapes.length; i++) {
			var shape = middleRowShapes[i];
			LK.effects.flashObject(shape, flashColor, 1000);
			// Add pulsing animation for winning shapes
			tween(shape, {
				scaleX: 1.3,
				scaleY: 1.3
			}, {
				duration: 300,
				easing: tween.easeOut,
				onFinish: function onFinish() {
					tween(shape, {
						scaleX: 1.0,
						scaleY: 1.0
					}, {
						duration: 200
					});
				}
			});
		}
		// Animate score text on big wins
		if (points >= 300) {
			tween(scoreTxt, {
				scaleX: 1.2,
				scaleY: 1.2
			}, {
				duration: 200,
				onFinish: function onFinish() {
					tween(scoreTxt, {
						scaleX: 1.0,
						scaleY: 1.0
					}, {
						duration: 200
					});
				}
			});
		}
	}
}
function cleanupOffscreenShapes() {
	for (var i = dropShapes.length - 1; i >= 0; i--) {
		var shape = dropShapes[i];
		// Clean up shapes that go beyond the bottom of the 4x5 grid
		if (shape.y > gridStartY + 5 * cellHeight) {
			shape.destroy();
			dropShapes.splice(i, 1);
		}
	}
}
var lastLeverPulled = false;
game.update = function () {
	// Check if lever was pulled
	var currentLeverPulled = lever.isPulled;
	if (!lastLeverPulled && currentLeverPulled) {
		spawnShapes();
	}
	lastLeverPulled = currentLeverPulled;
	// Check for completed drops and matches
	var allStopped = true;
	for (var i = 0; i < dropShapes.length; i++) {
		if (dropShapes[i].isMoving) {
			allStopped = false;
			break;
		}
	}
	if (allStopped && dropShapes.length > 0) {
		var hasNewlyStoppedShapes = false;
		for (var i = 0; i < dropShapes.length; i++) {
			if (dropShapes[i].row === 2 && !dropShapes[i].checked) {
				dropShapes[i].checked = true;
				hasNewlyStoppedShapes = true;
			}
		}
		if (hasNewlyStoppedShapes) {
			checkMatches();
		}
	}
	cleanupOffscreenShapes();
};
updateScore();

===================================================================
--- original.js
+++ change.js
@@ -372,9 +372,10 @@
 }
 function cleanupOffscreenShapes() {
 	for (var i = dropShapes.length - 1; i >= 0; i--) {
 		var shape = dropShapes[i];
-		if (shape.y > gridStartY + 5 * cellHeight + 100) {
+		// Clean up shapes that go beyond the bottom of the 4x5 grid
+		if (shape.y > gridStartY + 5 * cellHeight) {
 			shape.destroy();
 			dropShapes.splice(i, 1);
 		}
 	}