upb_generator/file_layout.h

   1 // Protocol Buffers - Google's data interchange format
   2 // Copyright 2023 Google LLC.  All rights reserved.
   3 //
   4 // Use of this source code is governed by a BSD-style
   5 // license that can be found in the LICENSE file or at
   6 // https://developers.google.com/open-source/licenses/bsd
   7
   8 #ifndef UPB_GENERATOR_FILE_LAYOUT_H
   9 #define UPB_GENERATOR_FILE_LAYOUT_H
  10
  11 #include <string>
  12 #include <vector>
  13
  14 #include "absl/container/flat_hash_map.h"
  15 #include "upb/base/status.hpp"
  16 #include "upb/mini_descriptor/decode.h"
  17 #include "upb/reflection/def.h"
  18 #include "upb/reflection/def.hpp"
  19 #include "upb/reflection/descriptor_bootstrap.h"
  20
  21 // Must be last
  22 #include "upb/port/def.inc"
  23
  24 namespace upb {
  25 namespace generator {
  26
  27 enum WhichEnums {
  28   kAllEnums = 0,
  29   kClosedEnums = 1,
  30 };
  31
  32 std::vector<upb::EnumDefPtr> SortedEnums(upb::FileDefPtr file,
  33                                          WhichEnums which);
  34
  35 // Ordering must match upb/def.c!
  36 //
  37 // The ordering is significant because each upb_MessageDef* will point at the
  38 // corresponding upb_MiniTable and we just iterate through the list without
  39 // any search or lookup.
  40 std::vector<upb::MessageDefPtr> SortedMessages(upb::FileDefPtr file);
  41
  42 // Ordering must match upb/def.c!
  43 //
  44 // The ordering is significant because each upb_FieldDef* will point at the
  45 // corresponding upb_MiniTableExtension and we just iterate through the list
  46 // without any search or lookup.
  47 std::vector<upb::FieldDefPtr> SortedExtensions(upb::FileDefPtr file);
  48
  49 std::vector<upb::FieldDefPtr> FieldNumberOrder(upb::MessageDefPtr message);
  50
  51 // DefPoolPair is a pair of DefPools: one for 32-bit and one for 64-bit.
  52 class DefPoolPair {
  53  public:
  54   DefPoolPair() {
  55     pool32_._SetPlatform(kUpb_MiniTablePlatform_32Bit);
  56     pool64_._SetPlatform(kUpb_MiniTablePlatform_64Bit);
  57   }
  58
  59   upb::FileDefPtr AddFile(const UPB_DESC(FileDescriptorProto) * file_proto,
  60                           upb::Status* status) {
  61     upb::FileDefPtr file32 = pool32_.AddFile(file_proto, status);
  62     upb::FileDefPtr file64 = pool64_.AddFile(file_proto, status);
  63     if (!file32) return file32;
  64     return file64;
  65   }
  66
  67   const upb_MiniTable* GetMiniTable32(upb::MessageDefPtr m) const {
  68     return pool32_.FindMessageByName(m.full_name()).mini_table();
  69   }
  70
  71   const upb_MiniTable* GetMiniTable64(upb::MessageDefPtr m) const {
  72     return pool64_.FindMessageByName(m.full_name()).mini_table();
  73   }
  74
  75   const upb_MiniTableField* GetField32(upb::FieldDefPtr f) const {
  76     return GetFieldFromPool(&pool32_, f);
  77   }
  78
  79   const upb_MiniTableField* GetField64(upb::FieldDefPtr f) const {
  80     return GetFieldFromPool(&pool64_, f);
  81   }
  82
  83  private:
  84   static const upb_MiniTableField* GetFieldFromPool(const upb::DefPool* pool,
  85                                                     upb::FieldDefPtr f) {
  86     if (f.is_extension()) {
  87       return pool->FindExtensionByName(f.full_name()).mini_table();
  88     } else {
  89       return pool->FindMessageByName(f.containing_type().full_name())
  90           .FindFieldByNumber(f.number())
  91           .mini_table();
  92     }
  93   }
  94
  95   upb::DefPool pool32_;
  96   upb::DefPool pool64_;
  97 };
  98
  99 }  // namespace generator
 100 }  // namespace upb
 101
 102 #include "upb/port/undef.inc"
 103
 104 #endif  // UPB_GENERATOR_FILE_LAYOUT_H