Tracing V2: Fully-functional plugin.

components/tracing/core/proto_zero_message.h

 // Copyright 2016 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #ifndef COMPONENTS_TRACING_CORE_PROTO_ZERO_MESSAGE_H_
 #define COMPONENTS_TRACING_CORE_PROTO_ZERO_MESSAGE_H_
 
 #include <stdint.h>
 
+#include <set>
 #include <type_traits>
 
 #include "base/compiler_specific.h"
 #include "base/gtest_prod_util.h"
 #include "base/logging.h"
 #include "base/macros.h"
 #include "base/template_util.h"
 #include "build/build_config.h"
 #include "components/tracing/core/proto_utils.h"
 #include "components/tracing/core/scattered_stream_writer.h"
@@ skipping 35 matching lines @@
   void inc_size_already_written(size_t size) { size_already_written_ += size; }
 
 #if DCHECK_IS_ON()
   // Only for ProtoZeroMessageHandleBase.
   void set_handle(ProtoZeroMessageHandleBase* handle) { handle_ = handle; }
 #endif
 
  protected:
   ProtoZeroMessage();
 
-  // Proto types: uint64, uint32, int64, int32, bool, enum.
+  // Proto types: uint64, uint32, int64, int32, enum (large).
   template <typename T>
   void AppendVarInt(uint32_t field_id, T value) {
     if (nested_message_)
       EndNestedMessage();
 
     uint8_t buffer[proto::kMaxSimpleFieldEncodedSize];
     uint8_t* pos = buffer;
 
     pos = proto::WriteVarInt(proto::MakeTagVarInt(field_id), pos);
     // WriteVarInt encodes signed values in two's complement form.
     pos = proto::WriteVarInt(value, pos);
     WriteToStream(buffer, pos);
   }
 
   // Proto types: sint64, sint32.
   template <typename T>
   void AppendSignedVarInt(uint32_t field_id, T value) {
     AppendVarInt(field_id, proto::ZigZagEncode(value));
   }
 
+  // Proto types: bool, enum (small).
+  // Faster version of AppendVarInt for tiny numbers.
+  void AppendTinyVarInt(uint32_t field_id, int32_t value) {
+    DCHECK(0 <= value && value < 0x80);
+    if (nested_message_)
+      EndNestedMessage();
+
+    uint8_t buffer[proto::kMaxSimpleFieldEncodedSize];
+    uint8_t* pos = buffer;
+    // MakeTagVarInt gets super optimized here for constexpr.
+    pos = proto::WriteVarInt(proto::MakeTagVarInt(field_id), pos);
+    *pos++ = static_cast<uint8_t>(value);
+    WriteToStream(buffer, pos);
+  }
+
   // Proto types: fixed64, sfixed64, fixed32, sfixed32, double, float.
   template <typename T>
   void AppendFixed(uint32_t field_id, T value) {
     if (nested_message_)
       EndNestedMessage();
 
     uint8_t buffer[proto::kMaxSimpleFieldEncodedSize];
     uint8_t* pos = buffer;
 
     pos = proto::WriteVarInt(proto::MakeTagFixed<T>(field_id), pos);
     memcpy(pos, &value, sizeof(T));
     pos += sizeof(T);
     // TODO(kraynov): Optimize memcpy performance, see http://crbug.com/624311 .
     WriteToStream(buffer, pos);
   }
 
-  // TODO(kraynov): Add AppendTinyInt.
   void AppendString(uint32_t field_id, const char* str);
   void AppendBytes(uint32_t field_id, const void* value, size_t size);
 
   // Begins a nested message, using the static storage provided by the parent
   // class (see comment in |nested_messages_arena_|). The nested message ends
   // either when Finalize() is called or when any other Append* method is called
   // in the parent class.
   // The template argument T is supposed to be a stub class auto generated from
   // a .proto, hence a subclass of ProtoZeroMessage.
   template <class T>
@@ skipping 77 matching lines @@
   // DO NOT add any fields below |nested_messages_arena_|. The memory layout of
   // nested messages would overflow the storage allocated by the root message.
 
   DISALLOW_COPY_AND_ASSIGN(ProtoZeroMessage);
 };
 
 }  // namespace v2
 }  // namespace tracing
 
 #endif  // COMPONENTS_TRACING_CORE_PROTO_ZERO_MESSAGE_H_