1#!/usr/bin/env python
2
3# Copyright (c) 2013-2014 ARM Limited
4# All rights reserved
5#
6# The license below extends only to copyright in the software and shall
7# not be construed as granting a license to any other intellectual
8# property including but not limited to intellectual property relating

--- 21 unchanged lines hidden (view full) ---

30# OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
31# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
32# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
33# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
34# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
35# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
36# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
37#
38# Authors: Andreas Hansson
39
40# This script is used to migrate ASCII packet traces to the protobuf
41# format currently used in gem5. It assumes that protoc has been
42# executed and already generated the Python package for the packet
43# messages. This can be done manually using:
44# protoc --python_out=. --proto_path=src/proto src/proto/packet.proto
45#
46# The ASCII trace format uses one line per request on the format cmd,
47# addr, size, tick. For example:
48# r,128,64,4000
49# w,232123,64,500000
50# This trace reads 64 bytes from decimal address 128 at tick 4000,
51# then writes 64 bytes to address 232123 at tick 500000.
52#
53# This script can of course also be used as a template to convert
54# other trace formats into the gem5 protobuf format
55
56import protolib
57import sys
58
59# Import the packet proto definitions. If they are not found, attempt
60# to generate them automatically. This assumes that the script is
61# executed from the gem5 root.
62try:
63 import packet_pb2
64except:

--- 10 unchanged lines hidden (view full) ---

75 print "Please install the Python protobuf module"
76 exit(-1)
77
78 import packet_pb2
79 else:
80 print "Failed to import packet proto definitions"
81 exit(-1)
82
83def main():
84 if len(sys.argv) != 3:
85 print "Usage: ", sys.argv[0], " <ASCII input> <protobuf output>"
86 exit(-1)
87
88 try:
89 ascii_in = open(sys.argv[1], 'r')
90 except IOError:

--- 10 unchanged lines hidden (view full) ---

101 # is done in src/proto/protoio.cc
102 proto_out.write("gem5")
103
104 # Add the packet header
105 header = packet_pb2.PacketHeader()
106 header.obj_id = "Converted ASCII trace " + sys.argv[1]
107 # Assume the default tick rate
108 header.tick_freq = 1000000000
109 protolib.encodeMessage(proto_out, header)
110
111 # For each line in the ASCII trace, create a packet message and
112 # write it to the encoded output
113 for line in ascii_in:
114 cmd, addr, size, tick = line.split(',')
115 packet = packet_pb2.Packet()
116 packet.tick = long(tick)
117 # ReadReq is 1 and WriteReq is 4 in src/mem/packet.hh Command enum
118 packet.cmd = 1 if cmd == 'r' else 4
119 packet.addr = long(addr)
120 packet.size = int(size)
121 protolib.encodeMessage(proto_out, packet)
122
123 # We're done
124 ascii_in.close()
125 proto_out.close()
126
127if __name__ == "__main__":
128 main()