#Powershell Magic Back once again with the Powershell 🥓 flavour

# Load DLLs
Add-Type -Path "C:\Music\CZ-1\NAudio.Core.dll"
Add-Type -Path "C:\Music\CZ-1\NAudio.Midi.dll"
Add-Type -Path "C:\Music\CZ-1\DryWetMIDI.dll"

function Invoke-Midi {
    param (
        [NAudio.Midi.MidiOut]$Device1,   # e.g. CZ-1
        [NAudio.Midi.MidiOut]$Device2,   # e.g. P-6
        [int]$BPM = 120,
        [string]$MidiFilePath,
        [int]$StartOffset = 0,           # delay in bars before CZ-1 plays
        [int]$BeatsPerBar = 4            # default 4/4 time
    )

    if (-not (Test-Path $MidiFilePath)) {
        Write-Warning "MIDI file not found: $MidiFilePath"
        return
    }

    # Calculate tick timing
    $msPerTick = 60000 / ($BPM * 24)

    # Calculate offset ticks from bars
    $cz1StartOffsetTicks = $StartOffset * $BeatsPerBar * 24

    # Load MIDI notes
    $midiFile = [Melanchall.DryWetMidi.Core.MidiFile]::Read($MidiFilePath)
    $ticksPerQuarterNote = $midiFile.TimeDivision.TicksPerQuarterNote
    $notes = [Melanchall.DryWetMidi.Interaction.NotesManagingUtilities]::GetNotes($midiFile)

    # Normalize timing to start at tick 0
    $minTime = ($notes | Measure-Object -Property Time -Minimum).Minimum

    # Convert notes to MIDI Clock tick schedule
    $noteData = foreach ($note in $notes) {
        $relativeTime = $note.Time - $minTime
        $startTick = [math]::Round(($relativeTime / $ticksPerQuarterNote) * 24)
        $durationTicks = [math]::Round(($note.Length / $ticksPerQuarterNote) * 24)
        [PSCustomObject]@{
            NoteNumber     = $note.NoteNumber
            Velocity       = $note.Velocity
            Channel        = $note.Channel
            StartTick      = $startTick
            DurationTicks  = $durationTicks
        }
    }
    $noteData = $noteData | Sort-Object StartTick

    # Send MIDI Start
    $Device1.Send(0xFA)
    $Device2.Send(0xFA)

    # Start clock and playback
    $sw = [System.Diagnostics.Stopwatch]::StartNew()
    $tickCount = 0
    $noteIndex = 0
    while ($noteIndex -lt $noteData.Count -or $pendingNotes.Count -gt 0) {
        $tickStart = $sw.Elapsed.TotalMilliseconds
        $Device2.Send(0xF8)   # send clock
        $tickCount++
        $pendingNotes = @()
        # Trigger notes
        while ($noteIndex -lt $noteData.Count -and $noteData[$noteIndex].StartTick -le $tickCount) {
            $note = $noteData[$noteIndex]

            # Only send CZ-1 notes after offset ticks
            if ($tickCount -ge $cz1StartOffsetTicks) {
                #Write-Host "Tick $tickCount : CZ-1 NoteOn $($note.NoteNumber) Velocity $($note.Velocity) DurationTicks $($note.DurationTicks)"
                $noteOn = [NAudio.Midi.MidiMessage]::StartNote($note.NoteNumber, $note.Velocity, 1).RawData
                $Device1.Send($noteOn)

                # Schedule note off
                $noteOffTick = $tickCount + $note.DurationTicks
                $pendingNotes += [PSCustomObject]@{
                    NoteNumber = $note.NoteNumber
                    NoteOff = [NAudio.Midi.MidiMessage]::StopNote($note.NoteNumber, $note.Velocity, 1).RawData
                    Tick    = $noteOffTick
                }
            }

            $noteIndex++
        }

        # Trigger pending note offs
        $toRemove = @()
        foreach ($n in $pendingNotes) {
            if ($tickCount -ge $n.Tick) {
                Write-Host "Tick $tickCount : CZ-1 NoteOff triggered"
                $Device1.Send($n.NoteOff)
                $toRemove += $n
            }
        }
        $pendingNotes = $pendingNotes | Where-Object { $toRemove -notcontains $_ }

        # Wait until next tick
        do {
            $tickNow = $sw.Elapsed.TotalMilliseconds
        } while (($tickNow - $tickStart) -lt $msPerTick)
    }

    # Flush any remaining NoteOffs
    foreach ($n in $pendingNotes) {
    $noteOffMsg = [NAudio.Midi.MidiMessage]::StopNote($n.NoteNumber, $n.Velocity, 1).RawData
    Write-Host "Final cleanup: CZ-1 NoteOff for note $($n.NoteNumber)"
    $midiOutCZ1.Send($noteOffMsg)
    }
    $pendingNotes = @()

    # Send MIDI Stop
    $Device1.Send(0xFC)
    $Device2.Send(0xFC)
}
function Set-CZ1Sound
{
    param(
            [int]$channel = 0,   # 0 = MIDI channel 1
            [int]$program = 0,    # 0 = pattern number 1
            [NAudio.Midi.MidiOut]$CZ1
        )
    $status = 0xC0 -bor $channel
    $message = $status -bor ($program -shl 8)
    $CZ1.Send($message)
    Write-Host "Sent Sound Change: Channel $($channel+1), Program $program"
}
# Open devices
$midiOutCZ1 = [NAudio.Midi.MidiOut]::new(1)  # CZ-1 Mini
$midiOutP6  = [NAudio.Midi.MidiOut]::new(2)  # Roland P-6 Compact

$files = @(
    "C:\music\airwolf\airwolf1.mid",
    "C:\music\airwolf\airwolf2.mid"
)
# Invoke playback
Set-CZ1Sound -program 14 -CZ1 $midiOutCZ1
Invoke-Midi -Device1 $midiOutCZ1 -Device2 $midiOutP6 -BPM 135 -MidiFilePath $files[0] -StartOffset 4
Set-CZ1Sound -program 8 -CZ1 $midiOutCZ1
Invoke-Midi -Device1 $midiOutCZ1 -Device2 $midiOutP6 -BPM 135 -MidiFilePath $files[1] -StartOffset 1
# Cleanup afterwards
$midiOutCZ1.Dispose()
$midiOutP6.Dispose()