third_party/upb/benchmarks/gen_protobuf_binary_cc.py

   1 #!/usr/bin/python
   2 #
   3 # Copyright (c) 2009-2021, Google LLC
   4 # All rights reserved.
   5 #
   6 # Redistribution and use in source and binary forms, with or without
   7 # modification, are permitted provided that the following conditions are met:
   8 #     * Redistributions of source code must retain the above copyright
   9 #       notice, this list of conditions and the following disclaimer.
  10 #     * Redistributions in binary form must reproduce the above copyright
  11 #       notice, this list of conditions and the following disclaimer in the
  12 #       documentation and/or other materials provided with the distribution.
  13 #     * Neither the name of Google LLC nor the
  14 #       names of its contributors may be used to endorse or promote products
  15 #       derived from this software without specific prior written permission.
  16 #
  17 # THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
  18 # ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
  19 # WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
  20 # DISCLAIMED. IN NO EVENT SHALL Google LLC BE LIABLE FOR ANY
  21 # DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
  22 # (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
  23 # LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
  24 # ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  25 # (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
  26 # SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  27
  28 import sys
  29 import re
  30
  31 include = sys.argv[1]
  32 msg_basename = sys.argv[2]
  33 count = 1
  34
  35 m = re.search(r'(.*\D)(\d+)$', sys.argv[2])
  36 if m:
  37   msg_basename = m.group(1)
  38   count = int(m.group(2))
  39
  40 print('''
  41 #include "{include}"
  42
  43 char buf[1];
  44
  45 int main() {{
  46 '''.format(include=include))
  47
  48 def RefMessage(name):
  49   print('''
  50   {{
  51     {name} proto;
  52     proto.ParseFromArray(buf, 0);
  53     proto.SerializePartialToArray(&buf[0], 0);
  54   }}
  55   '''.format(name=name))
  56
  57 RefMessage(msg_basename)
  58
  59 for i in range(2, count + 1):
  60   RefMessage(msg_basename + str(i))
  61
  62 print('''
  63   return 0;
  64 }''')