; PLASMA TREE

; based on original PLASMAGORIA code by French Touch

; =============================================================================
; ROUTINE MAIN
; =============================================================================

plasma_tree:

	lda	#0
	sta	FRAME

	lda	#4
	sta	DRAW_PAGE

	bit	PAGE2

	; remap the masks
	; $00->$00
	; $11->$40
	; $22->$80
	; $44->$C0


	ldy	#0
	sty	OUTL
	lda	#$40
	sta	OUTH
remap_mask:
	lda	(OUTL),Y
	and	#$7
	tax
	lda	remap_table,X
	sta	(OUTL),Y
	dey
	bne	remap_mask

	inc	OUTH
	lda	OUTH
	cmp	#$80
	bne	remap_mask


step3:

	; init

	lda	#02
	sta	COMPT2
	sta	PARAM1
	sta	PARAM2
	sta	PARAM3
	sta	PARAM4

bp3:
	jsr	precalc		; pre-calc
	jsr	display_normal	; display normal
	jsr	VBLANK

;	lda	#40
;	jsr	wait_for_pattern
;	bcc	keep_making_plasma
;	jmp	done_making_plasma

; update frame count

	inc	FRAMEL                                                  ; 5
	lda	FRAMEL                                                  ; 3
	and	#$3f                                                    ; 2
	sta	FRAMEL                                                  ; 3
	bne	frame_noflo5                                            ; 2/3
	inc	FRAMEH                                                  ; 5
frame_noflo5:


        lda	KEYPRESS
        bmi	done_making_plasma

        ; wait for_pattern / end

	lda	SOUND_STATUS
	and	#SOUND_MOCKINGBOARD
	beq	no_music5

;       lda     #1
;       cmp     current_pattern_smc+1
;       bcc     done_making_plasma
;       beq     done_making_plasma
;       jmp     done_music5

no_music5:
	lda	FRAMEH
	cmp	#8
	beq	done_making_plasma

done_music5:


keep_making_plasma:


	inc	COMPT1
	bne	bp3
	dec	COMPT2
	bne	bp3

	jmp	step3

done_making_plasma:
	bit	KEYRESET
	rts


; ============================================================================
; Precalculate some values
; ROUTINES PRE CALCUL
; ============================================================================
precalc:
	lda	PARAM1		; self modify various parts
	sta	pc_off1+1
	lda	PARAM2
	sta	pc_off2+1
	lda	PARAM3
	sta	pc_off3+1
	lda	PARAM4
	sta	pc_off4+1

	; Table1(X) = sin1(PARAM1+X)+sin2(PARAM1+X)
	; Table2(X) = sin3(PARAM3+X)+sin1(PARAM4+X)

	ldx	#$28		; 40
pc_b1:
pc_off1:
	lda	sin1
pc_off2:
	adc	sin2
	sta	Table1,X
pc_off3:
	lda	sin3
pc_off4:
	adc	sin1
	sta	Table2,X

	inc	pc_off1+1
	inc	pc_off2+1
	inc	pc_off3+1
	inc	pc_off4+1

 	dex
	bpl	pc_b1

 	inc	PARAM1
 	inc	PARAM1
	dec	PARAM2
 	inc	PARAM3
	dec	PARAM4

 	rts

; ============================================================================
; Display Routines
; ROUTINES AFFICHAGES
; ============================================================================

; Display "Normal"
; AFFICHAGE "NORMAL"

display_normal:
	bit	SET_GR			; gfx (lores)	why needed?

	ldx	#23			; lines 0-23	lignes 0-23

display_line_loop:
	lda	gr_lookup_low,X		; setup pointers for line
	sta	GRLINE
	lda	gr_lookup_high,X
	sta	GRLINE+1

	lda	gr_lookup_low,X		; setup pointers for mask
	sta	INL
	lda	gr_lookup_high,X
	clc
mask_src_smc:
	adc	#($70-$8)
	sta	INH


	ldy	#39			; col 0-39

	lda	Table2,X		; setup base sine value for row
	sta	display_row_sin_smc+1
display_col_loop:
	lda	Table1,Y		; load in column sine value
display_row_sin_smc:
	adc	#00			; add in row value

	and	#$3f
	ora	(INL),Y

	sta	display_lookup_smc+1	; patch in low byte of lookup
display_lookup_smc:
	lda	lores_colors_rgb	; attention: must be aligned

	sta	(GRLINE),Y
	dey
	bpl	display_col_loop
	dex
	bpl	display_line_loop

	rts


VBLANK:
	inc	FRAME

	lda	FRAME
	lsr
	lsr
;	lsr
;	lsr
	and	#$f
	tax
	lda	mask_src_table,X
	sta	mask_src_smc+1

	rts


;.align 256

gr_lookup_low:
	.byte $00,$80,$00,$80,$00,$80,$00,$80
	.byte $28,$A8,$28,$A8,$28,$A8,$28,$A8
	.byte $50,$D0,$50,$D0,$50,$D0,$50,$D0

gr_lookup_high:
	.byte $08,$08,$09,$09,$0A,$0A,$0B,$0B
	.byte $08,$08,$09,$09,$0A,$0A,$0B,$0B
	.byte $08,$08,$09,$09,$0A,$0A,$0B,$0B



.align 256


; This appears to be roughly 47+32*sin(x)+16*sin(2x)
sin1: ; 256
.byte $2E,$30,$32,$34,$35,$36,$38,$3A,$3C,$3C,$3E,$40,$41,$42,$44,$45,$47,$47,$49,$4A,$4B,$4C,$4D,$4E,$4F,$50,$51,$52,$53,$53,$54,$54
.byte $55,$55,$56,$57,$57,$58,$58,$57,$58,$58,$58,$58,$58,$58,$58,$58,$58,$57,$57,$57,$56,$56,$55,$54,$55,$54,$53,$52,$52,$51,$50,$4F
.byte $4E,$4E,$4D,$4C,$4B,$4B,$4A,$49,$48,$47,$46,$45,$45,$44,$42,$42,$41,$41,$3F,$3F,$3D,$3D,$3C,$3B,$3B,$39,$39,$39,$38,$38,$37,$36
.byte $36,$35,$35,$34,$34,$33,$32,$32,$32,$31,$31,$31,$30,$31,$30,$30,$30,$30,$2F,$2F,$30,$2F,$2F,$2F,$2F,$2F,$2F,$2F,$2E,$2F,$2F,$2F
.byte $2E,$2F,$2F,$2F,$2F,$2E,$2F,$2F,$2F,$2E,$2F,$2F,$2E,$2E,$2F,$2E,$2E,$2D,$2E,$2D,$2D,$2D,$2C,$2C,$2C,$2B,$2B,$2B,$2A,$2A,$29,$28
.byte $28,$27,$27,$26,$26,$25,$25,$23,$23,$22,$21,$21,$20,$1F,$1F,$1D,$1D,$1C,$1B,$1A,$19,$19,$17,$16,$16,$15,$14,$13,$13,$12,$11,$10
.byte $0F,$0F,$0E,$0D,$0C,$0C,$0B,$0A,$09,$09,$08,$08,$08,$07,$06,$07,$06,$06,$06,$06,$05,$06,$05,$05,$06,$05,$06,$06,$07,$07,$08,$08
.byte $09,$09,$0A,$0B,$0B,$0C,$0C,$0D,$0F,$0F,$10,$12,$12,$14,$15,$16,$17,$19,$1A,$1B,$1D,$1E,$20,$21,$22,$24,$26,$27,$28,$2A,$2C,$2E

; This appears to be roughly 47+32*sin(4x)+16*sin(3x)
sin2: ; 256
.byte $2E,$33,$38,$3C,$40,$43,$47,$4B,$4E,$51,$54,$56,$59,$5A,$5C,$5D,$5D,$5E,$5E,$5D,$5C,$5A,$59,$57,$55,$53,$4F,$4C,$49,$46,$42,$3E
.byte $3A,$36,$32,$2E,$2A,$26,$23,$1F,$1C,$18,$15,$12,$10,$0E,$0C,$0A,$09,$08,$07,$07,$07,$07,$09,$0A,$0B,$0D,$0F,$11,$13,$16,$19,$1C
.byte $1F,$22,$26,$29,$2C,$2F,$32,$36,$38,$3B,$3E,$3F,$42,$44,$46,$47,$48,$49,$4B,$4B,$4B,$4A,$4A,$49,$49,$48,$46,$44,$43,$41,$3F,$3C
.byte $3A,$38,$35,$33,$30,$2E,$2C,$2A,$28,$26,$24,$22,$21,$20,$1F,$1F,$1E,$1E,$1D,$1D,$1E,$1E,$1F,$20,$21,$22,$24,$25,$27,$29,$2B,$2D
.byte $2E,$30,$33,$35,$37,$38,$3A,$3C,$3D,$3E,$3F,$3F,$40,$40,$41,$40,$40,$3F,$3F,$3E,$3D,$3B,$3A,$38,$36,$34,$31,$2F,$2D,$2B,$29,$25
.byte $23,$21,$1F,$1D,$1B,$19,$18,$16,$15,$14,$14,$13,$13,$13,$13,$14,$16,$17,$18,$1A,$1C,$1D,$20,$23,$26,$28,$2C,$2E,$32,$35,$38,$3B
.byte $3E,$41,$45,$48,$4B,$4C,$4F,$51,$53,$54,$55,$55,$57,$57,$57,$56,$55,$53,$52,$50,$4E,$4B,$49,$45,$42,$3F,$3B,$37,$34,$30,$2C,$27
.byte $23,$1F,$1C,$18,$14,$11,$0E,$0B,$09,$07,$05,$03,$02,$01,$00,$00,$01,$01,$02,$03,$05,$07,$0A,$0D,$10,$13,$17,$1A,$1E,$22,$26,$2A

; This appears to be roughly 38+24*sin(3x)+16*sin(8x)
sin3: ; 256
.byte $26,$2C,$31,$35,$39,$3D,$40,$42,$44,$45,$45,$46,$45,$43,$42,$40,$3C,$3A,$38,$36,$33,$31,$30,$2F,$2F,$2E,$2F,$2F,$30,$33,$33,$36
.byte $37,$3A,$3C,$3C,$3E,$3E,$3D,$3D,$3B,$39,$36,$34,$30,$2B,$28,$23,$1D,$19,$14,$11,$0C,$09,$07,$04,$03,$03,$03,$03,$04,$07,$09,$0C
.byte $0F,$13,$16,$18,$1B,$1E,$20,$22,$22,$23,$24,$24,$23,$22,$21,$20,$1D,$1C,$1B,$1A,$19,$19,$19,$1A,$1C,$1E,$20,$23,$27,$2B,$2F,$33
.byte $37,$3D,$40,$44,$47,$4A,$4C,$4D,$4E,$4E,$4D,$4C,$4A,$47,$45,$41,$3C,$39,$35,$32,$2E,$2B,$28,$26,$25,$23,$23,$22,$22,$24,$24,$25
.byte $26,$29,$2A,$2A,$2B,$2C,$2B,$2B,$29,$28,$25,$23,$20,$1C,$19,$15,$10,$0D,$09,$07,$04,$02,$01,$00,$00,$00,$02,$03,$06,$0A,$0D,$11
.byte $15,$1B,$1F,$23,$27,$2B,$2D,$30,$32,$33,$34,$35,$35,$33,$33,$32,$30,$2E,$2D,$2C,$2B,$2A,$2A,$2A,$2B,$2C,$2E,$30,$32,$36,$38,$3B
.byte $3E,$42,$45,$47,$49,$4B,$4B,$4B,$4A,$49,$47,$45,$42,$3D,$3A,$35,$30,$2B,$26,$22,$1E,$1A,$17,$14,$13,$11,$10,$10,$10,$12,$12,$14
.byte $15,$18,$1A,$1B,$1D,$1E,$1F,$1F,$1F,$1F,$1E,$1D,$1B,$18,$16,$14,$10,$0E,$0C,$0B,$09,$08,$08,$09,$0A,$0C,$0E,$11,$14,$19,$1D,$22


; Lookup table for colors
; Note the sine tables point roughly to the middle and go to the edges



lores_colors_rgb: ; 256

; black
.byte $00,$00,$00,$00,$00,$00,$00,$00
.byte $00,$00,$00,$00,$00,$00,$00,$00
.byte $00,$00,$00,$00,$00,$00,$00,$00
.byte $00,$00,$00,$00,$00,$00,$00,$00
.byte $00,$00,$00,$00,$00,$00,$00,$00
.byte $00,$00,$00,$00,$00,$00,$00,$00
.byte $00,$00,$00,$00,$00,$00,$00,$00
.byte $00,$00,$00,$00,$00,$00,$00,$00


; red gradient
; $00, $11, $33, $BB, $FF, $BB, $33, $11

.byte $11,$11,$11,$11,$11,$11,$11,$11
.byte $11,$11,$33,$33,$33,$33,$33,$33
.byte $33,$33,$33,$33,$bb,$bb,$bb,$bb
.byte $bb,$bb,$bb,$bb,$bb,$bb,$ff,$ff
.byte $ff,$ff,$ff,$ff,$ff,$ff,$ff,$ff
.byte $bb,$bb,$bb,$bb,$bb,$bb,$bb,$bb
.byte $bb,$bb,$33,$33,$33,$33,$33,$33
.byte $33,$33,$33,$33,$11,$11,$11,$11

; red
.if 0
.byte $11,$11,$11,$11,$11,$33,$33,$33
.byte $33,$33,$bb,$bb,$bb,$bb,$bb,$ff
.byte $ff,$ff,$ff,$ff,$ff,$bb,$bb,$bb
.byte $bb,$bb,$33,$33,$33,$33,$33,$11
.byte $11,$11,$11,$11,$33,$33,$33,$33
.byte $bb,$bb,$bb,$bb,$bb,$ff,$ff,$ff
.byte $ff,$ff,$ff,$bb,$bb,$bb,$bb,$bb
.byte $33,$33,$33,$33,$33,$33,$11,$11
.endif

; $00, $22, $66, $77, $FF, $77, $66, $22

; 22 66 77 ff 77 66

; blue


.byte $22,$22,$22,$22,$22,$22,$22,$22
.byte $22,$22,$22,$66,$66,$66,$66,$66
.byte $66,$66,$66,$66,$66,$66,$77,$77
.byte $77,$77,$77,$77,$77,$77,$77,$77
.byte $77,$ff,$ff,$ff,$ff,$ff,$ff,$ff
.byte $ff,$ff,$ff,$ff,$77,$77,$77,$77
.byte $77,$77,$77,$77,$77,$77,$77,$66
.byte $66,$66,$66,$66,$66,$66,$66,$66


.if 0
.byte $22,$22,$22,$22,$22,$66,$66,$66
.byte $66,$66,$77,$77,$77,$77,$77,$ff
.byte $ff,$ff,$ff,$ff,$ff,$77,$77,$77
.byte $77,$77,$66,$66,$66,$66,$66,$22
.byte $22,$22,$22,$22,$66,$66,$66,$66
.byte $77,$77,$77,$77,$77,$ff,$ff,$ff
.byte $ff,$ff,$ff,$77,$77,$77,$77,$77
.byte $66,$66,$66,$66,$66,$66,$22,$22
.endif



; $00, $44, $CC, $DD, $FF, $DD, $CC, $44

; green
.byte $44,$44,$44,$44,$44,$cc,$cc,$cc
.byte $cc,$cc,$dd,$dd,$dd,$dd,$dd,$ff
.byte $ff,$ff,$ff,$ff,$ff,$dd,$dd,$dd
.byte $dd,$dd,$cc,$cc,$cc,$cc,$cc,$44
.byte $44,$44,$44,$44,$cc,$cc,$cc,$cc
.byte $dd,$dd,$dd,$dd,$dd,$ff,$ff,$ff
.byte $ff,$ff,$ff,$dd,$dd,$dd,$dd,$dd
.byte $cc,$cc,$cc,$cc,$cc,$cc,$44,$44
.if 0
.byte $00,$00,$00,$00,$00,$00,$00,$00
.byte $44,$44,$44,$44,$44,$44,$44,$44
.byte $CC,$CC,$CC,$CC,$CC,$CC,$CC,$CC
.byte $DD,$DD,$DD,$DD,$DD,$DD,$DD,$DD
.byte $ff,$ff,$ff,$ff,$ff,$ff,$ff,$ff
.byte $DD,$DD,$DD,$DD,$DD,$DD,$DD,$DD
.byte $CC,$CC,$CC,$CC,$CC,$CC,$CC,$CC
.byte $44,$44,$44,$44,$44,$44,$44,$44
.endif

; This table has relatively fine color bands
.if 0
lores_colors_fine: ; 256
.byte $00,$00,$00,$00,$88,$88,$88,$88
.byte $55,$55,$55,$55,$99,$99,$99,$99
.byte $ff,$ff,$ff,$ff,$bb,$bb,$bb,$bb
.byte $33,$33,$33,$33,$22,$22,$22,$22
.byte $66,$66,$66,$66,$77,$77,$77,$77
.byte $44,$44,$44,$44,$cc,$cc,$cc,$cc
.byte $ee,$ee,$ee,$ee,$dd,$dd,$dd,$dd
.byte $99,$99,$99,$99,$11,$11,$11,$11

.byte $00,$00,$00,$00,$88,$88,$88,$88
.byte $55,$55,$55,$55,$99,$99,$99,$99
.byte $ff,$ff,$ff,$ff,$bb,$bb,$bb,$bb
.byte $33,$33,$33,$33,$22,$22,$22,$22
.byte $66,$66,$66,$66,$77,$77,$77,$77
.byte $44,$44,$44,$44,$cc,$cc,$cc,$cc
.byte $ee,$ee,$ee,$ee,$dd,$dd,$dd,$dd
.byte $99,$99,$99,$99,$11,$11,$11,$11

.byte $00,$00,$00,$00,$88,$88,$88,$88
.byte $55,$55,$55,$55,$99,$99,$99,$99
.byte $ff,$ff,$ff,$ff,$bb,$bb,$bb,$bb
.byte $33,$33,$33,$33,$22,$22,$22,$22
.byte $66,$66,$66,$66,$77,$77,$77,$77
.byte $44,$44,$44,$44,$cc,$cc,$cc,$cc
.byte $ee,$ee,$ee,$ee,$dd,$dd,$dd,$dd
.byte $99,$99,$99,$99,$11,$11,$11,$11

.byte $00,$00,$00,$00,$88,$88,$88,$88
.byte $55,$55,$55,$55,$99,$99,$99,$99
.byte $ff,$ff,$ff,$ff,$bb,$bb,$bb,$bb
.byte $33,$33,$33,$33,$22,$22,$22,$22
.byte $66,$66,$66,$66,$77,$77,$77,$77
.byte $44,$44,$44,$44,$cc,$cc,$cc,$cc
.byte $ee,$ee,$ee,$ee,$dd,$dd,$dd,$dd
.byte $99,$99,$99,$99,$11,$11,$11,$11
.else
; This table has relatively wide color bands
lores_colors_wide: ; 256
.byte $00,$00,$00,$00,$00,$00,$00,$00
.byte $00,$00,$00,$00,$00,$00,$00,$00
.byte $88,$88,$88,$88,$88,$88,$88,$88
.byte $88,$88,$88,$88,$88,$88,$88,$88
.byte $55,$55,$55,$55,$55,$55,$55,$55
.byte $55,$55,$55,$55,$55,$55,$55,$55
.byte $22,$22,$22,$22,$22,$22,$22,$22
.byte $22,$22,$22,$22,$22,$22,$22,$22
.byte $ff,$ff,$ff,$ff,$ff,$ff,$ff,$ff
.byte $ff,$ff,$ff,$ff,$ff,$ff,$ff,$ff
.byte $bb,$bb,$bb,$bb,$bb,$bb,$bb,$bb
.byte $bb,$bb,$bb,$bb,$bb,$bb,$bb,$bb
.byte $33,$33,$33,$33,$33,$33,$33,$33
.byte $33,$33,$33,$33,$33,$33,$33,$33
.byte $22,$22,$22,$22,$22,$22,$22,$22
.byte $22,$22,$22,$22,$22,$22,$22,$22
.byte $66,$66,$66,$66,$66,$66,$66,$66
.byte $66,$66,$66,$66,$66,$66,$66,$66
.byte $77,$77,$77,$77,$77,$77,$77,$77
.byte $77,$77,$77,$77,$77,$77,$77,$77
.byte $44,$44,$44,$44,$44,$44,$44,$44
.byte $44,$44,$44,$44,$44,$44,$44,$44
.byte $cc,$cc,$cc,$cc,$cc,$cc,$cc,$cc
.byte $cc,$cc,$cc,$cc,$cc,$cc,$cc,$cc
.byte $ee,$ee,$ee,$ee,$ee,$ee,$ee,$ee
.byte $ee,$ee,$ee,$ee,$ee,$ee,$ee,$ee
.byte $dd,$dd,$dd,$dd,$dd,$dd,$dd,$dd
.byte $dd,$dd,$dd,$dd,$dd,$dd,$dd,$dd
.byte $99,$99,$99,$99,$99,$99,$99,$99
.byte $99,$99,$99,$99,$99,$99,$99,$99
.byte $11,$11,$11,$11,$11,$11,$11,$11
.byte $11,$11,$11,$11,$11,$11,$11,$11
.endif

Table1	=	$2000
Table2	=	$2000+64

remap_table:
	.byte $00,$40,$80,$00,$C0

mask_src_table:
	.byte	$40-8,$44-8,$48-8,$4C-8,$50-8,$54-8,$58-8,$5C-8
	.byte	$60-8,$64-8,$68-8,$6C-8,$70-8,$74-8,$78-8,$7C-8
;	.byte	$7C-8,$78-8,$74-8,$70-8,$6C-8,$68-8,$64-8,$60-8


mask01_data:
.incbin "graphics/tree01.gr.zx02"
mask02_data:
.incbin "graphics/tree02.gr.zx02"
mask03_data:
.incbin "graphics/tree03.gr.zx02"
mask04_data:
.incbin "graphics/tree04.gr.zx02"
mask05_data:
.incbin "graphics/tree05.gr.zx02"
mask06_data:
.incbin "graphics/tree06.gr.zx02"
mask07_data:
.incbin "graphics/tree07.gr.zx02"
mask08_data:
.incbin "graphics/tree08.gr.zx02"
mask09_data:
.incbin "graphics/tree09.gr.zx02"
mask10_data:
.incbin "graphics/tree10.gr.zx02"
mask11_data:
.incbin "graphics/tree11.gr.zx02"
mask12_data:
.incbin "graphics/tree12.gr.zx02"
mask13_data:
.incbin "graphics/tree13.gr.zx02"
mask14_data:
.incbin "graphics/tree14.gr.zx02"
mask15_data:
.incbin "graphics/tree15.gr.zx02"
mask16_data:
.incbin "graphics/tree16.gr.zx02"