-
-# check command line arguments
-if len(sys.argv) != 3:
- print "Usage: " + sys.argv[0] + " [pkernel mhx-file] [target mhx-file]"
- sys.exit(1)
-
-# read in data from mhx-files before starting
-try:
- bootloaderseqs = readMHXFile(sys.argv[1])
- pkernelseqs = readMHXFile(sys.argv[2])
-except IOError as error:
- print sys.argv[0] + ": Error - couldn't open file " + error.filename + "!"
- sys.exit(1)
-
-print "Initializing serial port..."
-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
-
-print "OK, trying to set baudrate..."
-# set baudrate
-cmdBAUDRATE(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)
- cmdWRITE(addr, len(seq.data), seq.data)
- tty.flush()
- sys.stdout.write(".")
- sys.stdout.flush()
-print
-
-# execute our pkernel finally and set pkernel conform baudrate
-cmdCALL(0x30000)
-time.sleep(0.5) # just to get sure that the pkernel is really running!
-del tty
-pkernelmode = True
-tty = SerialPort(DEVICE, None, KERNEL_BAUDRATE)
-
-print "Performing ChipErase..."
-pkernCHIPERASE()
-print "Chip erasing done."
-
-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
-print "Flashing done."
-
-sendByte(0x97) # exit and restart
-print "Program was started. Have fun!"
+def usage(execf):
+ """
+ print usage of frprog
+ """
+ print "Usage: " + execf + " <target mhx-file> [-d DEVICE]"
+
+def main(argv=None):
+ """
+ main function of frprog
+ """
+ # check command line arguments
+ if argv is None:
+ argv = sys.argv
+
+ if len(argv) == 2 and (argv[1] == "-v" or argv[1] == "--version"):
+ print "Version: %VERSION%"
+ return 0
+
+ if len(argv) != 2 and len(argv) != 4:
+ usage(argv[0])
+ return 1
+
+ # standard serial device to communicate with
+ device = "/dev/ttyUSB0"
+
+ # overrule standard device if provided with -d
+ if len(argv) == 4:
+ if argv[2] == "-d":
+ device = argv[3]
+ else:
+ usage(argv[0])
+ return 1
+
+ # read in data from mhx-files before starting
+ try:
+ try:
+ bootloaderseqs = readmhxfile("pkernel/pkernel.mhx")
+ except IOError as _:
+ 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
+
+ # save time at this point for evaluating the duration at the end
+ starttime = time.time()
+
+ print "OK, trying to set baudrate..."
+ # set baudrate
+ try:
+ bootrombaudrate(BOOTLOADER_BAUDRATE)
+ except SerialPortException:
+ print "timeout exception: try again ->"
+ bootrombaudrate(BOOTLOADER_BAUDRATE)
+ # just to get sure that the bootloader is really running in new baudrate mode!
+ time.sleep(0.1)
+ del tty
+ tty = SerialPort(device, 100, BOOTLOADER_BAUDRATE)
+
+ sdots = SPLIT
+ 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()
+
+ sdots = sdots - 1
+ if sdots == 0:
+ sys.stdout.write(".")
+ sys.stdout.flush()
+ sdots = SPLIT
+ 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()
+
+ sdots = SPLIT
+ 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()
+
+ sdots = sdots - 1
+ if sdots == 0:
+ sys.stdout.write(".")
+ sys.stdout.flush()
+ sdots = SPLIT
+ 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())