nesdev.com

.proc start_palette_checker
  rep #$30    ; A=16, X/Y=16
  ldx #$0000

palette_checker_loop1:
  lda PaletteTable,x         ;first 32 bytes specify what palette is in the palette slot
  cmp PaletteRequest         ;sees if the color in the palette slot matches with the one you are trying to add
  bne prepare_for_palette_checker_loop

done:
  lda PaletteTable+32,x      ;the last 32 bytes say how many objects are using that palette
  inc            ;so if it is 0, it is safe to overwrite it. a new object is using that palette,
  sta PaletteTable+32,x      ;so increase it by 1.

  sep #$20

  ldx
  ora Attributes+1   ;this will be direct paged for object slots, I just forgot how to do it without having to do a:.
  sta Attributes+1   ;Attributes is the exact same thing as bytes 3 and 4 in oam.
  rts

prepare_for_palette_checker_loop:
  inx               ;we're using 16 bits, so you increase x by 2.
  inx
  cpx #$0020            ;sees if we're at the second half of the palette table.
  bcc palette_checker_loop1

palette_checker_loop2:
  lda PaletteTable,x
  bne prepare_for_palette_checker_loop   ;0 means that no objects are using the palette slot,
               ;so it is safe to overwrite it.

done2:
  ldx
  ror         ;same thing as dividing by 2? How is asl any different than rol then?
  ora Attributes+1   ;this will be direct paged for object slots.
  sta Attributes+1   ;Attributes is the exact same thing as bytes 3 and 4 in oam.

  lda PaletteRequest
  sta PaletteTable-32,x      ;remember, x is twice as much here as it would be in done1 (64/2 = 32)
  lda PaletteTable,x      ;the last 32 bytes say how many objects are using that palette
  inc            ;so if it is 0, it is safe to overwrite it. a new object is using that palette,
  sta PaletteTable,x      ;so increase it by 1.

  sep #$20   ;how do you get an 8 bit x, because I'm pretty sure that's what you need here
  ldx      ;(by the way, this next section is going to be pretty rough, because I've never directly worked with DMA...)
  asl      ;since a palette is 32 bytes and we already have the palette number times 4,
  asl      ;we multiply it by 8 for the result.
  asl
  asl
  stx $2121   ;says which color to start at

  lda PaletteRequest
  sta $4302   ; Store data offset into DMA source offset (PaletteRequest is really an adress,
  lda #$01   ; and because I'm only using 1 bank for palettes, every number can be its own adress)
  sta $4304   ; Store data bank into DMA source bank (bank 1)
  lda #$10
  sta $4305   ; Store size of data block (32 bytes = 1 16 color palette)

  stz $4300   ; Set DMA Mode (byte, normal increment)
  lda #$22   ; Set destination register ($2122 - CGRAM Write)
  sta $4301
  lda #$01   ; Initiate DMA transfer
  sta $420B
  rts

.endproc

.proc start_palette_checker
  rep #$30
  sep #$20   
  ldx #$00

prepare_for_palette_checker_loop:
  inx
  cpx #$08
  bcs palette_slot_not_found

palette_checker_loop:
  lda OnscreenPaletteTable,x
  bne prepare_for_palette_checker_loop

done:
  ldx
  asl
  stx TotalPalettesTable,y

  lda #$01
  sta OnscreenPaletteTable,x

  lda PaletteAddressTable,y
  sta DMAPaletteRequestTable,x
  inx
  sta DMAPaletteRequestCounter

palette_slot_not_found:
  rts

.endproc

PaletteAddressTable:
  .byte $??,$??,$??,$??,etc...

.proc start_palette_uploader
  rep #$30
  sep #$20
  ldx #$00
  ldy #$00
  stz DMAPaletteRequestCounter
 
palette_uploader_loop:
  sty $2121

  lda DMAPaletteTable,x
  beq done:
  sta $4302
  lda #$01
  sta $4304
  lda #$10
  sta $4305

  stz $4300
  lda #$22
  sta $4301
  lda #$01
  sta $420B

  tya
  clc
  adc #$10
  tay
  inx
  cpx #$08
  bne palette_uploader_loop

done:
  rep #$30
  stz DMAPaletteTable
  stz DMAPaletteTable+2
  stz DMAPaletteTable+4
  stz DMAPaletteTable+6
  rts

.endproc

  ldy #$00
  jsr start_palette_checker

  ldy #$00
  jsr start_palette_checker
  ldy #$02
  jsr start_palette_checker

.setcpu "65816"
.smart

; See Game.asm
.importzp DMAPaletteRequestCounter
.import TotalPalettesTable
.import OnscreenPaletteTable
.import DMAPaletteTable
.import Player1Palette, BulletPalette

; Used by Game.asm
.export start_palette_checker

.segment "CODE"

;
; start_palette_checker
;
.proc start_palette_checker
  rep #$30
  sep #$10
  ldx #$00

palette_checker_loop:
  lda OnscreenPaletteTable,x
  beq done

prepare_for_palette_checker_loop:
  inx
  cpx #$08
  bcs palette_slot_not_found
  bra palette_checker_loop     ;Bug fix by UnDisbeliever

done:
  txa
  sta TotalPalettesTable,y
 
  lda #$01
  sta OnscreenPaletteTable,x

  rep #$30
  ldx DMAPaletteRequestCounter
  lda PaletteAddressTable,y
  sta DMAPaletteTable,x
  inx
  inx
  stx DMAPaletteRequestCounter

palette_slot_not_found:
  rts

.endproc

PaletteAddressTable:
  .word Player1Palette,BulletPalette (These can be whatever, of course.)

.setcpu "65816"
.smart

; See Game.asm
.importzp DMAPaletteRequestCounter
.import DMAPaletteTable

Player1Palette = 0
BulletPalette = 2

; Used by Game.asm
.export start_palette_uploader

.segment "CODE"

.proc start_palette_uploader
  sep #$30
  ldx #$00
  ldy #$80
  stz DMAPaletteRequestCounter
 
palette_uploader_loop:
  sty $2121

  rep #$30 
  lda DMAPaletteTable,x
  beq done
  sta $4302
  sep #$30
  lda #$00
  sta $4304
  lda #$20
  sta $4305

  stz $4300
  lda #$22
  sta $4301
  lda #$01
  sta $420B

  tya
  clc
  adc #$10
  tay
  inx
  inx
  cpx #$10
  bne palette_uploader_loop
  rep #$30

done:
  stz DMAPaletteTable
  stz DMAPaletteTable+2
  stz DMAPaletteTable+4
  stz DMAPaletteTable+6
  stz DMAPaletteTable+8
  stz DMAPaletteTable+10
  stz DMAPaletteTable+12
  stz DMAPaletteTable+14
  rts

.endproc

.proc start_palette_checker
  rep #$30
  sep #$10
  ldx #$00

palette_checker_loop:
  lda OnscreenPaletteTable,x
  beq done