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>;

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;
	}
}

class AtlasMenuItem extends MenuItem
{
	var atlas:FlxAtlasFrames;

	public function new(x = 0.0, y = 0.0, name:String, atlas:FlxAtlasFrames, 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);
	}

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

	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;
	}
}