#!/usr/bin/env python

###########################################################################

#

# Copyright 2019 Samsung Electronics All Rights Reserved.

#

# Licensed under the Apache License, Version 2.0 (the "License");

# you may not use this file except in compliance with the License.

# You may obtain a copy of the License at

#

# http://www.apache.org/licenses/LICENSE-2.0

#

# Unless required by applicable law or agreed to in writing,

# software distributed under the License is distributed on an

# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND,

# either express or implied. See the License for the specific

# language governing permissions and limitations under the License.

#

###########################################################################

from __future__ import print_function

from optparse import OptionParser

import sys

import re

if sys.version_info[0] < 3:

from sets import Set

parser = OptionParser()

parser.add_option("-f", "--file", dest="infilename",

help="map FILE needed to be parsed", metavar="INPUT_FILE")

parser.add_option("-o", "--output", dest="output",

help="Output written to this file. Default is stdout.", metavar="OUTPUT_FILE")

parser.add_option("-a", "--all", action="store_true", dest="all",

help="Print all information.", default=False)

parser.add_option("-s", "--size", action="store_true", dest="totsize",

help="Print total size of data segments.", default=False)

parser.add_option("-l", "--libsize", action="store_true",

dest="libsize", help="Print Library Sizes.", default=False)

parser.add_option("-d", "--details", action="store_true", dest="details",

help="Print details of Object Sizes.", default=False)

#Example: memstats_xtensa.py <option {-a / -s / -l / -d}> -f tinyara.map -o memory.txt.

(options, args) = parser.parse_args()

if not options.infilename:

parser.print_help()

sys.exit(1)

if (not options.all) and (not options.totsize) and (not options.libsize) and (not options.details):

parser.print_help()

sys.exit(1)

infile = open(options.infilename, 'r')

if options.output:

sys.stdout = open(options.output, 'w')

first = 0

currentSymbol = ""

level1 = {}

action = 0

subSymbol = 0

libObject = ""

librarySize = {}

objectSize = {}

allLibraries = {}

allmap = {}

Required = {

}

Symnamemap = {

}

key_list = list(Required.keys())

Symnamemap[".iram0.vectors"] = key_list[2] # text seg

Symnamemap["flash.rodata"] = key_list[2] # text seg

Symnamemap[".iram0.text"] = key_list[2] # text seg

Symnamemap[".flash.text"] = key_list[2] # text seg

Symnamemap[".dram0.data"] = key_list[0] # data seg

Symnamemap[".dram0.bss"] = key_list[1] # bss seg

def transform(k):

return k

def isValidSubSymbol(field1, dictKeys):

return None

def findParent(c):

return None

def getListfromString(s):

return res

def PLO(size, libObj, currSym, subSym):

allmap[library] = {dotO: {currSymParent: size}}

def printLibrarySizes():

print("\t" + str(librarySize[l]) + "\t" + l)

def sortByTotal(n):

return int(n[len(n)-2])

def sortPrintLibrarySizes():

print("\t" + r[len(r) - 1].strip())

def sortPrintall():

results = []

def printall():

print("\t" + str(objectSize[o]) + "\t" + o)

def printTotalSize():

print("\n")

for line in infile:

if line not in ['\n', '\r\n']:

# Ignore section before "Linker script and memory map" and after "/DISCARD/"

if line.strip() == "Linker script and memory map":

action = 1

continue

if line.strip() == "/DISCARD/":

action = 0

# Ignore section before "Linker script and memory map" and after "/DISCARD/"

if action == 0:

continue

# Ignore the first line of Linker script and Memory map

if re.search("__ebase = ORIGIN", line) != None:

continue

lsplit = line.split()

if (line[0] == '.'):

currentSymbol = lsplit[0]

subSymbol = 0

if currentSymbol == ".comment":

action = 0

level1[currentSymbol] = {}

continue

else:

if re.search('\*\(.*\)', line.strip()) != None:

level2string = getListfromString(line.strip())

for l2strs in level2string:

if l2strs in list(level1[currentSymbol].keys()):

continue

else:

level1[currentSymbol][l2strs] = 0

continue

else:

if currentSymbol == '':

continue

matchedkey = isValidSubSymbol(

lsplit[0], list(level1[currentSymbol].keys()))

if matchedkey != None:

subSymbol = matchedkey

if len(lsplit) > 2:

if int(lsplit[1], 16) != 0:

level1[currentSymbol][subSymbol] += int(

lsplit[2], 16)

libObject = lsplit[len(lsplit)-1]

PLO(int(lsplit[2], 16), libObject,

currentSymbol, subSymbol)

else:

line = next(infile)

line = line.strip()

lsplit = line.split()

if int(lsplit[0], 16) != 0:

level1[currentSymbol][subSymbol] += int(

lsplit[1], 16)

libObject = lsplit[len(lsplit)-1]

PLO(int(lsplit[1], 16), libObject,

currentSymbol, subSymbol)

elif re.search('\*fill\*', lsplit[0]) != None:

if int(lsplit[1], 16) != 0 and subSymbol != 0:

level1[currentSymbol][subSymbol] += int(lsplit[2], 16)

PLO(int(lsplit[2], 16), libObject,

currentSymbol, subSymbol)

infile.close()

if options.all:

options.totsize = options.libsize = options.details = True

if options.totsize:

print("######################################")

print("## Total sizes ##")

print("######################################")

printTotalSize()

print("######################################")

print("## Sizes Calculated Using: ##")

print("######################################")

for r in list(Required.keys()):

print(r+":\n\t", end=' ')

t = 1

for s in Required[r]:

if t % 4 == 0:

print(s + "\t\n\t", end=' ')

else:

print(s+"\t", end=' ')

t += 1

print("")

if options.libsize:

print("######################################")

print("## Library Sizes ##")

print("######################################")

#printLibrarySizes()

sortPrintLibrarySizes()

if options.details:

print("######################################")

print("## Details ##")

print("######################################")

#printall()

sortPrintall()

print("######################################\n")