Simplify and extend RtpHeaderExtensionMap

webrtc/modules/rtp_rtcp/source/rtp_header_extension.h

 /*
  *  Copyright (c) 2012 The WebRTC project authors. All Rights Reserved.
  *
  *  Use of this source code is governed by a BSD-style license
  *  that can be found in the LICENSE file in the root of the source
  *  tree. An additional intellectual property rights grant can be found
  *  in the file PATENTS.  All contributing project authors may
  *  be found in the AUTHORS file in the root of the source tree.
  */
 
 #ifndef WEBRTC_MODULES_RTP_RTCP_SOURCE_RTP_HEADER_EXTENSION_H_
 #define WEBRTC_MODULES_RTP_RTCP_SOURCE_RTP_HEADER_EXTENSION_H_
 
-#include <map>
+#include <initializer_list>
+#include <string>
 
+#include "webrtc/base/basictypes.h"
+#include "webrtc/base/checks.h"
+#include "webrtc/config.h"
 #include "webrtc/modules/rtp_rtcp/include/rtp_rtcp_defines.h"
-#include "webrtc/typedefs.h"
 
 namespace webrtc {
 
 const uint16_t kRtpOneByteHeaderExtensionId = 0xBEDE;
 
 const size_t kRtpOneByteHeaderLength = 4;
 const size_t kTransmissionTimeOffsetLength = 4;
 const size_t kAudioLevelLength = 2;
 const size_t kAbsoluteSendTimeLength = 4;
 const size_t kVideoRotationLength = 2;
 const size_t kTransportSequenceNumberLength = 3;
 const size_t kPlayoutDelayLength = 4;
 
 // Playout delay in milliseconds. A playout delay limit (min or max)
 // has 12 bits allocated. This allows a range of 0-4095 values which translates
 // to a range of 0-40950 in milliseconds.
 const int kPlayoutDelayGranularityMs = 10;
 // Maximum playout delay value in milliseconds.
 const int kPlayoutDelayMaxMs = 40950;
 
-struct HeaderExtension {
-  explicit HeaderExtension(RTPExtensionType extension_type)
-      : type(extension_type), length(0) {
-    Init();
-  }
-
-  void Init() {
-    // TODO(solenberg): Create handler classes for header extensions so we can
-    // get rid of switches like these as well as handling code spread out all
-    // over.
-    switch (type) {
-      case kRtpExtensionTransmissionTimeOffset:
-        length = kTransmissionTimeOffsetLength;
-        break;
-      case kRtpExtensionAudioLevel:
-        length = kAudioLevelLength;
-        break;
-      case kRtpExtensionAbsoluteSendTime:
-        length = kAbsoluteSendTimeLength;
-        break;
-      case kRtpExtensionVideoRotation:
-        length = kVideoRotationLength;
-        break;
-      case kRtpExtensionTransportSequenceNumber:
-        length = kTransportSequenceNumberLength;
-        break;
-      case kRtpExtensionPlayoutDelay:
-        length = kPlayoutDelayLength;
-        break;
-      default:
-        assert(false);
-    }
-  }
-
-  const RTPExtensionType type;
-  uint8_t length;
-};
-
 class RtpHeaderExtensionMap {
  public:
   static constexpr RTPExtensionType kInvalidType = kRtpExtensionNone;
   static constexpr uint8_t kInvalidId = 0;
+
   RtpHeaderExtensionMap();
-  ~RtpHeaderExtensionMap();
+  RtpHeaderExtensionMap(std::initializer_list<RtpExtension>);
 
-  void Erase();
+  template <typename Extension>
+  bool Register(uint8_t id) {
+    return Register(id, Extension::kId, Extension::kValueSizeBytes,
+                    Extension::kUri);
+  }
+  bool RegisterByType(uint8_t id, RTPExtensionType type);
+  bool RegisterByUri(uint8_t id, const std::string& uri);
 
-  int32_t Register(RTPExtensionType type, uint8_t id);
+  bool IsRegistered(RTPExtensionType type) const {
+    return GetId(type) != kInvalidId;
+  }
+  // Return kInvalidType if not found.
+  RTPExtensionType GetType(uint8_t id) const {
+    RTC_DCHECK_GE(id, kMinId);
+    RTC_DCHECK_LE(id, kMaxId);
+    return types_[id];
+  }
+  // Return kInvalidId if not found.
+  uint8_t GetId(RTPExtensionType type) const {
+    RTC_DCHECK_GT(type, kRtpExtensionNone);
+    RTC_DCHECK_LT(type, kRtpExtensionNumberOfExtensions);
+    return ids_[type];
+  }
 
-  int32_t Deregister(RTPExtensionType type);
-
-  bool IsRegistered(RTPExtensionType type) const;
-
-  int32_t GetType(uint8_t id, RTPExtensionType* type) const;
-  // Return kInvalidType if not found.
-  RTPExtensionType GetType(uint8_t id) const;
-
-  int32_t GetId(const RTPExtensionType type, uint8_t* id) const;
-  // Return kInvalidId if not found.
-  uint8_t GetId(RTPExtensionType type) const;
   size_t GetTotalLengthInBytes() const;
 
-  void GetCopy(RtpHeaderExtensionMap* map) const;
-
-  int32_t Size() const;
+  // TODO(danilchap): Remove use of the functions below.
+  void Erase() { *this = RtpHeaderExtensionMap(); }
+  int32_t Register(RTPExtensionType type, uint8_t id) {
+    return RegisterByType(id, type) ? 0 : -1;
+  }
+  int32_t Deregister(RTPExtensionType type);
+  int32_t GetType(uint8_t id, RTPExtensionType* type) const {
+    *type = GetType(id);
+    return *type == kInvalidType ? -1 : 0;
+  }
+  void GetCopy(RtpHeaderExtensionMap* copy) const { *copy = *this; }
 
  private:
-  std::map<uint8_t, HeaderExtension*> extensionMap_;
+  static constexpr uint8_t kMinId = 1;
+  static constexpr uint8_t kMaxId = 14;
+  bool Register(uint8_t id,
+                RTPExtensionType type,
+                size_t value_size,
+                const char* uri);
+
+  size_t total_values_size_bytes_ = 0;
+  RTPExtensionType types_[kMaxId + 1];
+  uint8_t ids_[kRtpExtensionNumberOfExtensions];
 };
 }  // namespace webrtc
 
 #endif  // WEBRTC_MODULES_RTP_RTCP_SOURCE_RTP_HEADER_EXTENSION_H_