source: Include/Classes/System/String.ab @ 182

Last change on this file since 182 was 182, checked in by dai, 16 years ago

_System_StrCmpのオーバーロードが解決できないバグを修正。
File size: 16.1 KB
Line 
1' Classes/System/String.ab
2
3#require <basic/function.sbp>
4
5#ifdef __STRING_IS_NOT_ALWAYS_UNICODE
6TypeDef StrChar = Char
7#ifndef UNICODE
8#define __STRING_IS_NOT_UNICODE
9#endif
10#else
11TypeDef StrChar = WCHAR
12#ifndef UNICODE
13#define __STRING_UNICODE_WINDOWS_ANSI
14#endif
15#endif
16
17Class String
18    m_Length As Long
19Public
20    Chars As *StrChar
21
22    Sub String()
23        Chars = 0
24        m_Length = 0
25    End Sub
26
27    Sub String(initStr As *Byte)
28        String()
29        Assign(initStr As PCTSTR)
30    End Sub
31
32    Sub String(initStr As PCSTR)
33        String()
34        Assign(initStr)
35    End Sub
36
37    Sub String(initStr As PCSTR, length As Long)
38        String()
39        Assign(initStr, length)
40    End Sub
41
42    Sub String(initStr As PCWSTR)
43        String()
44        Assign(initStr)
45    End Sub
46
47    Sub String(initStr As PCWSTR, length As Long)
48        String()
49        Assign(initStr, length)
50    End Sub
51
52    Sub String(ByRef initStr As String)
53        String()
54        Assign(initStr)
55    End Sub
56
57    Sub String(length As Long)
58        String()
59        ReSize(length)
60    End Sub
61
62    Sub String(initChar As StrChar, length As Long)
63        ReSize(length, initChar)
64    End Sub
65
66    Sub ~String()
67        _System_free(Chars)
68        Chars = 0
69#ifdef _DEBUG
70        m_Length = 0
71#endif
72    End Sub
73
74    Const Function Length() As Long
75        Return m_Length
76    End Function
77
78    Function Operator() As *StrChar
79        Return Chars
80    End Function
81
82    Sub Operator = (ByRef objString As String)
83        Assign(objString.Chars, objString.m_Length)
84    End Sub
85
86    Sub Operator = (text As *Byte)
87        Assign(text As PCTSTR)
88    End Sub
89
90    Sub Operator = (text As PCSTR)
91        Assign(text)
92    End Sub
93
94    Sub Operator = (text As PCWSTR)
95        Assign(text)
96    End Sub
97
98    Const Function Operator [] (n As Long) As StrChar
99#ifdef _DEBUG
100        If n > Length Then
101            'Throw ArgumentOutOfRangeException
102            Debug
103        End If
104#endif
105        Return Chars[n]
106    End Function
107
108    Sub Operator []= (n As Long, c As StrChar)
109#ifdef _DEBUG
110        If n >= Length Then
111            'Throw ArgumentOutOfRangeException
112            Debug
113        End If
114#endif
115        Chars[n] = c
116    End Sub
117
118    Const Function Operator + (text As *Byte) As String
119        Return Concat(text As PCTSTR, lstrlen(text))
120    End Function
121
122    Const Function Operator + (text As PCSTR) As String
123        Return Concat(text, lstrlenA(text))
124    End Function
125
126    Const Function Operator + (text As PCWSTR) As String
127        Return Concat(text, lstrlenW(text))
128    End Function
129
130    Const Function Operator + (ByRef objString As String) As String
131        Return Concat(objString.Chars, objString.m_Length)
132    End Function
133
134    Const Function Operator & (text As PCSTR) As String
135        Dim tempString = This + text
136        Return tempString
137    End Function
138
139    Const Function Operator & (text As PCWSTR) As String
140        Dim tempString = This + text
141        Return tempString
142    End Function
143
144    Const Function Operator & (ByRef objString As String) As String
145        Dim tempString = This + objString
146        Return tempString
147    End Function
148
149    Const Function Operator == (ByRef objString As String) As Boolean
150        Return _System_StrCmp(This.Chars, objString.Chars) = 0
151    End Function
152
153    Const Function Operator == (text As *StrChar) As Long
154        Return _System_StrCmp(This.Chars, text) = 0
155    End Function
156
157    Const Function Operator <> (ByRef objString As String) As Boolean
158        Return _System_StrCmp(This.Chars, objString.Chars) <> 0
159    End Function
160
161    Const Function Operator <> (text As *StrChar) As Boolean
162        Return _System_StrCmp(This.Chars, text) <> 0
163    End Function
164
165    Const Function Operator < (ByRef objString As String) As Boolean
166        Return _System_StrCmp(This.Chars, objString.Chars) < 0
167    End Function
168
169    Const Function Operator < (text As *StrChar) As Boolean
170        Return _System_StrCmp(This.Chars, text) < 0
171    End Function
172
173    Const Function Operator > (ByRef objString As String) As Boolean
174        Return _System_StrCmp(This.Chars, objString.Chars) > 0
175    End Function
176
177    Const Function Operator > (text As *StrChar) As Boolean
178        Return _System_StrCmp(This.Chars, text) > 0
179    End Function
180
181    Const Function Operator <= (ByRef objString As String) As Boolean
182        Return _System_StrCmp(This.Chars, objString.Chars) <= 0
183    End Function
184
185    Const Function Operator <= (text As *StrChar) As Boolean
186        Return _System_StrCmp(This.Chars, text) <= 0
187    End Function
188
189    Const Function Operator >= (ByRef objString As String) As Boolean
190        Return _System_StrCmp(This.Chars, objString.Chars) >= 0
191    End Function
192
193    Const Function Operator >= (text As *StrChar) As Boolean
194        Return _System_StrCmp(This.Chars, text) >= 0
195    End Function
196
197    Const Function StrPtr() As *StrChar
198        Return Chars
199    End Function
200
201    Sub ReSize(allocLength As Long)
202        If allocLength < 0 Then Exit Sub
203        If allocLength > m_Length Then
204            Dim oldLength As Long
205            oldLength = m_Length
206            If AllocStringBuffer(allocLength) <> 0 Then
207                ZeroMemory(VarPtr(Chars[oldLength]), SizeOf (StrChar) * (m_Length - oldLength + 1))
208            End If
209        Else
210            m_Length = allocLength
211            Chars[m_Length] = 0
212        End If
213    End Sub
214
215    Sub ReSize(allocLength As Long, c As StrChar)
216        If allocLength < 0 Then
217            Exit Sub
218        ElseIf allocLength > m_Length Then
219            Dim oldLength As Long
220            oldLength = m_Length
221            If AllocStringBuffer(allocLength) <> 0 Then
222                Dim p = VarPtr(Chars[oldLength]) As *StrChar
223                Dim fillLen = m_Length - oldLength
224                Dim i As Long
225                For i = 0 To ELM(fillLen)
226                    p[i] = c
227                Next
228            End If
229        Else
230            m_Length = allocLength
231        End If
232        Chars[m_Length] = 0
233    End Sub
234
235    Sub Assign(text As PCSTR, textLengthA As Long)
236#ifdef __STRING_IS_NOT_UNICODE
237        AssignFromStrChar(text, textLengthA)
238#else
239        Dim textLengthW = MultiByteToWideChar(CP_THREAD_ACP, 0, text, textLengthA, 0, 0)
240        If AllocStringBuffer(textLengthW) <> 0 Then
241            MultiByteToWideChar(CP_THREAD_ACP, 0, text, textLengthA, Chars, textLengthW)
242        End If
243#endif
244    End Sub
245
246    Sub Assign(text As PCWSTR, textLengthW As Long)
247#ifdef __STRING_IS_NOT_UNICODE
248        Dim textLengthA = WideCharToMultiByte(CP_THREAD_ACP, 0, text, textLengthW, 0, 0, 0, 0)
249        If AllocStringBuffer(textLengthA) <> 0 Then
250            WideCharToMultiByte(CP_THREAD_ACP, 0, text, textLengthW, Chars, textLengthA, 0, 0)
251        End If
252#else
253        AssignFromStrChar(text, textLengthW)
254#endif
255    End Sub
256
257    Sub Assign(ByRef objString As String)
258        Assign(objString.Chars, objString.m_Length)
259    End Sub
260
261    Sub Assign(text As PCSTR)
262        If text Then
263            Assign(text, lstrlenA(text))
264        Else
265            If Chars <> 0 Then
266                Chars[0] = 0
267            End If
268            m_Length = 0
269        End If
270    End Sub
271
272    Sub Assign(text As PCWSTR)
273        If text Then
274            Assign(text, lstrlenW(text))
275        Else
276            If Chars <> 0 Then
277                Chars[0] = 0
278            End If
279            m_Length = 0
280        End If
281    End Sub
282
283    Sub Append(text As *StrChar, textLength As Long)
284        Dim prevLen As Long
285        prevLen = m_Length
286        If AllocStringBuffer(m_Length + textLength) <> 0 Then
287            memcpy(VarPtr(Chars[prevLen]), text, SizeOf (StrChar) * textLength)
288            Chars[m_Length] = 0
289        End If
290    End Sub
291
292    Sub Append(text As *StrChar)
293        Append(text, lstrlen(text))
294    End Sub
295
296    Sub Append(ByRef str As String)
297        Append(str.Chars, str.m_Length)
298    End Sub
299
300    Const Function Clone() As String
301        Return This
302    End Function
303Private
304    Static Const Function ConcatStrChar(text1 As *StrChar, text1Length As Long, text2 As *StrChar, text2Length As Long) As String
305        With ConcatStrChar
306            .AllocStringBuffer(text1Length + text2Length)
307            memcpy(.Chars, text1, SizeOf (StrChar) * text1Length)
308            memcpy(VarPtr(.Chars[text1Length]), text2, SizeOf (StrChar) * text2Length)
309            .Chars[text1Length + text2Length] = 0
310        End With
311    End Function
312Public
313    Const Function Concat(text As PCSTR, len As Long) As String
314#ifdef __STRING_IS_NOT_UNICODE
315        Return ConcatStrChar(This.Chars, m_Length, text, len)
316#else
317        With Concat
318            Dim lenW = MultiByteToWideChar(CP_THREAD_ACP, 0, text, len, 0, 0)
319            .AllocStringBuffer(m_Length + lenW)
320            memcpy(.Chars, This.Chars, m_Length)
321            MultiByteToWideChar(CP_THREAD_ACP, 0, text, len, VarPtr(.Chars[m_Length]), lenW)
322            .Chars[m_Length + lenW] = 0
323        End With
324#endif
325    End Function
326
327    Const Function Concat(text As PCWSTR, len As Long) As String
328#ifdef __STRING_IS_NOT_UNICODE
329        With Concat
330            Dim lenA = MultiByteToWideChar(CP_THREAD_ACP, 0, text, len, 0, 0, 0, 0)
331            .AllocStringBuffer(m_Length + lenA)
332            memcpy(.Chars, This.Chars, m_Length)
333            MultiByteToWideChar(CP_THREAD_ACP, 0, text, len, VarPtr(.Chars[m_Length]), lenA, 0, 0)
334            .Chars[m_Length + lenA] = 0
335        End With
336#else
337        Return ConcatStrChar(This.Chars, m_Length, text, len)
338#endif
339    End Function
340    Const Function Contains(ByRef objString As String) As Boolean
341        Return IndexOf(objString, 0, m_Length) >= 0
342    End Function
343
344    Const Function Contains(lpszText As *StrChar) As Boolean
345        Return IndexOf(lpszText, 0, m_Length) >= 0
346    End Function
347
348    Const Function IndexOf(lpszText As *StrChar) As Long
349        Return IndexOf(lpszText, 0, m_Length)
350    End Function
351
352    Const Function IndexOf(lpszText As *StrChar, startIndex As Long) As Long
353        Return IndexOf(lpszText, startIndex, m_Length - startIndex)
354    End Function
355
356    Const Function IndexOf(lpszText As *StrChar, startIndex As Long, count As Long) As Long
357        Dim length = lstrlen(lpszText)
358
359        If startIndex < 0 Then Return -1
360        If count < 1 Or count + startIndex > m_Length Then Return -1
361        If length > m_Length Then Return -1
362
363        If length = 0 Then Return startIndex
364
365        Dim i As Long, j As Long
366        For i = startIndex To startIndex + count - 1
367            For j = 0 To length - 1
368                If Chars[i + j] = lpszText[j] Then
369                    If j = length - 1 Then Return i
370                Else
371                    Exit For
372                End If
373            Next
374        Next
375        Return -1
376    End Function
377
378    Const Function LastIndexOf(lpszText As *StrChar) As Long
379        Return LastIndexOf(lpszText, m_Length - 1, m_Length)
380    End Function
381
382    Const Function LastIndexOf(lpszText As *StrChar, startIndex As Long) As Long
383        Return LastIndexOf(lpszText As *StrChar, startIndex, startIndex + 1)
384    End Function
385
386    Const Function LastIndexOf(lpszText As *StrChar, startIndex As Long, count As Long) As Long
387        Dim length = lstrlen(lpszText)
388
389        If startIndex < 0 Or startIndex > m_Length - 1 Then Return -1
390        If count < 1 Or count > startIndex + 2 Then Return -1
391        If length > m_Length Then Return -1
392
393        If length = 0 Then Return startIndex
394
395        Dim i As Long, j As Long
396        For i = startIndex To  startIndex - count + 1 Step -1
397            For j = length - 1 To 0 Step -1
398                If Chars[i + j] = lpszText[j] Then
399                    If j = 0 Then Return i
400                Else
401                    Exit For
402                End If
403            Next
404        Next
405        Return -1
406    End Function
407
408    Const Function StartsWith(lpszText As *StrChar) As Boolean
409        Return IndexOf(lpszText) = 0
410    End Function
411
412    Const Function EndsWith(lpszText As *StrChar) As Boolean
413        Return LastIndexOf(lpszText) = m_Length - lstrlen(lpszText)
414    End Function
415
416    Const Function Insert(startIndex As Long, text As String) As String
417        Return Insert(startIndex, text.Chars, text.Length)
418    End Function
419
420    Const Function Insert(startIndex As Long, text As *StrChar) As String
421        Return Insert(startIndex, text, lstrlen(text))
422    End Function
423
424    Const Function Insert(startIndex As Long, text As *StrChar, length As Long) As String
425        If startIndex < 0 Or startIndex > m_Length Or length < 0 Then
426            Debug 'ArgumentOutOfRangeException
427
428        End If
429        Insert.ReSize(m_Length + length)
430        memcpy(Insert.Chars, Chars, SizeOf (StrChar) * startIndex)
431        memcpy(VarPtr(Insert.Chars[startIndex]), text, SizeOf (StrChar) * length)
432        memcpy(VarPtr(Insert.Chars[startIndex + length]), VarPtr(Chars[startIndex]), SizeOf (StrChar) * (m_Length - startIndex + 1))
433    End Function
434
435    Const Function SubString(startIndex As Long) As String
436        Return SubString(startIndex, m_Length - startIndex)
437    End Function
438
439    Const Function SubString(startIndex As Long, length As Long) As String
440        If startIndex < 0 Or length <= 0 Then Return ""
441        If startIndex + length > m_Length Then Return ""
442
443        Dim temp As String
444        temp.AllocStringBuffer(length)
445        memcpy(temp.Chars, VarPtr(Chars[startIndex]), SizeOf (StrChar) * length)
446        Chars[m_Length] = 0
447        Return temp
448    End Function
449
450    Const Function Remove(startIndex As Long) As String
451        If startIndex < 0 Or startIndex > m_Length Then
452            Debug 'ArgumentOutOfRangeException
453        End If
454
455        Remove.ReSize(startIndex)
456        memcpy(Remove.Chars, This.Chars, SizeOf (StrChar) * startIndex)
457    End Function
458
459    Const Function Remove(startIndex As Long, count As Long) As String
460        If startIndex < 0 Or count < 0 Or startIndex + count > m_Length Then
461            Debug 'ArgumentOutOfRangeException
462        End If
463
464        Remove.ReSize(m_Length - count)
465        memcpy(Remove.Chars, This.Chars, SizeOf (StrChar) * startIndex)
466        memcpy(VarPtr(Remove.Chars[startIndex]), VarPtr(This.Chars[startIndex + count]), SizeOf (StrChar) * (m_Length - startIndex - count))
467    End Function
468/*
469    Static Function IsNullOrEmpty(s As String) As Boolean
470        If s <> Nothing Then
471            If s.m_Length > 0 Then
472                Return True
473            End If
474        End If
475        Return False
476    End Function
477*/
478    Const Function Replace(oldChar As StrChar, newChar As StrChar) As String
479        Replace = Copy(This)
480        With Replace
481            Dim i As Long
482            For i = 0 To ELM(.m_Length)
483                If .Chars[i] = oldChar Then
484                    .Chars[i] = newChar
485                End If
486            Next
487        End With
488    End Function
489
490    Const Function Replace(ByRef oldStr As String, ByRef newStr As String) As String
491'       If oldStr = Nothing Then Throw ArgumentNullException
492'
493'       If newStr = Nothing Then
494'           Return ReplaceCore(oldStr, oldStr.m_Length, "", 0)
495'       Else
496            Return ReplaceCore(oldStr, oldStr.m_Length, newStr, newStr.m_Length)
497'       End If
498    End Function
499
500    Const Function Replace(oldStr As *StrChar, newStr As *StrChar)
501        If oldStr = 0 Then Debug 'Throw ArgumentNullException
502        If newStr = 0 Then newStr = ""
503        Return ReplaceCore(oldStr, lstrlen(oldStr), newStr, lstrlen(newStr)) As String
504    End Function
505
506    Const Function Replace(oldStr As *StrChar, oldLen As Long, newStr As *StrChar, newLen As Long) As String
507        If oldStr = 0 Then Debug 'Throw ArgumentNullException
508        If newStr = 0 Then
509            newStr = ""
510            newLen = 0
511        End If
512        Return ReplaceCore(oldStr, oldLen, newStr, newLen)
513    End Function
514
515    Const Function ToLower() As String
516        ToLower.ReSize(m_Length)
517        Dim i As Long
518        For i = 0 To ELM(m_Length)
519            ToLower.Chars[i] = _System_ASCII_ToLower(Chars[i])
520        Next
521    End Function
522
523    Const Function ToUpper() As String
524        ToUpper.ReSize(m_Length)
525        Dim i As Long
526        For i = 0 To ELM(m_Length)
527            ToUpper.Chars[i] = _System_ASCII_ToUpper(Chars[i])
528        Next
529    End Function
530
531    Sub Swap(ByRef x As String)
532        Dim tempLen As Long
533        Dim tempChars As *StrChar
534        tempLen = x.m_Length
535        tempChars = x.Chars
536        x.m_Length = This.m_Length
537        x.Chars = This.Chars
538        This.m_Length = tempLen
539        This.Chars = tempChars
540    End Sub
541
542    Override Function ToString() As String
543        Return This
544    End Function
545
546    Static Function Copy(s As String) As String
547        Copy.ReSize(s.m_Length)
548        memcpy(Copy.Chars, This.Chars, SizeOf (StrChar) * m_Length)
549    End Function
550
551    Override Function GetHashCode() As Long
552#ifdef __STRING_IS_NOT_UNICODE
553        Dim size = (m_Length + 1) >> 1
554#else
555        Dim size = m_Length
556#endif
557        Return _System_GetHashFromWordArray(Chars As *Word, size)
558    End Function
559Private
560    ' メモリ確保に失敗すると元の文字列は失われない。（例外安全でいう強い保障）
561    Function AllocStringBuffer(textLength As Long) As *StrChar
562        If textLength < 0 Then
563            Return 0
564        ElseIf textLength > m_Length Then
565            AllocStringBuffer = _System_realloc(Chars, SizeOf(StrChar) * (textLength + 1))
566            If AllocStringBuffer <> 0 Then
567                m_Length = textLength
568                Chars = AllocStringBuffer
569            End If
570        Else
571            m_Length = textLength
572            AllocStringBuffer = Chars
573        End If
574    End Function
575
576    Function ReplaceCore(oldStr As *StrChar, oldLen As Long, newStr As *StrChar, newLen As Long) As String
577        If oldLen = 0 Then
578            Debug 'Throw ArgumentException
579        End If
580        Dim tmp As String
581        With tmp
582            Dim current = 0 As Long
583            Do
584                Dim pos = IndexOf(oldStr, current)
585                If pos = -1 Then
586                    Exit Do
587                End If
588                .Append(VarPtr(Chars[current]), pos - current)
589                .Append(newStr, newLen)
590                current = pos + oldLen
591            Loop
592            .Append(VarPtr(Chars[current]), m_Length - current)
593        End With
594        Return tmp
595    End Function
596
597    Sub AssignFromStrChar(text As *StrChar, textLength As Long)
598        If text = Chars Then Exit Sub
599        If AllocStringBuffer(textLength) <> 0 Then
600            memcpy(Chars, text, SizeOf (StrChar) * textLength)
601            Chars[m_Length] = 0
602        End If
603    End Sub
604End Class
Note: See TracBrowser for help on using the repository browser.