Arduino_Tools/src/genpinmap/genpinmap_arduino.py
Frederic.Pillon 3078740849 Fix missing ',' for CAN
Signed-off-by: Frederic.Pillon <frederic.pillon@st.com>
2017-08-21 10:56:13 +02:00

667 lines
25 KiB
Python

import sys
import re
import os
from xml.dom import minidom
from xml.dom.minidom import parse, Node
io_list = [] #'PIN','name'
adclist = [] #'PIN','name','ADCSignal'
daclist = [] #'PIN','name','DACSignal'
i2cscl_list = [] #'PIN','name','I2CSCLSignal'
i2csda_list = [] #'PIN','name','I2CSDASignal'
pwm_list = [] #'PIN','name','PWM'
uarttx_list = [] #'PIN','name','UARTtx'
uartrx_list = [] #'PIN','name','UARTrx'
uartcts_list = [] #'PIN','name','UARTcts'
uartrts_list = [] #'PIN','name','UARTrts'
spimosi_list = [] #'PIN','name','SPIMOSI'
spimiso_list = [] #'PIN','name','SPIMISO'
spissel_list = [] #'PIN','name','SPISSEL'
spisclk_list = [] #'PIN','name','SPISCLK'
cantd_list = [] #'PIN','name','CANTD'
canrd_list = [] #'PIN','name','CANRD'
eth_list = [] #'PIN','name','ETH'
qspi_list = [] #'PIN','name','QUADSPI'
def find_gpio_file(xmldoc):
res = 'ERROR'
itemlist = xmldoc.getElementsByTagName('IP')
for s in itemlist:
a = s.attributes['Name'].value
if "GPIO" in a:
res = s.attributes['Version'].value
return res
def get_gpio_af_num(xml, pintofind, iptofind):
if 'STM32F10' in sys.argv[2]:
return get_gpio_af_numF1(xml, pintofind, iptofind)
# xml = parse('GPIO-STM32L051_gpio_v1_0_Modes.xml')
#xml = parse(gpiofile)
#DBG print ('pin to find ' + pintofind)
i=0
mygpioaf = 'NOTFOUND'
for n in xml.documentElement.childNodes:
i += 1
j = 0
if n.nodeType == Node.ELEMENT_NODE:
for firstlevel in n.attributes.items():
# if 'PB7' in firstlevel:
if pintofind == firstlevel[1]:
#DBG print (i , firstlevel)
#n = pin node found
for m in n.childNodes:
j += 1
k = 0
if m.nodeType == Node.ELEMENT_NODE:
for secondlevel in m.attributes.items():
k += 1
# if 'I2C1_SDA' in secondlevel:
if iptofind in secondlevel:
#DBG print (i, j, m.attributes.items())
# m = IP node found
for p in m.childNodes:
if p.nodeType == Node.ELEMENT_NODE:
#p node of 'Specific parameter'
#DBG print (i,j,k,p.attributes.items())
for myc in p.childNodes:
#DBG print (myc)
if myc.nodeType == Node.ELEMENT_NODE:
#myc = node of ALTERNATE
for mygpioaflist in myc.childNodes:
mygpioaf += ' ' + mygpioaflist.data
#print (mygpioaf)
if mygpioaf == 'NOTFOUND':
print ('GPIO AF not found in ' + gpiofile + ' for ' + pintofind + ' and the IP ' + iptofind)
#quit()
return mygpioaf.replace('NOTFOUND ', '')
def get_gpio_af_numF1(xml, pintofind, iptofind):
#print ('pin to find ' + pintofind + ' ip to find ' + iptofind)
i=0
mygpioaf = 'NOTFOUND'
for n in xml.documentElement.childNodes:
i += 1
j = 0
if n.nodeType == Node.ELEMENT_NODE:
for firstlevel in n.attributes.items():
#print ('firstlevel ' , firstlevel)
# if 'PB7' in firstlevel:
if pintofind == firstlevel[1]:
#print ('firstlevel ' , i , firstlevel)
#n = pin node found
for m in n.childNodes:
j += 1
k = 0
if m.nodeType == Node.ELEMENT_NODE:
for secondlevel in m.attributes.items():
#print ('secondlevel ' , i, j, k , secondlevel)
k += 1
# if 'I2C1_SDA' in secondlevel:
if iptofind in secondlevel:
# m = IP node found
#print (i, j, m.attributes.items())
if m.hasChildNodes() == False:
mygpioaf = 'AFIO_NONE'
else:
for p in m.childNodes:
#p node 'RemapBlock'
for s in p.childNodes:
if s.nodeType == Node.ELEMENT_NODE:
#s node 'Specific parameter'
#DBG print (i,j,k,p.attributes.items())
for myc in s.childNodes:
#DBG print (myc)
if myc.nodeType == Node.ELEMENT_NODE:
#myc = AF value
for mygpioaflist in myc.childNodes:
mygpioaf += ' ' + mygpioaflist.data.replace("__HAL_", "").replace("_REMAP", "")
#print mygpioaf
if mygpioaf == 'NOTFOUND':
print ('GPIO AF not found in ' + gpiofile + ' for ' + pintofind + ' and the IP ' + iptofind + ' set as AFIO_NONE')
mygpioaf = 'AFIO_NONE'
return mygpioaf.replace('NOTFOUND ', '')
def store_pin (pin, name):
#store pin I/O
p = [pin, name]
io_list.append(p)
#function to store ADC list
def store_adc (pin, name, signal):
adclist.append([pin,name,signal])
#function to store DAC list
def store_dac (pin, name, signal):
daclist.append([pin,name,signal])
#function to store I2C list
def store_i2c (pin, name, signal):
#is it SDA or SCL ?
if "_SCL" in signal:
i2cscl_list.append([pin,name,signal])
if "_SDA" in signal:
i2csda_list.append([pin,name,signal])
#function to store timers
def store_pwm(pin, name, signal):
if "_CH" in signal:
pwm_list.append([pin,name,signal])
#function to store Uart pins
def store_uart(pin, name, signal):
if "_TX" in signal:
uarttx_list.append([pin,name,signal])
if "_RX" in signal:
uartrx_list.append([pin,name,signal])
if "_CTS" in signal:
uartcts_list.append([pin,name,signal])
if "_RTS" in signal:
uartrts_list.append([pin,name,signal])
#function to store SPI pins
def store_spi(pin, name, signal):
if "_MISO" in signal:
spimiso_list.append([pin,name,signal])
if "_MOSI" in signal:
spimosi_list.append([pin,name,signal])
if "_SCK" in signal:
spisclk_list.append([pin,name,signal])
if "_NSS" in signal:
spissel_list.append([pin,name,signal])
#function to store CAN pins
def store_can(pin, name, signal):
if "_RX" in signal:
canrd_list.append([pin,name,signal])
if "_TX" in signal:
cantd_list.append([pin,name,signal])
#function to store ETH list
def store_eth (pin, name, signal):
eth_list.append([pin,name,signal])
#function to store QSPI pins
def store_qspi (pin, name, signal):
qspi_list.append([pin,name,signal])
def print_header():
s = ("""/*
*******************************************************************************
* Copyright (c) 2016, STMicroelectronics
* All rights reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions are met:
*
* 1. Redistributions of source code must retain the above copyright notice,
* this list of conditions and the following disclaimer.
* 2. Redistributions in binary form must reproduce the above copyright notice,
* this list of conditions and the following disclaimer in the documentation
* and/or other materials provided with the distribution.
* 3. Neither the name of STMicroelectronics nor the names of its contributors
* may be used to endorse or promote products derived from this software
* without specific prior written permission.
*
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
* AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
* DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE
* FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
* DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
* SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
* CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
* OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
* OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
*******************************************************************************
*/
#include "Arduino.h"
#include "%s.h"
// =====
// Note: Commented lines are alternative possibilities which are not used per default.
// If you change them, you will have to know what you do
// =====
""" % re.sub('\.c$', '', out_filename))
out_file.write( s)
def print_all_lists():
if print_list_header("ADC", "ADC", adclist, "ADC"):
print_adc()
if print_list_header("DAC", "DAC", daclist, "DAC"):
print_dac()
if print_list_header("I2C", "I2C_SDA", i2csda_list, "I2C"):
print_i2c(xml, i2csda_list)
if print_list_header("", "I2C_SCL", i2cscl_list, "I2C"):
print_i2c(xml, i2cscl_list)
if print_list_header("PWM", "PWM", pwm_list, "TIM"):
print_pwm(xml)
if print_list_header("SERIAL", "UART_TX", uarttx_list, "UART"):
print_uart(xml, uarttx_list)
if print_list_header("", "UART_RX", uartrx_list, "UART"):
print_uart(xml, uartrx_list)
if print_list_header("", "UART_RTS", uartrts_list, "UART"):
print_uart(xml, uartrts_list)
if print_list_header("", "UART_CTS", uartcts_list, "UART"):
print_uart(xml, uartcts_list)
if print_list_header("SPI", "SPI_MOSI", spimosi_list, "SPI"):
print_spi(xml, spimosi_list)
if print_list_header("", "SPI_MISO", spimiso_list, "SPI"):
print_spi(xml, spimiso_list)
if print_list_header("", "SPI_SCLK", spisclk_list, "SPI"):
print_spi(xml, spisclk_list)
if print_list_header("", "SPI_SSEL", spissel_list, "SPI"):
print_spi(xml, spissel_list)
if print_list_header("CAN", "CAN_RD", canrd_list, "CAN"):
print_can(xml, canrd_list)
if print_list_header("", "CAN_TD", cantd_list, "CAN"):
print_can(xml, cantd_list)
if print_list_header("ETHERNET", "Ethernet", eth_list, "ETH"):
print_eth(xml, eth_list)
if print_list_header("QUADSPI", "QUADSPI", qspi_list, "QSPI"):
print_qspi(xml, qspi_list)
def print_list_header(comment, name, l, switch):
if len(l)>0:
if comment:
s = ("""
//*** %s ***
""") % comment
else:
s = ""
s += ("""
#ifdef HAL_%s_MODULE_ENABLED
const PinMap PinMap_%s[] = {
""") % (switch, name)
else:
if comment:
s = ("""
//*** %s ***
""") % comment
else:
s = ""
s+=("""
//*** No %s ***
""") % name
out_file.write(s)
return len(l)
def print_adc():
i = 0
if len(adclist)>0:
# Check GPIO version (alternate or not)
s_pin_data = 'STM_PIN_DATA_EXT(STM_MODE_ANALOG, GPIO_NOPULL, 0, '
while i < len(adclist):
p=adclist[i]
if "IN" in p[2]:
s1 = "%-12s" % (" {" + p[0] + ',')
a = p[2].split('_')
inst = a[0].replace("ADC", "")
if len(inst) == 0:
inst = '1' #single ADC for this product
s1 += "%-7s" % ('ADC' + inst + ',')
chan = a[1].replace("IN", "")
s1 += s_pin_data + chan
s1 += ', 0)}, // ' + p[2] + '\n'
out_file.write(s1)
i += 1
out_file.write( """ {NC, NP, 0}
};
#endif
""")
def print_dac():
i = 0
if len(daclist)>0:
while i < len(daclist):
p=daclist[i]
b=p[2]
s1 = "%-12s" % (" {" + p[0] + ',')
#2nd element is the DAC signal
if b[3] == '_': # 1 DAC in this chip
s1 += 'DAC1, STM_PIN_DATA_EXT(STM_MODE_ANALOG, GPIO_NOPULL, 0, ' + b[7] + ', 0)}, // ' + b + '\n'
else:
s1 += 'DAC' + b[3] + ', STM_PIN_DATA_EXT(STM_MODE_ANALOG, GPIO_NOPULL, 0, ' + b[8] + ', 0)}, // ' + b + '\n'
out_file.write(s1)
i += 1
out_file.write( """ {NC, NP, 0}
};
#endif
""")
def print_i2c(xml, l):
i = 0
if len(l)>0:
while i < len(l):
p=l[i]
result = get_gpio_af_num(xml, p[1], p[2])
if result != 'NOTFOUND':
s1 = "%-12s" % (" {" + p[0] + ',')
#2nd element is the I2C XXX signal
b = p[2].split('_')[0]
s1 += b[:len(b)-1] + b[len(b)-1] + ', STM_PIN_DATA(STM_MODE_AF_OD, GPIO_NOPULL, '
r = result.split(' ')
for af in r:
s2 = s1 + af + ')},\n'
out_file.write(s2)
i += 1
out_file.write( """ {NC, NP, 0}
};
#endif
""")
def print_pwm(xml):
i=0
if len(pwm_list)>0:
while i < len(pwm_list):
p=pwm_list[i]
result = get_gpio_af_num(xml, p[1], p[2])
if result != 'NOTFOUND':
s1 = "%-12s" % (" {" + p[0] + ',')
#2nd element is the PWM signal
a = p[2].split('_')
inst = a[0]
if len(inst) == 3:
inst += '1'
s1 += "%-8s" % (inst + ',')
chan = a[1].replace("CH", "")
if chan.endswith('N'):
neg = ', 1'
chan = chan.strip('N')
else:
neg = ', 0'
s1 += 'STM_PIN_DATA_EXT(STM_MODE_AF_PP, GPIO_PULLUP, '
r = result.split(' ')
for af in r:
s2 = s1 + af + ', ' + chan + neg + ')}, // ' + p[2] + '\n'
out_file.write(s2)
i += 1
out_file.write( """ {NC, NP, 0}
};
#endif
""")
def print_uart(xml, l):
i=0
if len(l)>0:
while i < len(l):
p=l[i]
result = get_gpio_af_num(xml, p[1], p[2])
if result != 'NOTFOUND':
s1 = "%-12s" % (" {" + p[0] + ',')
#2nd element is the UART_XX signal
b=p[2].split('_')[0]
s1 += "%-9s" % (b[:len(b)-1] + b[len(b)-1:] + ',')
if 'STM32F10' in sys.argv[2] and l == uartrx_list:
s1 += 'STM_PIN_DATA(STM_MODE_INPUT, GPIO_PULLUP, '
else:
s1 += 'STM_PIN_DATA(STM_MODE_AF_PP, GPIO_PULLUP, '
r = result.split(' ')
for af in r:
s2 = s1 + af + ')},\n'
out_file.write(s2)
i += 1
out_file.write( """ {NC, NP, 0}
};
#endif
""")
def print_spi(xml, l):
i=0
if len(l)>0:
while i < len(l):
p=l[i]
result = get_gpio_af_num(xml, p[1], p[2])
if result != 'NOTFOUND':
s1 = "%-12s" % (" {" + p[0] + ',')
#2nd element is the SPI_XXXX signal
instance=p[2].split('_')[0].replace("SPI", "")
s1 += 'SPI' + instance + ', STM_PIN_DATA(STM_MODE_AF_PP, GPIO_PULLUP, '
r = result.split(' ')
for af in r:
s2 = s1 + af + ')},\n'
out_file.write(s2)
i += 1
out_file.write( """ {NC, NP, 0}
};
#endif
""")
def print_can(xml, l):
i=0
if len(l)>0:
while i < len(l):
p=l[i]
b=p[2]
result = get_gpio_af_num(xml, p[1], p[2])
if result != 'NOTFOUND':
s1 = "%-12s" % (" {" + p[0] + ',')
#2nd element is the CAN_XX signal
instance = p[2].split('_')[0].replace("CAN", "")
#if len(instance) == 0:
# instance = '1'
if 'STM32F10' in sys.argv[2] and l == canrd_list:
s1 += 'CAN' + instance + ', STM_PIN_DATA(STM_MODE_INPUT, GPIO_NOPULL, '
else:
s1 += 'CAN' + instance + ', STM_PIN_DATA(STM_MODE_AF_PP, GPIO_NOPULL, '
r = result.split(' ')
for af in r:
s2 = s1 + af + ')},\n'
out_file.write(s2)
i += 1
out_file.write( """ {NC, NP, 0}
};
#endif
""")
def print_eth(xml, l):
i=0
if len(l)>0:
prev_s = ''
while i < len(l):
p=l[i]
result = get_gpio_af_num(xml, p[1], p[2])
if result != 'NOTFOUND':
s1 = "%-12s" % (" {" + p[0] + ',')
#2nd element is the ETH_XXXX signal
s1 += 'ETH, STM_PIN_DATA(STM_MODE_AF_PP, GPIO_PULLUP, ' + result +')},'
#check duplicated lines, only signal differs
if (prev_s == s1):
s1 = '|' + p[2]
else:
if len(prev_s)>0:
out_file.write('\n')
prev_s = s1
s1 += ' // ' + p[2]
out_file.write(s1)
i += 1
out_file.write( """\n {NC, NP, 0}
};
#endif
""")
def print_qspi(xml, l):
i=0
if len(l)>0:
prev_s = ''
while i < len(l):
p=l[i]
result = get_gpio_af_num(xml, p[1], p[2])
if result != 'NOTFOUND':
s1 = "%-12s" % (" {" + p[0] + ',')
#2nd element is the QUADSPI_XXXX signal
s1 += 'QUADSPI, STM_PIN_DATA(STM_MODE_AF_PP, GPIO_PULLUP, ' + result +')},'
#check duplicated lines, only signal differs
if (prev_s == s1):
s1 = '|' + p[2]
else:
if len(prev_s)>0:
out_file.write('\n')
prev_s = s1
s1 += ' // ' + p[2]
out_file.write(s1)
i += 1
out_file.write( """\n {NC, NP, 0}
};
#endif
""")
tokenize = re.compile(r'(\d+)|(\D+)').findall
def natural_sortkey(list_2_elem):
return tuple(int(num) if num else alpha for num, alpha in tokenize(list_2_elem[0]))
def sort_my_lists():
adclist.sort(key=natural_sortkey)
daclist.sort(key=natural_sortkey)
i2cscl_list.sort(key=natural_sortkey)
i2csda_list.sort(key=natural_sortkey)
pwm_list.sort(key=natural_sortkey)
uarttx_list.sort(key=natural_sortkey)
uartrx_list.sort(key=natural_sortkey)
uartcts_list.sort(key=natural_sortkey)
uartrts_list.sort(key=natural_sortkey)
spimosi_list.sort(key=natural_sortkey)
spimiso_list.sort(key=natural_sortkey)
spissel_list.sort(key=natural_sortkey)
spisclk_list.sort(key=natural_sortkey)
cantd_list.sort(key=natural_sortkey)
canrd_list.sort(key=natural_sortkey)
eth_list.sort(key=natural_sortkey)
qspi_list.sort(key=natural_sortkey)
return
# START MAIN PROGRAM
#xmldoc = minidom.parse('STM32L051K(6-8)Tx.xml')
cur_dir = os.getcwd()
out_filename = 'PeripheralPins.c'
if len(sys.argv) < 3:
print("Usage: " + sys.argv[0] + " <BOARD_NAME> <product xml file name>")
print(" - <BOARD_NAME> is the name of the board as it will be named in mbed")
print(" - <product xml file name> is the STM32 file description in Cube MX")
print(" !!This xml file contains non alpha characters in its name, you should call it with quotes")
print("")
print(" This script is able to generate the %s for a specific board" % out_filename )
print(" After file generation, review it carefully and ")
print(" please report any issue to github:")
print(" https://github.com/fpistm/stm32_tools/issues")
print("")
print(" Once generated, you should comment a line if the pin is generated ")
print(" several times for the same IP")
print(" or if the pin should not be used (overlaid with some HW on the board, ")
print(" for instance)")
quit()
if sys.platform.startswith('win32'):
#print ("Windows env")
cubemxdir = 'C:\Program Files\STMicroelectronics\STM32Cube\STM32CubeMX\db\mcu'
cubemxdirIP = cubemxdir+"\\IP\\"
input_file_name = cubemxdir+"\\" + sys.argv[2]
out_path = cur_dir+'\\Arduino\\_'+sys.argv[1]
output_filename = out_path+"\\"+out_filename
else:
#print ("Linux env")
if sys.platform.startswith('linux'):
cubemxdir = os.getenv("HOME")+'/STM32CubeMX/db/mcu'
cubemxdirIP = cubemxdir+"/IP/"
input_file_name = cubemxdir+'/'+ sys.argv[2]
out_path = cur_dir+'/Arduino/'+sys.argv[1]
output_filename = out_path+'/'+out_filename
else:
#print ("Darwin env")
if sys.platform.startswith('darwin'):
print("Platform is Mac OSX")
cubemxdir = '/Applications/STMicroelectronics/STM32CubeMX.app/Contents/Resources/db/mcu'
cubemxdirIP = cubemxdir+"/IP/"
input_file_name = cubemxdir+'/'+ sys.argv[2]
out_path = cur_dir+'/Arduino/'+sys.argv[1]
output_filename = out_path+'/'+out_filename
else:
print ("Unsupported OS")
quit()
#open input file
#check input file exists
if not(os.path.isdir(cubemxdir)):
print ("\n ! ! ! Cube Mx seems not to be installed or not at the requested location")
print ("\n ! ! ! please check the value you set for cubemxdir variable at the top of " + sys.argv[0] + " file")
quit()
if not(os.path.isfile(input_file_name)):
print ('\n ! ! ! '+sys.argv[2] + ' file not found')
print ("\n ! ! ! Check in " + cubemxdir + " the correct name of this file")
print ("\n ! ! ! You may use double quotes for this file if it contains special characters")
quit()
print (" * * * Opening input file...")
if not(os.path.isdir(out_path)):
os.makedirs(out_path)
xmldoc = minidom.parse(input_file_name)
itemlist = xmldoc.getElementsByTagName('Pin')
#open output file
if (os.path.isfile(output_filename)):
print (" * * * * Requested %s file already exists and will be overwritten" % out_filename)
os.remove(output_filename)
out_file = open(output_filename, 'w')
gpiofile = find_gpio_file(xmldoc)
if gpiofile == 'ERROR':
quit()
xml = parse(cubemxdirIP + 'GPIO-' + gpiofile + '_Modes.xml')
print (" * * * Getting pins and Ips for the xml file...")
pinregex=r'^(P[A-Z][0-9][0-5]?)'
for s in itemlist:
m = re.match(pinregex, s.attributes['Name'].value)
if m:
pin = m.group(0)[:2] + '_' + m.group(0)[2:] # pin formatted P<port>_<number>: PF_O
name = s.attributes['Name'].value.strip() # full name: "PF0 / OSC_IN"
if s.attributes['Type'].value == "I/O":
store_pin(pin, name)
else:
continue
siglist = s.getElementsByTagName('Signal')
for a in siglist:
sig = a.attributes['Name'].value.strip()
if "ADC" in sig:
#store ADC pin
store_adc( pin, name, sig)
if all(["DAC" in sig, "_OUT" in sig]):
#store DAC
store_dac( pin, name, sig)
if "I2C" in sig:
#store DAC
store_i2c( pin, name, sig)
if re.match('^TIM', sig) is not None: #ignore HRTIM
#store PWM
store_pwm( pin, name, sig)
if re.match('^(LPU|US|U)ART', sig) is not None:
store_uart( pin, name, sig)
if "SPI" in sig:
store_spi( pin, name, sig)
if "CAN" in sig:
store_can( pin, name, sig)
if "ETH" in sig:
store_eth( pin, name, sig)
if "QUADSPI" in sig:
store_qspi( pin, name, sig)
print (" * * * Sorting lists...")
sort_my_lists()
print (" * * * Printing lists...")
print_header()
print_all_lists()
nb_pin = (len(io_list))
print ("nb of I/O pins: %i" % nb_pin)
print ('\n * * * ' + sys.argv[1]+' OK')