#include <stdio.h>
#include <unistd.h>
#ifdef __WIN32__
#define sleep(x) Sleep(x)
#endif
#include <ftdi.h>

#define HALF_CYCLE_USEC 800 // 500 can work sometimes

// Pinout for this:
// FTDI board | Floppy edge connector
// D7 | 6
// D5 | 10
// D4 | 12 drive select
// D3 | 14
// D2 | 16 drive motor enable
// D1 | 18 direction
// D0 | 20 step

int main(int argc, char **argv)
{
    struct ftdi_context ftdic;
    int f,i, count, dir;
    char buf[1];

	if (argc < 2)
	{
		printf("Usage: %s step-count\n   with positive for higher tracks, negative for lower (+/- 80 max)\n", argv[0]);
		return(-1);
	}

	count = atoi(argv[1]);
	if (count < 0)
	{
		count = - count;
		dir = 2;
	}
	else
		dir = 0;
	printf("will do %d steps, direction %d\n", count, dir);

    if (ftdi_init(&ftdic) < 0)
    {
        fprintf(stderr, "ftdi_init failed\n");
        return EXIT_FAILURE;
    }

    f = ftdi_usb_open(&ftdic, 0x0403, 0x6010);

    if (f < 0 && f != -5)
    {
        fprintf(stderr, "unable to open ftdi device: %d (%s)\n", f, ftdi_get_error_string(&ftdic));
        exit(-1);
    }

    printf("ftdi open succeeded: %d\n",f);

    printf("enabling bitbang mode\n");
    ftdi_set_bitmode(&ftdic, 0xFF, BITMODE_BITBANG);

    buf[0] = 0xFF;
    printf("turning everything off\n");
    f = ftdi_write_data(&ftdic, buf, 1);
    if (f < 0)
    {
        fprintf(stderr,"write failed for 0x%x, error %d (%s)\n",buf[0],f, ftdi_get_error_string(&ftdic));
    }

    sleep(1);

    buf[0] = 0x28; //DB;
    printf("turning motor on\n");
    f = ftdi_write_data(&ftdic, buf, 1);
    if (f < 0)
    {
        fprintf(stderr,"write failed for 0x%x, error %d (%s)\n",buf[0],f, ftdi_get_error_string(&ftdic));
    }

    sleep(1);

    buf[0] = 0x2D;
    printf("turning motor off\n");
    f = ftdi_write_data(&ftdic, buf, 1);
    if (f < 0)
    {
        fprintf(stderr,"write failed for 0x%x, error %d (%s)\n",buf[0],f, ftdi_get_error_string(&ftdic));
    }

    sleep(1);

	for (i = 0; i < count; ++i)
	{
		// direction only
		buf[0] = 0x2D | dir;
		ftdi_write_data(&ftdic, buf, 1);
		usleep(HALF_CYCLE_USEC);

		// step & direction
		buf[0] = 0x2C | dir;
		ftdi_write_data(&ftdic, buf, 1);
		usleep(HALF_CYCLE_USEC);
	}

    ftdi_disable_bitbang(&ftdic);

    ftdi_usb_close(&ftdic);
    ftdi_deinit(&ftdic);
}