Clean up CPDF_AnnotList.
[pdfium.git] / core / src / fpdfdoc / doc_bookmark.cpp
index 6cc954b..5ebd27e 100644 (file)
@@ -1,91 +1,91 @@
-// Copyright 2014 PDFium Authors. All rights reserved.\r
-// Use of this source code is governed by a BSD-style license that can be\r
-// found in the LICENSE file.\r
\r
-// Original code copyright 2014 Foxit Software Inc. http://www.foxitsoftware.com\r
-\r
-#include "../../include/fpdfdoc/fpdf_doc.h"\r
-CPDF_Bookmark CPDF_BookmarkTree::GetFirstChild(CPDF_Bookmark Parent)\r
-{\r
-    if (Parent.m_pDict == NULL) {\r
-        CPDF_Dictionary* pRoot = m_pDocument->GetRoot()->GetDict("Outlines");\r
-        if (pRoot == NULL) {\r
-            return NULL;\r
-        }\r
-        return pRoot->GetDict("First");\r
-    }\r
-    return Parent.m_pDict->GetDict("First");\r
-}\r
-CPDF_Bookmark CPDF_BookmarkTree::GetNextSibling(CPDF_Bookmark This)\r
-{\r
-    if (This.m_pDict == NULL) {\r
-        return NULL;\r
-    }\r
-    CPDF_Dictionary *pNext = This.m_pDict->GetDict("Next");\r
-    return pNext == This.m_pDict ? NULL : pNext;\r
-}\r
-FX_DWORD CPDF_Bookmark::GetColorRef()\r
-{\r
-    if (!m_pDict) {\r
-        return 0;\r
-    }\r
-    CPDF_Array* pColor = m_pDict->GetArray("C");\r
-    if (pColor == NULL) {\r
-        return FXSYS_RGB(0, 0, 0);\r
-    }\r
-    int r = FXSYS_round(pColor->GetNumber(0) * 255);\r
-    int g = FXSYS_round(pColor->GetNumber(1) * 255);\r
-    int b = FXSYS_round(pColor->GetNumber(2) * 255);\r
-    return FXSYS_RGB(r, g, b);\r
-}\r
-FX_DWORD CPDF_Bookmark::GetFontStyle()\r
-{\r
-    if (!m_pDict) {\r
-        return 0;\r
-    }\r
-    return m_pDict->GetInteger("F");\r
-}\r
-CFX_WideString CPDF_Bookmark::GetTitle()\r
-{\r
-    if (!m_pDict) {\r
-        return CFX_WideString();\r
-    }\r
-    CPDF_String* pString = (CPDF_String*)m_pDict->GetElementValue("Title");\r
-    if (pString == NULL || pString->GetType() != PDFOBJ_STRING) {\r
-        return CFX_WideString();\r
-    }\r
-    CFX_WideString title = pString->GetUnicodeText();\r
-    FX_LPWSTR buf = title.LockBuffer();\r
-    int len = title.GetLength(), i;\r
-    for (i = 0; i < len; i ++)\r
-        if (buf[i] < 0x20) {\r
-            buf[i] = 0x20;\r
-        }\r
-    title.ReleaseBuffer(len);\r
-    return title;\r
-}\r
-CPDF_Dest CPDF_Bookmark::GetDest(CPDF_Document* pDocument)\r
-{\r
-    if (!m_pDict) {\r
-        return NULL;\r
-    }\r
-    CPDF_Object* pDest = m_pDict->GetElementValue("Dest");\r
-    if (pDest == NULL) {\r
-        return NULL;\r
-    }\r
-    if (pDest->GetType() == PDFOBJ_STRING || pDest->GetType() == PDFOBJ_NAME) {\r
-        CPDF_NameTree name_tree(pDocument, FX_BSTRC("Dests"));\r
-        CFX_ByteStringC name = pDest->GetString();\r
-        return name_tree.LookupNamedDest(pDocument, name);\r
-    } else if (pDest->GetType() == PDFOBJ_ARRAY) {\r
-        return (CPDF_Array*)pDest;\r
-    }\r
-    return NULL;\r
-}\r
-CPDF_Action CPDF_Bookmark::GetAction()\r
-{\r
-    if (!m_pDict) {\r
-        return NULL;\r
-    }\r
-    return m_pDict->GetDict("A");\r
-}\r
+// Copyright 2014 PDFium Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+// Original code copyright 2014 Foxit Software Inc. http://www.foxitsoftware.com
+
+#include <vector>
+
+#include "../../../third_party/base/nonstd_unique_ptr.h"
+#include "../../include/fpdfdoc/fpdf_doc.h"
+
+CPDF_Bookmark CPDF_BookmarkTree::GetFirstChild(
+    const CPDF_Bookmark& parent) const {
+  if (!parent.m_pDict) {
+    CPDF_Dictionary* pRoot = m_pDocument->GetRoot()->GetDict("Outlines");
+    if (!pRoot) {
+      return CPDF_Bookmark();
+    }
+    return CPDF_Bookmark(pRoot->GetDict("First"));
+  }
+  return CPDF_Bookmark(parent.m_pDict->GetDict("First"));
+}
+CPDF_Bookmark CPDF_BookmarkTree::GetNextSibling(
+    const CPDF_Bookmark& bookmark) const {
+  if (!bookmark.m_pDict) {
+    return CPDF_Bookmark();
+  }
+  CPDF_Dictionary* pNext = bookmark.m_pDict->GetDict("Next");
+  return pNext == bookmark.m_pDict ? CPDF_Bookmark() : CPDF_Bookmark(pNext);
+}
+FX_DWORD CPDF_Bookmark::GetColorRef() const {
+  if (!m_pDict) {
+    return 0;
+  }
+  CPDF_Array* pColor = m_pDict->GetArray("C");
+  if (!pColor) {
+    return FXSYS_RGB(0, 0, 0);
+  }
+  int r = FXSYS_round(pColor->GetNumber(0) * 255);
+  int g = FXSYS_round(pColor->GetNumber(1) * 255);
+  int b = FXSYS_round(pColor->GetNumber(2) * 255);
+  return FXSYS_RGB(r, g, b);
+}
+FX_DWORD CPDF_Bookmark::GetFontStyle() const {
+  if (!m_pDict) {
+    return 0;
+  }
+  return m_pDict->GetInteger("F");
+}
+CFX_WideString CPDF_Bookmark::GetTitle() const {
+  if (!m_pDict) {
+    return CFX_WideString();
+  }
+  CPDF_String* pString = ToString(m_pDict->GetElementValue("Title"));
+  if (!pString)
+    return CFX_WideString();
+
+  CFX_WideString title = pString->GetUnicodeText();
+  int len = title.GetLength();
+  if (!len) {
+    return CFX_WideString();
+  }
+  nonstd::unique_ptr<FX_WCHAR[]> buf(new FX_WCHAR[len]);
+  for (int i = 0; i < len; i++) {
+    FX_WCHAR w = title[i];
+    buf[i] = w > 0x20 ? w : 0x20;
+  }
+  return CFX_WideString(buf.get(), len);
+}
+CPDF_Dest CPDF_Bookmark::GetDest(CPDF_Document* pDocument) const {
+  if (!m_pDict)
+    return CPDF_Dest();
+
+  CPDF_Object* pDest = m_pDict->GetElementValue("Dest");
+  if (!pDest)
+    return CPDF_Dest();
+  if (pDest->IsString() || pDest->IsName()) {
+    CPDF_NameTree name_tree(pDocument, FX_BSTRC("Dests"));
+    CFX_ByteStringC name = pDest->GetString();
+    return CPDF_Dest(name_tree.LookupNamedDest(pDocument, name));
+  }
+  if (CPDF_Array* pArray = pDest->AsArray())
+    return CPDF_Dest(pArray);
+  return CPDF_Dest();
+}
+CPDF_Action CPDF_Bookmark::GetAction() const {
+  if (!m_pDict) {
+    return CPDF_Action();
+  }
+  return CPDF_Action(m_pDict->GetDict("A"));
+}