#include <intuition/intuition.h>

#include <exec/interrupts.h>
#include <hardware/intbits.h>
#include <midi/camd.h>
#include <midi/mididefs.h>
#include <clib/camd_protos.h>
#include <pragmas/camd_pragmas.h>

#include <dos/dos.h>

#define BYNOTE			1
#define VELSCALE		150

#define	LEAVE			goto exitit;
#define	unless(x)		if ( !(x) )

struct Library			*IntuitionBase,
						*GfxBase,
						*CamdBase;
extern struct Library	*SysBase;

struct Window			*Window;
struct Screen			*Screen;
struct ViewPort			*vp;
struct RastPort			*rp;

USHORT colors[32] = {
	0x000,0xAAA,0x888,0x555,0x333,0xBB0,0x6B0,0x0F0,
	0xF99,0xF89,0xE8A,0xE8A,0xE7A,0xE7B,0xD6B,0xD6C,
	0xD5D,0xC5D,0xA5C,0x94C,0x84C,0x74B,0x53B,0x43B,
	0x333,0x444,0x555,0x666,0x777,0x888,0xEEE,0xFFF };

struct NewScreen NewScreen = { 0,200-47,320,47, 5, 0,1, NULL, CUSTOMSCREEN, };

extern struct Image		im1,
						im2,
						im3;

struct Gadget off_gadget = {
	NULL,	2,2,13,21,GFLG_GADGHIMAGE|GFLG_GADGIMAGE,GACT_RELVERIFY,GTYP_BOOLGADGET,(APTR)&im1,(APTR)&im2
};

struct NewWindow back_ground =
{	0,0,320,47, 0,1,
	VANILLAKEY | MOUSEBUTTONS | GADGETUP,
	NOCAREREFRESH | ACTIVATE | BORDERLESS | RMBTRAP,
	&off_gadget,NULL,NULL,NULL,NULL,320,200,320,200,CUSTOMSCREEN,
};

#if BYNOTE
#define VU_BARCOUNT		32
#define BARS_PER_COLOR	2
#else
#define VU_BARCOUNT		16
#define BARS_PER_COLOR	1
#endif

WORD					vu_levels[VU_BARCOUNT],
						prev_levels[VU_BARCOUNT];

struct Task				*myTask;

void draw_vu_levels(void);

int __interrupt __saveds VBlankInterface( void );

struct Interrupt VertBlank =
{	NULL,NULL,NT_INTERRUPT,-60,"VU VBlank",		/* node, pri = -60 */
	NULL,										/* data ptr, same as inputevent */
	(void *)VBlankInterface						/* code ptr */
};

WORD					vblank_ok=NULL;			/* TRUE if vblank int installed	*/

struct MidiNode			*midi;
struct MidiLink			*link;

struct MidiNode *CreateMidi(Tag tag, ...)
{	return CreateMidiA((struct TagItem *)&tag );
}

BOOL SetMidiAttrs(struct MidiNode *mi, Tag tag, ...)
{	return SetMidiAttrsA(mi, (struct TagItem *)&tag );
}

struct MidiLink *AddMidiLink(struct MidiNode *mi, LONG type, Tag tag, ...)
{	return AddMidiLinkA(mi, type, (struct TagItem *)&tag );
}

BOOL SetMidiLinkAttrs(struct MidiLink *mi, Tag tag, ...)
{	return SetMidiLinkAttrsA(mi, (struct TagItem *)&tag );
}

void main(int argc,char *argv[])
{	char	*inlinkname = "in.0";

	if (argc > 1) inlinkname = argv[1];

	myTask = FindTask(0);

	unless (IntuitionBase = OpenLibrary("intuition.library",0)) LEAVE;
	unless (GfxBase = OpenLibrary("graphics.library",0)) LEAVE;
	unless (CamdBase = OpenLibrary("camd.library",0)) LEAVE;

	if ((Screen = OpenScreen(&NewScreen)) == NULL) LEAVE;
	vp = &(Screen->ViewPort);
	LoadRGB4(vp,colors,32);

	back_ground.Screen = Screen;
	if ((Window = OpenWindow(&back_ground)) == NULL) LEAVE;
	rp = Window->RPort;

	DrawImage(rp,&im3,309,0);
	SetAPen(rp, 1); Move(rp,19, 0); Draw(rp,308, 0);
	SetAPen(rp,24); Move(rp,19, 1); Draw(rp,308, 1);
	SetAPen(rp, 3); Move(rp,19, 2); Draw(rp,308, 2);

	SetAPen(rp, 1); Move(rp,19,44); Draw(rp,308,44);
	SetAPen(rp,24); Move(rp,19,45); Draw(rp,308,45);
	SetAPen(rp, 3); Move(rp,19,46); Draw(rp,308,46);

	unless (midi = CreateMidi(
		MIDI_Name, "VU Meters",
		MIDI_RecvSignal, SIGBREAKB_CTRL_E,
		MIDI_MsgQueue,   100,
		MIDI_ErrFilter, CMEF_All,
		TAG_DONE))
			LEAVE;

	unless (link = AddMidiLink(midi, MLTYPE_Receiver,
		MLINK_Name, "VU Meter Link",
		MLINK_Location,	inlinkname,
		MLINK_EventMask, CMF_Note,
		MLINK_Comment,	"VU Meters [Input]",
		TAG_DONE))
			LEAVE;

	AddIntServer(INTB_VERTB, &VertBlank);
	vblank_ok = TRUE;

	while (TRUE)
	{	struct IntuiMessage *message;
		LONG			signals;

		signals = Wait((1 << Window->UserPort->mp_SigBit) | SIGBREAKF_CTRL_F | SIGBREAKF_CTRL_E);

		if (signals & SIGBREAKF_CTRL_F)
		{	int i;

			for (i=0; i<VU_BARCOUNT; i++)
			{	if (vu_levels[i] > 0)
					vu_levels[i]--;
			}
			draw_vu_levels();
		}

		if (signals & SIGBREAKF_CTRL_E)
		{
			MidiMsg		msg;

			while (GetMidi(midi,&msg))
			{
				if (noteon(&msg))
#if BYNOTE
				{	int		note = ((msg.mm_Data1-36)/2) % 31;
					int		velocity = msg.mm_Data2 * 39 / 127;

					velocity = velocity * VELSCALE / 100;
					if (velocity > 39) velocity = 39;

					if (vu_levels[note] < velocity) vu_levels[note] = velocity;
				}
#else
				{	int		channel = msg.mm_Status & 0x0f;
					int		velocity = msg.mm_Data2 * 39 / 127;

					velocity = velocity * VELSCALE / 100;
					if (velocity > 39) velocity = 39;

					if (vu_levels[channel] < velocity) vu_levels[channel] = velocity;
				}
#endif
			}
		}

		while (message = (struct IntuiMessage *)GetMsg(Window->UserPort))
		{	ULONG		class = message->Class;

			ReplyMsg(&message->ExecMessage);
			if (class == GADGETUP) LEAVE;
		}
	}
exitit:
	if (vblank_ok) RemIntServer(INTB_VERTB, &VertBlank);
	if (midi) DeleteMidi(midi);
	if (Window){ ClearMenuStrip(Window); CloseWindow(Window); }
	if (Screen) CloseScreen(Screen);
	if (CamdBase) CloseLibrary(CamdBase);
	if (GfxBase) CloseLibrary(GfxBase);
	if (IntuitionBase) CloseLibrary(IntuitionBase);
}

void draw_vu_levels(void)
{	int					i;

	int					ymin, ymax, y1, y2;
	int					xmin, xmax;

	for (i=0; i<VU_BARCOUNT; i++)
	{
		if (vu_levels[i] > prev_levels[i])
		{
			SetAPen(rp,i/BARS_PER_COLOR+8);

			ymin = prev_levels[i];
			ymax = vu_levels[i];
		}
		else
		{	SetAPen(rp,0);
			ymax = prev_levels[i];
			ymin = vu_levels[i];
		}
#if BYNOTE
		xmin = 21 + i * 9;
		xmax = xmin + 7;
#else
		xmin = 21 + i * 18;
		xmax = xmin + 15;
#endif

		y1 = 43 - ymax;
		y2 = 43 - ymin - 1;

		if (y1 <= y2) RectFill(rp,xmin,y1,xmax,y2);

		prev_levels[i] = vu_levels[i];
	}
}

int __interrupt __saveds VBlankInterface( void )
{	Signal(myTask,SIGBREAKF_CTRL_F);
	return 0;									/* server chain continues		*/
}