; Loader for MIST

.include "zp.inc"
.include "hardware.inc"

;.include "common_defines.inc"
.include "qboot.inc"

qload_start:

	; init the write code
;	lda	WHICH_SLOT
;	jsr	popwr_init

	; first time entry
	; start by loading text title

	lda	#0			; load LEMM engine
	sta	WHICH_LOAD

	lda	#1
	sta	CURRENT_DISK		; current disk number

	jsr	load_file

	jmp	$6000			; jump to LEMM

	;====================================
	; loads file specified by WHICH_LOAD
	;====================================
load_file:
	ldx	WHICH_LOAD

	lda	which_disk_array,X
	cmp	CURRENT_DISK
	bne	change_disk

load_file_no_diskcheck:
	lda	load_address_array,X
	sta	load_address

	lda	track_array,X
	sta	load_track

	lda	sector_array,X
	sta	load_sector

	lda	length_array,X
	sta	load_length

	jsr	load_new

	rts

	;===================================================
	;===================================================
	; change disk
	;===================================================
	;===================================================

change_disk:
.if 0
	; turn off disk drive light

	jsr	driveoff

	jsr	TEXT
	jsr	HOME

	lda	#<error_string
	sta	OUTL
	lda	#>error_string
	sta	OUTH

	ldx	WHICH_LOAD
	lda	which_disk_array,X
	clc
	adc	#48

	ldy	#19
	sta	(OUTL),Y

	ldy	#0

quick_print:
	lda	(OUTL),Y
	beq	quick_print_done
	jsr	COUT1
	iny
	jmp	quick_print

quick_print_done:

fnf_keypress:
	lda	KEYPRESS
	bpl	fnf_keypress
	bit	KEYRESET

	;==============================================
	; actually verify proper disk is there
	; read T0:S0 and verify proper disk

	lda	WHICH_LOAD
	pha

	ldx	#LOAD_FIRST_SECTOR	; load track 0 sector 0
	stx	WHICH_LOAD

	jsr	load_file_no_diskcheck

	pla
	sta	WHICH_LOAD
	tax

	; first sector now in $c00
	;	offset 59
	;		disk1 = $0a
	;		disk2 = $32 ('2')
	;		disk3 = $33 ('3')

	lda	$c59
	cmp	#$0a
	beq	is_disk1
	cmp	#$32
	beq	is_disk2
	cmp	#$33
	beq	is_disk3
	bne	change_disk		; unknown disk

is_disk1:
	lda	#1
	bne	disk_compare

is_disk2:
	lda	#2
	bne	disk_compare

is_disk3:
	lda	#3

disk_compare:
	cmp	which_disk_array,X
	bne	change_disk		; disk mismatch

	;==============================================
	; all good, retry original load

	jsr	HOME

	ldx	WHICH_LOAD
	lda	which_disk_array,X
	sta	CURRENT_DISK

	jmp	load_file

; offset for disk number is 19
error_string:
.byte "PLEASE INSERT DISK 1, PRESS RETURN",0

.endif

which_disk_array:
	.byte 1,1,1,1		; LEMM,   LEVEL1, LEVEL2, LEVEL3
	.byte 1,1,1,1		; LEVEL4, LEVEL5, LEVEL6, LEVEL7
	.byte 1,1,1,1		; LEVEL8, LEVEL9, LEVEL10

load_address_array:
        .byte $60,$90,$90,$90	; LEMM,   LEVEL1, LEVEL2, LEVEL3
	.byte $90,$90,$90,$90	; LEVEL4, LEVEL5, LEVEL6, LEVEL7
	.byte $90,$90,$90,$90	; LEVEL8, LEVEL9, LEVEL10

track_array:
        .byte  3, 6, 9,12	; LEMM,   LEVEL1, LEVEL2, LEVEL3
	.byte 15,18,21,24	; LEVEL4, LEVEL5, LEVEL6, LEVEL7
	.byte 27,30,33,33	; LEVEL8, LEVEL9, LEVEL10

sector_array:
        .byte  0, 0, 0, 0	; LEMM,   LEVEL1, LEVEL2, LEVEL3
	.byte  0, 0, 0, 0	; LEVEL4, LEVEL5, LEVEL6, LEVEL7
	.byte  0, 0, 0, 0	; LEVEL8, LEVEL9, LEVEL10

length_array:
        .byte  48, 46, 46, 46	; LEMM,   LEVEL1, LEVEL2, LEVEL3
	.byte  46, 46, 46, 46	; LEVEL4, LEVEL5, LEVEL6, LEVEL7
	.byte  46, 46, 46, 46	; LEVEL8, LEVEL9, LEVEL10


	.include	"wait.s"
	.include	"wait_a_bit.s"
	.include	"audio.s"
	.include	"decompress_fast_v2.s"
	.include	"gr_offsets.s"
	.include	"hgr_hlin.s"
	.include	"hgr_vlin.s"
	.include	"hgr_box.s"
	.include	"hgr_tables.s"
	.include	"hgr_partial_restore.s"
	.include	"hgr_14x14_sprite.s"

qload_end:

.assert (>qload_end - >qload_start) < $e , error, "loader too big"