1
0
mirror of https://bitbucket.org/librepilot/librepilot.git synced 2024-12-05 13:24:11 +01:00
LibrePilot/flight/libraries/PyMite/tools/bcode.py
2013-04-25 13:13:36 +03:00

107 lines
2.4 KiB
Python

#!/usr/bin/env python
# This file is Copyright 2003, 2006, 2007, 2009, 2010 Dean Hall.
#
# This file is part of the Python-on-a-Chip program.
# Python-on-a-Chip is free software: you can redistribute it and/or modify
# it under the terms of the GNU LESSER GENERAL PUBLIC LICENSE Version 2.1.
#
# Python-on-a-Chip 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.
# A copy of the GNU LESSER GENERAL PUBLIC LICENSE Version 2.1
# is seen in the file COPYING in this directory.
"""
Python Byte Code utility functions.
count(fn) - count the number of times each byte code
appears in a code object.
"""
## @file
# @copybrief bcode
## @package bcode
# @brief Python Byte Code utility functions.
#
# count(fn) - count the number of times each byte code
# appears in a code object.
import dis, types
#The highest numbered byte code
MAX_BCODE = 150
def count(fn):
"""
Compile the python source file named fn.
Count all the bytecodes in the file.
Return a list of counts indexed by bcode.
"""
#create a code object
src = open(fn).read()
co = compile(src, fn, "exec")
#init list
cnt = [0,] * MAX_BCODE
#iterate through the nested code objects
_rcount(co, cnt)
return cnt
def _rcount(co, count):
"""
Recursively descend through all the code objects
in the given code object, co.
Add byte code counts to the count dict.
"""
#current byte code string
str = co.co_code
strlen = len(str)
#count byte codes in current string
i = 0
while i < strlen:
bcode = ord(str[i])
i += 1
#incr count
count[bcode] += 1
#if bcode has arg, incr ptr by 2 bytes
if bcode >= 90:
i += 2
#count code objects in constant pool
for obj in co.co_consts:
if type(obj) == types.CodeType:
_rcount(obj, count)
return
def main(fn):
"""
Count the bytecodes in the file, fn,
and print them out in human-readable form.
"""
#count the bytecodes in this file
c = count(fn)
#print bcode name and count
for i in range(len(c)):
if c[i]:
print dis.opname[i], ":\t", c[i]
return
if __name__ == "__main__":
main("c:\\dwh\\tech\\cis\\py\\snap.py")