#include "xtc/xtc_editing.h"
#include <assert.h>
#include <stdio.h>

static void test_velocity(void)
{
    xtc_project *p=NULL;uint64_t t,c,n;xtc_note_info note;xtc_midi_transform edit={0};
    assert(xtc_project_create(&p)==XTC_OK && xtc_project_add_track(p,"Track",&t)==XTC_OK);
    assert(xtc_project_add_midi_clip(p,t,0,960,&c)==XTC_OK);
    assert(xtc_project_add_note(p,c,0,120,61,100,0,&n)==XTC_OK);
    edit.type=XTC_MIDI_TRANSFORM_VELOCITY;edit.velocity_percent=50;edit.velocity_offset=10;
    assert(xtc_project_transform_midi(p,&c,1,&edit)==XTC_OK);
    assert(xtc_project_get_note(p,0,&note)==XTC_OK && note.velocity==60 && note.pitch==61);
    edit.velocity_percent=400;assert(xtc_project_transform_midi(p,&c,1,&edit)==XTC_OK);
    assert(xtc_project_get_note(p,0,&note)==XTC_OK && note.velocity==127);
    edit.velocity_percent=0;edit.velocity_offset=-127;assert(xtc_project_transform_midi(p,&c,1,&edit)==XTC_OK);
    assert(xtc_project_get_note(p,0,&note)==XTC_OK && note.velocity==1 && note.id==n);
    edit.velocity_percent=401;assert(xtc_project_transform_midi(p,&c,1,&edit)==XTC_ERR_INVALID_ARGUMENT);
    xtc_project_destroy(p);
}
static void test_scale(void)
{
    xtc_project *p=NULL;uint64_t t,c,n;xtc_note_info note;xtc_midi_transform edit={0};
    assert(xtc_project_create(&p)==XTC_OK && xtc_project_add_track(p,"Track",&t)==XTC_OK);
    assert(xtc_project_add_midi_clip(p,t,0,960,&c)==XTC_OK);
    assert(xtc_project_add_note(p,c,0,120,61,100,0,&n)==XTC_OK);
    edit.type=XTC_MIDI_TRANSFORM_SCALE;edit.scale=1;
    assert(xtc_project_transform_midi(p,&c,1,&edit)==XTC_OK);
    assert(xtc_project_get_note(p,0,&note)==XTC_OK && note.pitch==60 && note.id==n);
    assert(xtc_project_edit_note(p,n,0,120,0,100,0)==XTC_OK);edit.root=11;
    assert(xtc_project_transform_midi(p,&c,1,&edit)==XTC_OK);
    assert(xtc_project_get_note(p,0,&note)==XTC_OK && note.pitch==1);
    edit.root=12;assert(xtc_project_transform_midi(p,&c,1,&edit)==XTC_ERR_INVALID_ARGUMENT);
    xtc_project_destroy(p);
}
static void test_gate_legato(void)
{
    xtc_project *p=NULL;uint64_t t,c,n;xtc_note_info note;xtc_midi_transform edit={0};
    assert(xtc_project_create(&p)==XTC_OK && xtc_project_add_track(p,"Track",&t)==XTC_OK);
    assert(xtc_project_add_midi_clip(p,t,0,960,&c)==XTC_OK);
    assert(xtc_project_add_note(p,c,0,120,60,100,0,&n)==XTC_OK);
    assert(xtc_project_add_note(p,c,0,120,64,100,0,&n)==XTC_OK);
    assert(xtc_project_add_note(p,c,240,120,67,100,0,&n)==XTC_OK);
    assert(xtc_project_add_note(p,c,120,120,72,100,1,&n)==XTC_OK);
    edit.type=XTC_MIDI_TRANSFORM_LEGATO;assert(xtc_project_transform_midi(p,&c,1,&edit)==XTC_OK);
    assert(xtc_project_get_note(p,0,&note)==XTC_OK && note.length_tick==240);
    assert(xtc_project_get_note(p,1,&note)==XTC_OK && note.length_tick==240);
    assert(xtc_project_get_note(p,2,&note)==XTC_OK && note.length_tick==720);
    assert(xtc_project_get_note(p,3,&note)==XTC_OK && note.length_tick==840);
    edit.type=XTC_MIDI_TRANSFORM_GATE;edit.gate_percent=200;
    assert(xtc_project_transform_midi(p,&c,1,&edit)==XTC_ERR_INVALID_ARGUMENT);
    assert(xtc_project_get_note(p,0,&note)==XTC_OK && note.length_tick==240);
    edit.gate_percent=50;assert(xtc_project_transform_midi(p,&c,1,&edit)==XTC_OK);
    assert(xtc_project_get_note(p,0,&note)==XTC_OK && note.length_tick==120);
    xtc_project_destroy(p);
}
static void test_swing(void)
{
    xtc_project *p=NULL;uint64_t t,c,n;xtc_note_info note;xtc_midi_transform edit={0};
    assert(xtc_project_create(&p)==XTC_OK && xtc_project_add_track(p,"Track",&t)==XTC_OK);
    assert(xtc_project_add_midi_clip(p,t,0,1920,&c)==XTC_OK);
    assert(xtc_project_add_note(p,c,230,120,60,100,0,&n)==XTC_OK);
    edit.grid=240;edit.swing=50;edit.strength=100;
    assert(xtc_project_transform_midi(p,&c,1,&edit)==XTC_OK);
    assert(xtc_project_get_note(p,0,&note)==XTC_OK && note.start_tick==300 && note.id==n && note.length_tick==120);
    edit.strength=50;edit.swing=0;
    assert(xtc_project_transform_midi(p,&c,1,&edit)==XTC_OK);
    assert(xtc_project_get_note(p,0,&note)==XTC_OK && note.start_tick==270);
    assert(xtc_project_add_note(p,c,1850,70,64,100,0,&n)==XTC_OK);
    edit.strength=100;
    assert(xtc_project_transform_midi(p,&c,1,&edit)==XTC_ERR_INVALID_ARGUMENT);
    assert(xtc_project_get_note(p,0,&note)==XTC_OK && note.start_tick==270);
    edit.grid=0;assert(xtc_project_transform_midi(p,&c,1,&edit)==XTC_ERR_INVALID_ARGUMENT);
    xtc_project_destroy(p);
}
static void test_align_slip(void)
{
    xtc_project *p=NULL;uint64_t t,ids[2];xtc_clip_info c;
    assert(xtc_project_create(&p)==XTC_OK && xtc_project_add_track(p,"Track",&t)==XTC_OK);
    assert(xtc_project_add_clip(p,t,"A",0,1920,&ids[0])==XTC_OK);
    assert(xtc_project_add_clip(p,t,"B",960,960,&ids[1])==XTC_OK);
    assert(xtc_project_edit_clip(p,ids[0],t,0,480,240)==XTC_OK);
    assert(xtc_project_edit_clip(p,ids[1],t,960,480,240)==XTC_OK);
    assert(xtc_project_transform_clips(p,ids,2,XTC_CLIPS_SLIP,240,NULL)==XTC_OK);
    assert(xtc_project_find_clip(p,ids[0],&c)==XTC_OK && c.offset_tick==480 && c.start_tick==0);
    /* First clip permits this slip, second does not: no partial publication. */
    assert(xtc_project_transform_clips(p,ids,2,XTC_CLIPS_SLIP,240,NULL)==XTC_ERR_INVALID_ARGUMENT);
    assert(xtc_project_find_clip(p,ids[0],&c)==XTC_OK && c.offset_tick==480);
    assert(xtc_project_transform_clips(p,ids,2,XTC_CLIPS_ALIGN,3840,NULL)==XTC_OK);
    assert(xtc_project_find_clip(p,ids[1],&c)==XTC_OK && c.start_tick==3840 && c.offset_tick==480);
    xtc_project_destroy(p);
}
static void test_repeat(void)
{
    xtc_project *p=NULL;uint64_t t,ids[2];xtc_clip_info c;size_t tracks,clips;
    assert(xtc_project_create(&p)==XTC_OK && xtc_project_add_track(p,"Track",&t)==XTC_OK);
    assert(xtc_project_add_clip(p,t,"A",0,240,&ids[0])==XTC_OK);
    assert(xtc_project_add_clip(p,t,"B",960,240,&ids[1])==XTC_OK);
    assert(xtc_project_repeat_clips(p,ids,2,3)==XTC_OK);
    assert(xtc_project_get_counts(p,&tracks,&clips)==XTC_OK && clips==8);
    assert(xtc_project_get_clip(p,7,&c)==XTC_OK && c.start_tick==4560);
    assert(xtc_project_repeat_clips(p,ids,2,1024)==XTC_ERR_CAPACITY);
    assert(xtc_project_get_counts(p,&tracks,&clips)==XTC_OK && clips==8);
    assert(xtc_project_repeat_clips(p,ids,2,0)==XTC_ERR_INVALID_ARGUMENT);
    xtc_project_destroy(p);
}
int main(void)
{
    test_repeat();test_align_slip();test_swing();test_gate_legato();test_scale();test_velocity();
    xtc_project *p=NULL;uint64_t track,a,b,n,ids[2],out[2]={999,999};xtc_clip_info c;
    size_t tracks,clips,notes;
    assert(xtc_project_create(&p)==XTC_OK);
    assert(xtc_project_add_track(p,"Keys",&track)==XTC_OK);
    assert(xtc_project_add_midi_clip(p,track,960,1920,&a)==XTC_OK);
    assert(xtc_project_add_note(p,a,0,240,60,100,0,&n)==XTC_OK);
    assert(xtc_project_add_clip(p,track,"Audio",3840,960,&b)==XTC_OK);
    ids[0]=a;ids[1]=b;
    assert(xtc_project_transform_clips(p,ids,2,XTC_CLIPS_MOVE,-1920,out)==XTC_ERR_INVALID_ARGUMENT);
    assert(out[0]==999 && xtc_project_find_clip(p,b,&c)==XTC_OK && c.start_tick==3840);
    assert(xtc_project_transform_clips(p,ids,2,XTC_CLIPS_MOVE,240,out)==XTC_OK);
    assert(xtc_project_find_clip(p,a,&c)==XTC_OK && c.start_tick==1200);
    assert(xtc_project_transform_clips(p,ids,2,XTC_CLIPS_DUPLICATE,4800,out)==XTC_OK);
    assert(out[0]!=a && out[1]!=b);
    assert(xtc_project_note_count(p,&notes)==XTC_OK && notes==2);
    assert(xtc_project_transform_clips(p,out,2,XTC_CLIPS_DELETE,0,NULL)==XTC_OK);
    assert(xtc_project_get_counts(p,&tracks,&clips)==XTC_OK && clips==2);
    ids[1]=a;assert(xtc_project_transform_clips(p,ids,2,XTC_CLIPS_DELETE,0,NULL)==XTC_ERR_INVALID_ARGUMENT);
    ids[1]=UINT64_MAX;assert(xtc_project_transform_clips(p,ids,2,XTC_CLIPS_MOVE,1,NULL)==XTC_ERR_NOT_FOUND);
    assert(xtc_project_transform_clips(p,ids,0,XTC_CLIPS_MOVE,1,NULL)==XTC_ERR_INVALID_ARGUMENT);
    {
        xtc_clipboard *board=NULL;xtc_project *dest=NULL;uint64_t dt;
        ids[0]=a;ids[1]=b;assert(xtc_clipboard_create(p,ids,2,&board)==XTC_OK);
        assert(xtc_clipboard_count(board)==2);assert(xtc_project_create(&dest)==XTC_OK);
        assert(xtc_project_add_track(dest,"Destination",&dt)==XTC_OK);
        assert(xtc_project_remove_clip(p,a)==XTC_OK);
        assert(xtc_clipboard_paste(board,dest,dt,0,out)==XTC_OK);
        assert(xtc_project_find_clip(dest,out[0],&c)==XTC_OK && c.start_tick==0 && c.midi);
        assert(xtc_project_note_count(dest,&notes)==XTC_OK && notes==1);
        assert(xtc_clipboard_paste(board,dest,dt,XTC_PROJECT_MAX_TICK,out)==XTC_ERR_INVALID_ARGUMENT);
        assert(xtc_project_get_counts(dest,&tracks,&clips)==XTC_OK && clips==2);
        xtc_clipboard_destroy(board);xtc_project_destroy(dest);
    }
    xtc_project_destroy(p);puts("Modern editing tests passed.");return 0;
}