| /* |
| * Copyright (C) 2019 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. |
| */ |
| |
| #ifndef SRC_TRACE_PROCESSOR_IMPORTERS_PROTO_TRACK_EVENT_MODULE_H_ |
| #define SRC_TRACE_PROCESSOR_IMPORTERS_PROTO_TRACK_EVENT_MODULE_H_ |
| |
| #include "perfetto/base/build_config.h" |
| #include "src/trace_processor/importers/proto/proto_importer_module.h" |
| #include "src/trace_processor/importers/proto/track_event_parser.h" |
| #include "src/trace_processor/importers/proto/track_event_tokenizer.h" |
| #include "src/trace_processor/timestamped_trace_piece.h" |
| |
| #include "protos/perfetto/config/trace_config.pbzero.h" |
| #include "protos/perfetto/trace/trace_packet.pbzero.h" |
| |
| namespace perfetto { |
| namespace trace_processor { |
| |
| class TrackEventModule : public ProtoImporterModuleBase</*IsEnabled=*/1> { |
| public: |
| explicit TrackEventModule(TraceProcessorContext* context) |
| : ProtoImporterModuleBase(context), |
| tokenizer_(context), |
| parser_(context) {} |
| |
| ModuleResult TokenizePacket( |
| const protos::pbzero::TracePacket::Decoder& decoder, |
| TraceBlobView* packet, |
| int64_t packet_timestamp, |
| PacketSequenceState* state) { |
| if (decoder.has_track_descriptor()) { |
| tokenizer_.TokenizeTrackDescriptorPacket(decoder); |
| return ModuleResult::Handled(); |
| } |
| |
| if (decoder.has_track_event()) { |
| tokenizer_.TokenizeTrackEventPacket(state, decoder, packet, |
| packet_timestamp); |
| return ModuleResult::Handled(); |
| } |
| |
| // TODO(eseckler): Remove these once Chrome has switched fully over to |
| // TrackDescriptors. |
| if (decoder.has_thread_descriptor()) { |
| tokenizer_.TokenizeThreadDescriptorPacket(state, decoder); |
| return ModuleResult::Handled(); |
| } |
| if (decoder.has_process_descriptor()) { |
| tokenizer_.TokenizeProcessDescriptorPacket(decoder); |
| return ModuleResult::Handled(); |
| } |
| |
| return ModuleResult::Ignored(); |
| } |
| |
| ModuleResult ParsePacket(const protos::pbzero::TracePacket::Decoder& decoder, |
| const TimestampedTracePiece& ttp) { |
| if (decoder.has_track_event()) { |
| parser_.ParseTrackEvent( |
| ttp.timestamp, ttp.thread_timestamp, ttp.thread_instruction_count, |
| ttp.packet_sequence_state, ttp.packet_sequence_state_generation, |
| decoder.track_event()); |
| return ModuleResult::Handled(); |
| } |
| return ModuleResult::Ignored(); |
| } |
| |
| private: |
| TrackEventTokenizer tokenizer_; |
| TrackEventParser parser_; |
| }; |
| |
| } // namespace trace_processor |
| } // namespace perfetto |
| |
| #endif // SRC_TRACE_PROCESSOR_IMPORTERS_PROTO_TRACK_EVENT_MODULE_H_ |