Funkin/source/funkin/ui/AtlasMenuList.hx

package funkin.ui;

import funkin.ui.MenuList;
import flixel.graphics.frames.FlxAtlasFrames;

typedef AtlasAsset = flixel.util.typeLimit.OneOfTwo<String, FlxAtlasFrames>;

/**
 * A menulist whose items share a single texture atlas.
 */
class AtlasMenuList extends MenuTypedList<AtlasMenuItem>
{
	public var atlas:FlxAtlasFrames;

	public function new(atlas, navControls:NavControls = Vertical, ?wrapMode)
	{
		super(navControls, wrapMode);

		if (Std.is(atlas, String))
			this.atlas = Paths.getSparrowAtlas(cast atlas);
		else
			this.atlas = cast atlas;
	}

	public function createItem(x = 0.0, y = 0.0, name, callback, fireInstantly = false)
	{
		var item = new AtlasMenuItem(x, y, name, atlas, callback);
		item.fireInstantly = fireInstantly;
		return addItem(name, item);
	}

	override function destroy()
	{
		super.destroy();
		atlas = null;
	}
}

/**
 * A menu list item which uses single texture atlas.
 */
class AtlasMenuItem extends MenuItem
{
	var atlas:FlxAtlasFrames;

	public var centered:Bool = false;

	public function new(x = 0.0, y = 0.0, name:String, atlas, callback)
	{
		this.atlas = atlas;
		super(x, y, name, callback);
	}

	override function setData(name:String, ?callback:Void->Void)
	{
		frames = atlas;
		animation.addByPrefix('idle', '$name idle', 24);
		animation.addByPrefix('selected', '$name selected', 24);

		super.setData(name, callback);
	}

	public function changeAnim(animName:String)
	{
		animation.play(animName);
		updateHitbox();

		if (centered)
		{
			// position by center
			centerOrigin();
			offset.copyFrom(origin);
		}
	}

	override function idle()
	{
		changeAnim('idle');
	}

	override function select()
	{
		changeAnim('selected');
	}

	override function get_selected()
	{
		return animation.curAnim != null && animation.curAnim.name == "selected";
	}

	override function destroy()
	{
		super.destroy();
		atlas = null;
	}
}