gecko-dev/xpcom/ds/nsAWritableString.h

/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
/*
 * The contents of this file are subject to the Netscape Public
 * License Version 1.1 (the "License"); you may not use this file
 * except in compliance with the License. You may obtain a copy of
 * the License at http://www.mozilla.org/NPL/
 *
 * Software distributed under the License is distributed on an "AS
 * IS" basis, WITHOUT WARRANTY OF ANY KIND, either express or
 * implied. See the License for the specific language governing
 * rights and limitations under the License.
 *
 * The Original Code is mozilla.org code.
 *
 * The Initial Developer of the Original Code is Netscape
 * Communications Corporation.  Portions created by Netscape are
 * Copyright (C) 1998 Netscape Communications Corporation. All
 * Rights Reserved.
 *
 * Contributor(s):
 *   Scott Collins <scc@netscape.com>
 */

#ifndef _nsAWritableString_h__
#define _nsAWritableString_h__

  // WORK IN PROGRESS

  // See also...
#include "nsAReadableString.h"


/*
  This file defines the abstract interfaces |nsAWritableString| and
  |nsAWritableCString|.

  |nsAWritableString| is a string of |PRUnichar|s.  |nsAWritableCString| (note the
  'C') is a string of |char|s.
*/

template <class CharT>
class basic_nsAWritableString
      : public basic_nsAReadableString<CharT>
    /*
      ...
    */
  {
    protected:
      typedef typename basic_nsAReadableString<CharT>::FragmentRequest  FragmentRequest;

      struct Fragment
        {
          CharT*    mStart;
          CharT*    mEnd;
          PRUint32  mFragmentIdentifier;

          Fragment()
              : mStart(0), mEnd(0), mFragmentIdentifier(0)
            {
              // nothing else to do here
            }
        };

    public:
      virtual CharT* GetFragment( Fragment&, FragmentRequest, PRUint32 = 0 ) = 0;

      friend class Iterator;
      class Iterator
          : public std::bidirectional_iterator_tag
        {
          public:
            typedef ptrdiff_t                   difference_type;
            typedef CharT                       value_type;
            typedef const CharT*                pointer;
            typedef const CharT&                reference;
            typedef bidirectional_iterator_tag  iterator_category;

          private:
            friend class basic_nsAWritableString<CharT>;

            Fragment  mFragment;
            CharT*    mPosition;
            basic_nsAWritableString<CharT>* mOwningString;

            void
            normalize_forward()
              {
                if ( mPosition == mFragment.mEnd )
                  if ( mOwningString->GetFragment(mFragment, kNextFragment) )
                    mPosition = mFragment.mStart;
              }

            void
            normalize_backward()
              {
                if ( mPosition == mFragment.mStart )
                  if ( mOwningString->GetFragment(mFragment, kPrevFragment) )
                    mPosition = mFragment.mEnd;
              }

            Iterator( Fragment& aFragment,
                      CharT* aStartingPosition,
                      basic_nsAWritableString<CharT>& aOwningString )
                : mFragment(aFragment),
                  mPosition(aStartingPosition),
                  mOwningString(&aOwningString)
              {
                // nothing else to do here
              }

          public:
            // Iterator( const Iterator& ); ...use default copy-constructor
            // Iterator& operator=( const Iterator& ); ...use default copy-assignment operator

            
            CharT&
            operator*()
              {
                return *mPosition;
              }

            Iterator&
            operator++()
              {
                ++mPosition;
                normalize_forward();
                return *this;
              }

            Iterator
            operator++( int )
              {
                Iterator result(*this);
                ++mPosition;
                normalize_forward();
                return result;
              }

            Iterator&
            operator--()
              {
                normalize_backward();
                --mPosition;
                return *this;
              }

            Iterator
            operator--( int )
              {
                Iterator result(*this);
                normalize_backward();
                --mPosition;
                return result;
              }

            PRBool
            operator==( const Iterator& rhs )
              {
                return mPosition == rhs.mPosition;
              }

            PRBool
            operator!=( const Iterator& rhs )
              {
                return mPosition != rhs.mPosition;
              }
        };

    public:

#ifdef HAVE_CPP_USING
      using basic_nsAReadableString<CharT>::Begin;
      using basic_nsAReadableString<CharT>::End;
#else
      basic_nsAReadableString<CharT>::ConstIterator
      Begin( PRUint32 aOffset = 0 ) const
        {
          return basic_nsAReadableString<CharT>::Begin(aOffset);
        }

      basic_nsAReadableString<CharT>::ConstIterator
      End( PRUint32 aOffset = 0 ) const
        {
          return basic_nsAReadableString<CharT>::End(aOffset);
        }
#endif

      Iterator
      Begin( PRUint32 aOffset = 0 )
        {
          Fragment fragment;
          CharT* startPos = GetFragment(fragment, kFragmentAt, aOffset);
          return Iterator(fragment, startPos, *this);
        }


      Iterator
      End( PRUint32 aOffset = 0 )
        {
          Fragment fragment;
          CharT* startPos = GetFragment(fragment, kFragmentAt, max(0U, Length()-aOffset));
          return Iterator(fragment, startPos, *this);
        }


      virtual void SetCapacity( PRUint32 ) = 0;
      virtual void SetLength( PRUint32 ) = 0;


        // ...a synonym for |SetLength()|
      void
      Truncate( PRUint32 aNewLength=0 )
        {
          SetLength(aNewLength);
        }


      // PRBool SetCharAt( char_type, index_type ) = 0;



      // void ToLowerCase();
      // void ToUpperCase();

      // void StripChars( const CharT* aSet );
      // void StripChar( ... );
      // void StripWhitespace();
      // void ReplaceChar( ... );
      // void ReplaceSubstring( ... );
      // void Trim( ... );
      // void CompressSet( ... );
      // void CompareWhitespace( ... );



      virtual void Assign( const basic_nsAReadableString<CharT>& rhs );
      // virtual void AssignChar( CharT ) = 0;

      virtual void Append( const basic_nsAReadableString<CharT>& );
      virtual void AppendChar( CharT );

      virtual void Insert( const basic_nsAReadableString<CharT>&, PRUint32 atPosition );
      // virtual void InsertChar( CharT, PRUint32 atPosition ) = 0;

      virtual void Cut( PRUint32 cutStart, PRUint32 cutLength );

      virtual void Replace( PRUint32 cutStart, PRUint32 cutLength, const basic_nsAReadableString<CharT>& );


      basic_nsAWritableString<CharT>&
      operator+=( const basic_nsAReadableString<CharT>& rhs )
        {
          Append(rhs);
          return *this;
        }

      basic_nsAWritableString<CharT>&
      operator+=( const basic_nsLiteralString<CharT>& rhs )
        {
          Append(rhs);
          return *this;
        }

      basic_nsAWritableString<CharT>&
      operator=( const basic_nsAReadableString<CharT>& rhs )
        {
          Assign(rhs);
          return *this;
        }

      basic_nsAWritableString<CharT>&
      operator=( const basic_nsLiteralString<CharT>& rhs )
        {
          Assign(rhs);
          return *this;
        }
  };

NS_DEF_STRING_COMPARISONS(basic_nsAWritableString<CharT>)

template <class CharT>
void
basic_nsAWritableString<CharT>::Assign( const basic_nsAReadableString<CharT>& rhs )
  {
    SetLength(rhs.Length());
    std::copy(rhs.Begin(), rhs.End(), Begin());
  }

template <class CharT>
void
basic_nsAWritableString<CharT>::Append( const basic_nsAReadableString<CharT>& rhs )
  {
    PRUint32 oldLength = Length();
    SetLength(oldLength + rhs.Length());
    std::copy(rhs.Begin(), rhs.End(), Begin(oldLength));
  }

template <class CharT>
void
basic_nsAWritableString<CharT>::AppendChar( CharT aChar )
  {
    SetLength(Length()+1);
    *End(1) = aChar;
  }

template <class CharT>
void
basic_nsAWritableString<CharT>::Insert( const basic_nsAReadableString<CharT>& aReadable, PRUint32 aPosition )
  {
    PRUint32 oldLength = Length();
    SetLength(oldLength + aReadable.Length());
    if ( aPosition < oldLength )
      std::copy_backward(Begin(aPosition), Begin(oldLength), End());
    else
      aPosition = oldLength;
    std::copy(aReadable.Begin(), aReadable.End(), Begin(aPosition));
  }

template <class CharT>
void
basic_nsAWritableString<CharT>::Cut( PRUint32 cutStart, PRUint32 cutLength )
  {
    std::copy(Begin(cutStart+cutLength), End(), Begin(cutStart));
    SetLength(Length()-cutLength);
  }

template <class CharT>
void
basic_nsAWritableString<CharT>::Replace( PRUint32 cutStart, PRUint32 cutLength, const basic_nsAReadableString<CharT>& aReplacement )
  {
    PRUint32 oldLength = Length();

    cutStart = min(cutStart, oldLength);
    cutLength = min(cutLength, oldLength-cutStart);
    PRUint32 cutEnd = cutStart + cutLength;

    PRUint32 replacementLength = aReplacement.Length();
    PRUint32 replacementEnd = cutStart + replacementLength;

    PRUint32 newLength = oldLength - cutLength + replacementLength;

    if ( cutLength > replacementLength )
      std::copy(Begin(cutEnd), End(), Begin(replacementEnd));
    SetLength(newLength);
    if ( cutLength < replacementLength )
      std::copy_backward(Begin(cutEnd), Begin(oldLength), Begin(replacementEnd));

    std::copy(aReplacement.Begin(), aReplacement.End(), Begin(cutStart));
  }

// operator>>
// getline (maybe)

typedef basic_nsAWritableString<PRUnichar> nsAWritableString;
typedef basic_nsAWritableString<char>      nsAWritableCString;

#endif // !defined(_nsAWritableString_h__)