1 /* see copyright notice in squirrel.h */
3 #include <sqstdstring.h>
10 #define scstrchr wcschr
11 #define scsnprintf wsnprintf
14 #define scstrchr strchr
15 #define scsnprintf snprintf
18 #define MAX_FORMAT_LEN 20
19 #define MAX_WFORMAT_LEN 3
20 #define ADDITIONAL_FORMAT_SPACE (100*sizeof(SQChar))
22 static int validate_format(HSQUIRRELVM v, SQChar *fmt, const SQChar *src, int n,int &width)
24 SQChar swidth[MAX_WFORMAT_LEN];
28 while (scstrchr(_SC("-+ #0"), src[n])) n++;
29 while (scisdigit(src[n])) {
33 if(wc>=MAX_WFORMAT_LEN)
34 return sq_throwerror(v,_SC("width format too long"));
38 width = scatoi(swidth);
46 while (scisdigit(src[n])) {
50 if(wc>=MAX_WFORMAT_LEN)
51 return sq_throwerror(v,_SC("precision format too long"));
55 width += scatoi(swidth);
58 if (n-start > MAX_FORMAT_LEN )
59 return sq_throwerror(v,_SC("format too long"));
60 memcpy(&fmt[1],&src[start],((n-start)+1)*sizeof(SQChar));
61 fmt[(n-start)+2] = '\0';
65 static int _string_format(HSQUIRRELVM v)
69 SQChar fmt[MAX_FORMAT_LEN];
70 sq_getstring(v,2,&format);
71 int allocated = (sq_getsize(v,2)+1)*sizeof(SQChar);
72 dest = sq_getscratchpad(v,allocated);
73 int n = 0,i = 0, nparam = 3, w;
74 while(format[n] != '\0') {
75 if(format[n] != '%') {
76 dest[i++] = format[n];
79 else if(format[++n] == '%') {
83 if( nparam > sq_gettop(v) )
84 return sq_throwerror(v,_SC("not enough paramters for the given format string"));
85 n = validate_format(v,fmt,format,n,w);
94 if(SQ_FAILED(sq_getstring(v,nparam,&ts)))
95 return sq_throwerror(v,_SC("string expected for the specified format"));
96 addlen = (sq_getsize(v,nparam)*sizeof(SQChar))+((w+1)*sizeof(SQChar));
99 case 'i': case 'd': case 'c':case 'o': case 'u': case 'x': case 'X':
100 if(SQ_FAILED(sq_getinteger(v,nparam,&ti)))
101 return sq_throwerror(v,_SC("integer expected for the specified format"));
102 addlen = (ADDITIONAL_FORMAT_SPACE)+((w+1)*sizeof(SQChar));
105 case 'f': case 'g': case 'G': case 'e': case 'E':
106 if(SQ_FAILED(sq_getfloat(v,nparam,&tf)))
107 return sq_throwerror(v,_SC("float expected for the specified format"));
108 addlen = (ADDITIONAL_FORMAT_SPACE)+((w+1)*sizeof(SQChar));
112 return sq_throwerror(v,_SC("invalid format"));
115 if((allocated-i) < addlen)
116 allocated += addlen - (allocated-i);
117 dest = sq_getscratchpad(v,allocated);
119 case 's': i += scsprintf(&dest[i],fmt,ts); break;
120 case 'i': i += scsprintf(&dest[i],fmt,ti); break;
121 case 'f': i += scsprintf(&dest[i],fmt,tf); break;
126 sq_pushstring(v,dest,i);
130 #define SETUP_REX(v) \
131 SQRex *self = NULL; \
132 sq_getinstanceup(v,1,(SQUserPointer *)&self,0);
134 static int _rexobj_releasehook(SQUserPointer p, int size)
136 SQRex *self = ((SQRex *)p);
137 sqstd_rex_free(self);
141 static int _regexp_match(HSQUIRRELVM v)
145 sq_getstring(v,2,&str);
146 if(sqstd_rex_match(self,str) == SQTrue)
154 static void _addrexmatch(HSQUIRRELVM v,const SQChar *str,const SQChar *begin,const SQChar *end)
157 sq_pushstring(v,_SC("begin"),-1);
158 sq_pushinteger(v,begin - str);
160 sq_pushstring(v,_SC("end"),-1);
161 sq_pushinteger(v,end - str);
165 static int _regexp_search(HSQUIRRELVM v)
168 const SQChar *str,*begin,*end;
170 sq_getstring(v,2,&str);
171 if(sq_gettop(v) > 2) sq_getinteger(v,3,&start);
172 if(sqstd_rex_search(self,str+start,&begin,&end) == SQTrue) {
173 _addrexmatch(v,str,begin,end);
179 static int _regexp_capture(HSQUIRRELVM v)
182 const SQChar *str,*begin,*end;
184 sq_getstring(v,2,&str);
185 if(sq_gettop(v) > 2) sq_getinteger(v,3,&start);
186 if(sqstd_rex_search(self,str+start,&begin,&end) == SQTrue) {
187 SQInteger n = sqstd_rex_getsubexpcount(self);
190 for(SQInteger i = 0;i < n; i++) {
191 sqstd_rex_getsubexp(self,i,&match);
193 _addrexmatch(v,str,match.begin,match.begin+match.len);
195 _addrexmatch(v,str,str,str); //empty match
196 sq_arrayappend(v,-2);
203 static int _regexp_subexpcount(HSQUIRRELVM v)
206 sq_pushinteger(v,sqstd_rex_getsubexpcount(self));
210 static int _regexp_constructor(HSQUIRRELVM v)
212 const SQChar *error,*pattern;
213 sq_getstring(v,2,&pattern);
214 SQRex *rex = sqstd_rex_compile(pattern,&error);
215 if(!rex) return sq_throwerror(v,error);
216 sq_setinstanceup(v,1,rex);
217 sq_setreleasehook(v,1,_rexobj_releasehook);
221 static int _regexp__typeof(HSQUIRRELVM v)
223 sq_pushstring(v,_SC("regexp"),-1);
227 #define _DECL_REX_FUNC(name,nparams,pmask) {_SC(#name),_regexp_##name,nparams,pmask}
228 static SQRegFunction rexobj_funcs[]={
229 _DECL_REX_FUNC(constructor,2,_SC(".s")),
230 _DECL_REX_FUNC(search,-2,_SC("xsn")),
231 _DECL_REX_FUNC(match,2,_SC("xs")),
232 _DECL_REX_FUNC(capture,-2,_SC("xsn")),
233 _DECL_REX_FUNC(subexpcount,1,_SC("x")),
234 _DECL_REX_FUNC(_typeof,1,_SC("x")),
238 #define _DECL_FUNC(name,nparams,pmask) {_SC(#name),_string_##name,nparams,pmask}
239 static SQRegFunction stringlib_funcs[]={
240 _DECL_FUNC(format,-2,_SC(".s")),
245 int sqstd_register_stringlib(HSQUIRRELVM v)
247 sq_pushstring(v,_SC("regexp"),-1);
248 sq_newclass(v,SQFalse);
250 while(rexobj_funcs[i].name != 0) {
251 SQRegFunction &f = rexobj_funcs[i];
252 sq_pushstring(v,f.name,-1);
253 sq_newclosure(v,f.f,0);
254 sq_setparamscheck(v,f.nparamscheck,f.typemask);
255 sq_setnativeclosurename(v,-1,f.name);
262 while(stringlib_funcs[i].name!=0)
264 sq_pushstring(v,stringlib_funcs[i].name,-1);
265 sq_newclosure(v,stringlib_funcs[i].f,0);
266 sq_setparamscheck(v,stringlib_funcs[i].nparamscheck,stringlib_funcs[i].typemask);
267 sq_setnativeclosurename(v,-1,stringlib_funcs[i].name);