uhd/fpga/usrp2/top/extract_usage.py
Lars Amsel 699cff5941 uhd: Surround all open() calls with with-statement
In addition to commit ab8ec99733
this adds `with` statements to remaining files in the UHD code base.
2026-02-20 12:20:10 +01:00

64 lines
1.8 KiB
Python
Executable File

#!/usr/bin/env python
#
# Copyright 2012 Ettus Research LLC
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
#
import os
import sys
ALL_MAP_FILES = """\
./N2x0/build-N210R4/u2plus_map.map N210
./N2x0/build-N200R4/u2plus_map.map N200
./USRP2/build/u2_rev3_map.map USRP2
./B100/build-B100/B100_map.map B100
"""
def extract_map_from_file(path):
output = ''
found = False
with open(path) as f:
for line in f:
if line.strip() == 'Mapping completed.': found = False
if line.strip() == 'Logic Utilization:': found = True
if found: output += line
return output
def extract_maps():
output = ''
for line in ALL_MAP_FILES.splitlines():
path, name = line.split()
if not os.path.exists(path):
print('DNE {} skipping...'.format(path))
output += """
########################################################################
## %s Usage Summary
########################################################################
%s"""%(name, extract_map_from_file(path).strip())
return output + '\n\n'
if __name__ == '__main__':
summary = extract_maps()
if len(sys.argv) == 1:
print(summary)
else:
with open(sys.argv[1], 'w') as f:
f.write(summary)