# Copyright (c) 2008 Nokia Corporation

#

# 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.

# This script calculates the code size of the dll, pyd's and the python runtime

# sis, which is displayed graphically in project metrics tab of release build.

import sys

import os

import time

# `metrics_data`: map format -> {binary_type: (path, extension)}

# 'binary_type' are the metrics for which code size is calculated.The value

# for these keys comprise of the path along with the extension(if specified),

# to the respective 'binary_type'.

metrics_data = {'Python25.dll':

                    ('\\epoc32\\release\\armv5\\urel\\python25.dll', ''),

                'Other PYDs': ('\\epoc32\\release\\armv5\\urel\\', 'pyd')}

binary_size = {}

def file_size(file_path):

    file_stat = os.stat(file_path)

    return (file_stat.st_size) / 1024.00   # Get size of file in kilobytes

def print_metrics(log_file, release):

    log = open(log_file, 'a+')

    now = time.strftime("%b %d %Y %H:%M:%S")  # Get date, time when writing log

    log_string = "Time=%s,Release=%s," % (now, release)

    for code in binary_size:

        log_string += "%s=%f," % (code, binary_size[code])

    log.write(log_string + "\n")

    log.close()

def updatelog(release, python_runtime, log_file):

# This function is called to log the code size, something like this

# code_size.log(platform, python_runtime, "C:\\logs\\code_size.log")

    global binary_size

    metrics_data['Python25_runtime.sis'] = (python_runtime, '')

    for binary_type in metrics_data:

        binary_path, file_ext = metrics_data[binary_type]

        if "pyd" in file_ext:

            all_files = os.listdir(binary_path)

            total_size = 0

            for afile in all_files:

                if afile.endswith(".pyd"):

                    total_size += file_size(os.path.join(binary_path, afile))

            binary_size[binary_type] = total_size

        elif not file_ext:  # Either sis or dll

            code_size = file_size(binary_path)

            binary_size[binary_type] = code_size

    print_metrics(log_file, release)