blob: 94218de22c4953aaf818b45a537f755d61ce29af [file] [log] [blame]
#!/usr/bin/env python3
# Copyright (C) 2022 The Android Open Source Project
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
import os
import sys
import re
import subprocess
import pathlib
import tempfile
import contextlib
import argparse
ROOT_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
TOOLS_DIR = os.path.join(ROOT_DIR, "tools")
OUT_DIR = os.path.join(ROOT_DIR, "out", "tools")
NINJA = os.path.join(TOOLS_DIR, "ninja")
GN = os.path.join(TOOLS_DIR, "gn")
PROTOC_PATH = os.path.join(OUT_DIR, "protoc")
DESCRIPTOR_PATH = os.path.join(ROOT_DIR, "src", "trace_processor", "importers",
"proto", "atoms.descriptor")
PROTOBUF_BUILTINS_DIR = os.path.join(ROOT_DIR, "buildtools", "protobuf", "src")
PROTO_LOGGING_URL = "https://android.googlesource.com/platform/frameworks/proto_logging.git"
ATOM_RE = r" message_type {\n. name: \"Atom\"(\n .+)+(\n })"
FIELD_RE = r" field {\n name: \"([^\"]+)\"\n number: ([0-9]+)"
ATOM_IDS_PATH = os.path.join(ROOT_DIR, "protos", "perfetto", "config", "statsd",
"atom_ids.proto")
ATOM_IDS_TEMPLATE = """/*
* Copyright (C) 2022 The Android Open Source Project
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
syntax = "proto2";
package perfetto.protos;
enum AtomId {{
{atoms}
}}
"""
def call(*cmd, stdin=None):
try:
return subprocess.check_output(cmd, stdin=stdin)
except subprocess.CalledProcessError as e:
print("Error running the command:")
print(" ".join(cmd))
print(e)
exit(1)
def main():
parser = argparse.ArgumentParser()
parser.add_argument("--atoms-checkout")
args = parser.parse_args()
call(GN, "gen", OUT_DIR, "--args=is_debug=false")
call(NINJA, "-C", OUT_DIR, "protoc")
with contextlib.ExitStack() as stack:
# Write the descriptor.
if args.atoms_checkout:
atoms_root = args.atoms_checkout
proto_logging_dir = os.path.join(atoms_root, "frameworks",
"proto_logging")
else:
atoms_root = stack.enter_context(tempfile.TemporaryDirectory())
proto_logging_dir = os.path.join(atoms_root, "frameworks",
"proto_logging")
pathlib.Path(proto_logging_dir).mkdir(parents=True, exist_ok=True)
call("git", "clone", PROTO_LOGGING_URL, proto_logging_dir)
atoms_path = os.path.join(proto_logging_dir, "stats", "atoms.proto")
call(PROTOC_PATH, f"--proto_path={PROTOBUF_BUILTINS_DIR}",
f"--proto_path={atoms_root}",
f"--descriptor_set_out={DESCRIPTOR_PATH}", "--include_imports",
atoms_path)
# Extract and update atom_ids.proto. To do this we regex the pbtext
# of the descriptor. This is hopefully:
# - more stable than regexing atom.proto directly
# - less complicated than parsing finding, importing, and using the
# Python protobuf library.
descriptor_in = stack.enter_context(open(DESCRIPTOR_PATH))
pbtext = call(
PROTOC_PATH,
f"--proto_path={PROTOBUF_BUILTINS_DIR}",
f"{PROTOBUF_BUILTINS_DIR}/google/protobuf/descriptor.proto",
"--decode=google.protobuf.FileDescriptorSet",
stdin=descriptor_in)
atom_pbtext = re.search(ATOM_RE, pbtext.decode("utf8"), re.MULTILINE)[0]
lines = []
for m in re.finditer(FIELD_RE, atom_pbtext):
name = m[1].upper()
field = m[2]
lines.append(f" {name} = {field};".format(name=name, field=field))
atom_ids_out = stack.enter_context(open(ATOM_IDS_PATH, "w"))
atom_ids_out.write(ATOM_IDS_TEMPLATE.format(atoms="\n".join(lines)))
if __name__ == "__main__":
sys.exit(main())