From cc8911261e11d67f2b8745289ed5ecbb89c9d917 Mon Sep 17 00:00:00 2001 From: Andreas Hansson Date: Thu, 30 May 2013 12:53:53 -0400 Subject: util: Auto generate the packet proto definitions This patch simplifies the usage of the packet trace encoder/decoder by attempting to automatically generating the packet proto definitions in case they cannot be found. --- util/encode_packet_trace.py | 18 +++++++++++++++++- 1 file changed, 17 insertions(+), 1 deletion(-) (limited to 'util/encode_packet_trace.py') diff --git a/util/encode_packet_trace.py b/util/encode_packet_trace.py index cbfc7d921..fc6ddf7fd 100755 --- a/util/encode_packet_trace.py +++ b/util/encode_packet_trace.py @@ -85,7 +85,23 @@ import struct import sys -import packet_pb2 + +# Import the packet proto definitions. If they are not found, attempt +# to generate them automatically. This assumes that the script is +# executed from the gem5 root. +try: + import packet_pb2 +except: + print "Did not find packet proto definitions, attempting to generate" + from subprocess import call + error = call(['protoc', '--python_out=util', '--proto_path=src/proto', + 'src/proto/packet.proto']) + if not error: + import packet_pb2 + print "Generated packet proto definitions" + else: + print "Failed to import packet proto definitions" + exit(-1) def EncodeVarint(out_file, value): """ -- cgit v1.2.3