using System;
using System.Collections.Generic;
using System.Linq;
using Melanchall.DryWetMidi.Smf;
using Melanchall.DryWetMidi.Smf.Interaction;

using SBRL.Utilities;

namespace MusicBoxConverter
{
    public class MusicBoxScoreGenerator
    {
		public bool Debug { get; set; } = false;

		public Vector2 Offset { get; set; } = new Vector2(10, 10);
		public Vector2 ScaleFactor { get; set; } = new Vector2(0.03f, 4f);

		public float HoleSize { get; set; } = 1f;

        private int trackLength;
        public int TrackLength
        {
            get
            {
                if (trackLength == 0)
                    trackLength = (int)AllNotes().Max((Note arg) => arg.Time);
                return trackLength;
            }
        }

		public int MaxNoteNumber {
			get {
				return SelectedMusicBox.HighestNote.NoteNumber;
			}
		}
		public int MinNoteNumber {
			get
			{
				return SelectedMusicBox.LowestNote.NoteNumber;
			}
		}

		public MusicBox SelectedMusicBox { get; private set; }

		private MidiFile midiFile;
		
		public MusicBoxScoreGenerator(string filename, MusicBox inMusicBox)
		{
            SelectedMusicBox = inMusicBox;
			midiFile = MidiFile.Read(filename);

			// Set the scale factor based on the strip height of the music box
			ScaleFactor = new Vector2(
				ScaleFactor.X,
				SelectedMusicBox.StripHeightMm / (SelectedMusicBox.NoteCount-1)
			);

            foreach(Note note in AllNotes()) {
                if (!SelectedMusicBox.IsValidNote(note))
                    Console.Error.WriteLine($"Warning: The note {note} at {note.Time} can't be played by the {SelectedMusicBox}.");
            }
		}

		public void Output(string destinationFilename)
		{
			Vector2 area = new Vector2(TrackLength, SelectedMusicBox.NoteCount-1).Multiply(ScaleFactor);
			Vector2 size = area.Add(Offset.Multiply(2));

			SvgWriter svg = new SvgWriter(
				destinationFilename,
				$"{size.X}mm", $"{size.Y}mm"
			);
			svg.UnitSuffix = "mm";

			for(float i = 0; i < area.Y; i += ScaleFactor.Y)
			{
				Vector2 start = Offset.Add(new Vector2(0, i));
				svg.WriteLine(start, start.Add(new Vector2(area.X, 0)), "darkgreen", 0.75f);
			}

			svg.WriteRectangle(Offset, area, "red", 0.75f);

			foreach(Note note in AllNotes())
			{
				if(Debug) {
					Console.WriteLine(
						"[Note] {0}: {1}{2}/{3}",
						note.Time,
						note.NoteName,
						note.Octave,
						note.NoteNumber
					);
				}

				svg.WriteCircle(
					new Vector2(
						Offset.X + note.Time * ScaleFactor.X,
						Offset.Y + ((SelectedMusicBox.NoteCount-1) - SelectedMusicBox.NoteToBoxNumber(note)) * ScaleFactor.Y
					),
					HoleSize // radius
				);
			}

			svg.Complete();
		}

		public IEnumerable<Note> AllNotes()
		{
			foreach(TrackChunk chunk in midiFile.Chunks.OfType<TrackChunk>())
			{
				using(NotesManager notesManager = new NotesManager(chunk.Events))
				{
					foreach(Note note in notesManager.Notes)
						yield return note;
				}
			}
		}
	}
}