From aa6611ff2f311f6982f50e6b6971d3f98b81540b Mon Sep 17 00:00:00 2001
From: Nic0Nic0Nii <sigonasr2@yahoo.com>
Date: Tue, 12 Oct 2021 06:05:12 +0000
Subject: [PATCH] Draw Editor Notes and Beat lines.

Co-authored-by: sigonasr2 <sigonasr2@gmail.com>
---
 LLSIG/src/main/java/LLSIG/Canvas.java | 161 +++++++++++++++-----------
 LLSIG/src/main/java/LLSIG/LLSIG.java  |  32 ++---
 2 files changed, 115 insertions(+), 78 deletions(-)

diff --git a/LLSIG/src/main/java/LLSIG/Canvas.java b/LLSIG/src/main/java/LLSIG/Canvas.java
index b207ef7..1e69cd5 100644
--- a/LLSIG/src/main/java/LLSIG/Canvas.java
+++ b/LLSIG/src/main/java/LLSIG/Canvas.java
@@ -6,6 +6,8 @@ import java.awt.geom.Rectangle2D;
 import java.awt.font.FontRenderContext;
 import java.awt.Font;
 import java.util.ArrayList;
+import java.util.List;
+import java.util.stream.Collectors;
 
 import javax.swing.JPanel;
 
@@ -55,80 +57,109 @@ public class Canvas extends JPanel implements AudioSpectrumListener{
 				SpectrographBar sb = spectrograph.get(i);
 				sb.draw(g,SPECTROBAR_SIZE*i,SPECTROBAR_SIZE,this.getHeight());
 			}
-
 			g.setColor(Color.WHITE);
 			g.drawString(Double.toString(LLSIG.game.musicPlayer.getPlayPosition()),0,32);
-			if (LLSIG.game.BPM_MEASURE) {
-				g.drawString("Average BPM: "+LLSIG.approximateBPM(),MIDDLE_X-128,MIDDLE_Y+64);
-			} else 
-			if (LLSIG.game.METRONOME) {
-				g.drawString("Offset: "+LLSIG.testOffset,MIDDLE_X-128,MIDDLE_Y+64);
+
+			if (LLSIG.game.EDITOR) {
+				final int MARGIN_X = 32;
+				final int BEAT_RANGE = 6;
+				final int BEAT_SPACING = 64;
+				final int NOTE_Y=(int)(MIDDLE_Y-NOTE_SIZE/2);
+				for (int y=-BEAT_RANGE;y<BEAT_RANGE;y++) {
+					g.setColor(Color.GRAY);
+					g.fillRect(MARGIN_X,(int)(NOTE_Y+(y)*BEAT_SPACING),this.getWidth()-MARGIN_X,4);
+					g.setColor(Color.DARK_GRAY);
+					for (int yy=0;yy<LLSIG.EDITOR_BEAT_DIVISIONS;yy++) {
+						g.fillRect(MARGIN_X*2,(int)(NOTE_Y+(y)*BEAT_SPACING)+(BEAT_SPACING/LLSIG.EDITOR_BEAT_DIVISIONS)*yy,this.getWidth()-MARGIN_X*2,2);
+					}
+				}
+				for (int i=0;i<9;i++) {
+					if (LLSIG.keyState[i]) {
+						g.setColor(Color.MAGENTA);
+					} else {
+						g.setColor(Color.GRAY);
+					}
+					int NOTE_X=(int)(((this.getWidth()-MARGIN_X*2)/9)*i-NOTE_SIZE/2+MARGIN_X);
+					g.fillOval(NOTE_X,NOTE_Y,NOTE_SIZE,NOTE_SIZE);
+					Lane lane = LLSIG.game.lanes.get(i);
+					List<Note> notes = lane.noteChart.stream().filter((note)->Math.abs(LLSIG.EDITOR_CURSOR_BEAT-note.beatSnapStart)<BEAT_RANGE).collect(Collectors.toList());
+					for (Note n : notes) {
+						g.fillOval(NOTE_X-NOTE_SIZE/2,(int)(NOTE_Y+(LLSIG.EDITOR_CURSOR_BEAT-n.beatSnapStart)*BEAT_SPACING-NOTE_SIZE/2),NOTE_SIZE,NOTE_SIZE);
+					}
+				}
 			} else {
-				g.setColor(LLSIG.game.musicPlayer.getPlayPosition()-LLSIG.LAST_PERFECT<500?Color.WHITE:Color.DARK_GRAY);g.drawString("PERFECT: "+LLSIG.PERFECT_COUNT,MIDDLE_X-128,MIDDLE_Y-96);
-				g.setColor(LLSIG.game.musicPlayer.getPlayPosition()-LLSIG.LAST_EXCELLENT<500?Color.WHITE:Color.DARK_GRAY);g.drawString("EXCELLENT: "+LLSIG.EXCELLENT_COUNT,MIDDLE_X-128,MIDDLE_Y-64);
-				g.setColor(LLSIG.game.musicPlayer.getPlayPosition()-LLSIG.LAST_GREAT<500?Color.WHITE:Color.DARK_GRAY);g.drawString("GREAT: "+LLSIG.GREAT_COUNT,MIDDLE_X-128,MIDDLE_Y-32);
-				g.setColor(LLSIG.game.musicPlayer.getPlayPosition()-LLSIG.LAST_EARLY<500?Color.WHITE:Color.DARK_GRAY);g.drawString("EARLY: "+LLSIG.EARLY_COUNT,MIDDLE_X-128,MIDDLE_Y);
-				g.setColor(LLSIG.game.musicPlayer.getPlayPosition()-LLSIG.LAST_LATE<500?Color.WHITE:Color.DARK_GRAY);g.drawString("LATE: "+LLSIG.LATE_COUNT,MIDDLE_X-128,MIDDLE_Y+32);
-				g.setColor(LLSIG.game.musicPlayer.getPlayPosition()-LLSIG.LAST_MISS<500?Color.WHITE:Color.DARK_GRAY);g.drawString("MISS: "+LLSIG.MISS_COUNT,MIDDLE_X-128,MIDDLE_Y+64);
-			}
-			g.setColor(Color.WHITE); 
-			String comboString = "x"+LLSIG.COMBO+" combo";
-			Rectangle2D bounds = calculateStringBoundsFont(comboString, g.getFont());
-			g.drawString(comboString,(int)(MIDDLE_X-bounds.getCenterX()),MIDDLE_Y+164);
-			for (int i=0;i<9;i++) {
-				int LANE_X_OFFSET = (i-5)*LANE_SPACING_X+LANE_SPACING_X/2+JUDGEMENT_LINE_WIDTH/2;
-				
-				if (LLSIG.keyState[i]) {
-					g.setColor(Color.MAGENTA);
+				if (LLSIG.game.BPM_MEASURE) {
+					g.drawString("Average BPM: "+LLSIG.approximateBPM(),MIDDLE_X-128,MIDDLE_Y+64);
+				} else 
+				if (LLSIG.game.METRONOME) {
+					g.drawString("Offset: "+LLSIG.testOffset,MIDDLE_X-128,MIDDLE_Y+64);
 				} else {
-					g.setColor(Color.GRAY);
+					g.setColor(LLSIG.game.musicPlayer.getPlayPosition()-LLSIG.LAST_PERFECT<500?Color.WHITE:Color.DARK_GRAY);g.drawString("PERFECT: "+LLSIG.PERFECT_COUNT,MIDDLE_X-128,MIDDLE_Y-96);
+					g.setColor(LLSIG.game.musicPlayer.getPlayPosition()-LLSIG.LAST_EXCELLENT<500?Color.WHITE:Color.DARK_GRAY);g.drawString("EXCELLENT: "+LLSIG.EXCELLENT_COUNT,MIDDLE_X-128,MIDDLE_Y-64);
+					g.setColor(LLSIG.game.musicPlayer.getPlayPosition()-LLSIG.LAST_GREAT<500?Color.WHITE:Color.DARK_GRAY);g.drawString("GREAT: "+LLSIG.GREAT_COUNT,MIDDLE_X-128,MIDDLE_Y-32);
+					g.setColor(LLSIG.game.musicPlayer.getPlayPosition()-LLSIG.LAST_EARLY<500?Color.WHITE:Color.DARK_GRAY);g.drawString("EARLY: "+LLSIG.EARLY_COUNT,MIDDLE_X-128,MIDDLE_Y);
+					g.setColor(LLSIG.game.musicPlayer.getPlayPosition()-LLSIG.LAST_LATE<500?Color.WHITE:Color.DARK_GRAY);g.drawString("LATE: "+LLSIG.LATE_COUNT,MIDDLE_X-128,MIDDLE_Y+32);
+					g.setColor(LLSIG.game.musicPlayer.getPlayPosition()-LLSIG.LAST_MISS<500?Color.WHITE:Color.DARK_GRAY);g.drawString("MISS: "+LLSIG.MISS_COUNT,MIDDLE_X-128,MIDDLE_Y+64);
 				}
-				//g.fillRect(MIDDLE_X-JUDGEMENT_LINE_WIDTH/2+LANE_X_OFFSET,MIDDLE_Y-JUDGEMENT_LINE_HEIGHT/2,JUDGEMENT_LINE_WIDTH,JUDGEMENT_LINE_HEIGHT);
-				int NOTE_X=(int)(MIDDLE_X-Math.cos(Math.toRadians(22.5*i))*NOTE_DISTANCE-NOTE_SIZE/2);
-				int NOTE_Y=(int)(MIDDLE_Y+Math.sin(Math.toRadians(22.5*i))*NOTE_DISTANCE-NOTE_SIZE/2);
-				g.fillOval(NOTE_X,NOTE_Y,NOTE_SIZE,NOTE_SIZE);
-				
-				Lane lane = LLSIG.game.lanes.get(i);
-				if (LLSIG.game.PLAYING) {
-					if (LLSIG.game.musicPlayer.getPlayPosition()-lane.lastNote<500) {
-						switch (lane.lastRating) {
-							case PERFECT:
-								g.setColor(colorList[(LLSIG.game.frameCount/10)%colorList.length]);
-								break;
-							case EXCELLENT:
-								g.setColor(Color.YELLOW);
-								break;
-							case GREAT:
-								g.setColor(Color.GREEN);
-								break;
-							case EARLY:
-								g.setColor(Color.CYAN);
-								break;
-							case LATE:
-								g.setColor(Color.MAGENTA);
-								break;
-							case MISS:
-								g.setColor(Color.RED);
-								break;
-							default:
-								break;
+				g.setColor(Color.WHITE); 
+				String comboString = "x"+LLSIG.COMBO+" combo";
+				Rectangle2D bounds = calculateStringBoundsFont(comboString, g.getFont());
+				g.drawString(comboString,(int)(MIDDLE_X-bounds.getCenterX()),MIDDLE_Y+164);
+				for (int i=0;i<9;i++) {
+					int LANE_X_OFFSET = (i-5)*LANE_SPACING_X+LANE_SPACING_X/2+JUDGEMENT_LINE_WIDTH/2;
+					
+					if (LLSIG.keyState[i]) {
+						g.setColor(Color.MAGENTA);
+					} else {
+						g.setColor(Color.GRAY);
+					}
+					//g.fillRect(MIDDLE_X-JUDGEMENT_LINE_WIDTH/2+LANE_X_OFFSET,MIDDLE_Y-JUDGEMENT_LINE_HEIGHT/2,JUDGEMENT_LINE_WIDTH,JUDGEMENT_LINE_HEIGHT);
+					int NOTE_X=(int)(MIDDLE_X-Math.cos(Math.toRadians(22.5*i))*NOTE_DISTANCE-NOTE_SIZE/2);
+					int NOTE_Y=(int)(MIDDLE_Y+Math.sin(Math.toRadians(22.5*i))*NOTE_DISTANCE-NOTE_SIZE/2);
+					g.fillOval(NOTE_X,NOTE_Y,NOTE_SIZE,NOTE_SIZE);
+					
+					Lane lane = LLSIG.game.lanes.get(i);
+					if (LLSIG.game.PLAYING) {
+						if (LLSIG.game.musicPlayer.getPlayPosition()-lane.lastNote<500) {
+							switch (lane.lastRating) {
+								case PERFECT:
+									g.setColor(colorList[(LLSIG.game.frameCount/10)%colorList.length]);
+									break;
+								case EXCELLENT:
+									g.setColor(Color.YELLOW);
+									break;
+								case GREAT:
+									g.setColor(Color.GREEN);
+									break;
+								case EARLY:
+									g.setColor(Color.CYAN);
+									break;
+								case LATE:
+									g.setColor(Color.MAGENTA);
+									break;
+								case MISS:
+									g.setColor(Color.RED);
+									break;
+								default:
+									break;
+							}
+							Rectangle2D textBounds = calculateStringBoundsFont(lane.lastRating.name(), g.getFont());
+							g.drawString(lane.lastRating.name(),(int)(NOTE_X-textBounds.getCenterX()),(int)(NOTE_Y+textBounds.getHeight()));
 						}
-						Rectangle2D textBounds = calculateStringBoundsFont(lane.lastRating.name(), g.getFont());
-						g.drawString(lane.lastRating.name(),(int)(NOTE_X-textBounds.getCenterX()),(int)(NOTE_Y+textBounds.getHeight()));
 					}
-				}
-				g.setColor(NOTE_COLOR);
-				int noteCounter = 0;
-				while (lane.noteExists(noteCounter)) {
-					Note n = lane.getNote(noteCounter);
-					if (n.active) {
-						double PLAYTIME_RATIO = (1-(((double)n.getStartFrame()-LLSIG.game.musicPlayer.getPlayPosition())/LLSIG.game.NOTE_SPEED));
-						if (n.getStartFrame()-LLSIG.game.musicPlayer.getPlayPosition()<LLSIG.game.NOTE_SPEED) {
-							g.fillOval((int)(MIDDLE_X-Math.cos(Math.toRadians(22.5*i))*(PLAYTIME_RATIO*NOTE_DISTANCE)-NOTE_SIZE/2),(int)(MIDDLE_Y+Math.sin(Math.toRadians(22.5*i))*(PLAYTIME_RATIO*NOTE_DISTANCE)-NOTE_SIZE/2),
-									(int)(Math.min(PLAYTIME_RATIO/2+0.5,1)*NOTE_SIZE),(int)(Math.min(PLAYTIME_RATIO/2+0.5,1)*NOTE_SIZE));
+					g.setColor(NOTE_COLOR);
+					int noteCounter = 0;
+					while (lane.noteExists(noteCounter)) {
+						Note n = lane.getNote(noteCounter);
+						if (n.active) {
+							double PLAYTIME_RATIO = (1-(((double)n.getStartFrame()-LLSIG.game.musicPlayer.getPlayPosition())/LLSIG.game.NOTE_SPEED));
+							if (n.getStartFrame()-LLSIG.game.musicPlayer.getPlayPosition()<LLSIG.game.NOTE_SPEED) {
+								g.fillOval((int)(MIDDLE_X-Math.cos(Math.toRadians(22.5*i))*(PLAYTIME_RATIO*NOTE_DISTANCE)-NOTE_SIZE/2),(int)(MIDDLE_Y+Math.sin(Math.toRadians(22.5*i))*(PLAYTIME_RATIO*NOTE_DISTANCE)-NOTE_SIZE/2),
+										(int)(Math.min(PLAYTIME_RATIO/2+0.5,1)*NOTE_SIZE),(int)(Math.min(PLAYTIME_RATIO/2+0.5,1)*NOTE_SIZE));
+							}
 						}
+						noteCounter++;
 					}
-					noteCounter++;
 				}
 			}
 		}
diff --git a/LLSIG/src/main/java/LLSIG/LLSIG.java b/LLSIG/src/main/java/LLSIG/LLSIG.java
index e325a15..ae6ca49 100644
--- a/LLSIG/src/main/java/LLSIG/LLSIG.java
+++ b/LLSIG/src/main/java/LLSIG/LLSIG.java
@@ -50,6 +50,12 @@ public class LLSIG implements KeyListener{
 	public boolean METRONOME = false;
 	public boolean BPM_MEASURE = false;
 	public boolean PLAYING = true; //Whether or not a song is loaded and playing.
+	public boolean EDITOR = false; //Whether or not we are in beatmap editing mode.
+
+	public static int EDITOR_CURSOR_BEAT = 0;
+	public static int EDITOR_BEAT_DIVISIONS = 4;
+	public static BeatTiming EDITOR_CURSOR_WINDOW;
+
 	public static int beatNumber = 0;
 
 	public static boolean[] lanePress = new boolean[9]; //A lane is being requested to being pressed.
@@ -81,10 +87,7 @@ public class LLSIG implements KeyListener{
 	
 	LLSIG(JFrame f) {
 		
-		Platform.startup(() ->
-		{
-		    // This block will be executed on JavaFX Thread
-		});
+		Platform.startup(()->{});
 		
 		this.window = f;
 		
@@ -113,7 +116,9 @@ public class LLSIG implements KeyListener{
 		PLAYING = new File("music/"+song+".mp3").exists();
 		if (PLAYING)  {
 			this.musicPlayer = new Player(Paths.get("music/"+song+".mp3").toUri().toString());
-			musicPlayer.play();
+			if (!EDITOR) {
+				musicPlayer.play();
+			}
 			
 			LoadSongData(song,lanes);
 		}
@@ -136,6 +141,7 @@ public class LLSIG implements KeyListener{
     							bpm=bt.bpm;
     							offset=bt.offset;
     							beatDelay = ((1/((double)bpm/60))*1000);
+								beatNumber=0;
     							System.out.println("BPM is "+bpm+". Delay is "+beatDelay);
     						}
     					}
@@ -298,15 +304,15 @@ public class LLSIG implements KeyListener{
 			case KeyEvent.VK_P:{if (LLSIG.game.PLAYING&&musicPlayer.isPaused()) {musicPlayer.resume();} else {musicPlayer.pause();}}break;
 			case KeyEvent.VK_Q:{if (LLSIG.game.PLAYING) {musicPlayer.pause();SaveSongData(song,lanes);}}break;
 		}
-		if (LLSIG.game.PLAYING&&lane!=-1&&EDITMODE) {
-			Note n = new Note(NoteType.NORMAL,musicPlayer.getPlayPosition());
-			n.active=false;
-			System.out.println(Math.round(((musicPlayer.getPlayPosition()-offset)/beatDelay)*4)/(double)4);
-			n.setBeatSnap(Math.round(((musicPlayer.getPlayPosition()-offset)/beatDelay)*4)/(double)4);
-			LLSIG.game.lanes.get(lane).addNote(n);
-		}
 		if (lane!=-1) {
-			if (PLAYING&&!EDITMODE) {
+			if (PLAYING&&EDITMODE) {
+				Note n = new Note(NoteType.NORMAL,musicPlayer.getPlayPosition());
+				n.active=false;
+				System.out.println(Math.round(((musicPlayer.getPlayPosition()-offset)/beatDelay)*4)/(double)4);
+				n.setBeatSnap(Math.round(((musicPlayer.getPlayPosition()-offset)/beatDelay)*4)/(double)4);
+				LLSIG.game.lanes.get(lane).addNote(n);
+			}
+			if (PLAYING&&!EDITMODE&&!EDITOR) {
 				Lane l = lanes.get(lane);
 				if (l.noteExists()) {
 					Note n = l.getNote();