-
-# check command line arguments
-if len(sys.argv) != 2:
- print "Usage: " + sys.argv[0] + " [mhx-file]"
- sys.exit(1)
-
-# read in data from mhx-file before starting
-try:
- fp = open(sys.argv[1], "r")
-except IOError:
- print sys.argv[0] + ": Error - couldn't open file " + sys.argv[1] + "!"
- sys.exit(1)
-flashseqs = readMHXFile(fp)
-fp.close()
-
-print "The following flash sequences have been read in:"
-for seq in flashseqs:
- print hex(seq.address) + ":", seq.data
-
-print "Initializing serial port..."
-tty = SerialPort(DEVICE, 100, INIT_BAUDRATE)
-
-print "Please press RESET on your 1337 board..."
-
-while 1:
- tty.write('V')
- tty.flush()
- try:
- if tty.read() == 'F':
- break
- except SerialPortException:
- # timeout happened, who cares ;-)
- pass
-
-print "OK, trying to set baudrate..."
-# set baudrate
-cmdBAUDRATE(REAL_BAUDRATE)
-tty = SerialPort(DEVICE, 100, REAL_BAUDRATE)
-
-# let the fun begin!
-for seq in flashseqs:
- if(seq.address <= 0x40000):
- addr = seq.address
- else:
- continue
- print "RAMing", len(seq.data), "bytes at address", hex(addr)
- cmdWRITE(addr, len(seq.data), seq.data)
- tty.flush()
-
-cmdCALL(0x30000);
-sys.exit(0)
-
-
-# some tests here.......
-"""
-# execute (existing) program in ram
-cmdCALL(0x00033ffc)
-sys.exit(0)
-"""
-
-# read something from the IRAM
-#print cmdREAD(0x00030000, 32)
-
-#data = []
-#for i in range(0, 32):
-# data.append(i)
-#cmdWRITE(0x00030000, 32, data)
-
-
-"""
-# write something to the begin of the IRAM
-data_wr = []
-checksum = 0
-for i in range(0, 0x400):
- value = i%256
- data_wr.append(value)
- checksum = (checksum + value) % (2**16)
-
-print "Calculated checksum:", checksum
-print "Writing", data_wr, "to the IRAM..."
-cmdWRITE(0x00030000, len(data_wr), data_wr)
-print "Received Checksum:", last_checksum
-print
-
-print "Reading from the IRAM again..."
-data_re = cmdREAD(0x00030000, len(data_wr))
-print "Received data:", data_re, "Checksum:", last_checksum
-"""
-
-"""
-# see whats in the iram at the moment
-data_wr = []
-print "Reading from the IRAM..."
-data_re = cmdREAD(0x00030000, 0x10000-4)
-print "Received data:", data_re, "Checksum:", last_checksum
-"""
-
-"""
-# see whats in the dram at the moment
-data_wr = []
-print "Reading from the DRAM..."
-data_re = cmdREAD(0x0002C000, 0x10000-0xC000-4)
-print "Received data:", data_re, "Checksum:", last_checksum
-"""
-
-"""
-# blank the iram
-data_wr = []
-for i in range(0, 0x10000-4):
- value = 0
- data_wr.append(value)
-
-print "Writing", data_wr, "to the IRAM..."
-cmdWRITE(0x00030000, len(data_wr), data_wr)
-print "Received Checksum:", last_checksum
-print
-"""
-
-"""
-# blank the dram
-data_wr = []
-for i in range(0, 0x10000-0xC000-4):
- value = 0
- data_wr.append(value)
-
-print "Writing", data_wr, "to the DRAM..."
-cmdWRITE(0x0002C000, len(data_wr), data_wr)
-print "Received Checksum:", last_checksum
-print
-"""
-
-"""
-# write some data in the iram and try to execute it
-data_wr =[
- 0x9B,0x00,
- 0x0D,0x4e,
- 0xcf,0xf1,
- 0x16,0x01,
- 0x9b,0x05,
- 0x04,0xc7,
- 0xc1,0x06,
- 0x16,0x56,
- 0xe0,0xfb, #branch
- 0x9f,0xa0,0x9f,0xa0,0x9f,0xa0, #nop
- 0x9f,0xa0,0x9f,0xa0,0x9f,0xa0,
- 0x9f,0xa0,0x9f,0xa0,0x9f,0xa0,
- 0x9f,0xa0,0x9f,0xa0,0x9f,0xa0,
- 0x9f,0xa0,0x9f,0xa0,0x9f,0xa0,
- 0x9f,0xa0,0x9f,0xa0,0x9f,0xa0,
- 0x9f,0xa0,0x9f,0xa0,0x9f,0xa0]
-print "Writing", data_wr, "to the IRAM..."
-cmdWRITE(0x00030000, len(data_wr), data_wr)
-print "Received Checksum:", last_checksum
-print
-cmdCALL(0x00030000)
-"""
+def main(argv=None):
+ # check command line arguments
+ if argv is None:
+ argv = sys.argv
+ if len(argv) != 2 and len(argv) != 4:
+ print "Usage: " + argv[0] + " <target mhx-file> [-d DEVICE]"
+ return 1
+
+ # standard serial device to communicate with
+ DEVICE="/dev/ttyUSB0"
+ if len(argv) == 4:
+ DEVICE = argv[3]
+
+ # read in data from mhx-files before starting
+ try:
+ try:
+ bootloaderseqs = readMHXFile("pkernel/pkernel.mhx")
+ except IOError as error1:
+ bootloaderseqs = readMHXFile("%PREFIX%/share/frprog/pkernel.mhx")
+ pkernelseqs = readMHXFile(argv[1])
+ except IOError as error:
+ print argv[0] + ": Error - couldn't open file " + error.filename + "!"
+ return 1
+
+ print "Initializing serial port..."
+ global tty
+ tty = SerialPort(DEVICE, 100, INIT_BAUDRATE)
+
+ print "Please press RESET on your board..."
+
+ while True:
+ tty.write('V')
+ tty.flush()
+ try:
+ if tty.read() == 'F':
+ break
+ except SerialPortException:
+ # timeout happened, who cares ;-)
+ pass
+
+ starttime = time.time() # save time at this point for evaluating the duration at the end
+
+ print "OK, trying to set baudrate..."
+ # set baudrate
+ bootromBAUDRATE(BOOTLOADER_BAUDRATE)
+ time.sleep(0.1) # just to get sure that the bootloader is really running in new baudrate mode!
+ del tty
+ tty = SerialPort(DEVICE, 100, BOOTLOADER_BAUDRATE)
+
+ print "Transfering pkernel program to IRAM",
+ # let the fun begin!
+ for seq in bootloaderseqs:
+ if(seq.address <= 0x40000):
+ addr = seq.address
+ else:
+ continue
+ #print "RAMing", len(seq.data), "bytes at address", hex(addr)
+ bootromWRITE(addr, len(seq.data), seq.data)
+ tty.flush()
+ sys.stdout.write(".")
+ sys.stdout.flush()
+ print
+
+ # execute our pkernel finally and set pkernel conform baudrate
+ bootromCALL(0x30000)
+ time.sleep(0.1) # just to get sure that the pkernel is really running!
+ del tty
+ tty = SerialPort(DEVICE, None, KERNEL_BAUDRATE)
+
+ print "Performing ChipErase..."
+ pkernCHIPERASE()
+
+ print "Flashing",
+ for seq in pkernelseqs:
+ # skip seqs only consisting of 0xffs
+ seqset = list(set(seq.data))
+ if len(seqset) == 1 and seqset[0] == 0xff:
+ continue
+ #print "Flashing", len(seq.data), "bytes at address", hex(seq.address)
+ pkernWRITE(seq.address, len(seq.data), seq.data)
+ tty.flush()
+ sys.stdout.write(".")
+ sys.stdout.flush()
+ print
+
+ duration = time.time() - starttime
+ print "Procedure complete, took", round(duration, 2), "seconds."
+
+ sendByte(0x97) # exit and restart
+ print "Program was started. Have fun!"
+
+
+if __name__ == '__main__':
+ sys.exit(main())