/*********************************************************************

	ap_dsk35.c

	Apple 3.5" disk images

	This code supports 3.5" 400k/800k disks used in early Macintoshes
	and the Apple IIgs.

	These disks have the following properties:
		
		400k:	80 tracks, 1 head
		800k:	80 tracks, 2 heads

		Tracks  0-15 have 12 sectors each
		Tracks 16-31 have 11 sectors each
		Tracks 32-47 have 10 sectors each
		Tracks 48-63 have  9 sectors each
		Tracks 64-79 have  8 sectors each

	Each sector has 524 bytes, 512 of which are really used by the Macintosh

	(80 tracks) * (avg of 10 sectors) * (512 bytes) * (2 sides) = 800 kB

	Data is nibblized : 3 data bytes -> 4 bytes on disk.

	In addition to 512 logical bytes, each sector contains 800 physical
	bytes.  Here is the layout of the physical sector:

		Pos
		0		0xFF (pad byte where head is turned on ???)
		1-35	self synch 0xFFs (7*5) (42 bytes actually written to media)
		36		0xD5
		37		0xAA
		38		0x96
		39		diskbytes[(track number) & 0x3F]
		40		diskbytes[(sector number)]
		41		diskbytes[("side")]
		42		diskbytes[("format byte")]
		43		diskbytes[("sum")]
		44		0xDE
		45		0xAA
		46		pad byte where head is turned off/on (0xFF here)
		47-51	self synch 0xFFs (6 bytes actually written to media)
		52		0xD5
		53		0xAA
		54		0xAD
		55		spare byte, generally diskbytes[(sector number)]
		56-754	"nibblized" sector data	...
		755-758	checksum
		759		0xDE
		760		0xAA
		761		pad byte where head is turned off (0xFF here)

	Note : "Self synch refers to a technique whereby two zeroes are inserted
	between each synch byte written to the disk.", i.e. "0xFF, 0xFF, 0xFF,
	0xFF, 0xFF" is actually "0xFF, 0x3F, 0xCF, 0xF3, 0xFC, 0xFF" on disk.
	Since the IWM assumes the data transfer is complete when the MSBit of its
	shift register is 1, we do read 4 0xFF, even though they are not
	contiguous on the disk media.  Some reflexion shows that 4 synch bytes
	allow the IWM to synchronize with the trailing data.

	Format byte codes:
		0x00	Apple II
		0x01	Lisa
		0x02	Mac MFS (single sided)?
		0x22	Mac MFS (double sided)?

*********************************************************************/

#include <stdio.h>
#include <assert.h>
#include "formats/ap_dsk35.h"
#include "utils.h"


#define APPLE35_TAG "a35_tag"

struct apple35_tag
{
	UINT32 data_offset;
	UINT32 data_size;
	UINT8 format_byte;
	UINT8 sides;

	/* stuff used in DiskCopy images */
	UINT32 tag_offset;
	UINT32 tag_size;
};



/* normal number of sector for each track */
const UINT8 apple35_tracklen_800kb[80] =
{
	12, 12, 12, 12, 12, 12, 12, 12, 12, 12, 12, 12, 12, 12, 12, 12,
	11, 11, 11, 11, 11, 11, 11, 11, 11, 11, 11, 11, 11, 11, 11, 11,
	10, 10, 10, 10, 10, 10, 10, 10, 10, 10, 10, 10, 10, 10, 10, 10,
	 9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,  9,
	 8,  8,  8,  8,  8,  8,  8,  8,  8,  8,  8,  8,  8,  8,  8,  8
};

/* blocks of data used to nibblize tracks */
static const UINT8 diskbytes[] =
{
	0x96, 0x97, 0x9A, 0x9B,  0x9D, 0x9E, 0x9F, 0xA6, /* 0x00 */
	0xA7, 0xAB, 0xAC, 0xAD,  0xAE, 0xAF, 0xB2, 0xB3,
	0xB4, 0xB5, 0xB6, 0xB7,  0xB9, 0xBA, 0xBB, 0xBC, /* 0x10 */
	0xBD, 0xBE, 0xBF, 0xCB,  0xCD, 0xCE, 0xCF, 0xD3,
	0xD6, 0xD7, 0xD9, 0xDA,  0xDB, 0xDC, 0xDD, 0xDE, /* 0x20 */
	0xDF, 0xE5, 0xE6, 0xE7,  0xE9, 0xEA, 0xEB, 0xEC,
	0xED, 0xEE, 0xEF, 0xF2,  0xF3, 0xF4, 0xF5, 0xF6, /* 0x30 */
	0xF7, 0xF9, 0xFA, 0xFB,  0xFC, 0xFD, 0xFE, 0xFF
};

/* reverse lookup of diskbytes */
static const INT16 rev_diskbytes[] =
{
	-1,   -1,   -1,   -1,    -1,   -1,   -1,   -1,   /* 0x00 */
	-1,   -1,   -1,   -1,    -1,   -1,   -1,   -1,
	-1,   -1,   -1,   -1,    -1,   -1,   -1,   -1,   /* 0x10 */
	-1,   -1,   -1,   -1,    -1,   -1,   -1,   -1,
	-1,   -1,   -1,   -1,    -1,   -1,   -1,   -1,   /* 0x20 */
	-1,   -1,   -1,   -1,    -1,   -1,   -1,   -1,
	-1,   -1,   -1,   -1,    -1,   -1,   -1,   -1,   /* 0x30 */
	-1,   -1,   -1,   -1,    -1,   -1,   -1,   -1,
	-1,   -1,   -1,   -1,    -1,   -1,   -1,   -1,   /* 0x40 */
	-1,   -1,   -1,   -1,    -1,   -1,   -1,   -1,
	-1,   -1,   -1,   -1,    -1,   -1,   -1,   -1,   /* 0x50 */
	-1,   -1,   -1,   -1,    -1,   -1,   -1,   -1,
	-1,   -1,   -1,   -1,    -1,   -1,   -1,   -1,   /* 0x60 */
	-1,   -1,   -1,   -1,    -1,   -1,   -1,   -1,
	-1,   -1,   -1,   -1,    -1,   -1,   -1,   -1,   /* 0x70 */
	-1,   -1,   -1,   -1,    -1,   -1,   -1,   -1,
	-1,   -1,   -1,   -1,    -1,   -1,   -1,   -1,   /* 0x80 */
	-1,   -1,   -1,   -1,    -1,   -1,   -1,   -1,
	-1,   -1,   -1,   -1,    -1,   -1,   0x00, 0x01, /* 0x90 */
	-1,   -1,   0x02, 0x03,  -1,   0x04, 0x05, 0x06,
	-1,   -1,   -1,   -1,    -1,   -1,   0x07, 0x08, /* 0xA0 */
	-1,   -1,   -1,   0x09,  0x0A, 0x0B, 0x0C, 0x0D,
	-1,   -1,   0x0E, 0x0F,  0x10, 0x11, 0x12, 0x13, /* 0xB0 */
	-1,   0x14, 0x15, 0x16,  0x17, 0x18, 0x19, 0x1A,
	-1,   -1,   -1,   -1,    -1,   -1,   -1,   -1,   /* 0xC0 */
	-1,   -1,   -1,   0x1B,  -1,   0x1C, 0x1D, 0x1E,
	-1,   -1,   -1,   0x1F,  -1,   -1,   0x20, 0x21, /* 0xD0 */
	-1,   0x22, 0x23, 0x24,  0x25, 0x26, 0x27, 0x28,
	-1,   -1,   -1,   -1,    -1,   0x29, 0x2A, 0x2B, /* 0xE0 */
	-1,   0x2C, 0x2D, 0x2E,  0x2F, 0x30, 0x31, 0x32,
	-1,   -1,   0x33, 0x34,  0x35, 0x36, 0x37, 0x38, /* 0xF0 */
	-1,   0x39, 0x3A, 0x3B,  0x3C, 0x3D, 0x3E, 0x3F
};

static const UINT8 blk1[] =
{
	/*0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF,
	0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF,
	0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF,
	0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF,
	0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF,
	0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF,
	0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF,
	0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF,
	0xFF, 0xFF, 0xD5, 0xAA, 0x96*/
	0xFF,
	0xFF, 0x3F, 0xCF, 0xF3, 0xFC, 0xFF,
	0xFF, 0x3F, 0xCF, 0xF3, 0xFC, 0xFF,
	0xFF, 0x3F, 0xCF, 0xF3, 0xFC, 0xFF,
	0xFF, 0x3F, 0xCF, 0xF3, 0xFC, 0xFF,
	0xFF, 0x3F, 0xCF, 0xF3, 0xFC, 0xFF,
	0xFF, 0x3F, 0xCF, 0xF3, 0xFC, 0xFF,
	0xFF, 0x3F, 0xCF, 0xF3, 0xFC, 0xFF,
	0xD5, 0xAA, 0x96
};
static const UINT8 blk2[] =
{
	0xDE, 0xAA, 0xFF, 0xFF, 0x3F, 0xCF, 0xF3, 0xFC, 0xFF, 0xD5, 0xAA, 0xAD
};
static const UINT8 blk3[] =
{
	0xDE, 0xAA, 0xFF
};



static struct apple35_tag *get_apple35_tag(floppy_image *floppy)
{
	struct apple35_tag *tag;
	tag = (struct apple35_tag *) floppy_tag(floppy, APPLE35_TAG);
	return tag;
}



/*
	converts data to its nibblized representation, and generate checksum

	now handles 524-byte-long sectors

	tag data IS important, since it allows data recovery when the catalog is trashed
*/
static void sony_nibblize35(const UINT8 *in, UINT8 *nib_ptr, UINT8 *csum)
{
	int	i, j;
	UINT32 c1, c2, c3, c4;
	UINT8 val;
	UINT8 w1, w2, w3, w4;
	UINT8 b1[175], b2[175], b3[175];

	/* Copy from the user's buffer to our buffer, while computing
	 * the three-byte data checksum
	 */

	i = 0;
	j = 0;
	c1 = 0;
	c2 = 0;
	c3 = 0;
	while (1)
	{
		c1 = (c1 & 0xFF) << 1;
		if (c1 & 0x0100)
			c1++;

		val = in[i++];
		c3 += val;
		if (c1 & 0x0100)
		{
			c3++;
			c1 &= 0xFF;
		}
		b1[j] = (val ^ c1) & 0xFF;

		val = in[i++];
		c2 += val;
		if (c3 > 0xFF)
		{
			c2++;
			c3 &= 0xFF;
		}
		b2[j] = (val ^ c3) & 0xFF;

		if (i == 524) break;

		val = in[i++];
		c1 += val;
		if (c2 > 0xFF)
		{
			c1++;
			c2 &= 0xFF;
		}
		b3[j] = (val ^ c2) & 0xFF;
		j++;
	}
	c4 =  ((c1 & 0xC0) >> 6) | ((c2 & 0xC0) >> 4) | ((c3 & 0xC0) >> 2);
	b3[174] = 0;

	j = 0;
	for (i = 0; i <= 174; i++)
	{
		w1 = b1[i] & 0x3F;
		w2 = b2[i] & 0x3F;
		w3 = b3[i] & 0x3F;
		w4 =  ((b1[i] & 0xC0) >> 2);
		w4 |= ((b2[i] & 0xC0) >> 4);
		w4 |= ((b3[i] & 0xC0) >> 6);

		nib_ptr[j++] = w4;
		nib_ptr[j++] = w1;
		nib_ptr[j++] = w2;

		if (i != 174) nib_ptr[j++] = w3;
	}

	csum[0] = c1 & 0x3F;
	csum[1] = c2 & 0x3F;
	csum[2] = c3 & 0x3F;
	csum[3] = c4 & 0x3F;
}



/*
	does the reverse process of sony_nibblize35
*/
static void sony_denibblize35(UINT8 *out, const UINT8 *nib_ptr, UINT8 *checksum)
{
	int	i, j;
	UINT32 c1,c2,c3,c4;
	UINT8 val;
	UINT8 w1,w2,w3=0,w4;
	UINT8 b1[175],b2[175],b3[175];

	j = 0;
	for (i=0; i<=174; i++)
	{
		w4 = nib_ptr[j++];
		w1 = nib_ptr[j++];
		w2 = nib_ptr[j++];

		if (i != 174) w3 = nib_ptr[j++];

		b1[i] = (w1 & 0x3F) | ((w4 << 2) & 0xC0);
		b2[i] = (w2 & 0x3F) | ((w4 << 4) & 0xC0);
		b3[i] = (w3 & 0x3F) | ((w4 << 6) & 0xC0);
	}

	/* Copy from the user's buffer to our buffer, while computing
	 * the three-byte data checksum
	 */

	i = 0;
	j = 0;
	c1 = 0;
	c2 = 0;
	c3 = 0;
	while (1)
	{
		c1 = (c1 & 0xFF) << 1;
		if (c1 & 0x0100)
			c1++;

		val = (b1[j] ^ c1) & 0xFF;
		c3 += val;
		if (c1 & 0x0100)
		{
			c3++;
			c1 &= 0xFF;
		}
		out[i++] = val;

		val = (b2[j] ^ c3) & 0xFF;
		c2 += val;
		if (c3 > 0xFF)
		{
			c2++;
			c3 &= 0xFF;
		}
		out[i++] = val;

		if (i == 524) break;

		val = (b3[j] ^ c2) & 0xFF;
		c1 += val;
		if (c2 > 0xFF)
		{
			c1++;
			c2 &= 0xFF;
		}
		out[i++] = val;
		j++;
	}
	c4 =  ((c1 & 0xC0) >> 6) | ((c2 & 0xC0) >> 4) | ((c3 & 0xC0) >> 2);
	b3[174] = 0;

	checksum[0] = c1 & 0x3F;
	checksum[1] = c2 & 0x3F;
	checksum[2] = c3 & 0x3F;
	checksum[3] = c4 & 0x3F;
}



void sony_filltrack(UINT8 *buffer, size_t buffer_len, size_t *pos, UINT8 data)
{
	buffer[*pos / 8] &= 0xFF << (8 - (*pos % 8));
	buffer[*pos / 8] |= data >> (*pos % 8);

	*pos += 8;
	*pos %= buffer_len * 8;

	buffer[*pos / 8] &= 0xFF >> (*pos % 8);
	buffer[*pos / 8] |= data << (8 - (*pos % 8));
}



UINT8 sony_fetchtrack(const UINT8 *buffer, size_t buffer_len, size_t *pos)
{
	UINT8 data;

	data = buffer[*pos / 8] << (*pos % 8);
	*pos += 8;
	*pos %= (buffer_len * 8);
	data |= buffer[*pos / 8] >> (8 - (*pos % 8));

	while ((data & 0x80) == 0)
	{
		/* this code looks weird because it isn't simply rotating the new bit
		 * in, but for some reason it won't work if I rotate the bit in; I
		 * have to match the algorithm used by the old code */
		data <<= 1;
		data |= (buffer[*pos / 8] >> (8 - ((*pos % 8) + 1)));
		(*pos)++;
		*pos %= (buffer_len * 8);
	}
	return data;
}



static UINT32 apple35_get_offset(floppy_image *floppy, int head, int track, int sector, UINT32 *tag_offset)
{
	int i;
	UINT32 sector_index = 0;
	struct apple35_tag *tag;

	tag = get_apple35_tag(floppy);

	if (track >= (sizeof(apple35_tracklen_800kb) / sizeof(apple35_tracklen_800kb[0])))
		return ~0;
	if (head >= tag->sides)
		return ~0;
	if (sector >= apple35_tracklen_800kb[track])
		return ~0;

	for (i = 0; i < track; i++)
		sector_index += apple35_tracklen_800kb[i];
	sector_index *= tag->sides;
	if (head)
		sector_index += apple35_tracklen_800kb[i];
	sector_index += sector;

	if (tag_offset)
	{
		*tag_offset = sector_index * 12 + tag->tag_offset;
		if (*tag_offset >= tag->tag_size)
			*tag_offset = ~0;
	}
	return sector_index * 0x200 + tag->data_offset;
}



static floperr_t apple35_read_sector(floppy_image *floppy, int head, int track, int sector, void *buffer, size_t buflen)
{
	UINT32 data_offset;
	data_offset = apple35_get_offset(floppy, head, track, sector, NULL);
	if (data_offset == ~0)
		return FLOPPY_ERROR_SEEKERROR;
	floppy_image_read(floppy, buffer, data_offset, buflen);
	return FLOPPY_ERROR_SUCCESS;
}



static floperr_t apple35_write_sector(floppy_image *floppy, int head, int track, int sector, const void *buffer, size_t buflen)
{
	UINT32 data_offset;

	data_offset = apple35_get_offset(floppy, head, track, sector, NULL);
	if (data_offset == ~0)
		return FLOPPY_ERROR_SEEKERROR;
	floppy_image_write(floppy, buffer, data_offset, buflen);
	return FLOPPY_ERROR_SUCCESS;
}



static floperr_t apple35_read_sector_td(floppy_image *floppy, int head, int track, int sector, void *buffer, size_t buflen)
{
	floperr_t err;
	UINT32 tag_offset;

	assert(buflen == 524);

	/* first read the sector */
	err = apple35_read_sector(floppy, head, track, sector, ((UINT8 *) buffer) + 12, 512);
	if (err)
		return err;

	/* read the tag data, if possible */
	memset(buffer, '\0', 12);
	apple35_get_offset(floppy, head, track, sector, &tag_offset);
	if (tag_offset != ~0)
		floppy_image_read(floppy, buffer, tag_offset, 12);

	return FLOPPY_ERROR_SUCCESS;
}



static floperr_t apple35_write_sector_td(floppy_image *floppy, int head, int track, int sector, const void *buffer, size_t buflen)
{
	floperr_t err;
	UINT32 tag_offset;

	assert(buflen == 524);

	/* first write the sector */
	err = apple35_write_sector(floppy, head, track, sector, ((const UINT8 *) buffer) + 12, 512);
	if (err)
		return err;

	/* write the tag data, if possible */
	apple35_get_offset(floppy, head, track, sector, &tag_offset);
	if (tag_offset != ~0)
		floppy_image_write(floppy, buffer, tag_offset, 12);

	return FLOPPY_ERROR_SUCCESS;
}



static floperr_t apple35_get_sector_length(floppy_image *floppy, int head, int track, int sector, UINT32 *sector_length)
{
	*sector_length = 512;
	return FLOPPY_ERROR_SUCCESS;
}



static int apple35_get_heads_per_disk(floppy_image *floppy)
{
	return get_apple35_tag(floppy)->sides;
}



static int apple35_get_tracks_per_disk(floppy_image *floppy)
{
	return 80;
}



UINT32 apple35_get_track_size(floppy_image *floppy, int head, int track)
{
	if ((track < 0) || (track >= 80))
		return 0;
	return apple35_tracklen_800kb[track] * 800;
}



static UINT8 calculate_side(int head, int track)
{
	UINT8 side;
	side = head ? 0x20 : 0x00;
	if (track & 0x40)
		side |= 0x01;
	return side;
}



static floperr_t apple35_read_track(floppy_image *floppy, int head, int track, UINT64 offset, void *buffer, size_t buflen)
{
	floperr_t err;
	size_t pos = 0;
	int sector_count, sector, i;
	UINT8 sum, side;
	struct apple35_tag *tag;
	UINT8 sector_data[524];
	UINT8 nibble_data[699];
	UINT8 checksum[4];

	tag = get_apple35_tag(floppy);

	if (track >= (sizeof(apple35_tracklen_800kb) / sizeof(apple35_tracklen_800kb[0])))
		return FLOPPY_ERROR_SEEKERROR;
	if (offset != 0)
		return FLOPPY_ERROR_UNSUPPORTED;

	memset(buffer, 0xFF, buflen);
	sector_count = apple35_tracklen_800kb[track];
	side = calculate_side(head, track);

	for (sector = 0; sector < sector_count; sector++)
	{
		/* read the sector */
		err = apple35_read_sector_td(floppy, head, track, sector, sector_data, sizeof(sector_data) / sizeof(sector_data[0]));
		if (err)
			return err;
		sony_nibblize35(sector_data, nibble_data, checksum);

		for (i = 0; i < (sizeof(blk1) / sizeof(blk1[0])); i++)
			sony_filltrack(buffer, buflen, &pos, blk1[i]);

		sum = (track ^ sector ^ side ^ tag->format_byte) & 0x3F;

		sony_filltrack(buffer, buflen, &pos, diskbytes[track & 0x3f]);
		sony_filltrack(buffer, buflen, &pos, diskbytes[sector]);
		sony_filltrack(buffer, buflen, &pos, diskbytes[side]);
		sony_filltrack(buffer, buflen, &pos, diskbytes[tag->format_byte]);
		sony_filltrack(buffer, buflen, &pos, diskbytes[sum]);

		for (i = 0; i < (sizeof(blk2) / sizeof(blk2[0])); i++)
			sony_filltrack(buffer, buflen, &pos, blk2[i]);

		sony_filltrack(buffer, buflen, &pos, diskbytes[sector]);

		for (i = 0; i < (sizeof(nibble_data) / sizeof(nibble_data[0])); i++)
			sony_filltrack(buffer, buflen, &pos, diskbytes[nibble_data[i]]);

		for (i = 3; i >= 0; i--)
			sony_filltrack(buffer, buflen, &pos, diskbytes[checksum[i]]);

		for (i = 0; i < (sizeof(blk3) / sizeof(blk3[0])); i++)
			sony_filltrack(buffer, buflen, &pos, blk3[i]);
	}
	return FLOPPY_ERROR_SUCCESS;
}



static floperr_t apple35_write_track(floppy_image *floppy, int head, int track, UINT64 offset, const void *buffer, size_t buflen)
{
	floperr_t err;
	size_t pos = 0;
	int sector_count, sector, i, j;
	struct apple35_tag *tag;
	UINT8 sum, format_byte, val, side;
	UINT32 found_sectors = 0;
	UINT8 sector_data[524];
	UINT8 nibble_data[699];
	UINT8 checksum[4];

	tag = get_apple35_tag(floppy);

	if (track >= (sizeof(apple35_tracklen_800kb) / sizeof(apple35_tracklen_800kb[0])))
		return FLOPPY_ERROR_SEEKERROR;
	if (offset != 0)
		return FLOPPY_ERROR_UNSUPPORTED;

	sector_count = apple35_tracklen_800kb[track];
	side = calculate_side(head, track);

	/* do 2 rotations, in case the bit slip stuff prevent us to read the first sector */
	for (j = 0; j < (buflen * 2); j++)
	{
		if (sony_fetchtrack(buffer, buflen, &pos) != 0xD5)
			continue;
		j++;

		if (sony_fetchtrack(buffer, buflen, &pos) != 0xAA)
			continue;
		j++;

		if (sony_fetchtrack(buffer, buflen, &pos) != 0x96)
			continue;
		j++;

		if (rev_diskbytes[sony_fetchtrack(buffer, buflen, &pos)] != (track & 0x3F))
			continue;
		j++;

		sector = rev_diskbytes[sony_fetchtrack(buffer, buflen, &pos)];
		if ((sector < 0) || (sector >= sector_count))
			continue;
		j++;

		if (rev_diskbytes[sony_fetchtrack(buffer, buflen, &pos)] != side)
			continue;
		j++;

		format_byte = rev_diskbytes[sony_fetchtrack(buffer, buflen, &pos)];
		if (format_byte != tag->format_byte)
		{
			/* this is an error, but not THAT critical, I guess */
		}
		j++;

		sum = track ^ sector ^ side ^ format_byte;
		if (rev_diskbytes[sony_fetchtrack(buffer, buflen, &pos)] != sum)
			continue;
		j++;

		if (sony_fetchtrack(buffer, buflen, &pos) != 0xDE)
			continue;
		j++;

		if (sony_fetchtrack(buffer, buflen, &pos) != 0xAA)
			continue;
		j++;

		while((val = sony_fetchtrack(buffer, buflen, &pos)) == 0xFF)
			j++;
		if (val != 0xD5)
			continue;	/* lost bit slip mark! */
		j++;

		if (sony_fetchtrack(buffer, buflen, &pos) != 0xAA)
			continue;
		j++;

		if (sony_fetchtrack(buffer, buflen, &pos) != 0xAD)
			continue;
		j++;

		/* should this be regarded as a critical error ??? */
		if (rev_diskbytes[sony_fetchtrack(buffer, buflen, &pos)] != sector)
			continue;
		j++;

		for (i = 0; i < (sizeof(nibble_data) / sizeof(nibble_data[0])); i++)
		{
			nibble_data[i] = rev_diskbytes[sony_fetchtrack(buffer, buflen, &pos)];
			j++;
		}

		for (i = 3; i >= 0; i--)
		{
			/* should be checking checksum */
			sony_fetchtrack(buffer, buflen, &pos);
		}

		sony_fetchtrack(buffer, buflen, &pos);	/* should get 0xDE */
		sony_fetchtrack(buffer, buflen, &pos);	/* should get 0xAA */
		sony_fetchtrack(buffer, buflen, &pos);	/* should get 0xFF */

		/* did we already write this sector? */
		if ((found_sectors & (1 << sector)) == 0)
		{
			sony_denibblize35(sector_data, nibble_data, checksum);
	
			/* write the sector */
			err = apple35_write_sector_td(floppy, head, track, sector, sector_data, sizeof(sector_data) / sizeof(sector_data[0]));
			if (err)
				return err;

			found_sectors |= 1 << sector;
		}
	}
	return FLOPPY_ERROR_SUCCESS;
}



static floperr_t apple35_construct(floppy_image *floppy, UINT32 data_offset, UINT32 data_size,
	UINT32 tag_offset, UINT32 tag_size, INT16 format_byte, UINT8 sides)
{
	struct apple35_tag *tag;
	struct FloppyCallbacks *format;

	/* figure out format byte if not specified */
	if (format_byte < 0)
	{
		switch(sides)
		{
			case 1:
				format_byte = 0x02;
				break;
			case 2:
				format_byte = 0x22;
				break;
			default:
				return FLOPPY_ERROR_INVALIDIMAGE;
		}
	}

	/* create tag */
	tag = (struct apple35_tag *) floppy_create_tag(floppy, APPLE35_TAG, sizeof(struct apple35_tag));
	if (!tag)
		return FLOPPY_ERROR_OUTOFMEMORY;
	tag->data_offset = data_offset;
	tag->data_size = data_size;
	tag->tag_offset = tag_offset;
	tag->tag_size = tag_size;
	tag->format_byte = (UINT8) format_byte;
	tag->sides = sides;


	/* set up format callbacks */
	format = floppy_callbacks(floppy);
	format->read_sector = apple35_read_sector;
	format->write_sector = apple35_write_sector;
	format->read_track = apple35_read_track;
	format->write_track = apple35_write_track;
	format->get_sector_length = apple35_get_sector_length;
	format->get_heads_per_disk = apple35_get_heads_per_disk;
	format->get_tracks_per_disk = apple35_get_tracks_per_disk;
	format->get_track_size = apple35_get_track_size;
	return FLOPPY_ERROR_SUCCESS;
}



/* ----------------------------------------------------------------------- */

static FLOPPY_IDENTIFY(apple35_raw_identify)
{
	UINT64 size;
	size = floppy_image_size(floppy);
	*vote = ((size == 80*1*10*512) || (size == 80*2*10*512)) ? 100 : 0;
	return FLOPPY_ERROR_SUCCESS;
}




static FLOPPY_CONSTRUCT(apple35_raw_construct)
{
	UINT64 size;
	UINT8 sides;

	if (params)
	{
		/* create */
		sides = option_resolution_lookup_int(params, PARAM_HEADS);
		size = 80*sides*10*512;
	}
	else
	{
		/* load */
		size = floppy_image_size(floppy);
		if (size == 80*1*10*512)
			sides = 1;
		else if (size == 80*2*10*512)
			sides = 2;
		else
			return FLOPPY_ERROR_INVALIDIMAGE;
	}

	return apple35_construct(floppy, 0, (UINT32) size, 0, 0, -1, sides);
}



/* ----------------------------------------------------------------------- */

struct header_diskcopy
{
	UINT8 disk_name[64];	/* name of the disk */
	UINT32 data_size;		/* total size of data for all sectors (512*number_of_sectors) */
	UINT32 tag_size;		/* total size of tag data for all sectors (12*number_of_sectors for GCR 3.5" floppies, 20*number_of_sectors for HD20, 0 otherwise) */
	UINT32 data_checksum;	/* CRC32 checksum of all sector data */
	UINT32 tag_checksum;	/* CRC32 checksum of all tag data */
	UINT8 disk_format;		/* 0 = 400K, 1 = 800K, 2 = 720K, 3 = 1440K  (other values reserved) */
	UINT8 format_byte;		/* should be $00 Apple II, $01 Lisa, $02 Mac MFS ??? */
							/* $12 = 400K, $22 = >400K Macintosh (DiskCopy uses this value for
							   all Apple II disks not 800K in size, and even for some of those),
							   $24 = 800K Apple II disk */
	UINT16 magic;			/* always $0100 (otherwise, the file may be in a different format. */
};



static floperr_t apple35_diskcopy_headerdecode(floppy_image *floppy, UINT32 *data_offset,
	UINT32 *data_size, UINT32 *tag_offset, UINT32 *tag_size, UINT8 *format_byte, UINT8 *sides)
{
	UINT64 size;
	struct header_diskcopy header;

	if (data_offset)
		*data_offset = 0;
	if (data_size)
		*data_size = 0;
	if (tag_offset)
		*tag_offset = 0;
	if (tag_size)
		*tag_size = 0;
	if (format_byte)
		*format_byte = 0;
	if (sides)
		*sides = 0;

	size = floppy_image_size(floppy);
	if (size < sizeof(struct header_diskcopy))
		return FLOPPY_ERROR_INVALIDIMAGE;

	floppy_image_read(floppy, &header, 0, sizeof(header));

	header.data_size = BIG_ENDIANIZE_INT32(header.data_size);
	header.tag_size = BIG_ENDIANIZE_INT32(header.tag_size);
	header.data_checksum = BIG_ENDIANIZE_INT32(header.data_checksum);
	header.tag_checksum = BIG_ENDIANIZE_INT32(header.tag_checksum);
	header.magic = BIG_ENDIANIZE_INT16(header.magic);

	if (header.disk_name[0] >= sizeof(header.disk_name))
		return FLOPPY_ERROR_INVALIDIMAGE;
	if (header.magic != 0x0100)
		return FLOPPY_ERROR_INVALIDIMAGE;
	if (size != (header.data_size + header.tag_size + sizeof(header)))
		return FLOPPY_ERROR_INVALIDIMAGE;

	if (header.data_size == 80*1*10*512)
	{
		if (sides)
			*sides = 1;
	}
	else if (header.data_size == 80*2*10*512)
	{
		if (sides)
			*sides = 2;
	}
	else
		return FLOPPY_ERROR_INVALIDIMAGE;

	if (data_offset)
		*data_offset = sizeof(header);
	if (data_size)
		*data_size = header.data_size;
	if (tag_offset)
		*tag_offset = sizeof(header) + header.data_size;
	if (tag_size)
		*tag_size = header.tag_size;
	if (format_byte)
		*format_byte = header.format_byte;
	return FLOPPY_ERROR_SUCCESS;
}



static FLOPPY_IDENTIFY(apple35_diskcopy_identify)
{
	*vote = apple35_diskcopy_headerdecode(floppy, NULL, NULL, NULL, NULL, NULL, NULL) ? 0 : 100;
	return FLOPPY_ERROR_SUCCESS;
}




static FLOPPY_CONSTRUCT(apple35_diskcopy_construct)
{
	floperr_t err;
	UINT8 format_byte, sides;
	UINT32 data_offset, data_size;
	UINT32 tag_offset, tag_size;
	INT16 format_byte_param = -1;
	struct header_diskcopy header;

	if (params)
	{
		/* create */
		sides = option_resolution_lookup_int(params, PARAM_HEADS);

		data_size = 80*sides*10*512;
		tag_size = 80*sides*10*12;

		memset(&header, 0, sizeof(header));
		header.data_size = BIG_ENDIANIZE_INT32(data_size);
		header.tag_size = BIG_ENDIANIZE_INT32(tag_size);
		header.disk_format = (sides > 1) ? 1 : 0;
		header.magic = BIG_ENDIANIZE_INT16(0x100);

		floppy_image_write(floppy, &header, 0, sizeof(header));
		floppy_image_write_filler(floppy, 0, sizeof(header), data_size + tag_size); 
	}

	/* load */
	err = apple35_diskcopy_headerdecode(floppy, &data_offset, &data_size,
		&tag_offset, &tag_size, &format_byte, &sides);
	if (err)
		return err;

	format_byte_param = format_byte;

	return apple35_construct(floppy, data_offset, data_size,
		tag_offset, tag_size, format_byte_param, sides);
}



/* ----------------------------------------------------------------------- */

struct header_2img
{
	char magic[4];			/* '2IMG' */
	char creator[4];		/* signature; 'MESS' for MESS */
	UINT16 header_length;
	UINT16 version;
	UINT32 image_format;
	UINT32 flags;
	UINT32 block_count;
	UINT32 data_offset;
	UINT32 data_length;
	UINT32 comment_offset;
	UINT32 comment_length;
	UINT32 creator_offset;
	UINT32 creator_length;
	UINT32 padding[4];
};

#define IMAGE_FORMAT_DO		0
#define IMAGE_FORMAT_PO		1
#define IMAGE_FORMAT_NIB	2
#define IMAGE_FLAGS_LOCKED	0x80000000



static floperr_t apple35_2img_decode(floppy_image *floppy, UINT32 *image_format,
	UINT32 *data_offset, UINT32 *data_length)
{
	struct header_2img header;
	UINT64 size;

	if (image_format)
		*image_format = 0;
	if (data_offset)
		*data_offset = 0;
	if (data_length)
		*data_length = 0;

	size = floppy_image_size(floppy);
	if (size < sizeof(header))
		return FLOPPY_ERROR_INVALIDIMAGE;

	floppy_image_read(floppy, &header, 0, sizeof(header));

	if (memcmp(header.magic, "2IMG", 4))
		return FLOPPY_ERROR_INVALIDIMAGE;

	header.header_length	= LITTLE_ENDIANIZE_INT16(header.header_length);
	header.version			= LITTLE_ENDIANIZE_INT16(header.version);
	header.image_format		= LITTLE_ENDIANIZE_INT32(header.image_format);
	header.flags			= LITTLE_ENDIANIZE_INT32(header.flags);
	header.block_count		= LITTLE_ENDIANIZE_INT32(header.block_count);
	header.data_offset		= LITTLE_ENDIANIZE_INT32(header.data_offset);
	header.data_length		= LITTLE_ENDIANIZE_INT32(header.data_length);
	header.comment_offset	= LITTLE_ENDIANIZE_INT32(header.comment_offset);
	header.comment_length	= LITTLE_ENDIANIZE_INT32(header.comment_length);
	header.creator_offset	= LITTLE_ENDIANIZE_INT32(header.creator_offset);
	header.creator_length	= LITTLE_ENDIANIZE_INT32(header.creator_length);

	if ((((UINT64) header.data_offset) + header.data_length) > size)
		return FLOPPY_ERROR_INVALIDIMAGE;
	if ((((UINT64) header.comment_offset) + header.comment_length) > size)
		return FLOPPY_ERROR_INVALIDIMAGE;
	if ((((UINT64) header.creator_offset) + header.creator_length) > size)
		return FLOPPY_ERROR_INVALIDIMAGE;
	if ((header.image_format != IMAGE_FORMAT_DO) &&
			(header.image_format != IMAGE_FORMAT_PO) &&
			(header.image_format != IMAGE_FORMAT_NIB))
		return FLOPPY_ERROR_INVALIDIMAGE;

	if (image_format)
		*image_format = header.image_format;
	if (data_offset)
		*data_offset = header.data_offset;
	if (data_length)
		*data_length = header.data_length;
	return FLOPPY_ERROR_SUCCESS;
}



static FLOPPY_IDENTIFY(apple35_2img_identify)
{
	*vote = apple35_2img_decode(floppy, NULL, NULL, NULL) ? 0 : 100;
	return FLOPPY_ERROR_SUCCESS;
}



static FLOPPY_CONSTRUCT(apple35_2img_construct)
{
	floperr_t err;
	UINT32 image_format;
	UINT32 data_offset;
	UINT32 data_size;
	UINT8 sides = 2;
	struct header_2img header;

	if (params)
	{
		/* create */
		sides = option_resolution_lookup_int(params, PARAM_HEADS);

		data_offset = sizeof(header);
		data_size = 80*sides*10*512;

		memset(&header, 0, sizeof(header));
		header.header_length	= LITTLE_ENDIANIZE_INT16(sizeof(header));
		header.block_count		= LITTLE_ENDIANIZE_INT32(80*sides*10);
		header.data_offset		= LITTLE_ENDIANIZE_INT32(data_offset);
		header.data_length		= LITTLE_ENDIANIZE_INT32(data_size);

		floppy_image_write(floppy, &header, 0, sizeof(header));
		floppy_image_write_filler(floppy, 0, sizeof(header), data_size);
	}
	else
	{
		/* load */
		err = apple35_2img_decode(floppy, &image_format, &data_offset, &data_size);
		if (err)
			return err;

		if (data_size == 80*1*10*512)
			sides = 1;	/* single sided */
		else if (data_size == 80*2*10*512)
			sides = 2;	/* double sided */
		else
			sides = 2;	/* unknown... what to do... */
	}

	return apple35_construct(floppy, data_offset, data_size,
		0, 0, -1, sides);
}



FLOPPY_OPTIONS_START( apple35_mac )
	FLOPPY_OPTION( apple35_raw, "dsk\0img\0image\0",	"Apple raw 3.5\" disk image",	apple35_raw_identify,		apple35_raw_construct,
		HEADS([1]-2)
		TRACKS([80])
		SECTOR_LENGTH([512])
		FIRST_SECTOR_ID([0]))
	FLOPPY_OPTION( apple35_dc, "dc\0dsk\0img\0image\0",	"Apple DiskCopy disk image",	apple35_diskcopy_identify,	apple35_diskcopy_construct,
		HEADS([1]-2)
		TRACKS([80])
		SECTOR_LENGTH([512])
		FIRST_SECTOR_ID([0]))
FLOPPY_OPTIONS_END

FLOPPY_OPTIONS_START( apple35_iigs )
	FLOPPY_OPTION( apple35_raw, "dsk\0img\0image\0",	"Apple raw 3.5\" disk image",	apple35_raw_identify,		apple35_raw_construct,
		HEADS([1]-2)
		TRACKS([80])
		SECTOR_LENGTH([512])
		FIRST_SECTOR_ID([0]))
	FLOPPY_OPTION( apple35_dc, "dc\0dsk\0img\0image\0",	"Apple DiskCopy disk image",	apple35_diskcopy_identify,	apple35_diskcopy_construct,	
		HEADS([1]-2)
		TRACKS([80])
		SECTOR_LENGTH([512])
		FIRST_SECTOR_ID([0]))
	FLOPPY_OPTION( apple35_2img, "2img\0002mg\0",		"Apple ][gs 2IMG disk image",	apple35_2img_identify,		apple35_2img_construct,
		HEADS([1]-2)
		TRACKS([80])
		SECTOR_LENGTH([512])
		FIRST_SECTOR_ID([0]))
FLOPPY_OPTIONS_END
