dlls/jscript/math.c

   1 /*
   2  * Copyright 2008 Jacek Caban for CodeWeavers
   3  * Copyright 2009 Piotr Caban
   4  *
   5  * This library is free software; you can redistribute it and/or
   6  * modify it under the terms of the GNU Lesser General Public
   7  * License as published by the Free Software Foundation; either
   8  * version 2.1 of the License, or (at your option) any later version.
   9  *
  10  * This library is distributed in the hope that it will be useful,
  11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  13  * Lesser General Public License for more details.
  14  *
  15  * You should have received a copy of the GNU Lesser General Public
  16  * License along with this library; if not, write to the Free Software
  17  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  18  */
  19
  20 #include "config.h"
  21 #include "wine/port.h"
  22
  23 #include <math.h>
  24 #include <limits.h>
  25
  26 #include "jscript.h"
  27 #include "ntsecapi.h"
  28
  29 #include "wine/debug.h"
  30
  31 WINE_DEFAULT_DEBUG_CHANNEL(jscript);
  32
  33 static const WCHAR EW[] = {'E',0};
  34 static const WCHAR LOG2EW[] = {'L','O','G','2','E',0};
  35 static const WCHAR LOG10EW[] = {'L','O','G','1','0','E',0};
  36 static const WCHAR LN2W[] = {'L','N','2',0};
  37 static const WCHAR LN10W[] = {'L','N','1','0',0};
  38 static const WCHAR PIW[] = {'P','I',0};
  39 static const WCHAR SQRT2W[] = {'S','Q','R','T','2',0};
  40 static const WCHAR SQRT1_2W[] = {'S','Q','R','T','1','_','2',0};
  41 static const WCHAR absW[] = {'a','b','s',0};
  42 static const WCHAR acosW[] = {'a','c','o','s',0};
  43 static const WCHAR asinW[] = {'a','s','i','n',0};
  44 static const WCHAR atanW[] = {'a','t','a','n',0};
  45 static const WCHAR atan2W[] = {'a','t','a','n','2',0};
  46 static const WCHAR ceilW[] = {'c','e','i','l',0};
  47 static const WCHAR cosW[] = {'c','o','s',0};
  48 static const WCHAR expW[] = {'e','x','p',0};
  49 static const WCHAR floorW[] = {'f','l','o','o','r',0};
  50 static const WCHAR logW[] = {'l','o','g',0};
  51 static const WCHAR maxW[] = {'m','a','x',0};
  52 static const WCHAR minW[] = {'m','i','n',0};
  53 static const WCHAR powW[] = {'p','o','w',0};
  54 static const WCHAR randomW[] = {'r','a','n','d','o','m',0};
  55 static const WCHAR roundW[] = {'r','o','u','n','d',0};
  56 static const WCHAR sinW[] = {'s','i','n',0};
  57 static const WCHAR sqrtW[] = {'s','q','r','t',0};
  58 static const WCHAR tanW[] = {'t','a','n',0};
  59
  60 /* ECMA-262 3rd Edition    15.8.2.12 */
  61 static HRESULT Math_abs(script_ctx_t *ctx, vdisp_t *jsthis, WORD flags, DISPPARAMS *dp,
  62         VARIANT *retv, jsexcept_t *ei)
  63 {
  64     VARIANT v;
  65     DOUBLE d;
  66     HRESULT hres;
  67
  68     TRACE("\n");
  69
  70     if(!arg_cnt(dp)) {
  71         if(retv)
  72             num_set_nan(retv);
  73         return S_OK;
  74     }
  75
  76     hres = to_number(ctx, get_arg(dp, 0), ei, &v);
  77     if(FAILED(hres))
  78         return hres;
  79
  80     d = num_val(&v);
  81     if(retv)
  82         num_set_val(retv, d < 0.0 ? -d : d);
  83     return S_OK;
  84 }
  85
  86 static HRESULT Math_acos(script_ctx_t *ctx, vdisp_t *jsthis, WORD flags, DISPPARAMS *dp,
  87         VARIANT *retv, jsexcept_t *ei)
  88 {
  89     VARIANT v;
  90     HRESULT hres;
  91
  92     TRACE("\n");
  93
  94     if(!arg_cnt(dp)) {
  95         if(retv) num_set_nan(retv);
  96         return S_OK;
  97     }
  98
  99     hres = to_number(ctx, get_arg(dp, 0), ei, &v);
 100     if(FAILED(hres))
 101         return hres;
 102
 103     if(retv) {
 104         DOUBLE x = num_val(&v);
 105         if(x < -1.0 || x > 1.0)
 106             num_set_nan(retv);
 107         else
 108             num_set_val(retv, acos(x));
 109     }
 110     return S_OK;
 111 }
 112
 113 static HRESULT Math_asin(script_ctx_t *ctx, vdisp_t *jsthis, WORD flags, DISPPARAMS *dp,
 114         VARIANT *retv, jsexcept_t *ei)
 115 {
 116     VARIANT v;
 117     HRESULT hres;
 118
 119     TRACE("\n");
 120
 121     if(!arg_cnt(dp)) {
 122         if(retv) num_set_nan(retv);
 123         return S_OK;
 124     }
 125
 126     hres = to_number(ctx, get_arg(dp, 0), ei, &v);
 127     if(FAILED(hres))
 128         return hres;
 129
 130     if(retv) {
 131         DOUBLE x = num_val(&v);
 132         if(x < -1.0 || x > 1.0)
 133             num_set_nan(retv);
 134         else
 135             num_set_val(retv, asin(x));
 136     }
 137     return S_OK;
 138 }
 139
 140 static HRESULT Math_atan(script_ctx_t *ctx, vdisp_t *jsthis, WORD flags, DISPPARAMS *dp,
 141         VARIANT *retv, jsexcept_t *ei)
 142 {
 143     VARIANT v;
 144     HRESULT hres;
 145
 146     TRACE("\n");
 147
 148     if(!arg_cnt(dp)) {
 149         if(retv) num_set_nan(retv);
 150         return S_OK;
 151     }
 152
 153     hres = to_number(ctx, get_arg(dp, 0), ei, &v);
 154     if(FAILED(hres))
 155         return hres;
 156
 157     if(retv) num_set_val(retv, atan(num_val(&v)));
 158     return S_OK;
 159 }
 160
 161 static HRESULT Math_atan2(script_ctx_t *ctx, vdisp_t *jsthis, WORD flags, DISPPARAMS *dp,
 162         VARIANT *retv, jsexcept_t *ei)
 163 {
 164     VARIANT v1, v2;
 165     HRESULT hres;
 166
 167     TRACE("\n");
 168
 169     if(arg_cnt(dp)<2) {
 170         if(retv) num_set_nan(retv);
 171         return S_OK;
 172     }
 173
 174     hres = to_number(ctx, get_arg(dp, 0), ei, &v1);
 175     if(FAILED(hres))
 176         return hres;
 177
 178     hres = to_number(ctx, get_arg(dp, 1), ei, &v2);
 179     if(FAILED(hres))
 180         return hres;
 181
 182     if(retv) num_set_val(retv, atan2(num_val(&v1), num_val(&v2)));
 183     return S_OK;
 184 }
 185
 186 /* ECMA-262 3rd Edition    15.8.2.6 */
 187 static HRESULT Math_ceil(script_ctx_t *ctx, vdisp_t *jsthis, WORD flags, DISPPARAMS *dp,
 188         VARIANT *retv, jsexcept_t *ei)
 189 {
 190     VARIANT v;
 191     HRESULT hres;
 192
 193     TRACE("\n");
 194
 195     if(!arg_cnt(dp)) {
 196         if(retv)
 197             num_set_nan(retv);
 198         return S_OK;
 199     }
 200
 201     hres = to_number(ctx, get_arg(dp, 0), ei, &v);
 202     if(FAILED(hres))
 203         return hres;
 204
 205     if(retv)
 206         num_set_val(retv, ceil(num_val(&v)));
 207     return S_OK;
 208 }
 209
 210 static HRESULT Math_cos(script_ctx_t *ctx, vdisp_t *jsthis, WORD flags, DISPPARAMS *dp,
 211         VARIANT *retv, jsexcept_t *ei)
 212 {
 213     VARIANT v;
 214     HRESULT hres;
 215
 216     TRACE("\n");
 217
 218     if(!arg_cnt(dp)) {
 219         if(retv) num_set_nan(retv);
 220         return S_OK;
 221     }
 222
 223     hres = to_number(ctx, get_arg(dp, 0), ei, &v);
 224     if(FAILED(hres))
 225         return hres;
 226
 227     if(retv) num_set_val(retv, cos(num_val(&v)));
 228     return S_OK;
 229 }
 230
 231 static HRESULT Math_exp(script_ctx_t *ctx, vdisp_t *jsthis, WORD flags, DISPPARAMS *dp,
 232         VARIANT *retv, jsexcept_t *ei)
 233 {
 234     VARIANT v;
 235     HRESULT hres;
 236
 237     TRACE("\n");
 238
 239     if(!arg_cnt(dp)) {
 240         if(retv) num_set_nan(retv);
 241         return S_OK;
 242     }
 243
 244     hres = to_number(ctx, get_arg(dp, 0), ei, &v);
 245     if(FAILED(hres))
 246         return hres;
 247
 248     if(retv) num_set_val(retv, exp(num_val(&v)));
 249     return S_OK;
 250 }
 251
 252 static HRESULT Math_floor(script_ctx_t *ctx, vdisp_t *jsthis, WORD flags, DISPPARAMS *dp,
 253         VARIANT *retv, jsexcept_t *ei)
 254 {
 255     VARIANT v;
 256     HRESULT hres;
 257
 258     TRACE("\n");
 259
 260     if(!arg_cnt(dp)) {
 261         if(retv)
 262             num_set_nan(retv);
 263         return S_OK;
 264     }
 265
 266     hres = to_number(ctx, get_arg(dp, 0), ei, &v);
 267     if(FAILED(hres))
 268         return hres;
 269
 270     if(retv)
 271         num_set_val(retv, floor(num_val(&v)));
 272     return S_OK;
 273 }
 274
 275 static HRESULT Math_log(script_ctx_t *ctx, vdisp_t *jsthis, WORD flags, DISPPARAMS *dp,
 276         VARIANT *retv, jsexcept_t *ei)
 277 {
 278     VARIANT v;
 279     HRESULT hres;
 280
 281     TRACE("\n");
 282
 283     if(!arg_cnt(dp)) {
 284         if(retv)
 285             num_set_nan(retv);
 286         return S_OK;
 287     }
 288
 289     hres = to_number(ctx, get_arg(dp, 0), ei, &v);
 290     if(FAILED(hres))
 291         return hres;
 292
 293     if(retv) {
 294         DOUBLE x = num_val(&v);
 295         if(x < -0.0)
 296             num_set_nan(retv);
 297         else
 298             num_set_val(retv, log(x));
 299     }
 300     return S_OK;
 301 }
 302
 303 /* ECMA-262 3rd Edition    15.8.2.11 */
 304 static HRESULT Math_max(script_ctx_t *ctx, vdisp_t *jsthis, WORD flags, DISPPARAMS *dp,
 305         VARIANT *retv, jsexcept_t *ei)
 306 {
 307     DOUBLE max, d;
 308     VARIANT v;
 309     DWORD i;
 310     HRESULT hres;
 311
 312     TRACE("\n");
 313
 314     if(!arg_cnt(dp)) {
 315         if(retv)
 316             num_set_inf(retv, FALSE);
 317         return S_OK;
 318     }
 319
 320     hres = to_number(ctx, get_arg(dp, 0), ei, &v);
 321     if(FAILED(hres))
 322         return hres;
 323
 324     max = num_val(&v);
 325     for(i=1; i < arg_cnt(dp); i++) {
 326         hres = to_number(ctx, get_arg(dp, i), ei, &v);
 327         if(FAILED(hres))
 328             return hres;
 329
 330         d = num_val(&v);
 331         if(d > max || isnan(d))
 332             max = d;
 333     }
 334
 335     if(retv)
 336         num_set_val(retv, max);
 337     return S_OK;
 338 }
 339
 340 /* ECMA-262 3rd Edition    15.8.2.12 */
 341 static HRESULT Math_min(script_ctx_t *ctx, vdisp_t *jsthis, WORD flags, DISPPARAMS *dp,
 342         VARIANT *retv, jsexcept_t *ei)
 343 {
 344     DOUBLE min, d;
 345     VARIANT v;
 346     DWORD i;
 347     HRESULT hres;
 348
 349     TRACE("\n");
 350
 351     if(!arg_cnt(dp)) {
 352         if(retv)
 353             num_set_inf(retv, TRUE);
 354         return S_OK;
 355     }
 356
 357     hres = to_number(ctx, get_arg(dp, 0), ei, &v);
 358     if(FAILED(hres))
 359         return hres;
 360
 361     min = num_val(&v);
 362     for(i=1; i < arg_cnt(dp); i++) {
 363         hres = to_number(ctx, get_arg(dp, i), ei, &v);
 364         if(FAILED(hres))
 365             return hres;
 366
 367         d = num_val(&v);
 368         if(d < min || isnan(d))
 369             min = d;
 370     }
 371
 372     if(retv)
 373         num_set_val(retv, min);
 374     return S_OK;
 375 }
 376
 377 /* ECMA-262 3rd Edition    15.8.2.13 */
 378 static HRESULT Math_pow(script_ctx_t *ctx, vdisp_t *jsthis, WORD flags, DISPPARAMS *dp,
 379         VARIANT *retv, jsexcept_t *ei)
 380 {
 381     VARIANT x, y;
 382     HRESULT hres;
 383
 384     TRACE("\n");
 385
 386     if(arg_cnt(dp) < 2) {
 387         if(retv) num_set_nan(retv);
 388         return S_OK;
 389     }
 390
 391     hres = to_number(ctx, get_arg(dp, 0), ei, &x);
 392     if(FAILED(hres))
 393         return hres;
 394
 395     hres = to_number(ctx, get_arg(dp, 1), ei, &y);
 396     if(FAILED(hres))
 397         return hres;
 398
 399     if(retv)
 400         num_set_val(retv, pow(num_val(&x), num_val(&y)));
 401     return S_OK;
 402 }
 403
 404 /* ECMA-262 3rd Edition    15.8.2.14 */
 405 static HRESULT Math_random(script_ctx_t *ctx, vdisp_t *jsthis, WORD flags, DISPPARAMS *dp,
 406         VARIANT *retv, jsexcept_t *ei)
 407 {
 408     UINT r;
 409
 410     TRACE("\n");
 411
 412     if(!RtlGenRandom(&r, sizeof(r)))
 413         return E_UNEXPECTED;
 414
 415     if(retv)
 416         num_set_val(retv, (DOUBLE)r/(DOUBLE)UINT_MAX);
 417
 418     return S_OK;
 419 }
 420
 421 /* ECMA-262 3rd Edition    15.8.2.15 */
 422 static HRESULT Math_round(script_ctx_t *ctx, vdisp_t *jsthis, WORD flags, DISPPARAMS *dp,
 423         VARIANT *retv, jsexcept_t *ei)
 424 {
 425     VARIANT v;
 426     HRESULT hres;
 427
 428     TRACE("\n");
 429
 430     if(!arg_cnt(dp)) {
 431         num_set_nan(retv);
 432         return S_OK;
 433     }
 434
 435     hres = to_number(ctx, get_arg(dp, 0), ei, &v);
 436     if(FAILED(hres))
 437         return hres;
 438
 439     if(retv)
 440         num_set_val(retv, floor(num_val(&v)+0.5));
 441     return S_OK;
 442 }
 443
 444 static HRESULT Math_sin(script_ctx_t *ctx, vdisp_t *jsthis, WORD flags, DISPPARAMS *dp,
 445         VARIANT *retv, jsexcept_t *ei)
 446 {
 447     VARIANT v;
 448     HRESULT hres;
 449
 450     TRACE("\n");
 451
 452     if(!arg_cnt(dp)) {
 453         if(retv) num_set_nan(retv);
 454         return S_OK;
 455     }
 456
 457     hres = to_number(ctx, get_arg(dp, 0), ei, &v);
 458     if(FAILED(hres))
 459         return hres;
 460
 461     if(retv) num_set_val(retv, sin(num_val(&v)));
 462     return S_OK;
 463 }
 464
 465 static HRESULT Math_sqrt(script_ctx_t *ctx, vdisp_t *jsthis, WORD flags, DISPPARAMS *dp,
 466         VARIANT *retv, jsexcept_t *ei)
 467 {
 468     VARIANT v;
 469     HRESULT hres;
 470
 471     TRACE("\n");
 472
 473     if(!arg_cnt(dp)) {
 474         if(retv) num_set_nan(retv);
 475         return S_OK;
 476     }
 477
 478     hres = to_number(ctx, get_arg(dp, 0), ei, &v);
 479     if(FAILED(hres))
 480         return hres;
 481
 482     if(retv) num_set_val(retv, sqrt(num_val(&v)));
 483     return S_OK;
 484 }
 485
 486 static HRESULT Math_tan(script_ctx_t *ctx, vdisp_t *jsthis, WORD flags, DISPPARAMS *dp,
 487         VARIANT *retv, jsexcept_t *ei)
 488 {
 489     VARIANT v;
 490     HRESULT hres;
 491
 492     TRACE("\n");
 493
 494     if(!arg_cnt(dp)) {
 495         if(retv) num_set_nan(retv);
 496         return S_OK;
 497     }
 498
 499     hres = to_number(ctx, get_arg(dp, 0), ei, &v);
 500     if(FAILED(hres))
 501         return hres;
 502
 503     if(retv) num_set_val(retv, tan(num_val(&v)));
 504     return S_OK;
 505 }
 506
 507 static const builtin_prop_t Math_props[] = {
 508     {absW,      Math_abs,      PROPF_METHOD|1},
 509     {acosW,     Math_acos,     PROPF_METHOD|1},
 510     {asinW,     Math_asin,     PROPF_METHOD|1},
 511     {atanW,     Math_atan,     PROPF_METHOD|1},
 512     {atan2W,    Math_atan2,    PROPF_METHOD|2},
 513     {ceilW,     Math_ceil,     PROPF_METHOD|1},
 514     {cosW,      Math_cos,      PROPF_METHOD|1},
 515     {expW,      Math_exp,      PROPF_METHOD|1},
 516     {floorW,    Math_floor,    PROPF_METHOD|1},
 517     {logW,      Math_log,      PROPF_METHOD|1},
 518     {maxW,      Math_max,      PROPF_METHOD|2},
 519     {minW,      Math_min,      PROPF_METHOD|2},
 520     {powW,      Math_pow,      PROPF_METHOD|2},
 521     {randomW,   Math_random,   PROPF_METHOD},
 522     {roundW,    Math_round,    PROPF_METHOD|1},
 523     {sinW,      Math_sin,      PROPF_METHOD|1},
 524     {sqrtW,     Math_sqrt,     PROPF_METHOD|1},
 525     {tanW,      Math_tan,      PROPF_METHOD|1}
 526 };
 527
 528 static const builtin_info_t Math_info = {
 529     JSCLASS_MATH,
 530     {NULL, NULL, 0},
 531     sizeof(Math_props)/sizeof(*Math_props),
 532     Math_props,
 533     NULL,
 534     NULL
 535 };
 536
 537 HRESULT create_math(script_ctx_t *ctx, jsdisp_t **ret)
 538 {
 539     jsdisp_t *math;
 540     unsigned i;
 541     VARIANT v;
 542     HRESULT hres;
 543
 544     struct {
 545         const WCHAR *name;
 546         DOUBLE val;
 547     }constants[] = {
 548         {EW,        M_E},        /* ECMA-262 3rd Edition    15.8.1.1 */
 549         {LN10W,     M_LN10},     /* ECMA-262 3rd Edition    15.8.1.2 */
 550         {LN2W,      M_LN2},      /* ECMA-262 3rd Edition    15.8.1.3 */
 551         {LOG2EW,    M_LOG2E},    /* ECMA-262 3rd Edition    15.8.1.4 */
 552         {LOG10EW,   M_LOG10E},   /* ECMA-262 3rd Edition    15.8.1.5 */
 553         {PIW,       M_PI},       /* ECMA-262 3rd Edition    15.8.1.6 */
 554         {SQRT1_2W,  M_SQRT1_2},  /* ECMA-262 3rd Edition    15.8.1.7 */
 555         {SQRT2W,    M_SQRT2},    /* ECMA-262 3rd Edition    15.8.1.8 */
 556     };
 557
 558     math = heap_alloc_zero(sizeof(jsdisp_t));
 559     if(!math)
 560         return E_OUTOFMEMORY;
 561
 562     hres = init_dispex_from_constr(math, ctx, &Math_info, ctx->object_constr);
 563     if(FAILED(hres)) {
 564         heap_free(math);
 565         return hres;
 566     }
 567
 568     V_VT(&v) = VT_R8;
 569     for(i=0; i < sizeof(constants)/sizeof(*constants); i++) {
 570         V_R8(&v) = constants[i].val;
 571         hres = jsdisp_propput_const(math, constants[i].name, &v);
 572         if(FAILED(hres)) {
 573             jsdisp_release(math);
 574             return hres;
 575         }
 576     }
 577
 578     *ret = math;
 579     return S_OK;
 580 }