michael@0: /* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 4 -*-
michael@0:  * vim: set ts=8 sts=4 et sw=4 tw=99:
michael@0:  * This Source Code Form is subject to the terms of the Mozilla Public
michael@0:  * License, v. 2.0. If a copy of the MPL was not distributed with this
michael@0:  * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
michael@0: 
michael@0: #include "vm/StringBuffer.h"
michael@0: 
michael@0: #include "jsobjinlines.h"
michael@0: 
michael@0: #include "vm/String-inl.h"
michael@0: 
michael@0: using namespace js;
michael@0: 
michael@0: jschar *
michael@0: StringBuffer::extractWellSized()
michael@0: {
michael@0:     size_t capacity = cb.capacity();
michael@0:     size_t length = cb.length();
michael@0: 
michael@0:     jschar *buf = cb.extractRawBuffer();
michael@0:     if (!buf)
michael@0:         return nullptr;
michael@0: 
michael@0:     /* For medium/big buffers, avoid wasting more than 1/4 of the memory. */
michael@0:     JS_ASSERT(capacity >= length);
michael@0:     if (length > CharBuffer::sMaxInlineStorage && capacity - length > length / 4) {
michael@0:         size_t bytes = sizeof(jschar) * (length + 1);
michael@0:         ExclusiveContext *cx = context();
michael@0:         jschar *tmp = (jschar *)cx->realloc_(buf, bytes);
michael@0:         if (!tmp) {
michael@0:             js_free(buf);
michael@0:             return nullptr;
michael@0:         }
michael@0:         buf = tmp;
michael@0:     }
michael@0: 
michael@0:     return buf;
michael@0: }
michael@0: 
michael@0: JSFlatString *
michael@0: StringBuffer::finishString()
michael@0: {
michael@0:     ExclusiveContext *cx = context();
michael@0:     if (cb.empty())
michael@0:         return cx->names().empty;
michael@0: 
michael@0:     size_t length = cb.length();
michael@0:     if (!JSString::validateLength(cx, length))
michael@0:         return nullptr;
michael@0: 
michael@0:     JS_STATIC_ASSERT(JSFatInlineString::MAX_FAT_INLINE_LENGTH < CharBuffer::InlineLength);
michael@0:     if (JSFatInlineString::lengthFits(length))
michael@0:         return NewFatInlineString<CanGC>(cx, TwoByteChars(cb.begin(), length));
michael@0: 
michael@0:     if (!cb.append('\0'))
michael@0:         return nullptr;
michael@0: 
michael@0:     jschar *buf = extractWellSized();
michael@0:     if (!buf)
michael@0:         return nullptr;
michael@0: 
michael@0:     JSFlatString *str = js_NewString<CanGC>(cx, buf, length);
michael@0:     if (!str)
michael@0:         js_free(buf);
michael@0:     return str;
michael@0: }
michael@0: 
michael@0: JSAtom *
michael@0: StringBuffer::finishAtom()
michael@0: {
michael@0:     ExclusiveContext *cx = context();
michael@0: 
michael@0:     size_t length = cb.length();
michael@0:     if (length == 0)
michael@0:         return cx->names().empty;
michael@0: 
michael@0:     JSAtom *atom = AtomizeChars(cx, cb.begin(), length);
michael@0:     cb.clear();
michael@0:     return atom;
michael@0: }
michael@0: 
michael@0: bool
michael@0: js::ValueToStringBufferSlow(JSContext *cx, const Value &arg, StringBuffer &sb)
michael@0: {
michael@0:     RootedValue v(cx, arg);
michael@0:     if (!ToPrimitive(cx, JSTYPE_STRING, &v))
michael@0:         return false;
michael@0: 
michael@0:     if (v.isString())
michael@0:         return sb.append(v.toString());
michael@0:     if (v.isNumber())
michael@0:         return NumberValueToStringBuffer(cx, v, sb);
michael@0:     if (v.isBoolean())
michael@0:         return BooleanToStringBuffer(v.toBoolean(), sb);
michael@0:     if (v.isNull())
michael@0:         return sb.append(cx->names().null);
michael@0:     JS_ASSERT(v.isUndefined());
michael@0:     return sb.append(cx->names().undefined);
michael@0: }