Bu::FString now has insert and remove functions, yay!

2 files changed, 69 insertions, 0 deletions

diff --git a/src/fstring.h b/src/fstring.h
index 8cccd5c..8b512e2 100644
--- a/src/fstring.h
+++ b/src/fstring.h
@@ -226,6 +226,45 @@ namespace Bu
                         prependChunk( pNew );
                 }
 
+                void insert( long nPos, const chr *pData, long nLen )
+                {
+                        if( nLen <= 0 )
+                                return;
+                        flatten();
+                        if( nPos <= 0 )
+                        {
+                                prepend( pData, nLen );
+                        }
+                        else if( nPos >= nLength )
+                        {
+                                append( pData, nLen );
+                        }
+                        else
+                        {
+                                Chunk *p1 = newChunk( nPos );
+                                Chunk *p2 = newChunk( nLen );
+                                Chunk *p3 = newChunk( nLength-nPos );
+                                cpy( p1->pData, pFirst->pData, nPos );
+                                cpy( p2->pData, pData, nLen );
+                                cpy( p3->pData, pFirst->pData+nPos, nLength-nPos );
+                                clear();
+                                appendChunk( p1 );
+                                appendChunk( p2 );
+                                appendChunk( p3 );
+                        }
+                }
+
+                void remove( long nPos, long nLen )
+                {
+                        if( nLen <= 0 || nPos < 0 || nPos >= nLength )
+                                return;
+                        if( nLen >= nLength-nPos )
+                                nLen = nLength-nPos-1;
+                        flatten();
+                        cpy( pFirst->pData+nPos, pFirst->pData+nPos+nLen, nLen+1 );
+                        nLength -= nPos;
+                }
+
                 /**
                  *@todo void prepend( const chr &cData )
                  */
diff --git a/src/unit/fstring.cpp b/src/unit/fstring.cpp
index 69c7e0a..03df9c3 100644
--- a/src/unit/fstring.cpp
+++ b/src/unit/fstring.cpp
@@ -18,6 +18,8 @@ public:
                 addTest( Unit::compare2 );
                 addTest( Unit::appendSingle );
                 addTest( Unit::shared1 );
+                addTest( Unit::insert );
+                addTest( Unit::remove );
         }
 
         virtual ~Unit()
@@ -57,6 +59,34 @@ public:
                 a = b;
                 unitTest( a.getStr() == b.getStr() );
         }
+
+        void insert()
+        {
+                Bu::FString a("abcd");
+                a.insert( 2, "-!-", 3 );
+                unitTest( a == "ab-!-cd" );
+                
+                a.insert( 0, "!!", 2 );
+                unitTest( a == "!!ab-!-cd" );
+                
+                a.insert( -10, "789", 3 );
+                unitTest( a == "789!!ab-!-cd" );
+                
+                a.insert( 12, "89", 2 );
+                unitTest( a == "789!!ab-!-cd89" );
+                
+                a.insert( 1203, "12", 2 );
+                unitTest( a == "789!!ab-!-cd8912" );
+        }
+
+        void remove()
+        {
+                Bu::FString a("abHEYcd");
+                a.remove( 2, 3 );
+                unitTest( a == "abcd" );
+                a.remove( 2, 5 );
+                unitTest( a == "ab" );
+        }
 };
 
 int main( int argc, char *argv[] )