mojo::ArrayTraits: Add Support for Iterators

mojo/public/cpp/bindings/lib/array_serialization.h

 // Copyright 2014 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 MOJO_PUBLIC_CPP_BINDINGS_LIB_ARRAY_SERIALIZATION_H_
 #define MOJO_PUBLIC_CPP_BINDINGS_LIB_ARRAY_SERIALIZATION_H_
 
 #include <stddef.h>
 #include <string.h>  // For |memcpy()|.
 
@@ skipping 38 matching lines @@
                  ? ArraySerializerType::POINTER
                  : (IsHandle<DataElement>::value
                         ? ArraySerializerType::HANDLE
                         : (std::is_same<Element, bool>::value
                                ? ArraySerializerType::BOOLEAN
                                : (std::is_enum<Element>::value
                                       ? ArraySerializerType::ENUM
                                       : ArraySerializerType::POD))));
 };
 
-// Used as the UserTypeReader template parameter of ArraySerializer.
-template <typename MaybeConstUserType>
-class ArrayReader {
+template <typename Traits,
+          typename MaybeConstUserType,
+          bool HasGetBegin =
+              HasGetBeginMethod<Traits, MaybeConstUserType>::value>
+class ArrayIterator {};
+
+// Used as the UserTypeIterator template parameter of ArraySerializer.
+template <typename Traits, typename MaybeConstUserType>
+class ArrayIterator<Traits, MaybeConstUserType, true> {
  public:
-  using UserType = typename std::remove_const<MaybeConstUserType>::type;
-  using Traits = ArrayTraits<UserType>;
+  using IteratorType = decltype(
+      CallGetBeginIfExists<Traits>(std::declval<MaybeConstUserType&>()));
 
-  explicit ArrayReader(MaybeConstUserType& input) : input_(input) {}
-  ~ArrayReader() {}
+  explicit ArrayIterator(MaybeConstUserType& input)
+      : input_(input), iter_(CallGetBeginIfExists<Traits>(input)) {}
+  ~ArrayIterator() {}
+
+  size_t GetSize() const { return Traits::GetSize(input_); }
+
+  using GetNextResult =
+      decltype(Traits::GetValue(std::declval<IteratorType&>()));
+  GetNextResult GetNext() {
+    auto& value = Traits::GetValue(iter_);
+    Traits::AdvanceIterator(iter_);
+    return value;
+  }
+
+  using GetDataIfExistsResult = decltype(
+      CallGetDataIfExists<Traits>(std::declval<MaybeConstUserType&>()));
+  GetDataIfExistsResult GetDataIfExists() {
+    return CallGetDataIfExists<Traits>(input_);
+  }
+
+ private:
+  MaybeConstUserType& input_;
+  IteratorType iter_;
+};
+
+// Used as the UserTypeIterator template parameter of ArraySerializer.
+template <typename Traits, typename MaybeConstUserType>
+class ArrayIterator<Traits, MaybeConstUserType, false> {
+ public:
+  explicit ArrayIterator(MaybeConstUserType& input) : input_(input), iter_(0) {}
+  ~ArrayIterator() {}
 
   size_t GetSize() const { return Traits::GetSize(input_); }
 
   using GetNextResult =
       decltype(Traits::GetAt(std::declval<MaybeConstUserType&>(), 0));
   GetNextResult GetNext() {
-    DCHECK_LT(index_, Traits::GetSize(input_));
-    return Traits::GetAt(input_, index_++);
+    DCHECK_LT(iter_, Traits::GetSize(input_));
+    return Traits::GetAt(input_, iter_++);
   }
 
   using GetDataIfExistsResult = decltype(
       CallGetDataIfExists<Traits>(std::declval<MaybeConstUserType&>()));
   GetDataIfExistsResult GetDataIfExists() {
     return CallGetDataIfExists<Traits>(input_);
   }
 
  private:
   MaybeConstUserType& input_;
-  size_t index_ = 0;
+  size_t iter_;
 };
 
 // ArraySerializer is also used to serialize map keys and values. Therefore, it
-// has a UserTypeReader parameter which is an adaptor for reading to hide the
+// has a UserTypeIterator parameter which is an adaptor for reading to hide the
 // difference between ArrayTraits and MapTraits.
 template <typename MojomType,
           typename MaybeConstUserType,
-          typename UserTypeReader,
+          typename UserTypeIterator,
           ArraySerializerType type = GetArraySerializerType<MojomType>::value>
 struct ArraySerializer;
 
 // Handles serialization and deserialization of arrays of pod types.
 template <typename MojomType,
           typename MaybeConstUserType,
-          typename UserTypeReader>
+          typename UserTypeIterator>
 struct ArraySerializer<MojomType,
                        MaybeConstUserType,
-                       UserTypeReader,
+                       UserTypeIterator,
                        ArraySerializerType::POD> {
   using UserType = typename std::remove_const<MaybeConstUserType>::type;
   using Data = typename MojomType::Data_;
   using DataElement = typename Data::Element;
   using Element = typename MojomType::Element;
   using Traits = ArrayTraits<UserType>;
+  using IteratorType =

[yzshen1, 2016/06/13 15:18:05]

please remove this line.

[Fady Samuel, 2016/06/13 15:38:11]

Done.

+      decltype(CallGetBeginIfExists<Traits>(std::declval<UserType&>()));
 
   static_assert(std::is_same<Element, DataElement>::value,
                 "Incorrect array serializer");
   static_assert(std::is_same<Element, typename Traits::Element>::value,
                 "Incorrect array serializer");
 
-  static size_t GetSerializedSize(UserTypeReader* input,
+  static size_t GetSerializedSize(UserTypeIterator* input,
                                   SerializationContext* context) {
     return sizeof(Data) + Align(input->GetSize() * sizeof(DataElement));
   }
 
-  static void SerializeElements(UserTypeReader* input,
+  static void SerializeElements(UserTypeIterator* input,
                                 Buffer* buf,
                                 Data* output,
                                 const ContainerValidateParams* validate_params,
                                 SerializationContext* context) {
     DCHECK(!validate_params->element_is_nullable)
         << "Primitive type should be non-nullable";
     DCHECK(!validate_params->element_validate_params)
         << "Primitive type should not have array validate params";
 
     size_t size = input->GetSize();
     if (size == 0)
       return;
 
     auto data = input->GetDataIfExists();
     if (data) {
       memcpy(output->storage(), data, size * sizeof(DataElement));
     } else {
       for (size_t i = 0; i < size; ++i)
         output->at(i) = input->GetNext();
     }
   }
 
   static bool DeserializeElements(Data* input,
                                   UserType* output,
                                   SerializationContext* context) {
     if (!Traits::Resize(*output, input->size()))
       return false;
+    ArrayIterator<Traits, UserType> iterator(*output);
     if (input->size()) {
-      auto data = CallGetDataIfExists<Traits>(*output);
+      auto data = iterator.GetDataIfExists();
       if (data) {
         memcpy(data, input->storage(), input->size() * sizeof(DataElement));
       } else {
         for (size_t i = 0; i < input->size(); ++i)
-          Traits::GetAt(*output, i) = input->at(i);
+          iterator.GetNext() = input->at(i);
       }
     }
     return true;
   }
 };
 
 // Handles serialization and deserialization of arrays of enum types.
 template <typename MojomType,
           typename MaybeConstUserType,
-          typename UserTypeReader>
+          typename UserTypeIterator>
 struct ArraySerializer<MojomType,
                        MaybeConstUserType,
-                       UserTypeReader,
+                       UserTypeIterator,
                        ArraySerializerType::ENUM> {
   using UserType = typename std::remove_const<MaybeConstUserType>::type;
   using Data = typename MojomType::Data_;
   using DataElement = typename Data::Element;
   using Element = typename MojomType::Element;
   using Traits = ArrayTraits<UserType>;
 
   static_assert(sizeof(Element) == sizeof(DataElement),
                 "Incorrect array serializer");
 
-  static size_t GetSerializedSize(UserTypeReader* input,
+  static size_t GetSerializedSize(UserTypeIterator* input,
                                   SerializationContext* context) {
     return sizeof(Data) + Align(input->GetSize() * sizeof(DataElement));
   }
 
-  static void SerializeElements(UserTypeReader* input,
+  static void SerializeElements(UserTypeIterator* input,
                                 Buffer* buf,
                                 Data* output,
                                 const ContainerValidateParams* validate_params,
                                 SerializationContext* context) {
     DCHECK(!validate_params->element_is_nullable)
         << "Primitive type should be non-nullable";
     DCHECK(!validate_params->element_validate_params)
         << "Primitive type should not have array validate params";
 
     size_t size = input->GetSize();
     for (size_t i = 0; i < size; ++i)
       Serialize<Element>(input->GetNext(), output->storage() + i);
   }
 
   static bool DeserializeElements(Data* input,
                                   UserType* output,
                                   SerializationContext* context) {
     if (!Traits::Resize(*output, input->size()))
       return false;
+    ArrayIterator<Traits, UserType> iterator(*output);
     for (size_t i = 0; i < input->size(); ++i) {
-      if (!Deserialize<Element>(input->at(i), &Traits::GetAt(*output, i)))
+      if (!Deserialize<Element>(input->at(i), &iterator.GetNext()))
         return false;
     }
     return true;
   }
 };
 
 // Serializes and deserializes arrays of bools.
 template <typename MojomType,
           typename MaybeConstUserType,
-          typename UserTypeReader>
+          typename UserTypeIterator>
 struct ArraySerializer<MojomType,
                        MaybeConstUserType,
-                       UserTypeReader,
+                       UserTypeIterator,
                        ArraySerializerType::BOOLEAN> {
   using UserType = typename std::remove_const<MaybeConstUserType>::type;
   using Traits = ArrayTraits<UserType>;
   using Data = typename MojomType::Data_;
 
   static_assert(std::is_same<bool, typename UserType::Element>::value,
                 "Incorrect array serializer");
 
-  static size_t GetSerializedSize(UserTypeReader* input,
+  static size_t GetSerializedSize(UserTypeIterator* input,
                                   SerializationContext* context) {
     return sizeof(Data) + Align((input->GetSize() + 7) / 8);
   }
 
-  static void SerializeElements(UserTypeReader* input,
+  static void SerializeElements(UserTypeIterator* input,
                                 Buffer* buf,
                                 Data* output,
                                 const ContainerValidateParams* validate_params,
                                 SerializationContext* context) {
     DCHECK(!validate_params->element_is_nullable)
         << "Primitive type should be non-nullable";
     DCHECK(!validate_params->element_validate_params)
         << "Primitive type should not have array validate params";
 
     size_t size = input->GetSize();
     for (size_t i = 0; i < size; ++i)
       output->at(i) = input->GetNext();
   }
   static bool DeserializeElements(Data* input,
                                   UserType* output,
                                   SerializationContext* context) {
     if (!Traits::Resize(*output, input->size()))
       return false;
+    ArrayIterator<Traits, UserType> iterator(*output);
     for (size_t i = 0; i < input->size(); ++i)
-      Traits::GetAt(*output, i) = input->at(i);
+      iterator.GetNext() = input->at(i);
     return true;
   }
 };
 
 // Serializes and deserializes arrays of handles.
 template <typename MojomType,
           typename MaybeConstUserType,
-          typename UserTypeReader>
+          typename UserTypeIterator>
 struct ArraySerializer<MojomType,
                        MaybeConstUserType,
-                       UserTypeReader,
+                       UserTypeIterator,
                        ArraySerializerType::HANDLE> {
   using UserType = typename std::remove_const<MaybeConstUserType>::type;
   using Data = typename MojomType::Data_;
   using Element = typename MojomType::Element;
   using Traits = ArrayTraits<UserType>;
 
   static_assert(std::is_same<Element, typename Traits::Element>::value,
                 "Incorrect array serializer");
 
-  static size_t GetSerializedSize(UserTypeReader* input,
+  static size_t GetSerializedSize(UserTypeIterator* input,
                                   SerializationContext* context) {
     return sizeof(Data) +
            Align(input->GetSize() * sizeof(typename Data::Element));
   }
 
-  static void SerializeElements(UserTypeReader* input,
+  static void SerializeElements(UserTypeIterator* input,
                                 Buffer* buf,
                                 Data* output,
                                 const ContainerValidateParams* validate_params,
                                 SerializationContext* context) {
     DCHECK(!validate_params->element_validate_params)
         << "Handle type should not have array validate params";
 
     size_t size = input->GetSize();
     for (size_t i = 0; i < size; ++i) {
       // Transfer ownership of the handle.
       output->at(i) = context->handles.AddHandle(input->GetNext().release());
       MOJO_INTERNAL_DLOG_SERIALIZATION_WARNING(
           !validate_params->element_is_nullable && !output->at(i).is_valid(),
           VALIDATION_ERROR_UNEXPECTED_INVALID_HANDLE,
           MakeMessageWithArrayIndex(
               "invalid handle in array expecting valid handles", size, i));
     }
   }
   static bool DeserializeElements(Data* input,
                                   UserType* output,
                                   SerializationContext* context) {
     using HandleType = typename Element::RawHandleType;
     if (!Traits::Resize(*output, input->size()))
       return false;
+    ArrayIterator<Traits, UserType> iterator(*output);
     for (size_t i = 0; i < input->size(); ++i) {
-      Traits::GetAt(*output, i) = MakeScopedHandle(
+      iterator.GetNext() = MakeScopedHandle(
           HandleType(context->handles.TakeHandle(input->at(i)).value()));
     }
     return true;
   }
 };
 
 // This template must only apply to pointer mojo entity (strings, structs,
 // arrays and maps).
 template <typename MojomType,
           typename MaybeConstUserType,
-          typename UserTypeReader>
+          typename UserTypeIterator>
 struct ArraySerializer<MojomType,
                        MaybeConstUserType,
-                       UserTypeReader,
+                       UserTypeIterator,
                        ArraySerializerType::POINTER> {
   using UserType = typename std::remove_const<MaybeConstUserType>::type;
   using Data = typename MojomType::Data_;
   using DataElement = typename Data::Element;
   using Element = typename MojomType::Element;
   using Traits = ArrayTraits<UserType>;
 
-  static size_t GetSerializedSize(UserTypeReader* input,
+  static size_t GetSerializedSize(UserTypeIterator* input,
                                   SerializationContext* context) {
     size_t element_count = input->GetSize();
     size_t size =
         sizeof(Data) +
         element_count *
             sizeof(Pointer<typename std::remove_pointer<DataElement>::type>);
     for (size_t i = 0; i < element_count; ++i)
       size += PrepareToSerialize<Element>(input->GetNext(), context);
     return size;
   }
 
-  static void SerializeElements(UserTypeReader* input,
+  static void SerializeElements(UserTypeIterator* input,
                                 Buffer* buf,
                                 Data* output,
                                 const ContainerValidateParams* validate_params,
                                 SerializationContext* context) {
     size_t size = input->GetSize();
     for (size_t i = 0; i < size; ++i) {
       DataElement element;
       SerializeCaller<Element>::Run(input->GetNext(), buf, &element,
                                     validate_params->element_validate_params,
                                     context);
       output->at(i) = element;
       MOJO_INTERNAL_DLOG_SERIALIZATION_WARNING(
           !validate_params->element_is_nullable && !element,
           VALIDATION_ERROR_UNEXPECTED_NULL_POINTER,
           MakeMessageWithArrayIndex("null in array expecting valid pointers",
                                     size, i));
     }
   }
   static bool DeserializeElements(Data* input,
                                   UserType* output,
                                   SerializationContext* context) {
     bool success = true;
     if (!Traits::Resize(*output, input->size()))
       return false;
+    ArrayIterator<Traits, UserType> iterator(*output);
     for (size_t i = 0; i < input->size(); ++i) {
       // Note that we rely on complete deserialization taking place in order to
       // transfer ownership of all encoded handles. Therefore we don't
       // short-circuit on failure here.
-      if (!Deserialize<Element>(input->at(i), &Traits::GetAt(*output, i),
-                                context)) {
+      if (!Deserialize<Element>(input->at(i), &iterator.GetNext(), context)) {
         success = false;
       }
     }
     return success;
   }
 
  private:
   template <typename T,
             bool is_array_or_map = IsSpecializationOf<Array, T>::value ||
                                    IsSpecializationOf<Map, T>::value>
@@ skipping 18 matching lines @@
                     SerializationContext* context) {
       Serialize<T>(std::forward<InputElementType>(input), buf, output,
                    validate_params, context);
     }
   };
 };
 
 // Handles serialization and deserialization of arrays of unions.
 template <typename MojomType,
           typename MaybeConstUserType,
-          typename UserTypeReader>
+          typename UserTypeIterator>
 struct ArraySerializer<MojomType,
                        MaybeConstUserType,
-                       UserTypeReader,
+                       UserTypeIterator,
                        ArraySerializerType::UNION> {
   using UserType = typename std::remove_const<MaybeConstUserType>::type;
   using Data = typename MojomType::Data_;
   using Element = typename MojomType::Element;
   using Traits = ArrayTraits<UserType>;
 
   static_assert(std::is_same<typename MojomType::Element,
                              typename Traits::Element>::value,
                 "Incorrect array serializer");
 
-  static size_t GetSerializedSize(UserTypeReader* input,
+  static size_t GetSerializedSize(UserTypeIterator* input,
                                   SerializationContext* context) {
     size_t element_count = input->GetSize();
     size_t size = sizeof(Data);
     for (size_t i = 0; i < element_count; ++i) {
       // Call with |inlined| set to false, so that it will account for both the
       // data in the union and the space in the array used to hold the union.
       size += PrepareToSerialize<Element>(input->GetNext(), false, context);
     }
     return size;
   }
 
-  static void SerializeElements(UserTypeReader* input,
+  static void SerializeElements(UserTypeIterator* input,
                                 Buffer* buf,
                                 Data* output,
                                 const ContainerValidateParams* validate_params,
                                 SerializationContext* context) {
     size_t size = input->GetSize();
     for (size_t i = 0; i < size; ++i) {
       typename Data::Element* result = output->storage() + i;
       Serialize<Element>(input->GetNext(), buf, &result, true, context);
       MOJO_INTERNAL_DLOG_SERIALIZATION_WARNING(
           !validate_params->element_is_nullable && output->at(i).is_null(),
           VALIDATION_ERROR_UNEXPECTED_NULL_POINTER,
           MakeMessageWithArrayIndex("null in array expecting valid unions",
                                     size, i));
     }
   }
 
   static bool DeserializeElements(Data* input,
                                   UserType* output,
                                   SerializationContext* context) {
     bool success = true;
     if (!Traits::Resize(*output, input->size()))
       return false;
+    ArrayIterator<Traits, UserType> iterator(*output);
     for (size_t i = 0; i < input->size(); ++i) {
       // Note that we rely on complete deserialization taking place in order to
       // transfer ownership of all encoded handles. Therefore we don't
       // short-circuit on failure here.
-      if (!Deserialize<Element>(&input->at(i), &Traits::GetAt(*output, i),
-                                context)) {
+      if (!Deserialize<Element>(&input->at(i), &iterator.GetNext(), context)) {
         success = false;
       }
     }
     return success;
   }
 };
 
 template <typename Element, typename MaybeConstUserType>
 struct Serializer<Array<Element>, MaybeConstUserType> {
   using UserType = typename std::remove_const<MaybeConstUserType>::type;
+  using Traits = ArrayTraits<UserType>;
   using Impl = ArraySerializer<Array<Element>,
                                MaybeConstUserType,
-                               ArrayReader<MaybeConstUserType>>;
-  using Traits = ArrayTraits<UserType>;
+                               ArrayIterator<Traits, MaybeConstUserType>>;
   using Data = typename Array<Element>::Data_;
 
   static size_t PrepareToSerialize(MaybeConstUserType& input,
                                    SerializationContext* context) {
     if (CallIsNullIfExists<Traits>(input))
       return 0;
-    ArrayReader<MaybeConstUserType> reader(input);
-    return Impl::GetSerializedSize(&reader, context);
+    ArrayIterator<Traits, MaybeConstUserType> iterator(input);
+    return Impl::GetSerializedSize(&iterator, context);
   }
 
   static void Serialize(MaybeConstUserType& input,
                         Buffer* buf,
                         Data** output,
                         const ContainerValidateParams* validate_params,
                         SerializationContext* context) {
     if (!CallIsNullIfExists<Traits>(input)) {
       MOJO_INTERNAL_DLOG_SERIALIZATION_WARNING(
           validate_params->expected_num_elements != 0 &&
               Traits::GetSize(input) != validate_params->expected_num_elements,
           internal::VALIDATION_ERROR_UNEXPECTED_ARRAY_HEADER,
           internal::MakeMessageWithExpectedArraySize(
               "fixed-size array has wrong number of elements",
               Traits::GetSize(input), validate_params->expected_num_elements));
       Data* result = Data::New(Traits::GetSize(input), buf);
       if (result) {
-        ArrayReader<MaybeConstUserType> reader(input);
-        Impl::SerializeElements(&reader, buf, result, validate_params, context);
+        ArrayIterator<Traits, MaybeConstUserType> iterator(input);
+        Impl::SerializeElements(&iterator, buf, result, validate_params,
+                                context);
       }
       *output = result;
     } else {
       *output = nullptr;
     }
   }
 
   static bool Deserialize(Data* input,
                           UserType* output,
                           SerializationContext* context) {
     if (!input)
       return CallSetToNullIfExists<Traits>(output);
     return Impl::DeserializeElements(input, output, context);
   }
 };
 
 }  // namespace internal
 }  // namespace mojo
 
 #endif  // MOJO_PUBLIC_CPP_BINDINGS_LIB_ARRAY_SERIALIZATION_H_