Cura/plugins/USBPrinting/avr_isp/ispBase.py

"""
General interface for Isp based AVR programmers.
The ISP AVR programmer can load firmware into AVR chips. Which are commonly used on 3D printers.

 Needs to be subclassed to support different programmers.
 Currently only the stk500v2 subclass exists.
 This is a python 3 conversion of the code created by David Braam for the Cura project.
"""

from . import chipDB
from UM.Logger import Logger

class IspBase():
    """
    Base class for ISP based AVR programmers.
    Functions in this class raise an IspError when something goes wrong.
    """
    def programChip(self, flash_data):
        """ Program a chip with the given flash data. """
        self.cur_ext_addr = -1
        self.chip = chipDB.getChipFromDB(self.getSignature())
        if not self.chip:
            raise IspError("Chip with signature: " + str(self.getSignature()) + "not found")
        self.chipErase()
        
        Logger.log("d", "Flashing %i bytes", len(flash_data))
        self.writeFlash(flash_data)
        Logger.log("d", "Verifying %i bytes", len(flash_data))
        self.verifyFlash(flash_data)
        Logger.log("d", "Completed")

    def getSignature(self):
        """
        Get the AVR signature from the chip. This is a 3 byte array which describes which chip we are connected to.
        This is important to verify that we are programming the correct type of chip and that we use proper flash block sizes.
        """
        sig = []
        sig.append(self.sendISP([0x30, 0x00, 0x00, 0x00])[3])
        sig.append(self.sendISP([0x30, 0x00, 0x01, 0x00])[3])
        sig.append(self.sendISP([0x30, 0x00, 0x02, 0x00])[3])
        return sig

    def chipErase(self):
        """
        Do a full chip erase, clears all data, and lockbits.
        """
        self.sendISP([0xAC, 0x80, 0x00, 0x00])

    def writeFlash(self, flash_data):
        """
        Write the flash data, needs to be implemented in a subclass.
        """
        raise IspError("Called undefined writeFlash")

    def verifyFlash(self, flash_data):
        """
        Verify the flash data, needs to be implemented in a subclass.
        """
        raise IspError("Called undefined verifyFlash")


class IspError(Exception):
    def __init__(self, value):
        self.value = value

    def __str__(self):
        return repr(self.value)