#!/usr/bin/env python3
# -*- coding: cp1251 -*-
import jlinkarm as jl
import os, sys, time, struct

dllfilename = 'D:/MCU/SEGGER/JLink_V612i/JLinkARM.dll'

if __name__ == '__main__':
	if len(sys.argv) >= 2:
		if sys.argv[1] == '-h':
			print 'Usage: ldram.py ram_all.bin'
			exit(0)
	imgfilename = 'build/bin/ram_all.bin'

	if len(sys.argv) > 1: 
		if sys.argv[1]:
			imgfilename = sys.argv[1]
	try:
		ff = open(imgfilename, "rb")
	except:
		print "Error file open " + imgfilename
		exit(1)
	jl.loadJLinkARMdll(dllfilename)
	jl.open()
	if jl.is_connected == 0:
		raise RuntimeError('Jlink not connected')
	jl.exec_command('device Cortex-M3', 0, 0)
	jl.exec_command('endian little', 0, 0)
	jl.tif_select(1)
	jl.set_speed(1000)
	jl.clear_RESET()
	jl.clear_TRST()
	time.sleep(0.01)
	jl.set_RESET()
	jl.set_TRST()
	jl.reset()
	jl.halt()
	jl.set_speed(3500)
	bin = ff.read()
	get_dword = lambda address: struct.unpack('<L', bin[address:address + 4])[0]
	i1_address = 0x0
	i1_length = get_dword(0x10)
	jl.write_mem(0x10000bc8, bin[i1_address + 0x20: i1_address + 0x20 + i1_length])
	i2_address = (get_dword(0x18) & 0xffff) * 1024
	i2_length = get_dword(i2_address) + 0x10
	jl.write_mem(0x10006000, bin[i2_address + 0x10: i2_address + 0x10 + i2_length])
	jl.reset()
	jl.write_u32(0x40000210, 0x20111113)
	jl.go()
	jl.close()

	ff.close()
	exit(0)