CodenameEngine/source/funkin/game/Strum.hx

package funkin.game;

import flixel.math.FlxPoint;
import funkin.backend.system.Conductor;

class Strum extends FlxSprite {
	public var extra:Map<String, Dynamic> = [];

	/**
	 * Which animation suffix on characters that should be used when hitting notes.
	 */
	public var animSuffix:String = "";

	public var cpu = false; // Unused
	public var lastHit:Float = -5000;

	public var scrollSpeed:Null<Float> = null; // custom scroll speed per strum
	public var noteAngle:Null<Float> = null;

	public var lastDrawCameras(default, null):Array<FlxCamera> = [];

	public var getPressed:StrumLine->Bool = null;
	public var getJustPressed:StrumLine->Bool = null;
	public var getJustReleased:StrumLine->Bool = null;

	public inline function __getPressed(strumLine:StrumLine):Bool {
		return getPressed != null ? getPressed(strumLine) : switch(ID) {
			case 0: strumLine.controls.NOTE_LEFT;
			case 1: strumLine.controls.NOTE_DOWN;
			case 2: strumLine.controls.NOTE_UP;
			case 3: strumLine.controls.NOTE_RIGHT;
			default: false;
		}
	}
	public inline function __getJustPressed(strumLine:StrumLine) {
		return getJustPressed != null ? getJustPressed(strumLine) : switch(ID) {
			case 0: strumLine.controls.NOTE_LEFT_P;
			case 1: strumLine.controls.NOTE_DOWN_P;
			case 2: strumLine.controls.NOTE_UP_P;
			case 3: strumLine.controls.NOTE_RIGHT_P;
			default: false;
		}
	}
	public inline function __getJustReleased(strumLine:StrumLine) {
		return getJustReleased != null ? getJustReleased(strumLine) : switch(ID) {
			case 0: strumLine.controls.NOTE_LEFT_R;
			case 1: strumLine.controls.NOTE_DOWN_R;
			case 2: strumLine.controls.NOTE_UP_R;
			case 3: strumLine.controls.NOTE_RIGHT_R;
			default: false;
		}
	}

	public inline function getScrollSpeed(?note:Note):Float {
		if (note != null && note.scrollSpeed != null) return note.scrollSpeed;
		if (scrollSpeed != null) return scrollSpeed;
		if (PlayState.instance != null) return PlayState.instance.scrollSpeed;
		return 1;
	}

	public inline function getNotesAngle(?note:Note):Float {
		if (note != null && note.noteAngle != null) return note.noteAngle;
		if (noteAngle != null) return noteAngle;
		return angle;
	}

	public override function update(elapsed:Float) {
		super.update(elapsed);
		if (cpu) {
			if (lastHit + (Conductor.crochet / 2) < Conductor.songPosition && getAnim() == "confirm") {
				playAnim("static");
			}
		}
	}

	public override function draw() {
		lastDrawCameras = [for(c in cameras) c];
		super.draw();
	}

	@:noCompletion public static final PIX180:Float = 565.4866776461628; // 180 * Math.PI
	@:noCompletion public static final N_WIDTHDIV2:Float = Note.swagWidth / 2;

	public function updateNotePosition(daNote:Note) {
		if (!daNote.exists) return;

		daNote.__strumCameras = lastDrawCameras;
		daNote.__strum = this;
		daNote.scrollFactor.set(scrollFactor.x, scrollFactor.y);
		daNote.__noteAngle = getNotesAngle(daNote);
		daNote.angle = daNote.isSustainNote ? daNote.__noteAngle : angle;

		updateNotePos(daNote);
	}

	private inline function updateNotePos(daNote:Note) {
		if (daNote.strumRelativePos) {
			daNote.setPosition((this.width - daNote.width) / 2, (daNote.strumTime - Conductor.songPosition) * (0.45 * CoolUtil.quantize(getScrollSpeed(daNote), 100)));
			if (daNote.isSustainNote) daNote.y += N_WIDTHDIV2;
		} else {
			var offset = FlxPoint.get(0, (Conductor.songPosition - daNote.strumTime) * (0.45 * CoolUtil.quantize(getScrollSpeed(daNote), 100)));
			var realOffset = FlxPoint.get(0, 0);

			if (daNote.isSustainNote) offset.y -= N_WIDTHDIV2;

			if (Std.int(daNote.__noteAngle % 360) != 0) {
				var noteAngleCos = FlxMath.fastCos(daNote.__noteAngle / PIX180);
				var noteAngleSin = FlxMath.fastSin(daNote.__noteAngle / PIX180);

				var aOffset:FlxPoint = FlxPoint.get(
					(daNote.origin.x / daNote.scale.x) - daNote.offset.x,
					(daNote.origin.y / daNote.scale.y) - daNote.offset.y
				);
				realOffset.x = -aOffset.x + (noteAngleCos * (offset.x + aOffset.x)) + (noteAngleSin * (offset.y + aOffset.y));
				realOffset.y = -aOffset.y + (noteAngleSin * (offset.x + aOffset.x)) + (noteAngleCos * (offset.y + aOffset.y));

				aOffset.put();
			} else {
				realOffset.x = offset.x;
				realOffset.y = offset.y;
			}
			realOffset.y *= -1;

			daNote.setPosition(x + realOffset.x, y + realOffset.y);

			offset.put();
			realOffset.put();
		}
	}

	public inline function updateSustain(daNote:Note) {
		if (!daNote.isSustainNote) return;
		daNote.updateSustain(this);
	}

	public function updatePlayerInput(pressed:Bool, justPressed:Bool, justReleased:Bool) {
		switch(getAnim()) {
			case "confirm":
				if (justReleased || !pressed)
					playAnim("static");
			case "pressed":
				if (justReleased || !pressed)
					playAnim("static");
			case "static":
				if (justPressed || pressed)
					playAnim("pressed");
			case null:
				playAnim("static");
		}
	}

	public inline function press(time:Float) {
		lastHit = time;
		playAnim("confirm");
	}

	public function playAnim(anim:String, force:Bool = true) {
		animation.play(anim, force);
		centerOffsets();
		centerOrigin();
	}
	public function getAnim() {
		return animation.name;
	}
}