/*

 * Copyright (c) 2004

 * Francois Dumont

 *

 * This material is provided "as is", with absolutely no warranty expressed

 * or implied. Any use is at your own risk.

 *

 * Permission to use or copy this software for any purpose is hereby granted

 * without fee, provided the above notices are retained on all copies.

 * Permission to modify the code and to distribute modified code is granted,

 * provided the above notices are retained, and a notice that the code was

 * modified is included with the above copyright notice.

 *

 */

 /*

  * This is an internal string for the STLport own iostream implementation.

  * The only diference rely on the allocator used to instanciate the basic_string.

  * Its goals is to improve performance limitating the number of dynamic allocation

  * that could occur when requesting a big float ouput for instance. This allocator

  * is not standard conformant as it has an internal state (the static buffer)

  */

#ifndef _STLP_INTERNAL_IOSTREAM_STRING_H

#define _STLP_INTERNAL_IOSTREAM_STRING_H

#ifndef _STLP_INTERNAL_ALLOC_H

#  include <stl/_alloc.h>

#endif /* _STLP_INTERNAL_ALLOC_H */

#ifndef _STLP_INTERNAL_STRING_H

#  include <stl/_string.h>

#endif /* _STLP_INTERNAL_STRING_H */

_STLP_BEGIN_NAMESPACE

_STLP_MOVE_TO_PRIV_NAMESPACE

template <class _CharT>

class __iostring_allocator : public allocator<_CharT> {

public:

  enum { _STR_SIZE = 256 };

private:

  enum { _BUF_SIZE = _STR_SIZE + 1 };

  typedef allocator<_CharT> _Base;

  _CharT _M_static_buf[_BUF_SIZE];

public:

  typedef typename _Base::size_type size_type;

  typedef typename _Base::pointer pointer;

#if defined (_STLP_MEMBER_TEMPLATE_CLASSES)

  template <class _Tp1> struct rebind {

#  if !defined (_STLP_MSVC) || (_STLP_MSVC >= 1300)

    typedef __iostring_allocator<_Tp1> other;

#  else

    typedef _STLP_PRIV __iostring_allocator<_Tp1> other;

#  endif

  };

#endif

  _CharT* allocate(size_type __n, const void* __ptr = 0) {

    if (__n > _BUF_SIZE) {

      return _Base::allocate(__n, __ptr);

    }

    return _M_static_buf;

  }

  void deallocate(pointer __p, size_type __n) {

    if (__p != _M_static_buf) _Base::deallocate(__p, __n);

  }

};

#if defined (_STLP_DONT_SUPPORT_REBIND_MEMBER_TEMPLATE) || !defined (_STLP_MEMBER_TEMPLATES)

/*

 * As the __iostring_allocator allocator will only be used in the basic_string implementation

 * we known that it is never going to be bound to another type that the one used to instantiate

 * the basic_string. This is why the associated __stl_alloc_rebind has only one template

 * parameter.

 */

_STLP_MOVE_TO_STD_NAMESPACE

template <class _Tp>

inline _STLP_PRIV __iostring_allocator<_Tp>& _STLP_CALL

__stl_alloc_rebind(_STLP_PRIV __iostring_allocator<_Tp>& __a, const _Tp*)

{ return __a; }

template <class _Tp>

inline _STLP_PRIV __iostring_allocator<_Tp> _STLP_CALL

__stl_alloc_create(const _STLP_PRIV __iostring_allocator<_Tp>&, const _Tp*)

{ return _STLP_PRIV __iostring_allocator<_Tp>(); }

_STLP_MOVE_TO_PRIV_NAMESPACE

#endif /* _STLP_DONT_SUPPORT_REBIND_MEMBER_TEMPLATE */

#if !defined (_STLP_DEBUG)

template <class _CharT>

struct __basic_iostring : public basic_string<_CharT, char_traits<_CharT>, __iostring_allocator<_CharT> > {

  /*

   * A consequence of the non standard conformant allocator is that a string using it

   * must always be presized to the allocator static buffer size because the basic_string implementation

   * do not manage an allocator returning always the same memory adress as long as the

   * requested memory block size is under a certain value.

   */

  typedef __basic_iostring<_CharT> _Self;

  typedef basic_string<_CharT, char_traits<_CharT>, __iostring_allocator<_CharT> > _Base;

  typedef typename _Base::_Reserve_t _Reserve_t;

  __basic_iostring() : _Base(_Reserve_t(), __iostring_allocator<_CharT>::_STR_SIZE)

  {}

  _Self& operator=(const _CharT* __s) {

    _Base::operator=(__s);

    return *this;

  }

};

typedef __basic_iostring<char> __iostring;

#  if !defined (_STLP_NO_WCHAR_T)

typedef __basic_iostring<wchar_t> __iowstring;

#  endif

#  define _STLP_BASIC_IOSTRING(_CharT) _STLP_PRIV __basic_iostring<_CharT>

#else

typedef string __iostring;

#  if !defined (_STLP_NO_WCHAR_T)

typedef wstring __iowstring;

#  endif

#  define _STLP_BASIC_IOSTRING(_CharT) basic_string<_CharT, char_traits<_CharT>, allocator<_CharT> >

#endif

_STLP_MOVE_TO_STD_NAMESPACE

_STLP_END_NAMESPACE

#endif /* _STLP_INTERNAL_IOSTREAM_STRING_H */