src/cpp-common/bt2/field-path.hpp

   1 /*
   2  * Copyright (c) 2020 Philippe Proulx <pproulx@efficios.com>
   3  *
   4  * SPDX-License-Identifier: MIT
   5  */
   6
   7 #ifndef BABELTRACE_CPP_COMMON_BT2_FIELD_PATH_HPP
   8 #define BABELTRACE_CPP_COMMON_BT2_FIELD_PATH_HPP
   9
  10 #include <cstdint>
  11
  12 #include <babeltrace2/babeltrace.h>
  13
  14 #include "common/assert.h"
  15
  16 #include "borrowed-object-iterator.hpp"
  17 #include "borrowed-object.hpp"
  18 #include "shared-object.hpp"
  19
  20 namespace bt2 {
  21
  22 class ConstIndexFieldPathItem;
  23
  24 enum class FieldPathItemType
  25 {
  26     INDEX = BT_FIELD_PATH_ITEM_TYPE_INDEX,
  27     CURRENT_ARRAY_ELEMENT = BT_FIELD_PATH_ITEM_TYPE_CURRENT_ARRAY_ELEMENT,
  28     CURRENT_OPTION_CONTENT = BT_FIELD_PATH_ITEM_TYPE_CURRENT_OPTION_CONTENT,
  29 };
  30
  31 class ConstFieldPathItem : public BorrowedObject<const bt_field_path_item>
  32 {
  33 public:
  34     explicit ConstFieldPathItem(const LibObjPtr libObjPtr) noexcept :
  35         _ThisBorrowedObject {libObjPtr}
  36     {
  37     }
  38
  39     FieldPathItemType type() const noexcept
  40     {
  41         return static_cast<FieldPathItemType>(this->_libType());
  42     }
  43
  44     bool isIndex() const noexcept
  45     {
  46         return this->_libType() == BT_FIELD_PATH_ITEM_TYPE_INDEX;
  47     }
  48
  49     bool isCurrentArrayElement() const noexcept
  50     {
  51         return this->_libType() == BT_FIELD_PATH_ITEM_TYPE_CURRENT_ARRAY_ELEMENT;
  52     }
  53
  54     bool isCurrentOptionContent() const noexcept
  55     {
  56         return this->_libType() == BT_FIELD_PATH_ITEM_TYPE_CURRENT_OPTION_CONTENT;
  57     }
  58
  59     ConstIndexFieldPathItem asIndex() const noexcept;
  60
  61 private:
  62     bt_field_path_item_type _libType() const noexcept
  63     {
  64         return bt_field_path_item_get_type(this->libObjPtr());
  65     }
  66 };
  67
  68 class ConstIndexFieldPathItem final : public ConstFieldPathItem
  69 {
  70 public:
  71     explicit ConstIndexFieldPathItem(const LibObjPtr libObjPtr) noexcept :
  72         ConstFieldPathItem {libObjPtr}
  73     {
  74         BT_ASSERT_DBG(this->isIndex());
  75     }
  76
  77     std::uint64_t index() const noexcept
  78     {
  79         return bt_field_path_item_index_get_index(this->libObjPtr());
  80     }
  81 };
  82
  83 inline ConstIndexFieldPathItem ConstFieldPathItem::asIndex() const noexcept
  84 {
  85     BT_ASSERT_DBG(this->isIndex());
  86     return ConstIndexFieldPathItem {this->libObjPtr()};
  87 }
  88
  89 namespace internal {
  90
  91 struct FieldPathRefFuncs final
  92 {
  93     static void get(const bt_field_path * const libObjPtr) noexcept
  94     {
  95         bt_field_path_get_ref(libObjPtr);
  96     }
  97
  98     static void put(const bt_field_path * const libObjPtr) noexcept
  99     {
 100         bt_field_path_put_ref(libObjPtr);
 101     }
 102 };
 103
 104 } /* namespace internal */
 105
 106 class ConstFieldPath final : public BorrowedObject<const bt_field_path>
 107 {
 108 public:
 109     using Shared = SharedObject<ConstFieldPath, const bt_field_path, internal::FieldPathRefFuncs>;
 110     using Iterator = BorrowedObjectIterator<ConstFieldPath>;
 111
 112     enum class Scope
 113     {
 114         PACKET_CONTEXT = BT_FIELD_PATH_SCOPE_PACKET_CONTEXT,
 115         EVENT_COMMON_CONTEXT = BT_FIELD_PATH_SCOPE_EVENT_COMMON_CONTEXT,
 116         EVENT_SPECIFIC_CONTEXT = BT_FIELD_PATH_SCOPE_EVENT_SPECIFIC_CONTEXT,
 117         EVENT_PAYLOAD = BT_FIELD_PATH_SCOPE_EVENT_PAYLOAD,
 118     };
 119
 120     explicit ConstFieldPath(const LibObjPtr libObjPtr) noexcept : _ThisBorrowedObject {libObjPtr}
 121     {
 122     }
 123
 124     Scope rootScope() const noexcept
 125     {
 126         return static_cast<Scope>(bt_field_path_get_root_scope(this->libObjPtr()));
 127     }
 128
 129     std::uint64_t length() const noexcept
 130     {
 131         return bt_field_path_get_item_count(this->libObjPtr());
 132     }
 133
 134     ConstFieldPathItem operator[](const std::uint64_t index) const noexcept
 135     {
 136         return ConstFieldPathItem {
 137             bt_field_path_borrow_item_by_index_const(this->libObjPtr(), index)};
 138     }
 139
 140     Iterator begin() const noexcept
 141     {
 142         return Iterator {*this, 0};
 143     }
 144
 145     Iterator end() const noexcept
 146     {
 147         return Iterator {*this, this->length()};
 148     }
 149
 150     Shared shared() const noexcept
 151     {
 152         return Shared::createWithRef(*this);
 153     }
 154 };
 155
 156 } /* namespace bt2 */
 157
 158 #endif /* BABELTRACE_CPP_COMMON_BT2_FIELD_PATH_HPP */