jscript: Optimize GetDispID usage.
[wine] / dlls / jscript / number.c
1 /*
2  * Copyright 2008 Jacek Caban for CodeWeavers
3  *
4  * This library is free software; you can redistribute it and/or
5  * modify it under the terms of the GNU Lesser General Public
6  * License as published by the Free Software Foundation; either
7  * version 2.1 of the License, or (at your option) any later version.
8  *
9  * This library is distributed in the hope that it will be useful,
10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
12  * Lesser General Public License for more details.
13  *
14  * You should have received a copy of the GNU Lesser General Public
15  * License along with this library; if not, write to the Free Software
16  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
17  */
18
19 #include "jscript.h"
20
21 #include "wine/debug.h"
22
23 WINE_DEFAULT_DEBUG_CHANNEL(jscript);
24
25 typedef struct {
26     DispatchEx dispex;
27
28     VARIANT num;
29 } NumberInstance;
30
31 static const WCHAR toStringW[] = {'t','o','S','t','r','i','n','g',0};
32 static const WCHAR toLocaleStringW[] = {'t','o','L','o','c','a','l','e','S','t','r','i','n','g',0};
33 static const WCHAR toFixedW[] = {'t','o','F','i','x','e','d',0};
34 static const WCHAR toExponentialW[] = {'t','o','E','x','p','o','n','e','n','t','i','a','l',0};
35 static const WCHAR toPrecisionW[] = {'t','o','P','r','e','c','i','s','i','o','n',0};
36 static const WCHAR valueOfW[] = {'v','a','l','u','e','O','f',0};
37 static const WCHAR hasOwnPropertyW[] = {'h','a','s','O','w','n','P','r','o','p','e','r','t','y',0};
38 static const WCHAR propertyIsEnumerableW[] =
39     {'p','r','o','p','e','r','t','y','I','s','E','n','u','m','e','r','a','b','l','e',0};
40 static const WCHAR isPrototypeOfW[] = {'i','s','P','r','o','t','o','t','y','p','e','O','f',0};
41
42 /* ECMA-262 3rd Edition    15.7.4.2 */
43 static HRESULT Number_toString(DispatchEx *dispex, LCID lcid, WORD flags, DISPPARAMS *dp,
44         VARIANT *retv, jsexcept_t *ei, IServiceProvider *sp)
45 {
46     NumberInstance *number;
47     BSTR str;
48     HRESULT hres;
49
50     TRACE("\n");
51
52     if(!is_class(dispex, JSCLASS_NUMBER)) {
53         FIXME("throw TypeError\n");
54         return E_FAIL;
55     }
56
57     number = (NumberInstance*)dispex;
58
59     if(arg_cnt(dp) != 0) {
60         FIXME("unsupported args\n");
61         return E_NOTIMPL;
62     }
63
64     hres = to_string(dispex->ctx, &number->num, ei, &str);
65     if(FAILED(hres))
66         return hres;
67
68     if(retv) {
69         V_VT(retv) = VT_BSTR;
70         V_BSTR(retv) = str;
71     }else {
72         SysFreeString(str);
73     }
74     return S_OK;
75 }
76
77 static HRESULT Number_toLocaleString(DispatchEx *dispex, LCID lcid, WORD flags, DISPPARAMS *dp,
78         VARIANT *retv, jsexcept_t *ei, IServiceProvider *sp)
79 {
80     FIXME("\n");
81     return E_NOTIMPL;
82 }
83
84 static HRESULT Number_toFixed(DispatchEx *dispex, LCID lcid, WORD flags, DISPPARAMS *dp,
85         VARIANT *retv, jsexcept_t *ei, IServiceProvider *sp)
86 {
87     FIXME("\n");
88     return E_NOTIMPL;
89 }
90
91 static HRESULT Number_toExponential(DispatchEx *dispex, LCID lcid, WORD flags, DISPPARAMS *dp,
92         VARIANT *retv, jsexcept_t *ei, IServiceProvider *sp)
93 {
94     FIXME("\n");
95     return E_NOTIMPL;
96 }
97
98 static HRESULT Number_toPrecision(DispatchEx *dispex, LCID lcid, WORD flags, DISPPARAMS *dp,
99         VARIANT *retv, jsexcept_t *ei, IServiceProvider *sp)
100 {
101     FIXME("\n");
102     return E_NOTIMPL;
103 }
104
105 static HRESULT Number_valueOf(DispatchEx *dispex, LCID lcid, WORD flags, DISPPARAMS *dp,
106         VARIANT *retv, jsexcept_t *ei, IServiceProvider *sp)
107 {
108     FIXME("\n");
109     return E_NOTIMPL;
110 }
111
112 static HRESULT Number_hasOwnProperty(DispatchEx *dispex, LCID lcid, WORD flags, DISPPARAMS *dp,
113         VARIANT *retv, jsexcept_t *ei, IServiceProvider *sp)
114 {
115     FIXME("\n");
116     return E_NOTIMPL;
117 }
118
119 static HRESULT Number_propertyIsEnumerable(DispatchEx *dispex, LCID lcid, WORD flags, DISPPARAMS *dp,
120         VARIANT *retv, jsexcept_t *ei, IServiceProvider *sp)
121 {
122     FIXME("\n");
123     return E_NOTIMPL;
124 }
125
126 static HRESULT Number_isPrototypeOf(DispatchEx *dispex, LCID lcid, WORD flags, DISPPARAMS *dp,
127         VARIANT *retv, jsexcept_t *ei, IServiceProvider *sp)
128 {
129     FIXME("\n");
130     return E_NOTIMPL;
131 }
132
133 static HRESULT Number_value(DispatchEx *dispex, LCID lcid, WORD flags, DISPPARAMS *dp,
134         VARIANT *retv, jsexcept_t *ei, IServiceProvider *sp)
135 {
136     NumberInstance *number = (NumberInstance*)dispex;
137
138     switch(flags) {
139     case DISPATCH_PROPERTYGET:
140         *retv = number->num;
141         break;
142
143     default:
144         FIXME("flags %x\n", flags);
145         return E_NOTIMPL;
146     }
147
148     return S_OK;
149 }
150
151 static const builtin_prop_t Number_props[] = {
152     {hasOwnPropertyW,        Number_hasOwnProperty,        PROPF_METHOD},
153     {isPrototypeOfW,         Number_isPrototypeOf,         PROPF_METHOD},
154     {propertyIsEnumerableW,  Number_propertyIsEnumerable,  PROPF_METHOD},
155     {toExponentialW,         Number_toExponential,         PROPF_METHOD},
156     {toFixedW,               Number_toFixed,               PROPF_METHOD},
157     {toLocaleStringW,        Number_toLocaleString,        PROPF_METHOD},
158     {toPrecisionW,           Number_toPrecision,           PROPF_METHOD},
159     {toStringW,              Number_toString,              PROPF_METHOD},
160     {valueOfW,               Number_valueOf,               PROPF_METHOD}
161 };
162
163 static const builtin_info_t Number_info = {
164     JSCLASS_NUMBER,
165     {NULL, Number_value, 0},
166     sizeof(Number_props)/sizeof(*Number_props),
167     Number_props,
168     NULL,
169     NULL
170 };
171
172 static HRESULT NumberConstr_value(DispatchEx *dispex, LCID lcid, WORD flags, DISPPARAMS *dp,
173         VARIANT *retv, jsexcept_t *ei, IServiceProvider *sp)
174 {
175     VARIANT num;
176     HRESULT hres;
177
178     TRACE("\n");
179
180     switch(flags) {
181     case INVOKE_FUNC:
182         if(!arg_cnt(dp)) {
183             if(retv) {
184                 V_VT(retv) = VT_I4;
185                 V_I4(retv) = 0;
186             }
187             return S_OK;
188         }
189
190         hres = to_number(dispex->ctx, get_arg(dp, 0), ei, &num);
191         if(FAILED(hres))
192             return hres;
193
194         if(retv)
195             *retv = num;
196         break;
197
198     case DISPATCH_CONSTRUCT: {
199         DispatchEx *obj;
200
201         switch(arg_cnt(dp)) {
202         case 0:
203             V_VT(&num) = VT_I4;
204             V_I4(&num) = 0;
205             break;
206         case 1:
207             hres = to_number(dispex->ctx, get_arg(dp, 0), ei, &num);
208             if(FAILED(hres))
209                 return hres;
210             break;
211         default:
212             FIXME("unimplemented args\n");
213             return E_NOTIMPL;
214         }
215
216         hres = create_number(dispex->ctx, &num, &obj);
217         if(FAILED(hres))
218             return hres;
219
220         V_VT(retv) = VT_DISPATCH;
221         V_DISPATCH(retv) = (IDispatch*)_IDispatchEx_(obj);
222         break;
223     }
224     default:
225         FIXME("unimplemented flags %x\n", flags);
226         return E_NOTIMPL;
227     }
228
229     return S_OK;
230 }
231
232 static HRESULT alloc_number(script_ctx_t *ctx, BOOL use_constr, NumberInstance **ret)
233 {
234     NumberInstance *number = heap_alloc_zero(sizeof(NumberInstance));
235     HRESULT hres;
236
237     if(use_constr)
238         hres = init_dispex_from_constr(&number->dispex, ctx, &Number_info, ctx->number_constr);
239     else
240         hres = init_dispex(&number->dispex, ctx, &Number_info, NULL);
241     if(FAILED(hres))
242         return hres;
243
244     *ret = number;
245     return S_OK;
246 }
247
248 HRESULT create_number_constr(script_ctx_t *ctx, DispatchEx **ret)
249 {
250     NumberInstance *number;
251     HRESULT hres;
252
253     hres = alloc_number(ctx, FALSE, &number);
254     if(FAILED(hres))
255         return hres;
256
257     hres = create_builtin_function(ctx, NumberConstr_value, PROPF_CONSTR, &number->dispex, ret);
258
259     jsdisp_release(&number->dispex);
260     return hres;
261 }
262
263 HRESULT create_number(script_ctx_t *ctx, VARIANT *num, DispatchEx **ret)
264 {
265     NumberInstance *number;
266     HRESULT hres;
267
268     hres = alloc_number(ctx, TRUE, &number);
269     if(FAILED(hres))
270         return hres;
271
272     number->num = *num;
273
274     *ret = &number->dispex;
275     return S_OK;
276 }