Copying scripts from 2.4x without 2.5x changes
[blender.git] / intern / string / STR_String.h
1 /**
2  * $Id$
3  * ***** BEGIN GPL LICENSE BLOCK *****
4  *
5  * This program is free software; you can redistribute it and/or
6  * modify it under the terms of the GNU General Public License
7  * as published by the Free Software Foundation; either version 2
8  * of the License, or (at your option) any later version.
9  *
10  * This program is distributed in the hope that it will be useful,
11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
13  * GNU General Public License for more details.
14  *
15  * You should have received a copy of the GNU General Public License
16  * along with this program; if not, write to the Free Software Foundation,
17  * Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
18  *
19  * The Original Code is Copyright (C) 2001-2002 by NaN Holding BV.
20  * All rights reserved.
21  *
22  * The Original Code is: all of this file.
23  *
24  * Contributor(s): none yet.
25  *
26  * ***** END GPL LICENSE BLOCK *****
27  */
28
29 /**
30
31  * $Id$
32  * Copyright (C) 2001 NaN Technologies B.V.
33  * This file was formerly known as: GEN_StdString.h.
34  * @date        April, 25, 2001
35  */
36
37 #ifndef _STR_String_H_
38 #define _STR_String_H_
39
40 #ifndef STR_NO_ASSERTD
41 #undef  assertd
42 #define assertd(exp)                    ((void)NULL)
43 #endif
44
45 #include <vector>
46 #include <limits.h>
47
48 #include <cstring>
49 #include <cstdlib>
50
51 using namespace std;
52
53 #ifdef WITH_CXX_GUARDEDALLOC
54 #include "MEM_guardedalloc.h"
55 #endif
56
57 #ifdef _WIN32
58 #define stricmp _stricmp
59 #endif
60
61 class STR_String;
62
63 typedef  unsigned long dword;
64 typedef const STR_String& rcSTR_String;
65 typedef unsigned char byte;
66
67 /**
68  *      Smart String Value class. Is used by parser when an expression tree is build containing string.
69 */
70
71 class STR_String
72 {
73 public:
74         // Initialization
75         STR_String();
76         STR_String(char c);
77         STR_String(char c, int len);
78         STR_String(const char *str);
79         STR_String(const char *str, int len);
80         STR_String(const STR_String &str);
81         STR_String(const STR_String & str, int len);
82         STR_String(const char *src1, int src1_len, const char *src2, int src2_len);
83         explicit STR_String(int val);
84         explicit STR_String(dword val);
85         explicit STR_String(float val);
86         explicit STR_String(double val);
87         inline ~STR_String()                                                                                            { delete[] pData; }
88
89         // Operations
90         STR_String&                     Format(const char *fmt, ...);                           // Set formatted text to string
91         STR_String&                     FormatAdd(const char *fmt, ...);                        // Add formatted text to string
92         inline void                     Clear()                                                                         { Len = pData[0] = 0; }
93         inline const STR_String & Reverse()                                                                     
94         { 
95             for (int i1=0, i2=Len-1; i1<i2; i1++, i2--) 
96                 swap(pData[i1], pData[i2]); return *this; 
97         }
98
99         // Properties
100         bool                            IsUpper() const;
101         bool                            IsLower() const;
102         inline bool                     IsEmpty() const                                                         { return Len==0; }
103         inline int                      Length() const                                                          { return Len; }
104
105         // Data access
106         inline STR_String&      SetLength(int len)                                                      { AllocBuffer(len, true); Len=len; pData[len]=0; return *this; }
107         inline char                     GetAt(int pos) const                                            { assertd(pos<Len); return pData[pos]; }
108         inline void                     SetAt(int pos, char c)                                          { assertd(pos<Len); pData[pos]=c; }
109         inline void                     SetAt(int pos, rcSTR_String str);
110         inline void                     SetAt(int pos, int num, rcSTR_String str);
111         void                            Replace(int pos, rcSTR_String str);
112         void                            Replace(int pos, int num, rcSTR_String str);
113
114         // Substrings
115         inline STR_String       Left(int num) const                                                             { num = (num < Len ? num:Len ); return STR_String(pData, num); }
116         inline STR_String       Right(int num) const                                                    { num = (num < Len ? num:Len ); return STR_String(pData+Len-num, num); }
117         inline STR_String       Mid(int pos, int num = INT_MAX) const                   { pos = (pos < Len ? pos:Len ); num = (num < (Len - pos) ? num : (Len - pos)); return STR_String(pData+pos, num); }
118
119         // Comparison
120         int                             Compare(rcSTR_String rhs) const;
121         int                             CompareNoCase(rcSTR_String rhs) const;
122         inline bool             IsEqual(rcSTR_String rhs) const                                 { return (Compare(rhs)==0); }
123         inline bool             IsEqualNoCase(rcSTR_String rhs) const                           { return (CompareNoCase(rhs)==0); }
124
125         // Search/replace
126         int                             Find(char c, int pos = 0) const;
127         int                             Find(const char *str, int pos = 0) const;
128         int                             Find(rcSTR_String str, int pos = 0) const;
129         int                             RFind(char c) const;
130         int                             FindOneOf(const char *set, int pos = 0) const;
131         int                             RFindOneOf(const char *set, int pos = 0) const;
132
133         vector<STR_String>      Explode(char c) const;
134
135         // Formatting
136         STR_String&                     Upper();
137         STR_String&                     Lower();
138         STR_String&                     Capitalize();
139         STR_String&                     TrimLeft();
140         STR_String&                     TrimLeft(char *set);
141         STR_String&                     TrimRight();
142         STR_String&                     TrimRight(char *set);
143         STR_String&                     Trim();
144         STR_String&                     Trim(char *set);
145         STR_String&                     TrimQuotes();
146
147         // Conversions
148 //      inline operator char*()                                                                                         { return pData; }
149         inline operator const char *() const                                                            { return pData; }
150         inline char *Ptr()                                                                                                      { return pData; }
151         inline const char *ReadPtr() const                                                                      { return pData; }
152         inline float    ToFloat() const                                                                         { float x=(float)(atof(pData)); return x; }
153         inline int              ToInt() const                                                                           { return atoi(pData); }
154
155         // Operators
156         inline rcSTR_String     operator=(const byte *rhs)                                              { return Copy((const char *)rhs, strlen((const char *)rhs)); }
157         inline rcSTR_String     operator=(rcSTR_String rhs)                                             { return Copy(rhs.ReadPtr(), rhs.Length()); }
158         inline rcSTR_String     operator=(char rhs)                                                             { return Copy(&rhs, 1); }
159         inline rcSTR_String     operator=(const char *rhs)                                              { return Copy(rhs, strlen(rhs)); }
160
161         inline rcSTR_String     operator+=(const char *rhs)                                             { return Concat(rhs, strlen(rhs)); }
162         inline rcSTR_String     operator+=(rcSTR_String rhs)                                            { return Concat(rhs.ReadPtr(), rhs.Length()); }
163         inline rcSTR_String     operator+=(char rhs)                                                    { return Concat(&rhs, 1); }
164
165         
166         inline friend bool operator<(rcSTR_String      lhs, rcSTR_String                rhs)    { return (strcmp(lhs, rhs)<0); }
167         inline friend bool operator<(rcSTR_String      lhs, const char          *rhs)   { return (strcmp(lhs, rhs)<0); };
168         inline friend bool operator<(const char         *lhs, rcSTR_String     rhs)     { return (strcmp(lhs, rhs)<0); }
169         inline friend bool operator>(rcSTR_String      lhs, rcSTR_String                rhs)    { return (strcmp(lhs, rhs)>0); }
170         inline friend bool operator>(rcSTR_String      lhs, const char          *rhs)   { return (strcmp(lhs, rhs)>0); }
171         inline friend bool operator>(const char         *lhs, rcSTR_String     rhs)     { return (strcmp(lhs, rhs)>0); }
172         inline friend bool operator<=(rcSTR_String     lhs, rcSTR_String                rhs)    { return (strcmp(lhs, rhs)<=0); }
173         inline friend bool operator<=(rcSTR_String     lhs, const char          *rhs)   { return (strcmp(lhs, rhs)<=0); }
174         inline friend bool operator<=(const char        *lhs, rcSTR_String     rhs)     { return (strcmp(lhs, rhs)<=0); }
175         inline friend bool operator>=(rcSTR_String     lhs, rcSTR_String                rhs)    { return (strcmp(lhs, rhs)>=0); }
176         inline friend bool operator>=(rcSTR_String     lhs, const char          *rhs)   { return (strcmp(lhs, rhs)>=0); }
177         inline friend bool operator>=(const char        *lhs, rcSTR_String     rhs)     { return (strcmp(lhs, rhs)>=0); }
178         inline friend bool operator==(rcSTR_String     lhs, rcSTR_String                rhs)    { return ((lhs.Length() == rhs.Length()) && (memcmp(lhs, rhs, lhs.Length())==0)); }
179         inline friend bool operator==(rcSTR_String     lhs, const char          *rhs)   { return (memcmp(lhs, rhs, lhs.Length()+1)==0); }
180         inline friend bool operator==(const char        *lhs, rcSTR_String     rhs)     { return (memcmp(lhs, rhs, rhs.Length()+1)==0); }
181         inline friend bool operator!=(rcSTR_String     lhs, rcSTR_String                rhs)    { return ((lhs.Length() != rhs.Length()) || (memcmp(lhs, rhs, lhs.Length())!=0)); }
182         inline friend bool operator!=(rcSTR_String     lhs, const char          *rhs)   { return (memcmp(lhs, rhs, lhs.Length()+1)!=0); }
183         inline friend bool operator!=(const char        *lhs, rcSTR_String     rhs)     { return (memcmp(lhs, rhs, rhs.Length()+1)!=0); }
184
185         // serializing
186         //int                   Serialize(pCStream stream);
187
188 protected:
189         // Implementation
190         void    AllocBuffer(int len, bool keep_contents);
191         rcSTR_String Copy(const char *src, int len);
192         rcSTR_String Concat(const char *data, int len);
193
194         static bool             isLower(char c)                                                                 { return !isUpper(c); }
195         static bool             isUpper(char c)                                                                 { return (c>='A') && (c <= 'Z'); }
196         static bool             isSpace(char c)                                                                 { return (c==' ') || (c=='\t'); }
197
198         char   *pData;                                                                                                  // -> STR_String data
199         int        Len;                                                                                                         // Data length
200         int        Max;                                                                                                         // Space in data buffer
201
202
203 #ifdef WITH_CXX_GUARDEDALLOC
204 public:
205         void *operator new( unsigned int num_bytes) { return MEM_mallocN(num_bytes, "CXX:STR_String"); }
206         void operator delete( void *mem ) { MEM_freeN(mem); }
207 #endif
208 };
209
210 inline  STR_String operator+(rcSTR_String    lhs, rcSTR_String   rhs)   { return STR_String(lhs.ReadPtr(), lhs.Length(), rhs.ReadPtr(), rhs.Length()); }
211 inline  STR_String operator+(rcSTR_String    lhs, char        rhs)      { return STR_String(lhs.ReadPtr(), lhs.Length(), &rhs, 1); }
212 inline  STR_String operator+(char         lhs, rcSTR_String   rhs)      { return STR_String(&lhs, 1, rhs.ReadPtr(), rhs.Length()); }
213 inline  STR_String operator+(rcSTR_String    lhs, const char *rhs)      { return STR_String(lhs.ReadPtr(), lhs.Length(), rhs, strlen(rhs)); }
214 inline  STR_String operator+(const char  *lhs, rcSTR_String   rhs)      { return STR_String(lhs, strlen(lhs), rhs.ReadPtr(), rhs.Length()); }
215
216
217 #endif //_STR_String_H_
218