Funkin/source/ui/AtlasMenuList.hx

package ui;

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