ObjectDerivatives.cpp 6.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173
  1. /*
  2. * Copyright (c) 2021-2022, Matthew Olsson <mattco@serenityos.org>
  3. *
  4. * SPDX-License-Identifier: BSD-2-Clause
  5. */
  6. #include <AK/Hex.h>
  7. #include <LibPDF/Document.h>
  8. #include <LibPDF/ObjectDerivatives.h>
  9. namespace PDF {
  10. PDFErrorOr<NonnullRefPtr<Object>> ArrayObject::get_object_at(Document* document, size_t index) const
  11. {
  12. return document->resolve_to<Object>(at(index));
  13. }
  14. PDFErrorOr<NonnullRefPtr<Object>> DictObject::get_object(Document* document, DeprecatedFlyString const& key) const
  15. {
  16. return document->resolve_to<Object>(get_value(key));
  17. }
  18. #define DEFINE_ACCESSORS(class_name, snake_name) \
  19. PDFErrorOr<NonnullRefPtr<class_name>> ArrayObject::get_##snake_name##_at(Document* document, size_t index) const \
  20. { \
  21. if (index >= m_elements.size()) \
  22. return Error { Error::Type::Internal, "Out of bounds array access" }; \
  23. return document->resolve_to<class_name>(m_elements[index]); \
  24. } \
  25. \
  26. NonnullRefPtr<class_name> ArrayObject::get_##snake_name##_at(size_t index) const \
  27. { \
  28. VERIFY(index < m_elements.size()); \
  29. return cast_to<class_name>(m_elements[index]); \
  30. } \
  31. \
  32. PDFErrorOr<NonnullRefPtr<class_name>> DictObject::get_##snake_name(Document* document, DeprecatedFlyString const& key) const \
  33. { \
  34. return document->resolve_to<class_name>(get_value(key)); \
  35. } \
  36. \
  37. NonnullRefPtr<class_name> DictObject::get_##snake_name(DeprecatedFlyString const& key) const \
  38. { \
  39. return cast_to<class_name>(get_value(key)); \
  40. }
  41. ENUMERATE_OBJECT_TYPES(DEFINE_ACCESSORS)
  42. #undef DEFINE_INDEXER
  43. static void append_indent(StringBuilder& builder, int indent)
  44. {
  45. for (int i = 0; i < indent; i++)
  46. builder.append(" "sv);
  47. }
  48. DeprecatedString StringObject::to_deprecated_string(int) const
  49. {
  50. if (is_binary())
  51. return DeprecatedString::formatted("<{}>", encode_hex(string().bytes()).to_uppercase());
  52. return DeprecatedString::formatted("({})", string());
  53. }
  54. DeprecatedString NameObject::to_deprecated_string(int) const
  55. {
  56. StringBuilder builder;
  57. builder.appendff("/{}", this->name());
  58. return builder.to_deprecated_string();
  59. }
  60. Vector<float> ArrayObject::float_elements() const
  61. {
  62. Vector<float> values;
  63. values.ensure_capacity(m_elements.size());
  64. for (auto const& value : m_elements) {
  65. values.append(value.to_float());
  66. }
  67. return values;
  68. }
  69. DeprecatedString ArrayObject::to_deprecated_string(int indent) const
  70. {
  71. StringBuilder builder;
  72. builder.append("[\n"sv);
  73. bool first = true;
  74. for (auto& element : elements()) {
  75. if (!first)
  76. builder.append("\n"sv);
  77. first = false;
  78. append_indent(builder, indent + 1);
  79. builder.appendff("{}", element.to_deprecated_string(indent));
  80. }
  81. builder.append('\n');
  82. append_indent(builder, indent);
  83. builder.append(']');
  84. return builder.to_deprecated_string();
  85. }
  86. DeprecatedString DictObject::to_deprecated_string(int indent) const
  87. {
  88. StringBuilder builder;
  89. append_indent(builder, indent);
  90. builder.append("<<\n"sv);
  91. bool first = true;
  92. for (auto& [key, value] : map()) {
  93. if (!first)
  94. builder.append("\n"sv);
  95. first = false;
  96. append_indent(builder, indent + 1);
  97. builder.appendff("/{} ", key);
  98. builder.appendff("{}", value.to_deprecated_string(indent + 1));
  99. }
  100. builder.append('\n');
  101. append_indent(builder, indent);
  102. builder.append(">>"sv);
  103. return builder.to_deprecated_string();
  104. }
  105. DeprecatedString StreamObject::to_deprecated_string(int indent) const
  106. {
  107. StringBuilder builder;
  108. builder.appendff("{}\n", dict()->to_deprecated_string(indent));
  109. builder.append("stream\n"sv);
  110. size_t ascii_count = 0;
  111. for (auto c : bytes()) {
  112. if (c < 128)
  113. ++ascii_count;
  114. }
  115. size_t percentage_ascii = 100;
  116. if (bytes().size())
  117. percentage_ascii = ascii_count * 100 / bytes().size();
  118. bool is_mostly_text = percentage_ascii > 95;
  119. if (is_mostly_text) {
  120. for (auto c : bytes()) {
  121. if (c < 128)
  122. builder.append(c);
  123. else
  124. builder.appendff("\\{:03o}", c);
  125. }
  126. } else {
  127. auto string = encode_hex(bytes());
  128. while (string.length() > 60) {
  129. builder.appendff("{}\n", string.substring(0, 60));
  130. append_indent(builder, indent);
  131. string = string.substring(60);
  132. }
  133. builder.appendff("{}\n", string);
  134. }
  135. builder.append("endstream"sv);
  136. return builder.to_deprecated_string();
  137. }
  138. DeprecatedString IndirectValue::to_deprecated_string(int indent) const
  139. {
  140. StringBuilder builder;
  141. builder.appendff("{} {} obj\n", index(), generation_index());
  142. append_indent(builder, indent + 1);
  143. builder.append(value().to_deprecated_string(indent + 1));
  144. builder.append('\n');
  145. append_indent(builder, indent);
  146. builder.append("endobj"sv);
  147. return builder.to_deprecated_string();
  148. }
  149. }