From b9c668a0401e25210b208f7df6e951b2918ab958 Mon Sep 17 00:00:00 2001
From: Michael Buesch <mb@bu3sch.de>
Date: Mon, 8 Feb 2010 03:18:05 +0100
Subject: atmega8dip28: Implement Flash writing

Signed-off-by: Michael Buesch <mb@bu3sch.de>
---
 chip_atmega8dip28.py | 68 ++++++++++++++++++++++++++++++++++++++++++++++------
 toprammer            |  6 ++---
 2 files changed, 64 insertions(+), 10 deletions(-)

diff --git a/chip_atmega8dip28.py b/chip_atmega8dip28.py
index f51ffb7..c83d525 100644
--- a/chip_atmega8dip28.py
+++ b/chip_atmega8dip28.py
@@ -56,12 +56,7 @@ class ATMega8DIP28(Chip):
 		self.top.blockCommands()
 		self.__loadCommand(self.CMD_CHIPERASE)
 		self.__pulseWR()
-		self.top.unblockCommands()
-		time.sleep(0.1)
-		self.top.blockCommands()
-		while not self.__getRDY():
-			self.printInfo(".", newline=False)
-			time.sleep(0.1)
+		self.__waitForRDY()
 		self.top.unblockCommands()
 		self.printInfo("100%")
 
@@ -103,7 +98,28 @@ class ATMega8DIP28(Chip):
 		self.__checkDUTPresence()
 		self.__initPins()
 
-		#TODO
+		self.printInfo("Writing Flash ", newline=False)
+		self.top.blockCommands()
+		for chunk in range(0, 256, 2):
+			if chunk % 8 == 0:
+				percent = (chunk * 100 / 256)
+				if percent % 25 == 0:
+					self.printInfo("%d%%" % percent, newline=False)
+				else:
+					self.printInfo(".", newline=False)
+			for word in range(0, 32):
+				self.__loadCommand(self.CMD_WRITEFLASH)
+				addr = (chunk << 4) + word
+				self.__loadAddr(addr)
+				addr *= 2
+				data = image[addr : addr + 2]
+				self.__loadData(ord(data[0]) | (ord(data[1]) << 8))
+				self.__setBS1(1)
+				self.__pulsePAGEL()
+			self.__setBS1(0)
+			self.__pulseWR()
+			self.__waitForRDY()
+		self.top.unblockCommands()
 
 	def readEEPROM(self):
 		self.__checkDUTPresence()
@@ -304,6 +320,27 @@ class ATMega8DIP28(Chip):
 		self.__setOE(1)
 		self.__setReadMode(0)
 
+	def __loadData(self, data):
+		"""Load a data word."""
+		self.__loadDataLow(data)
+		self.__loadDataHigh(data >> 8)
+
+	def __loadDataLow(self, dataLow):
+		"""Load the low data byte."""
+		self.__setBS1(0)
+		self.__setXA0(1)
+		self.__setXA1(0)
+		self.top.cmdFPGAWrite(0x10, dataLow & 0xFF)
+		self.__pulseXTAL1()
+
+	def __loadDataHigh(self, dataHigh):
+		"""Load the high data byte."""
+		self.__setBS1(1)
+		self.__setXA0(1)
+		self.__setXA1(0)
+		self.top.cmdFPGAWrite(0x10, dataHigh & 0xFF)
+		self.__pulseXTAL1()
+
 	def __loadAddr(self, addr):
 		"""Load an address word."""
 		self.__loadAddrLow(addr)
@@ -342,6 +379,16 @@ class ATMega8DIP28(Chip):
 			value |= 0x80
 		self.top.cmdFPGAWrite(0x12, value)
 
+	def __waitForRDY(self):
+		"""Wait for the RDY pin to go high."""
+		self.top.flushCommands()
+		time.sleep(0.01)
+		for i in range(0, 50):
+			if self.__getRDY():
+				return
+			time.sleep(0.01)
+		self.throwError("Timeout waiting for READY signal from chip.")
+
 	def __getRDY(self):
 		"""Read the state of the RDY/BSY pin."""
 		self.top.cmdFPGAReadRaw(0x17)
@@ -412,6 +459,13 @@ class ATMega8DIP28(Chip):
 			value |= 0x80
 		self.top.cmdFPGAWrite(0x12, value)
 
+	def __pulsePAGEL(self, count=1):
+		"""Do a positive pulse on the PAGEL pin of the DUT"""
+		while count > 0:
+			self.__setPAGEL(1)
+			self.__setPAGEL(0)
+			count -= 1
+
 	def __setBS2(self, high):
 		"""Set the BS2 pin of the DUT"""
 		value = 0x0A
diff --git a/toprammer b/toprammer
index f24a9af..571df51 100755
--- a/toprammer
+++ b/toprammer
@@ -368,7 +368,7 @@ class TOP:
 	def receive(self, size):
 		"""Receive 'size' bytes on the bulk-in ep."""
 		# If there are blocked commands in the queue, send them now.
-		self.__flushCommands()
+		self.flushCommands()
 		try:
 			ep = self.bulkIn.address
 			data = ""
@@ -397,9 +397,9 @@ class TOP:
 			return
 		assert(self.commandsBlocked)
 		self.commandsBlocked = False
-		self.__flushCommands()
+		self.flushCommands()
 
-	def __flushCommands(self):
+	def flushCommands(self):
 		"""Flush the command queue, but don't unblock it."""
 		command = ""
 		for oneCommand in self.commandQueue:
-- 
cgit v1.2.3