ObjectDerivatives.cpp 6.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159
  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. builder.append("<<\n"sv);
  90. bool first = true;
  91. for (auto& [key, value] : map()) {
  92. if (!first)
  93. builder.append(",\n"sv);
  94. first = false;
  95. append_indent(builder, indent + 1);
  96. builder.appendff("/{} ", key);
  97. builder.appendff("{}", value.to_deprecated_string(indent + 1));
  98. }
  99. builder.append('\n');
  100. append_indent(builder, indent);
  101. builder.append(">>"sv);
  102. return builder.to_deprecated_string();
  103. }
  104. DeprecatedString StreamObject::to_deprecated_string(int indent) const
  105. {
  106. StringBuilder builder;
  107. builder.append("stream\n"sv);
  108. append_indent(builder, indent);
  109. builder.appendff("{}\n", dict()->to_deprecated_string(indent + 1));
  110. append_indent(builder, indent + 1);
  111. auto string = encode_hex(bytes());
  112. while (true) {
  113. if (string.length() > 60) {
  114. builder.appendff("{}\n", string.substring(0, 60));
  115. append_indent(builder, indent);
  116. string = string.substring(60);
  117. continue;
  118. }
  119. builder.appendff("{}\n", string);
  120. break;
  121. }
  122. append_indent(builder, indent);
  123. builder.append("endstream"sv);
  124. return builder.to_deprecated_string();
  125. }
  126. DeprecatedString IndirectValue::to_deprecated_string(int indent) const
  127. {
  128. StringBuilder builder;
  129. builder.appendff("{} {} obj\n", index(), generation_index());
  130. append_indent(builder, indent + 1);
  131. builder.append(value().to_deprecated_string(indent + 1));
  132. builder.append('\n');
  133. append_indent(builder, indent);
  134. builder.append("endobj"sv);
  135. return builder.to_deprecated_string();
  136. }
  137. }