123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365 |
- //-----------------------------------------------------------------------------
- // Copyright (c) 2013 GarageGames, LLC
- //
- // Permission is hereby granted, free of charge, to any person obtaining a copy
- // of this software and associated documentation files (the "Software"), to
- // deal in the Software without restriction, including without limitation the
- // rights to use, copy, modify, merge, publish, distribute, sublicense, and/or
- // sell copies of the Software, and to permit persons to whom the Software is
- // furnished to do so, subject to the following conditions:
- //
- // The above copyright notice and this permission notice shall be included in
- // all copies or substantial portions of the Software.
- //
- // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
- // IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
- // FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
- // AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
- // LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
- // FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
- // IN THE SOFTWARE.
- //-----------------------------------------------------------------------------
- /*!
- @defgroup StringFunctions String
- @ingroup TorqueScriptFunctions
- @{
- */
- /*! Use the strcmp function to do a lexicographic case sensitive string comparison between string1 and string2.
- @param string1 String to be compared to string2.
- @param string2 String to be compared to string1.
- @return Returns a numeric value: <-1 (string1 is less than string2, including case), 0 (string1 is equal to string2, including case), 1 (string1 is greater than string2, including case)>.
- @sa see stricmp, strstr
- */
- ConsoleFunctionWithDocs(strcmp, ConsoleInt, 3, 3, ( string1 , string2 ))
- {
- TORQUE_UNUSED( argc );
- return dStrcmp(argv[1], argv[2]);
- }
- /*! Use the stricmp function to do a lexicographic case in-sensitive string comparison between string1 and string2.
- @param string1 String to be compared to string2.
- @param string2 String to be compared to string1.
- @return Returns a numeric value: <-1 (string1 is less than string2, ignoring case), 0 (string1 is equal to string2, ignoring case), 1 (string1 is greater than string2, ignoring case)>.
- @sa see strcmp, strstr
- */
- ConsoleFunctionWithDocs(stricmp, ConsoleInt, 3, 3, ( string1 , string2 ))
- {
- TORQUE_UNUSED( argc );
- return dStricmp(argv[1], argv[2]);
- }
- /*! Use the strlen function to determine how many characters there are in string.
- @param string The string to count characters for.
- @return Returns the number of characters in string, or 0 if string is invalid or a NULL string
- */
- ConsoleFunctionWithDocs(strlen, ConsoleInt, 2, 2, ( string ))
- {
- TORQUE_UNUSED( argc );
- return dStrlen(argv[1]);
- }
- /*! Use the strstr function to locate the first instance of searchString in sourceString.
- @param sourceString The string in which to search for searchString.
- @param searchString The string for which to search for in sourceString.
- @return Returns a numeric character index representing the position in sourceString at which searchString was found, or -1 to indicate that no instance of searchString was found.
- @sa strpos
- */
- ConsoleFunctionWithDocs(strstr, ConsoleInt, 3, 3, ( sourceString , searchString ))
- {
- TORQUE_UNUSED( argc );
- // returns the start of the sub string argv[2] in argv[1]
- // or -1 if not found.
- const char *retpos = dStrstr(argv[1], argv[2]);
- if(!retpos)
- return -1;
- return (S32)(retpos - argv[1]);
- }
- /*! Use the strPos function to locate the first instance of searchString in sourceString, starting at character 0, or at an optional offset.
- @param sourceString The string in which to search for searchString.
- @param searchString The string for which to search for in sourceString.
- @param offset An optional non-negative integer value representing the character offset within sourceString at which to begin the search.
- @return Returns a numeric character index representing the postion in sourceString at which searchString was found, or -1 to indicate that no instance of searchString was found.
- @sa strstr
- */
- ConsoleFunctionWithDocs(strpos, ConsoleInt, 3, 4, ( sourceString , searchString, [offset]? ))
- {
- S32 start = 0;
- if(argc == 4)
- start = dAtoi(argv[3]);
- U32 sublen = dStrlen(argv[2]);
- U32 strlen = dStrlen(argv[1]);
- if(start < 0)
- return -1;
- if(sublen + start > strlen)
- return -1;
- for(; start + sublen <= strlen; start++)
- if(!dStrncmp(argv[1] + start, argv[2], sublen))
- return start;
- return -1;
- }
- /*! Use the ltrim function to strip the leading white space from sourceString.
- White space is any character in this set: spaces, TABs, and NULL strings.
- @param sourceString The string to be trimmed.
- @return Returns sourceString with all the leading white spaces removed.
- @sa stripChars, stripMLControlChars, stripTrailingSpaces, rtrim, trim
- */
- ConsoleFunctionWithDocs(ltrim, ConsoleString,2,2, ( sourceString ))
- {
- TORQUE_UNUSED( argc );
- const char *ret = argv[1];
- while(*ret == ' ' || *ret == '\n' || *ret == '\t')
- ret++;
- return ret;
- }
- /*! Use the rtrim function to strip the trailing white space from sourceString.
- White space is any character in this set: spaces, TABs, and NULL strings.
- @param sourceString The string to be trimmed.
- @return Returns sourceString with all the trailing white spaces removed.
- @sa stripChars, stripMLControlChars, stripTrailingSpaces, ltrim, trim
- */
- ConsoleFunctionWithDocs(rtrim, ConsoleString,2,2, ( sourceString ))
- {
- TORQUE_UNUSED( argc );
- S32 firstWhitespace = 0;
- S32 pos = 0;
- const char *str = argv[1];
- while(str[pos])
- {
- if(str[pos] != ' ' && str[pos] != '\n' && str[pos] != '\t')
- firstWhitespace = pos + 1;
- pos++;
- }
- char *ret = Con::getReturnBuffer(firstWhitespace + 1);
- dStrncpy(ret, argv[1], firstWhitespace);
- ret[firstWhitespace] = 0;
- return ret;
- }
- /*! Use the trim function to strip the leading and trailing white space from sourceString.
- White space is any character in this set: spaces, TABs, and NULL strings.
- @param sourceString The string to be trimmed.
- @return Returns sourceString with all the leading and trailing white spaces removed.
- @sa stripChars, stripMLControlChars, stripTrailingSpaces, ltrim, rtrim
- */
- ConsoleFunctionWithDocs(trim, ConsoleString,2,2, ( sourceString ))
- {
- TORQUE_UNUSED( argc );
- const char *ptr = argv[1];
- while(*ptr == ' ' || *ptr == '\n' || *ptr == '\t')
- ptr++;
- S32 firstWhitespace = 0;
- S32 pos = 0;
- const char *str = ptr;
- while(str[pos])
- {
- if(str[pos] != ' ' && str[pos] != '\n' && str[pos] != '\t')
- firstWhitespace = pos + 1;
- pos++;
- }
- char *ret = Con::getReturnBuffer(firstWhitespace + 1);
- dStrncpy(ret, ptr, firstWhitespace);
- ret[firstWhitespace] = 0;
- return ret;
- }
- /*! Use the stripChars function to remove chars from sourceString.
- @param sourceString The string to be modified.
- @param chars The character or characters to search for and remove.
- @return Returns a copy of sourceString, from which all instances of chars have been removed. This may be the original sourceString, if chars was not found.
- @sa stripMLControlChars, stripTrailingSpaces
- */
- ConsoleFunctionWithDocs(stripChars, ConsoleString, 3, 3, ( sourceString , chars ))
- {
- TORQUE_UNUSED( argc );
- char* ret = Con::getReturnBuffer( dStrlen( argv[1] ) + 1 );
- dStrcpy( ret, argv[1] );
- U32 pos = dStrcspn( ret, argv[2] );
- while ( pos < dStrlen( ret ) )
- {
- dStrcpy( ret + pos, ret + pos + 1 );
- pos = dStrcspn( ret, argv[2] );
- }
- return( ret );
- }
- /*!
- remove TorqueML color codes from the string.
- @param stringtoString The string from which to remove TorqueML color codes
- @return A string consisting of the original string minus color codes
- */
- ConsoleFunctionWithDocs(stripColorCodes, ConsoleString, 2,2, (stringtoStrip))
- {
- char* ret = Con::getReturnBuffer( dStrlen( argv[1] ) + 1 );
- dStrcpy(ret, argv[1]);
- Con::stripColorChars(ret);
- return ret;
- }
- /*! Use the strlwr function to convert all alpha characters in sourceString to lower-case equivalents.
- @param sourceString The string to be modified.
- @return Returns a copy of sourceString in which all upper-case characters have been converted to lower-case letters.
- @sa strupr
- */
- ConsoleFunctionWithDocs(strlwr, ConsoleString,2,2, ( sourceString ))
- {
- TORQUE_UNUSED( argc );
- char *ret = Con::getReturnBuffer(dStrlen(argv[1]) + 1);
- dStrcpy(ret, argv[1]);
- return dStrlwr(ret);
- }
- /*! Use the strupr function to convert all alpha characters in sourceString to upper-case equivalents.
- @param sourceString The string to be modified.
- @return Returns a copy of sourceString in which all lower-case characters have been converted to upper-case letters.
- @sa strlwr
- */
- ConsoleFunctionWithDocs(strupr, ConsoleString,2,2, ( sourceString ))
- {
- TORQUE_UNUSED( argc );
- char *ret = Con::getReturnBuffer(dStrlen(argv[1]) + 1);
- dStrcpy(ret, argv[1]);
- return dStrupr(ret);
- }
- /*! Use the strchr function to extract a sub-string of sourceString, where the sub-string is equal to the first occurence of char in sourceString followed by the remainder of sourceString.
- @param sourceString The string from which to extract a sub-string.
- @param char The character to search for in sourceString.
- @return Returns a string composed of first instance of char in sourceString, and all of the characters after it. If char is not found, a NULL string is returned.
- @sa getSubStr
- */
- ConsoleFunctionWithDocs(strchr, ConsoleString,3,3, ( sourceString , char ))
- {
- TORQUE_UNUSED( argc );
- const char *ret = dStrchr(argv[1], argv[2][0]);
- return ret ? ret : "";
- }
- /*! strrchr searches the sourceString for the last occurance of the given char
- @param sourceString The string to search
- @return Either a string consisting of the given string from the last occurance of the given char on or an empty string if not found
- */
- ConsoleFunctionWithDocs(strrchr, ConsoleString,3,3, ( sourceString , char ))
- {
- TORQUE_UNUSED( argc );
- const char *ret = dStrrchr(argv[1], argv[2][0]);
- return ret ? ret : "";
- }
- /*! Use the strreplace function to replace every instance of from in sourceString with to.
- This function is case-sensitive and only does exact matching
- @param sourceString The string to do replacement operations on.
- @param from The old value to be replaced.
- @param to The new value to replace old values with.
- @return Returns a new version of sourceString in which every instance of the value in from was replaced with the value in to.
- */
- ConsoleFunctionWithDocs(strreplace, ConsoleString, 4, 4, ( sourceString , from , to ))
- {
- TORQUE_UNUSED( argc );
- S32 fromLen = dStrlen(argv[2]);
- if(!fromLen)
- return argv[1];
- S32 toLen = dStrlen(argv[3]);
- S32 count = 0;
- const char *scan = argv[1];
- while(scan)
- {
- scan = dStrstr(scan, argv[2]);
- if(scan)
- {
- scan += fromLen;
- count++;
- }
- }
- char *ret = Con::getReturnBuffer(dStrlen(argv[1]) + 1 + (toLen - fromLen) * count);
- U32 scanp = 0;
- U32 dstp = 0;
- for(;;)
- {
- const char *scan = dStrstr(argv[1] + scanp, argv[2]);
- if(!scan)
- {
- dStrcpy(ret + dstp, argv[1] + scanp);
- return ret;
- }
- U32 len = (U32)(scan - (argv[1] + scanp));
- dStrncpy(ret + dstp, argv[1] + scanp, len);
- dstp += len;
- dStrcpy(ret + dstp, argv[3]);
- dstp += toLen;
- scanp += len + fromLen;
- }
- return ret;
- }
- /*! Use the getSubStr function to get a sub-string of sourceString, starting at character index start and ending at character index start + count, or the end-of-string, which ever comes first.
- If start + count is greater than the length of sourceString, the extraction will return a string shorter than count.
- @param sourceString The string from which to extract a sub-string.
- @param start The character index at which the extraction starts.
- @param count The length of the sub-string to extract.
- @return Returns a string made up of the character at start in sourceString and ending at the end of the original sourceString, or start + count, whichever comes first.
- @sa strchr
- */
- ConsoleFunctionWithDocs(getSubStr, ConsoleString, 4, 4, ( sourceString , start , count ))
- {
- TORQUE_UNUSED( argc );
- // Returns the substring of argv[1], starting at argv[2], and continuing
- // to either the end of the string, or argv[3] characters, whichever
- // comes first.
- //
- S32 startPos = dAtoi(argv[2]);
- S32 desiredLen = dAtoi(argv[3]);
- if (startPos < 0 || desiredLen < 0) {
- Con::errorf(ConsoleLogEntry::Script, "getSubStr(...): error, starting position and desired length must be >= 0: (%d, %d)", startPos, desiredLen);
- return "";
- }
- S32 baseLen = dStrlen(argv[1]);
- if (baseLen < startPos)
- return "";
- U32 actualLen = desiredLen;
- if (startPos + desiredLen > baseLen)
- actualLen = baseLen - startPos;
- char *ret = Con::getReturnBuffer(actualLen + 1);
- dStrncpy(ret, argv[1] + startPos, actualLen);
- ret[actualLen] = '\0';
- return ret;
- }
- // Used?
- /*! Removes all spaces after the final
- @param string from which to remove trailing spaces
- @return the source string minus trailing spaces
- */
- ConsoleFunctionWithDocs( stripTrailingSpaces, ConsoleString, 2, 2, ( string ))
- {
- TORQUE_UNUSED( argc );
- S32 temp = S32(dStrlen( argv[1] ));
- if ( temp )
- {
- while ( temp >= 1 && ( argv[1][temp - 1] == ' ' || argv[1][temp - 1] == '_' ) )
- temp--;
- if ( temp )
- {
- char* returnString = Con::getReturnBuffer( temp + 1 );
- dStrncpy( returnString, argv[1], U32(temp) );
- returnString[temp] = '\0';
- return( returnString );
- }
- }
- return( "" );
- }
- /*! @} */ // group StringFunctions
|