// Get all the "standard" definitions from the official boot.h
#include <avr/boot.h>


/*
   Implement some optimized versions that will use OUT instead
     of STS to write SPMCSR.
     (However, omit the *_extended_short, since by the time you
     need _extended_, the extra byte shouldn't be relevant any more)

   The C preprocessor can not determine at compile time whether SPMCSR is
     "out of range" of the OUT instruction, but we CAN do that in the
     assembler.  We can even make it pretty with a macro.
   With this modification, the _short functions should work on cpus
     (like ATmega128) where STS is required.
*/

asm(".macro __wr_spmcsr p, v \n\t"
    ".if \\p > 0x57     \n\t"
    "sts \\p, \\v       \n\t"
    ".else              \n\t"
    "out \\p-0x20, \\v  \n\t"
    ".endif             \n\t"
    ".endm              \n");


#if defined(__SPM_REG)

#define __boot_page_fill_short(address, data)    \
  (__extension__({                                 \
    __asm__ __volatile__                         \
    (                                            \
        "movw  r0, %3\n\t"                       \
        "__wr_spmcsr %0, %1\n\t"                 \
        "spm\n\t"                                \
        "clr  r1\n\t"                            \
        :                                        \
        : "i" (_SFR_MEM_ADDR(__SPM_REG)),         \
        "r" ((uint8_t)__BOOT_PAGE_FILL),       \
        "z" ((uint16_t)address),               \
        "r" ((uint16_t)data)                   \
        : "r0"                                   \
    );                                           \
  }))

#define __boot_page_erase_short(address)         \
  (__extension__({                                 \
    __asm__ __volatile__                         \
    (                                            \
        "__wr_spmcsr %0, %1\n\t"                 \
        "spm\n\t"                                \
        :                                        \
        : "i" (_SFR_MEM_ADDR(__SPM_REG)),         \
        "r" ((uint8_t)__BOOT_PAGE_ERASE),      \
        "z" ((uint16_t)address)                \
    );                                           \
  }))

#define __boot_page_write_short(address)         \
  (__extension__({                                 \
    __asm__ __volatile__                         \
    (                                            \
        "__wr_spmcsr %0, %1\n\t"                 \
        "spm\n\t"                                \
        :                                        \
        : "i" (_SFR_MEM_ADDR(__SPM_REG)),         \
        "r" ((uint8_t)__BOOT_PAGE_WRITE),      \
        "z" ((uint16_t)address)                \
    );                                           \
  }))

#define __boot_rww_enable_short()                \
  (__extension__({                                 \
    __asm__ __volatile__                         \
    (                                            \
        "__wr_spmcsr %0, %1\n\t"                 \
        "spm\n\t"                                \
        :                                        \
        : "i" (_SFR_MEM_ADDR(__SPM_REG)),         \
        "r" ((uint8_t)__BOOT_RWW_ENABLE)       \
    );                                           \
  }))

#endif // __SPM_REG

#ifndef __boot_page_erase_short

  /*
  if __SPM_REG didn't get defined by now, but we didn't exit it means
  we have some sort of new-fangled chip that post-dates the version
  of boot.h that we know about.  In this case, it's possible that the
  standard boot.h still has workable functions, so we'll alias those.
  */

  #define __boot_page_fill_short(address, data) boot_page_fill(address, data)
  #define __boot_page_erase_short(address) boot_page_erase(address)
  #define __boot_page_write_short(address) boot_page_write(address)
  #define __boot_rww_enable_short() boot_rww_enable()

#endif