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 Vector2 offset { get; set; } = new Vector2(10, 10);
        public Vector2 scaleFactor { get; set; } = new Vector2(0.1f, 4f);

        public int holeSize { get; set; } = 3;

        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 MusicBox.HighestNote.NoteNumber;
			}
		}
		public int MinNoteNumber {
			get
			{
				return MusicBox.LowestNote.NoteNumber;
			}
		}

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

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

		public void Output(string destinationFilename)
		{
            Vector2 area = new Vector2(TrackLength, MaxNoteNumber - MinNoteNumber).Multiply(scaleFactor);
			Vector2 size = area.Add(offset.Multiply(2)).Add(new Vector2(100000, 1000));

			SvgWriter svg = new SvgWriter(destinationFilename, size.X.ToString(), size.Y.ToString());
			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(TrackLength * scaleFactor.X, 0)), "darkgreen", 1);
			}
			svg.WriteRectangle(offset, area);

			foreach(Note note in AllNotes())
			{
				Console.WriteLine("[{0}] {1}{2}/{3}", note.Time, note.NoteName, note.Octave, note.NoteNumber);
				svg.WriteCircle(
					new Vector2(
						offset.X + note.Time * scaleFactor.X,
						offset.Y + ((127 - note.NoteNumber) - (127 - MaxNoteNumber)) * 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;
				}
			}
		}
	}
}