// 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");

#ifdef LOWERCAL

// For a few parts, we need to support reading OSCCAL setting
// __BOOT_SIGROW_READ macro from boot.h doesn't work here
// so using hardcoded value which works for all of the parts
// in question - 0x21

#define boot_signature_byte_get_short(addr) \
(__extension__({                      \
      uint8_t __result;                         \
      __asm__ __volatile__                      \
      (                                         \
        "__wr_spmcsr %1, %2\n\t"                \
        "lpm %0, Z" "\n\t"                      \
        : "=r" (__result)                       \
        : "i" (_SFR_MEM_ADDR(__SPM_REG)),       \
          "r" ((uint8_t)(0x21)),  \
          "z" ((uint16_t)(addr))                \
      );                                        \
      __result;                                 \
}))

#endif


#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

