3 # Expects either "prepare" or "clean" argument
10 print('Please specify "prepare <SIZE>" or "clean <SIZE>" as first argument.')
13 if (len(sys
.argv
) <= 1):
16 if (len(sys
.argv
) <= 2):
18 array_len
= int(sys
.argv
[2])
20 tracedir_name
= './trace-' + str(array_len
) + '/'
21 output_metadata
= tracedir_name
+ 'metadata'
22 output_stream
= tracedir_name
+ 'stream'
25 def write_binary(f
, arr
):
28 def generate_metadata(array_len
):
33 'typealias integer {{ size = 8; align = 8; signed = false; base = 10; }} := uint8_t;\n'
34 'typealias integer {{ size = 32; align = 8; signed = false; base = hex; }} := uint32_t;\n'
35 'typealias integer {{ size = 64; align = 8; signed = false; base = hex; }} := uint64_t;\n'
40 ' uuid = "2a6422d0-6cee-11e0-8c08-cb07d7b3a564";\n'
42 ' packet.header := struct {{\n'
44 ' uint8_t uuid[16];\n'
49 ' event.header := struct {{\n'
61 metadata_f
= open(output_metadata
, 'w')
62 metadata_f
.write(metadata_str
)
63 for i
in range(array_len
):
64 metadata_f
.write(metadata_event
.format(i
, i
))
67 def generate_stream(array_len
):
69 stream_packet_header
= [
70 0xC1, 0x1F, 0xFC, 0xC1, # magic
71 0x2A, 0x64, 0x22, 0xD0, 0x6C, 0xEE, 0x11, 0xE0,
72 0x8C, 0x08, 0xCB, 0x07, 0xD7, 0xB3, 0xA5, 0x64, # uuid
74 stream_f
= open(output_stream
, 'wb')
75 write_binary(stream_f
, stream_packet_header
)
76 # generate 8 bytes per event
77 for i
in range(array_len
):
78 write_binary(stream_f
, struct
.pack('<Q', i
))
82 print('Preparing test for ' + str(array_len
) + ' events')
83 os
.mkdir(tracedir_name
)
84 generate_metadata(array_len
)
85 generate_stream(array_len
)
88 print('Cleaning up test for ' + str(array_len
) + ' events')
90 os
.remove(output_metadata
)
94 os
.remove(output_stream
)
98 os
.rmdir(tracedir_name
)
102 if (mode
== "prepare"):
105 elif (mode
== "clean"):
This page took 0.031443 seconds and 4 git commands to generate.