#!/usr/bin/python3

# Copyright 2017 The Emscripten Authors. All rights reserved.

# Emscripten is available under two separate licenses, the MIT license and the

# University of Illinois/NCSA Open Source License. Both these licenses can be

# found in the LICENSE file.

from __future__ import print_function

import sys, re

def print_usage():

print("Usage: ll-strip.py file from to", file=sys.stderr)

print("\tStrip function bodies in all ll file in the [from, to] range", file=sys.stderr)

sys.exit(1)

try:

range_from = int(sys.argv[2])

range_to = int(sys.argv[3])

if range_from >= range_to:

raise Exception("error")

file = open(sys.argv[1])

except:

print_usage()

func_start = re.compile(r"^define\s")

func_end = re.compile("^}$")

function_counter = 0

in_function = False

line_number = 0

skip = False

dummy_stmt = "unreachable"

for orig_line in file:

line = orig_line.strip()

if func_start.match(line):

if in_function:

print("Discovered a function inside another function!", file=sys.stderr)

sys.exit(1)

in_function = True

line_number = 0

skip = False

function_counter = function_counter + 1

elif func_end.match(line):

if not in_function:

print("Discovered a function end without a matching beginning!", file=sys.stderr)

sys.exit(1)

in_function = False

line_number = 0

skip = False

elif in_function:

line_number = line_number + 1

if not skip and line_number == 1:

if line == dummy_stmt:

function_counter = function_counter - 1

if range_from <= function_counter <= range_to and line_number == 1:

#import pdb;pdb.set_trace()

if line != dummy_stmt:

skip = True

print(dummy_stmt)

if not skip:

print(orig_line.rstrip("\n"))