dlls/jscript/math.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 "config.h"
  20 #include "wine/port.h"
  21
  22 #include <math.h>
  23 #include <limits.h>
  24
  25 #include "jscript.h"
  26 #include "ntsecapi.h"
  27
  28 #include "wine/debug.h"
  29
  30 WINE_DEFAULT_DEBUG_CHANNEL(jscript);
  31
  32 static const WCHAR EW[] = {'E',0};
  33 static const WCHAR LOG2EW[] = {'L','O','G','2','E',0};
  34 static const WCHAR LOG10EW[] = {'L','O','G','1','0','E',0};
  35 static const WCHAR LN2W[] = {'L','N','2',0};
  36 static const WCHAR LN10W[] = {'L','N','1','0',0};
  37 static const WCHAR PIW[] = {'P','I',0};
  38 static const WCHAR SQRT2W[] = {'S','Q','R','T','2',0};
  39 static const WCHAR SQRT1_2W[] = {'S','Q','R','T','1','_','2',0};
  40 static const WCHAR absW[] = {'a','b','s',0};
  41 static const WCHAR acosW[] = {'a','c','o','s',0};
  42 static const WCHAR asinW[] = {'a','s','i','n',0};
  43 static const WCHAR atanW[] = {'a','t','a','n',0};
  44 static const WCHAR atan2W[] = {'a','t','a','n','2',0};
  45 static const WCHAR ceilW[] = {'c','e','i','l',0};
  46 static const WCHAR cosW[] = {'c','o','s',0};
  47 static const WCHAR expW[] = {'e','x','p',0};
  48 static const WCHAR floorW[] = {'f','l','o','o','r',0};
  49 static const WCHAR logW[] = {'l','o','g',0};
  50 static const WCHAR maxW[] = {'m','a','x',0};
  51 static const WCHAR minW[] = {'m','i','n',0};
  52 static const WCHAR powW[] = {'p','o','w',0};
  53 static const WCHAR randomW[] = {'r','a','n','d','o','m',0};
  54 static const WCHAR roundW[] = {'r','o','u','n','d',0};
  55 static const WCHAR sinW[] = {'s','i','n',0};
  56 static const WCHAR sqrtW[] = {'s','q','r','t',0};
  57 static const WCHAR tanW[] = {'t','a','n',0};
  58
  59 static HRESULT math_constant(DOUBLE val, WORD flags, VARIANT *retv)
  60 {
  61     switch(flags) {
  62     case DISPATCH_PROPERTYGET:
  63         V_VT(retv) = VT_R8;
  64         V_R8(retv) = val;
  65         return S_OK;
  66     case DISPATCH_PROPERTYPUT:
  67         return S_OK;
  68     }
  69
  70     FIXME("unhandled flags %x\n", flags);
  71     return E_NOTIMPL;
  72 }
  73
  74 /* ECMA-262 3rd Edition    15.8.1.1 */
  75 static HRESULT Math_E(DispatchEx *dispex, LCID lcid, WORD flags, DISPPARAMS *dp,
  76         VARIANT *retv, jsexcept_t *ei, IServiceProvider *sp)
  77 {
  78     TRACE("\n");
  79     return math_constant(M_E, flags, retv);
  80 }
  81
  82 /* ECMA-262 3rd Edition    15.8.1.4 */
  83 static HRESULT Math_LOG2E(DispatchEx *dispex, LCID lcid, WORD flags, DISPPARAMS *dp,
  84         VARIANT *retv, jsexcept_t *ei, IServiceProvider *sp)
  85 {
  86     TRACE("\n");
  87     return math_constant(M_LOG2E, flags, retv);
  88 }
  89
  90 /* ECMA-262 3rd Edition    15.8.1.4 */
  91 static HRESULT Math_LOG10E(DispatchEx *dispex, LCID lcid, WORD flags, DISPPARAMS *dp,
  92         VARIANT *retv, jsexcept_t *ei, IServiceProvider *sp)
  93 {
  94     TRACE("\n");
  95     return math_constant(M_LOG10E, flags, retv);
  96 }
  97
  98 static HRESULT Math_LN2(DispatchEx *dispex, LCID lcid, WORD flags, DISPPARAMS *dp,
  99         VARIANT *retv, jsexcept_t *ei, IServiceProvider *sp)
 100 {
 101     TRACE("\n");
 102     return math_constant(M_LN2, flags, retv);
 103 }
 104
 105 static HRESULT Math_LN10(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 /* ECMA-262 3rd Edition    15.8.1.6 */
 113 static HRESULT Math_PI(DispatchEx *dispex, LCID lcid, WORD flags, DISPPARAMS *dp,
 114         VARIANT *retv, jsexcept_t *ei, IServiceProvider *sp)
 115 {
 116     TRACE("\n");
 117     return math_constant(M_PI, flags, retv);
 118 }
 119
 120 static HRESULT Math_SQRT2(DispatchEx *dispex, LCID lcid, WORD flags, DISPPARAMS *dp,
 121         VARIANT *retv, jsexcept_t *ei, IServiceProvider *sp)
 122 {
 123     FIXME("\n");
 124     return E_NOTIMPL;
 125 }
 126
 127 static HRESULT Math_SQRT1_2(DispatchEx *dispex, LCID lcid, WORD flags, DISPPARAMS *dp,
 128         VARIANT *retv, jsexcept_t *ei, IServiceProvider *sp)
 129 {
 130     FIXME("\n");
 131     return E_NOTIMPL;
 132 }
 133
 134 /* ECMA-262 3rd Edition    15.8.2.12 */
 135 static HRESULT Math_abs(DispatchEx *dispex, LCID lcid, WORD flags, DISPPARAMS *dp,
 136         VARIANT *retv, jsexcept_t *ei, IServiceProvider *sp)
 137 {
 138     VARIANT v;
 139     DOUBLE d;
 140     HRESULT hres;
 141
 142     TRACE("\n");
 143
 144     if(!arg_cnt(dp)) {
 145         if(retv)
 146             num_set_nan(retv);
 147         return S_OK;
 148     }
 149
 150     hres = to_number(dispex->ctx, get_arg(dp, 0), ei, &v);
 151     if(FAILED(hres))
 152         return hres;
 153
 154     d = num_val(&v);
 155     if(retv)
 156         num_set_val(retv, d < 0.0 ? -d : d);
 157     return S_OK;
 158 }
 159
 160 static HRESULT Math_acos(DispatchEx *dispex, LCID lcid, WORD flags, DISPPARAMS *dp,
 161         VARIANT *retv, jsexcept_t *ei, IServiceProvider *sp)
 162 {
 163     FIXME("\n");
 164     return E_NOTIMPL;
 165 }
 166
 167 static HRESULT Math_asin(DispatchEx *dispex, LCID lcid, WORD flags, DISPPARAMS *dp,
 168         VARIANT *retv, jsexcept_t *ei, IServiceProvider *sp)
 169 {
 170     FIXME("\n");
 171     return E_NOTIMPL;
 172 }
 173
 174 static HRESULT Math_atan(DispatchEx *dispex, LCID lcid, WORD flags, DISPPARAMS *dp,
 175         VARIANT *retv, jsexcept_t *ei, IServiceProvider *sp)
 176 {
 177     FIXME("\n");
 178     return E_NOTIMPL;
 179 }
 180
 181 static HRESULT Math_atan2(DispatchEx *dispex, LCID lcid, WORD flags, DISPPARAMS *dp,
 182         VARIANT *retv, jsexcept_t *ei, IServiceProvider *sp)
 183 {
 184     FIXME("\n");
 185     return E_NOTIMPL;
 186 }
 187
 188 /* ECMA-262 3rd Edition    15.8.2.6 */
 189 static HRESULT Math_ceil(DispatchEx *dispex, LCID lcid, WORD flags, DISPPARAMS *dp,
 190         VARIANT *retv, jsexcept_t *ei, IServiceProvider *sp)
 191 {
 192     VARIANT v;
 193     HRESULT hres;
 194
 195     TRACE("\n");
 196
 197     if(!arg_cnt(dp)) {
 198         if(retv)
 199             num_set_nan(retv);
 200         return S_OK;
 201     }
 202
 203     hres = to_number(dispex->ctx, get_arg(dp, 0), ei, &v);
 204     if(FAILED(hres))
 205         return hres;
 206
 207     if(retv)
 208         num_set_val(retv, ceil(num_val(&v)));
 209     return S_OK;
 210 }
 211
 212 static HRESULT Math_cos(DispatchEx *dispex, LCID lcid, WORD flags, DISPPARAMS *dp,
 213         VARIANT *retv, jsexcept_t *ei, IServiceProvider *sp)
 214 {
 215     VARIANT v;
 216     HRESULT hres;
 217
 218     TRACE("\n");
 219
 220     if(!arg_cnt(dp)) {
 221         if(retv) num_set_nan(retv);
 222         return S_OK;
 223     }
 224
 225     hres = to_number(dispex->ctx, get_arg(dp, 0), ei, &v);
 226     if(FAILED(hres))
 227         return hres;
 228
 229     if(retv) num_set_val(retv, cos(num_val(&v)));
 230     return S_OK;
 231 }
 232
 233 static HRESULT Math_exp(DispatchEx *dispex, LCID lcid, WORD flags, DISPPARAMS *dp,
 234         VARIANT *retv, jsexcept_t *ei, IServiceProvider *sp)
 235 {
 236     FIXME("\n");
 237     return E_NOTIMPL;
 238 }
 239
 240 static HRESULT Math_floor(DispatchEx *dispex, LCID lcid, WORD flags, DISPPARAMS *dp,
 241         VARIANT *retv, jsexcept_t *ei, IServiceProvider *sp)
 242 {
 243     VARIANT v;
 244     HRESULT hres;
 245
 246     TRACE("\n");
 247
 248     if(!arg_cnt(dp)) {
 249         if(retv)
 250             num_set_nan(retv);
 251         return S_OK;
 252     }
 253
 254     hres = to_number(dispex->ctx, get_arg(dp, 0), ei, &v);
 255     if(FAILED(hres))
 256         return hres;
 257
 258     if(retv)
 259         num_set_val(retv, floor(num_val(&v)));
 260     return S_OK;
 261 }
 262
 263 static HRESULT Math_log(DispatchEx *dispex, LCID lcid, WORD flags, DISPPARAMS *dp,
 264         VARIANT *retv, jsexcept_t *ei, IServiceProvider *sp)
 265 {
 266     FIXME("\n");
 267     return E_NOTIMPL;
 268 }
 269
 270 /* ECMA-262 3rd Edition    15.8.2.11 */
 271 static HRESULT Math_max(DispatchEx *dispex, LCID lcid, WORD flags, DISPPARAMS *dp,
 272         VARIANT *retv, jsexcept_t *ei, IServiceProvider *sp)
 273 {
 274     DOUBLE max, d;
 275     VARIANT v;
 276     DWORD i;
 277     HRESULT hres;
 278
 279     TRACE("\n");
 280
 281     if(!arg_cnt(dp)) {
 282         if(retv)
 283             num_set_inf(retv, FALSE);
 284         return S_OK;
 285     }
 286
 287     hres = to_number(dispex->ctx, get_arg(dp, 0), ei, &v);
 288     if(FAILED(hres))
 289         return hres;
 290
 291     max = num_val(&v);
 292     for(i=1; i < arg_cnt(dp); i++) {
 293         hres = to_number(dispex->ctx, get_arg(dp, i), ei, &v);
 294         if(FAILED(hres))
 295             return hres;
 296
 297         d = num_val(&v);
 298         if(d > max || isnan(d))
 299             max = d;
 300     }
 301
 302     if(retv)
 303         num_set_val(retv, max);
 304     return S_OK;
 305 }
 306
 307 /* ECMA-262 3rd Edition    15.8.2.12 */
 308 static HRESULT Math_min(DispatchEx *dispex, LCID lcid, WORD flags, DISPPARAMS *dp,
 309         VARIANT *retv, jsexcept_t *ei, IServiceProvider *sp)
 310 {
 311     DOUBLE min, d;
 312     VARIANT v;
 313     DWORD i;
 314     HRESULT hres;
 315
 316     TRACE("\n");
 317
 318     if(!arg_cnt(dp)) {
 319         if(retv)
 320             num_set_inf(retv, TRUE);
 321         return S_OK;
 322     }
 323
 324     hres = to_number(dispex->ctx, get_arg(dp, 0), ei, &v);
 325     if(FAILED(hres))
 326         return hres;
 327
 328     min = num_val(&v);
 329     for(i=1; i < arg_cnt(dp); i++) {
 330         hres = to_number(dispex->ctx, get_arg(dp, i), ei, &v);
 331         if(FAILED(hres))
 332             return hres;
 333
 334         d = num_val(&v);
 335         if(d < min || isnan(d))
 336             min = d;
 337     }
 338
 339     if(retv)
 340         num_set_val(retv, min);
 341     return S_OK;
 342 }
 343
 344 /* ECMA-262 3rd Edition    15.8.2.13 */
 345 static HRESULT Math_pow(DispatchEx *dispex, LCID lcid, WORD flags, DISPPARAMS *dp,
 346         VARIANT *retv, jsexcept_t *ei, IServiceProvider *sp)
 347 {
 348     VARIANT x, y;
 349     HRESULT hres;
 350
 351     TRACE("\n");
 352
 353     if(arg_cnt(dp) < 2) {
 354         FIXME("unimplemented arg_cnt %d\n", arg_cnt(dp));
 355         return E_NOTIMPL;
 356     }
 357
 358     hres = to_number(dispex->ctx, get_arg(dp, 0), ei, &x);
 359     if(FAILED(hres))
 360         return hres;
 361
 362     hres = to_number(dispex->ctx, get_arg(dp, 1), ei, &y);
 363     if(FAILED(hres))
 364         return hres;
 365
 366     if(retv)
 367         num_set_val(retv, pow(num_val(&x), num_val(&y)));
 368     return S_OK;
 369 }
 370
 371 /* ECMA-262 3rd Edition    15.8.2.14 */
 372 static HRESULT Math_random(DispatchEx *dispex, LCID lcid, WORD flags, DISPPARAMS *dp,
 373         VARIANT *retv, jsexcept_t *ei, IServiceProvider *sp)
 374 {
 375     UINT r;
 376
 377     TRACE("\n");
 378
 379     if(!RtlGenRandom(&r, sizeof(r)))
 380         return E_UNEXPECTED;
 381
 382     if(retv)
 383         num_set_val(retv, (DOUBLE)r/(DOUBLE)UINT_MAX);
 384
 385     return S_OK;
 386 }
 387
 388 /* ECMA-262 3rd Edition    15.8.2.15 */
 389 static HRESULT Math_round(DispatchEx *dispex, LCID lcid, WORD flags, DISPPARAMS *dp,
 390         VARIANT *retv, jsexcept_t *ei, IServiceProvider *sp)
 391 {
 392     VARIANT v;
 393     HRESULT hres;
 394
 395     TRACE("\n");
 396
 397     if(!arg_cnt(dp)) {
 398         num_set_nan(retv);
 399         return S_OK;
 400     }
 401
 402     hres = to_number(dispex->ctx, get_arg(dp, 0), ei, &v);
 403     if(FAILED(hres))
 404         return hres;
 405
 406     if(retv)
 407         num_set_val(retv, floor(num_val(&v)+0.5));
 408     return S_OK;
 409 }
 410
 411 static HRESULT Math_sin(DispatchEx *dispex, LCID lcid, WORD flags, DISPPARAMS *dp,
 412         VARIANT *retv, jsexcept_t *ei, IServiceProvider *sp)
 413 {
 414     FIXME("\n");
 415     return E_NOTIMPL;
 416 }
 417
 418 static HRESULT Math_sqrt(DispatchEx *dispex, LCID lcid, WORD flags, DISPPARAMS *dp,
 419         VARIANT *retv, jsexcept_t *ei, IServiceProvider *sp)
 420 {
 421     FIXME("\n");
 422     return E_NOTIMPL;
 423 }
 424
 425 static HRESULT Math_tan(DispatchEx *dispex, LCID lcid, WORD flags, DISPPARAMS *dp,
 426         VARIANT *retv, jsexcept_t *ei, IServiceProvider *sp)
 427 {
 428     FIXME("\n");
 429     return E_NOTIMPL;
 430 }
 431
 432 static const builtin_prop_t Math_props[] = {
 433     {EW,        Math_E,        0},
 434     {LN10W,     Math_LN10,     0},
 435     {LN2W,      Math_LN2,      0},
 436     {LOG10EW,   Math_LOG10E,   0},
 437     {LOG2EW,    Math_LOG2E,    0},
 438     {PIW,       Math_PI,       0},
 439     {SQRT1_2W,  Math_SQRT1_2,  0},
 440     {SQRT2W,    Math_SQRT2,    0},
 441     {absW,      Math_abs,      PROPF_METHOD},
 442     {acosW,     Math_acos,     PROPF_METHOD},
 443     {asinW,     Math_asin,     PROPF_METHOD},
 444     {atanW,     Math_atan,     PROPF_METHOD},
 445     {atan2W,    Math_atan2,    PROPF_METHOD},
 446     {ceilW,     Math_ceil,     PROPF_METHOD},
 447     {cosW,      Math_cos,      PROPF_METHOD},
 448     {expW,      Math_exp,      PROPF_METHOD},
 449     {floorW,    Math_floor,    PROPF_METHOD},
 450     {logW,      Math_log,      PROPF_METHOD},
 451     {maxW,      Math_max,      PROPF_METHOD},
 452     {minW,      Math_min,      PROPF_METHOD},
 453     {powW,      Math_pow,      PROPF_METHOD},
 454     {randomW,   Math_random,   PROPF_METHOD},
 455     {roundW,    Math_round,    PROPF_METHOD},
 456     {sinW,      Math_sin,      PROPF_METHOD},
 457     {sqrtW,     Math_sqrt,     PROPF_METHOD},
 458     {tanW,      Math_tan,      PROPF_METHOD}
 459 };
 460
 461 static const builtin_info_t Math_info = {
 462     JSCLASS_MATH,
 463     {NULL, NULL, 0},
 464     sizeof(Math_props)/sizeof(*Math_props),
 465     Math_props,
 466     NULL,
 467     NULL
 468 };
 469
 470 HRESULT create_math(script_ctx_t *ctx, DispatchEx **ret)
 471 {
 472     return create_dispex(ctx, &Math_info, NULL, ret);
 473 }