Merge to XFA: Make CFX_WideString::LockBuffer() completely unused.
[pdfium.git] / core / src / fpdfdoc / doc_bookmark.cpp
index 0ae649e..6ba98e6 100644 (file)
@@ -4,33 +4,37 @@
  
 // 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(CPDF_Bookmark Parent)
+
+CPDF_Bookmark CPDF_BookmarkTree::GetFirstChild(const CPDF_Bookmark& parent) const
 {
-    if (Parent.m_pDict == NULL) {
+    if (!parent.m_pDict) {
         CPDF_Dictionary* pRoot = m_pDocument->GetRoot()->GetDict("Outlines");
-        if (pRoot == NULL) {
-            return NULL;
+        if (!pRoot) {
+            return CPDF_Bookmark();
         }
-        return pRoot->GetDict("First");
+        return CPDF_Bookmark(pRoot->GetDict("First"));
     }
-    return Parent.m_pDict->GetDict("First");
+    return CPDF_Bookmark(parent.m_pDict->GetDict("First"));
 }
-CPDF_Bookmark CPDF_BookmarkTree::GetNextSibling(CPDF_Bookmark This)
+CPDF_Bookmark CPDF_BookmarkTree::GetNextSibling(const CPDF_Bookmark& bookmark) const
 {
-    if (This.m_pDict == NULL) {
-        return NULL;
+    if (!bookmark.m_pDict) {
+        return CPDF_Bookmark();
     }
-    CPDF_Dictionary *pNext = This.m_pDict->GetDict("Next");
-    return pNext == This.m_pDict ? NULL : pNext;
+    CPDF_Dictionary *pNext = bookmark.m_pDict->GetDict("Next");
+    return pNext == bookmark.m_pDict ? CPDF_Bookmark() : CPDF_Bookmark(pNext);
 }
-FX_DWORD CPDF_Bookmark::GetColorRef()
+FX_DWORD CPDF_Bookmark::GetColorRef() const
 {
     if (!m_pDict) {
         return 0;
     }
     CPDF_Array* pColor = m_pDict->GetArray("C");
-    if (pColor == NULL) {
+    if (!pColor) {
         return FXSYS_RGB(0, 0, 0);
     }
     int r = FXSYS_round(pColor->GetNumber(0) * 255);
@@ -38,54 +42,59 @@ FX_DWORD CPDF_Bookmark::GetColorRef()
     int b = FXSYS_round(pColor->GetNumber(2) * 255);
     return FXSYS_RGB(r, g, b);
 }
-FX_DWORD CPDF_Bookmark::GetFontStyle()
+FX_DWORD CPDF_Bookmark::GetFontStyle() const
 {
     if (!m_pDict) {
         return 0;
     }
     return m_pDict->GetInteger("F");
 }
-CFX_WideString CPDF_Bookmark::GetTitle()
+CFX_WideString CPDF_Bookmark::GetTitle() const
 {
     if (!m_pDict) {
         return CFX_WideString();
     }
     CPDF_String* pString = (CPDF_String*)m_pDict->GetElementValue("Title");
-    if (pString == NULL || pString->GetType() != PDFOBJ_STRING) {
+    if (!pString || pString->GetType() != PDFOBJ_STRING) {
         return CFX_WideString();
     }
     CFX_WideString title = pString->GetUnicodeText();
-    FX_LPWSTR buf = title.LockBuffer();
-    int len = title.GetLength(), i;
-    for (i = 0; i < len; i ++)
-        if (buf[i] < 0x20) {
-            buf[i] = 0x20;
-        }
-    title.ReleaseBuffer(len);
-    return title;
+    int len = title.GetLength();
+    if (!len) {
+        return CFX_WideString();
+    }
+    nonstd::unique_ptr<std::vector<FX_WCHAR> > vec;
+    vec.reset(new std::vector<FX_WCHAR>(len));
+    FX_WCHAR* buf = &vec->front();
+    for (int i = 0; i < len; i++) {
+        FX_WCHAR w = title[i];
+        buf[i] = w > 0x20 ? w : 0x20;
+    }
+    return CFX_WideString(buf, len);
 }
-CPDF_Dest CPDF_Bookmark::GetDest(CPDF_Document* pDocument)
+CPDF_Dest CPDF_Bookmark::GetDest(CPDF_Document* pDocument) const
 {
     if (!m_pDict) {
-        return NULL;
+        return CPDF_Dest();
     }
     CPDF_Object* pDest = m_pDict->GetElementValue("Dest");
-    if (pDest == NULL) {
-        return NULL;
+    if (!pDest) {
+        return CPDF_Dest();
     }
     if (pDest->GetType() == PDFOBJ_STRING || pDest->GetType() == PDFOBJ_NAME) {
         CPDF_NameTree name_tree(pDocument, FX_BSTRC("Dests"));
         CFX_ByteStringC name = pDest->GetString();
-        return name_tree.LookupNamedDest(pDocument, name);
-    } else if (pDest->GetType() == PDFOBJ_ARRAY) {
-        return (CPDF_Array*)pDest;
+        return CPDF_Dest(name_tree.LookupNamedDest(pDocument, name));
+    }
+    if (pDest->GetType() == PDFOBJ_ARRAY) {
+        return CPDF_Dest((CPDF_Array*)pDest);
     }
-    return NULL;
+    return CPDF_Dest();
 }
-CPDF_Action CPDF_Bookmark::GetAction()
+CPDF_Action CPDF_Bookmark::GetAction() const
 {
     if (!m_pDict) {
-        return NULL;
+        return CPDF_Action();
     }
-    return m_pDict->GetDict("A");
+    return CPDF_Action(m_pDict->GetDict("A"));
 }