src/compiler/x86/arith.lisp

   1 ;;;; the VM definition of arithmetic VOPs for the x86
   2
   3 ;;;; This software is part of the SBCL system. See the README file for
   4 ;;;; more information.
   5 ;;;;
   6 ;;;; This software is derived from the CMU CL system, which was
   7 ;;;; written at Carnegie Mellon University and released into the
   8 ;;;; public domain. The software is in the public domain and is
   9 ;;;; provided with absolutely no warranty. See the COPYING and CREDITS
  10 ;;;; files for more information.
  11
  12 (in-package "SB!VM")
  13 \f
  14 ;;;; unary operations
  15
  16 (define-vop (fast-safe-arith-op)
  17   (:policy :fast-safe)
  18   (:effects)
  19   (:affected))
  20
  21 (define-vop (fixnum-unop fast-safe-arith-op)
  22   (:args (x :scs (any-reg) :target res))
  23   (:results (res :scs (any-reg)))
  24   (:note "inline fixnum arithmetic")
  25   (:arg-types tagged-num)
  26   (:result-types tagged-num))
  27
  28 (define-vop (signed-unop fast-safe-arith-op)
  29   (:args (x :scs (signed-reg) :target res))
  30   (:results (res :scs (signed-reg)))
  31   (:note "inline (signed-byte 32) arithmetic")
  32   (:arg-types signed-num)
  33   (:result-types signed-num))
  34
  35 (define-vop (fast-negate/fixnum fixnum-unop)
  36   (:translate %negate)
  37   (:generator 1
  38     (move res x)
  39     (inst neg res)))
  40
  41 (define-vop (fast-negate/signed signed-unop)
  42   (:translate %negate)
  43   (:generator 2
  44     (move res x)
  45     (inst neg res)))
  46
  47 (define-vop (fast-lognot/fixnum fixnum-unop)
  48   (:translate lognot)
  49   (:generator 2
  50     (move res x)
  51     (inst xor res (fixnumize -1))))
  52
  53 (define-vop (fast-lognot/signed signed-unop)
  54   (:translate lognot)
  55   (:generator 1
  56     (move res x)
  57     (inst not res)))
  58 \f
  59 ;;;; binary fixnum operations
  60
  61 ;;; Assume that any constant operand is the second arg...
  62
  63 (define-vop (fast-fixnum-binop fast-safe-arith-op)
  64   (:args (x :target r :scs (any-reg)
  65             :load-if (not (and (sc-is x control-stack)
  66                                (sc-is y any-reg)
  67                                (sc-is r control-stack)
  68                                (location= x r))))
  69          (y :scs (any-reg control-stack)))
  70   (:arg-types tagged-num tagged-num)
  71   (:results (r :scs (any-reg) :from (:argument 0)
  72                :load-if (not (and (sc-is x control-stack)
  73                                   (sc-is y any-reg)
  74                                   (sc-is r control-stack)
  75                                   (location= x r)))))
  76   (:result-types tagged-num)
  77   (:note "inline fixnum arithmetic"))
  78
  79 (define-vop (fast-unsigned-binop fast-safe-arith-op)
  80   (:args (x :target r :scs (unsigned-reg)
  81             :load-if (not (and (sc-is x unsigned-stack)
  82                                (sc-is y unsigned-reg)
  83                                (sc-is r unsigned-stack)
  84                                (location= x r))))
  85          (y :scs (unsigned-reg unsigned-stack)))
  86   (:arg-types unsigned-num unsigned-num)
  87   (:results (r :scs (unsigned-reg) :from (:argument 0)
  88             :load-if (not (and (sc-is x unsigned-stack)
  89                                (sc-is y unsigned-reg)
  90                                (sc-is r unsigned-stack)
  91                                (location= x r)))))
  92   (:result-types unsigned-num)
  93   (:note "inline (unsigned-byte 32) arithmetic"))
  94
  95 (define-vop (fast-signed-binop fast-safe-arith-op)
  96   (:args (x :target r :scs (signed-reg)
  97             :load-if (not (and (sc-is x signed-stack)
  98                                (sc-is y signed-reg)
  99                                (sc-is r signed-stack)
 100                                (location= x r))))
 101          (y :scs (signed-reg signed-stack)))
 102   (:arg-types signed-num signed-num)
 103   (:results (r :scs (signed-reg) :from (:argument 0)
 104             :load-if (not (and (sc-is x signed-stack)
 105                                (sc-is y signed-reg)
 106                                (sc-is r signed-stack)
 107                                (location= x r)))))
 108   (:result-types signed-num)
 109   (:note "inline (signed-byte 32) arithmetic"))
 110
 111 (define-vop (fast-fixnum-binop-c fast-safe-arith-op)
 112   (:args (x :target r :scs (any-reg control-stack)))
 113   (:info y)
 114   (:arg-types tagged-num (:constant (signed-byte 30)))
 115   (:results (r :scs (any-reg)
 116                :load-if (not (location= x r))))
 117   (:result-types tagged-num)
 118   (:note "inline fixnum arithmetic"))
 119
 120 (define-vop (fast-unsigned-binop-c fast-safe-arith-op)
 121   (:args (x :target r :scs (unsigned-reg unsigned-stack)))
 122   (:info y)
 123   (:arg-types unsigned-num (:constant (unsigned-byte 32)))
 124   (:results (r :scs (unsigned-reg)
 125                :load-if (not (location= x r))))
 126   (:result-types unsigned-num)
 127   (:note "inline (unsigned-byte 32) arithmetic"))
 128
 129 (define-vop (fast-signed-binop-c fast-safe-arith-op)
 130   (:args (x :target r :scs (signed-reg signed-stack)))
 131   (:info y)
 132   (:arg-types signed-num (:constant (signed-byte 32)))
 133   (:results (r :scs (signed-reg)
 134                :load-if (not (location= x r))))
 135   (:result-types signed-num)
 136   (:note "inline (signed-byte 32) arithmetic"))
 137
 138 (macrolet ((define-binop (translate untagged-penalty op)
 139              `(progn
 140                 (define-vop (,(symbolicate "FAST-" translate "/FIXNUM=>FIXNUM")
 141                              fast-fixnum-binop)
 142                   (:translate ,translate)
 143                   (:generator 2
 144                               (move r x)
 145                               (inst ,op r y)))
 146                 (define-vop (,(symbolicate 'fast- translate '-c/fixnum=>fixnum)
 147                              fast-fixnum-binop-c)
 148                   (:translate ,translate)
 149                   (:generator 1
 150                   (move r x)
 151                   (inst ,op r (fixnumize y))))
 152                 (define-vop (,(symbolicate "FAST-" translate "/SIGNED=>SIGNED")
 153                              fast-signed-binop)
 154                   (:translate ,translate)
 155                   (:generator ,(1+ untagged-penalty)
 156                   (move r x)
 157                   (inst ,op r y)))
 158                 (define-vop (,(symbolicate 'fast- translate '-c/signed=>signed)
 159                              fast-signed-binop-c)
 160                   (:translate ,translate)
 161                   (:generator ,untagged-penalty
 162                   (move r x)
 163                   (inst ,op r y)))
 164                 (define-vop (,(symbolicate "FAST-"
 165                                            translate
 166                                            "/UNSIGNED=>UNSIGNED")
 167                 fast-unsigned-binop)
 168                   (:translate ,translate)
 169                   (:generator ,(1+ untagged-penalty)
 170                   (move r x)
 171                   (inst ,op r y)))
 172                 (define-vop (,(symbolicate 'fast-
 173                                            translate
 174                                            '-c/unsigned=>unsigned)
 175                              fast-unsigned-binop-c)
 176                   (:translate ,translate)
 177                   (:generator ,untagged-penalty
 178                   (move r x)
 179                   ,(if (eq translate 'logand)
 180                        ;; for the -C/UNSIGNED=>UNSIGNED VOP, this case
 181                        ;; is optimized away as an identity somewhere
 182                        ;; along the lines.  However, this VOP is used in
 183                        ;; -C/SIGNED=>UNSIGNED, below, when the
 184                        ;; higher-level lisp code can't optimize away the
 185                        ;; non-trivial identity.
 186                        `(unless (= y #.(1- (ash 1 n-word-bits)))
 187                           (inst ,op r y))
 188                        `(inst ,op r y)))))))
 189   (define-binop - 4 sub)
 190   (define-binop logand 2 and)
 191   (define-binop logior 2 or)
 192   (define-binop logxor 2 xor))
 193
 194 ;;; Special handling of add on the x86; can use lea to avoid a
 195 ;;; register load, otherwise it uses add.
 196 (define-vop (fast-+/fixnum=>fixnum fast-safe-arith-op)
 197   (:translate +)
 198   (:args (x :scs (any-reg) :target r
 199             :load-if (not (and (sc-is x control-stack)
 200                                (sc-is y any-reg)
 201                                (sc-is r control-stack)
 202                                (location= x r))))
 203          (y :scs (any-reg control-stack)))
 204   (:arg-types tagged-num tagged-num)
 205   (:results (r :scs (any-reg) :from (:argument 0)
 206                :load-if (not (and (sc-is x control-stack)
 207                                   (sc-is y any-reg)
 208                                   (sc-is r control-stack)
 209                                   (location= x r)))))
 210   (:result-types tagged-num)
 211   (:note "inline fixnum arithmetic")
 212   (:generator 2
 213     (cond ((and (sc-is x any-reg) (sc-is y any-reg) (sc-is r any-reg)
 214                 (not (location= x r)))
 215            (inst lea r (make-ea :dword :base x :index y :scale 1)))
 216           (t
 217            (move r x)
 218            (inst add r y)))))
 219
 220 (define-vop (fast-+-c/fixnum=>fixnum fast-safe-arith-op)
 221   (:translate +)
 222   (:args (x :target r :scs (any-reg control-stack)))
 223   (:info y)
 224   (:arg-types tagged-num (:constant (signed-byte 30)))
 225   (:results (r :scs (any-reg)
 226                :load-if (not (location= x r))))
 227   (:result-types tagged-num)
 228   (:note "inline fixnum arithmetic")
 229   (:generator 1
 230     (cond ((and (sc-is x any-reg) (sc-is r any-reg) (not (location= x r)))
 231            (inst lea r (make-ea :dword :base x :disp (fixnumize y))))
 232           (t
 233            (move r x)
 234            (inst add r (fixnumize y))))))
 235
 236 (define-vop (fast-+/signed=>signed fast-safe-arith-op)
 237   (:translate +)
 238   (:args (x :scs (signed-reg) :target r
 239             :load-if (not (and (sc-is x signed-stack)
 240                                (sc-is y signed-reg)
 241                                (sc-is r signed-stack)
 242                                (location= x r))))
 243          (y :scs (signed-reg signed-stack)))
 244   (:arg-types signed-num signed-num)
 245   (:results (r :scs (signed-reg) :from (:argument 0)
 246                :load-if (not (and (sc-is x signed-stack)
 247                                   (sc-is y signed-reg)
 248                                   (location= x r)))))
 249   (:result-types signed-num)
 250   (:note "inline (signed-byte 32) arithmetic")
 251   (:generator 5
 252     (cond ((and (sc-is x signed-reg) (sc-is y signed-reg) (sc-is r signed-reg)
 253                 (not (location= x r)))
 254            (inst lea r (make-ea :dword :base x :index y :scale 1)))
 255           (t
 256            (move r x)
 257            (inst add r y)))))
 258
 259 ;;;; Special logand cases: (logand signed unsigned) => unsigned
 260
 261 (define-vop (fast-logand/signed-unsigned=>unsigned
 262              fast-logand/unsigned=>unsigned)
 263   (:args (x :target r :scs (signed-reg)
 264             :load-if (not (and (sc-is x signed-stack)
 265                                (sc-is y unsigned-reg)
 266                                (sc-is r unsigned-stack)
 267                                (location= x r))))
 268          (y :scs (unsigned-reg unsigned-stack)))
 269   (:arg-types signed-num unsigned-num))
 270
 271 (define-vop (fast-logand-c/signed-unsigned=>unsigned
 272              fast-logand-c/unsigned=>unsigned)
 273   (:args (x :target r :scs (signed-reg signed-stack)))
 274   (:arg-types signed-num (:constant (unsigned-byte 32))))
 275
 276 (define-vop (fast-logand/unsigned-signed=>unsigned
 277              fast-logand/unsigned=>unsigned)
 278   (:args (x :target r :scs (unsigned-reg)
 279             :load-if (not (and (sc-is x unsigned-stack)
 280                                (sc-is y signed-reg)
 281                                (sc-is r unsigned-stack)
 282                                (location= x r))))
 283          (y :scs (signed-reg signed-stack)))
 284   (:arg-types unsigned-num signed-num))
 285 \f
 286
 287 (define-vop (fast-+-c/signed=>signed fast-safe-arith-op)
 288   (:translate +)
 289   (:args (x :target r :scs (signed-reg signed-stack)))
 290   (:info y)
 291   (:arg-types signed-num (:constant (signed-byte 32)))
 292   (:results (r :scs (signed-reg)
 293                :load-if (not (location= x r))))
 294   (:result-types signed-num)
 295   (:note "inline (signed-byte 32) arithmetic")
 296   (:generator 4
 297     (cond ((and (sc-is x signed-reg) (sc-is r signed-reg)
 298                 (not (location= x r)))
 299            (inst lea r (make-ea :dword :base x :disp y)))
 300           (t
 301            (move r x)
 302            (if (= y 1)
 303                (inst inc r)
 304              (inst add r y))))))
 305
 306 (define-vop (fast-+/unsigned=>unsigned fast-safe-arith-op)
 307   (:translate +)
 308   (:args (x :scs (unsigned-reg) :target r
 309             :load-if (not (and (sc-is x unsigned-stack)
 310                                (sc-is y unsigned-reg)
 311                                (sc-is r unsigned-stack)
 312                                (location= x r))))
 313          (y :scs (unsigned-reg unsigned-stack)))
 314   (:arg-types unsigned-num unsigned-num)
 315   (:results (r :scs (unsigned-reg) :from (:argument 0)
 316                :load-if (not (and (sc-is x unsigned-stack)
 317                                   (sc-is y unsigned-reg)
 318                                   (sc-is r unsigned-stack)
 319                                   (location= x r)))))
 320   (:result-types unsigned-num)
 321   (:note "inline (unsigned-byte 32) arithmetic")
 322   (:generator 5
 323     (cond ((and (sc-is x unsigned-reg) (sc-is y unsigned-reg)
 324                 (sc-is r unsigned-reg) (not (location= x r)))
 325            (inst lea r (make-ea :dword :base x :index y :scale 1)))
 326           (t
 327            (move r x)
 328            (inst add r y)))))
 329
 330 (define-vop (fast-+-c/unsigned=>unsigned fast-safe-arith-op)
 331   (:translate +)
 332   (:args (x :target r :scs (unsigned-reg unsigned-stack)))
 333   (:info y)
 334   (:arg-types unsigned-num (:constant (unsigned-byte 32)))
 335   (:results (r :scs (unsigned-reg)
 336                :load-if (not (location= x r))))
 337   (:result-types unsigned-num)
 338   (:note "inline (unsigned-byte 32) arithmetic")
 339   (:generator 4
 340     (cond ((and (sc-is x unsigned-reg) (sc-is r unsigned-reg)
 341                 (not (location= x r)))
 342            (inst lea r (make-ea :dword :base x :disp y)))
 343           (t
 344            (move r x)
 345            (if (= y 1)
 346                (inst inc r)
 347              (inst add r y))))))
 348 \f
 349 ;;;; multiplication and division
 350
 351 (define-vop (fast-*/fixnum=>fixnum fast-safe-arith-op)
 352   (:translate *)
 353   ;; We need different loading characteristics.
 354   (:args (x :scs (any-reg) :target r)
 355          (y :scs (any-reg control-stack)))
 356   (:arg-types tagged-num tagged-num)
 357   (:results (r :scs (any-reg) :from (:argument 0)))
 358   (:result-types tagged-num)
 359   (:note "inline fixnum arithmetic")
 360   (:generator 4
 361     (move r x)
 362     (inst sar r 2)
 363     (inst imul r y)))
 364
 365 (define-vop (fast-*-c/fixnum=>fixnum fast-safe-arith-op)
 366   (:translate *)
 367   ;; We need different loading characteristics.
 368   (:args (x :scs (any-reg control-stack)))
 369   (:info y)
 370   (:arg-types tagged-num (:constant (signed-byte 30)))
 371   (:results (r :scs (any-reg)))
 372   (:result-types tagged-num)
 373   (:note "inline fixnum arithmetic")
 374   (:generator 3
 375     (inst imul r x y)))
 376
 377 (define-vop (fast-*/signed=>signed fast-safe-arith-op)
 378   (:translate *)
 379   ;; We need different loading characteristics.
 380   (:args (x :scs (signed-reg) :target r)
 381          (y :scs (signed-reg signed-stack)))
 382   (:arg-types signed-num signed-num)
 383   (:results (r :scs (signed-reg) :from (:argument 0)))
 384   (:result-types signed-num)
 385   (:note "inline (signed-byte 32) arithmetic")
 386   (:generator 5
 387     (move r x)
 388     (inst imul r y)))
 389
 390 (define-vop (fast-*-c/signed=>signed fast-safe-arith-op)
 391   (:translate *)
 392   ;; We need different loading characteristics.
 393   (:args (x :scs (signed-reg signed-stack)))
 394   (:info y)
 395   (:arg-types signed-num (:constant (signed-byte 32)))
 396   (:results (r :scs (signed-reg)))
 397   (:result-types signed-num)
 398   (:note "inline (signed-byte 32) arithmetic")
 399   (:generator 4
 400     (inst imul r x y)))
 401
 402 (define-vop (fast-*/unsigned=>unsigned fast-safe-arith-op)
 403   (:translate *)
 404   (:args (x :scs (unsigned-reg) :target eax)
 405          (y :scs (unsigned-reg unsigned-stack)))
 406   (:arg-types unsigned-num unsigned-num)
 407   (:temporary (:sc unsigned-reg :offset eax-offset :target r
 408                    :from (:argument 0) :to :result) eax)
 409   (:temporary (:sc unsigned-reg :offset edx-offset
 410                    :from :eval :to :result) edx)
 411   (:ignore edx)
 412   (:results (r :scs (unsigned-reg)))
 413   (:result-types unsigned-num)
 414   (:note "inline (unsigned-byte 32) arithmetic")
 415   (:vop-var vop)
 416   (:save-p :compute-only)
 417   (:generator 6
 418     (move eax x)
 419     (inst mul eax y)
 420     (move r eax)))
 421
 422
 423 (define-vop (fast-truncate/fixnum=>fixnum fast-safe-arith-op)
 424   (:translate truncate)
 425   (:args (x :scs (any-reg) :target eax)
 426          (y :scs (any-reg control-stack)))
 427   (:arg-types tagged-num tagged-num)
 428   (:temporary (:sc signed-reg :offset eax-offset :target quo
 429                    :from (:argument 0) :to (:result 0)) eax)
 430   (:temporary (:sc unsigned-reg :offset edx-offset :target rem
 431                    :from (:argument 0) :to (:result 1)) edx)
 432   (:results (quo :scs (any-reg))
 433             (rem :scs (any-reg)))
 434   (:result-types tagged-num tagged-num)
 435   (:note "inline fixnum arithmetic")
 436   (:vop-var vop)
 437   (:save-p :compute-only)
 438   (:generator 31
 439     (let ((zero (generate-error-code vop division-by-zero-error x y)))
 440       (if (sc-is y any-reg)
 441           (inst test y y)  ; smaller instruction
 442           (inst cmp y 0))
 443       (inst jmp :eq zero))
 444     (move eax x)
 445     (inst cdq)
 446     (inst idiv eax y)
 447     (if (location= quo eax)
 448         (inst shl eax 2)
 449         (inst lea quo (make-ea :dword :index eax :scale 4)))
 450     (move rem edx)))
 451
 452 (define-vop (fast-truncate-c/fixnum=>fixnum fast-safe-arith-op)
 453   (:translate truncate)
 454   (:args (x :scs (any-reg) :target eax))
 455   (:info y)
 456   (:arg-types tagged-num (:constant (signed-byte 30)))
 457   (:temporary (:sc signed-reg :offset eax-offset :target quo
 458                    :from :argument :to (:result 0)) eax)
 459   (:temporary (:sc any-reg :offset edx-offset :target rem
 460                    :from :eval :to (:result 1)) edx)
 461   (:temporary (:sc any-reg :from :eval :to :result) y-arg)
 462   (:results (quo :scs (any-reg))
 463             (rem :scs (any-reg)))
 464   (:result-types tagged-num tagged-num)
 465   (:note "inline fixnum arithmetic")
 466   (:vop-var vop)
 467   (:save-p :compute-only)
 468   (:generator 30
 469     (move eax x)
 470     (inst cdq)
 471     (inst mov y-arg (fixnumize y))
 472     (inst idiv eax y-arg)
 473     (if (location= quo eax)
 474         (inst shl eax 2)
 475         (inst lea quo (make-ea :dword :index eax :scale 4)))
 476     (move rem edx)))
 477
 478 (define-vop (fast-truncate/unsigned=>unsigned fast-safe-arith-op)
 479   (:translate truncate)
 480   (:args (x :scs (unsigned-reg) :target eax)
 481          (y :scs (unsigned-reg signed-stack)))
 482   (:arg-types unsigned-num unsigned-num)
 483   (:temporary (:sc unsigned-reg :offset eax-offset :target quo
 484                    :from (:argument 0) :to (:result 0)) eax)
 485   (:temporary (:sc unsigned-reg :offset edx-offset :target rem
 486                    :from (:argument 0) :to (:result 1)) edx)
 487   (:results (quo :scs (unsigned-reg))
 488             (rem :scs (unsigned-reg)))
 489   (:result-types unsigned-num unsigned-num)
 490   (:note "inline (unsigned-byte 32) arithmetic")
 491   (:vop-var vop)
 492   (:save-p :compute-only)
 493   (:generator 33
 494     (let ((zero (generate-error-code vop division-by-zero-error x y)))
 495       (if (sc-is y unsigned-reg)
 496           (inst test y y)  ; smaller instruction
 497           (inst cmp y 0))
 498       (inst jmp :eq zero))
 499     (move eax x)
 500     (inst xor edx edx)
 501     (inst div eax y)
 502     (move quo eax)
 503     (move rem edx)))
 504
 505 (define-vop (fast-truncate-c/unsigned=>unsigned fast-safe-arith-op)
 506   (:translate truncate)
 507   (:args (x :scs (unsigned-reg) :target eax))
 508   (:info y)
 509   (:arg-types unsigned-num (:constant (unsigned-byte 32)))
 510   (:temporary (:sc unsigned-reg :offset eax-offset :target quo
 511                    :from :argument :to (:result 0)) eax)
 512   (:temporary (:sc unsigned-reg :offset edx-offset :target rem
 513                    :from :eval :to (:result 1)) edx)
 514   (:temporary (:sc unsigned-reg :from :eval :to :result) y-arg)
 515   (:results (quo :scs (unsigned-reg))
 516             (rem :scs (unsigned-reg)))
 517   (:result-types unsigned-num unsigned-num)
 518   (:note "inline (unsigned-byte 32) arithmetic")
 519   (:vop-var vop)
 520   (:save-p :compute-only)
 521   (:generator 32
 522     (move eax x)
 523     (inst xor edx edx)
 524     (inst mov y-arg y)
 525     (inst div eax y-arg)
 526     (move quo eax)
 527     (move rem edx)))
 528
 529 (define-vop (fast-truncate/signed=>signed fast-safe-arith-op)
 530   (:translate truncate)
 531   (:args (x :scs (signed-reg) :target eax)
 532          (y :scs (signed-reg signed-stack)))
 533   (:arg-types signed-num signed-num)
 534   (:temporary (:sc signed-reg :offset eax-offset :target quo
 535                    :from (:argument 0) :to (:result 0)) eax)
 536   (:temporary (:sc signed-reg :offset edx-offset :target rem
 537                    :from (:argument 0) :to (:result 1)) edx)
 538   (:results (quo :scs (signed-reg))
 539             (rem :scs (signed-reg)))
 540   (:result-types signed-num signed-num)
 541   (:note "inline (signed-byte 32) arithmetic")
 542   (:vop-var vop)
 543   (:save-p :compute-only)
 544   (:generator 33
 545     (let ((zero (generate-error-code vop division-by-zero-error x y)))
 546       (if (sc-is y signed-reg)
 547           (inst test y y)  ; smaller instruction
 548           (inst cmp y 0))
 549       (inst jmp :eq zero))
 550     (move eax x)
 551     (inst cdq)
 552     (inst idiv eax y)
 553     (move quo eax)
 554     (move rem edx)))
 555
 556 (define-vop (fast-truncate-c/signed=>signed fast-safe-arith-op)
 557   (:translate truncate)
 558   (:args (x :scs (signed-reg) :target eax))
 559   (:info y)
 560   (:arg-types signed-num (:constant (signed-byte 32)))
 561   (:temporary (:sc signed-reg :offset eax-offset :target quo
 562                    :from :argument :to (:result 0)) eax)
 563   (:temporary (:sc signed-reg :offset edx-offset :target rem
 564                    :from :eval :to (:result 1)) edx)
 565   (:temporary (:sc signed-reg :from :eval :to :result) y-arg)
 566   (:results (quo :scs (signed-reg))
 567             (rem :scs (signed-reg)))
 568   (:result-types signed-num signed-num)
 569   (:note "inline (signed-byte 32) arithmetic")
 570   (:vop-var vop)
 571   (:save-p :compute-only)
 572   (:generator 32
 573     (move eax x)
 574     (inst cdq)
 575     (inst mov y-arg y)
 576     (inst idiv eax y-arg)
 577     (move quo eax)
 578     (move rem edx)))
 579
 580
 581 \f
 582 ;;;; Shifting
 583 (define-vop (fast-ash-c/fixnum=>fixnum)
 584   (:translate ash)
 585   (:policy :fast-safe)
 586   (:args (number :scs (any-reg) :target result
 587                  :load-if (not (and (sc-is number any-reg control-stack)
 588                                     (sc-is result any-reg control-stack)
 589                                     (location= number result)))))
 590   (:info amount)
 591   (:arg-types tagged-num (:constant integer))
 592   (:results (result :scs (any-reg)
 593                     :load-if (not (and (sc-is number control-stack)
 594                                        (sc-is result control-stack)
 595                                        (location= number result)))))
 596   (:result-types tagged-num)
 597   (:note "inline ASH")
 598   (:generator 2
 599     (cond ((and (= amount 1) (not (location= number result)))
 600            (inst lea result (make-ea :dword :index number :scale 2)))
 601           ((and (= amount 2) (not (location= number result)))
 602            (inst lea result (make-ea :dword :index number :scale 4)))
 603           ((and (= amount 3) (not (location= number result)))
 604            (inst lea result (make-ea :dword :index number :scale 8)))
 605           (t
 606            (move result number)
 607            (cond ((plusp amount)
 608                   ;; We don't have to worry about overflow because of the
 609                   ;; result type restriction.
 610                   (inst shl result amount))
 611                  (t
 612                   ;; If the amount is greater than 31, only shift by 31. We
 613                   ;; have to do this because the shift instructions only look
 614                   ;; at the low five bits of the result.
 615                   (inst sar result (min 31 (- amount)))
 616                   ;; Fixnum correction.
 617                   (inst and result #xfffffffc)))))))
 618
 619 (define-vop (fast-ash-left/fixnum=>fixnum)
 620   (:translate ash)
 621   (:args (number :scs (any-reg) :target result
 622                  :load-if (not (and (sc-is number control-stack)
 623                                     (sc-is result control-stack)
 624                                     (location= number result))))
 625          (amount :scs (unsigned-reg) :target ecx))
 626   (:arg-types tagged-num positive-fixnum)
 627   (:temporary (:sc unsigned-reg :offset ecx-offset :from (:argument 1)) ecx)
 628   (:results (result :scs (any-reg) :from (:argument 0)
 629                     :load-if (not (and (sc-is number control-stack)
 630                                        (sc-is result control-stack)
 631                                        (location= number result)))))
 632   (:result-types tagged-num)
 633   (:policy :fast-safe)
 634   (:note "inline ASH")
 635   (:generator 3
 636     (move result number)
 637     (move ecx amount)
 638     ;; The result-type ensures us that this shift will not overflow.
 639     (inst shl result :cl)))
 640
 641 (define-vop (fast-ash-c/signed=>signed)
 642   (:translate ash)
 643   (:policy :fast-safe)
 644   (:args (number :scs (signed-reg) :target result
 645                  :load-if (not (and (sc-is number signed-stack)
 646                                     (sc-is result signed-stack)
 647                                     (location= number result)))))
 648   (:info amount)
 649   (:arg-types signed-num (:constant integer))
 650   (:results (result :scs (signed-reg)
 651                     :load-if (not (and (sc-is number signed-stack)
 652                                        (sc-is result signed-stack)
 653                                        (location= number result)))))
 654   (:result-types signed-num)
 655   (:note "inline ASH")
 656   (:generator 3
 657     (cond ((and (= amount 1) (not (location= number result)))
 658            (inst lea result (make-ea :dword :index number :scale 2)))
 659           ((and (= amount 2) (not (location= number result)))
 660            (inst lea result (make-ea :dword :index number :scale 4)))
 661           ((and (= amount 3) (not (location= number result)))
 662            (inst lea result (make-ea :dword :index number :scale 8)))
 663           (t
 664            (move result number)
 665            (cond ((plusp amount) (inst shl result amount))
 666                  (t (inst sar result (min 31 (- amount)))))))))
 667
 668 (define-vop (fast-ash-c/unsigned=>unsigned)
 669   (:translate ash)
 670   (:policy :fast-safe)
 671   (:args (number :scs (unsigned-reg) :target result
 672                  :load-if (not (and (sc-is number unsigned-stack)
 673                                     (sc-is result unsigned-stack)
 674                                     (location= number result)))))
 675   (:info amount)
 676   (:arg-types unsigned-num (:constant integer))
 677   (:results (result :scs (unsigned-reg)
 678                     :load-if (not (and (sc-is number unsigned-stack)
 679                                        (sc-is result unsigned-stack)
 680                                        (location= number result)))))
 681   (:result-types unsigned-num)
 682   (:note "inline ASH")
 683   (:generator 3
 684     (cond ((and (= amount 1) (not (location= number result)))
 685            (inst lea result (make-ea :dword :index number :scale 2)))
 686           ((and (= amount 2) (not (location= number result)))
 687            (inst lea result (make-ea :dword :index number :scale 4)))
 688           ((and (= amount 3) (not (location= number result)))
 689            (inst lea result (make-ea :dword :index number :scale 8)))
 690           (t
 691            (move result number)
 692            (cond ((< -32 amount 32)
 693                   ;; this code is used both in ASH and ASH-MOD32, so
 694                   ;; be careful
 695                   (if (plusp amount)
 696                       (inst shl result amount)
 697                       (inst shr result (- amount))))
 698                  (t (if (sc-is result unsigned-reg)
 699                         (inst xor result result)
 700                         (inst mov result 0))))))))
 701
 702 (define-vop (fast-ash-left/signed=>signed)
 703   (:translate ash)
 704   (:args (number :scs (signed-reg) :target result
 705                  :load-if (not (and (sc-is number signed-stack)
 706                                     (sc-is result signed-stack)
 707                                     (location= number result))))
 708          (amount :scs (unsigned-reg) :target ecx))
 709   (:arg-types signed-num positive-fixnum)
 710   (:temporary (:sc unsigned-reg :offset ecx-offset :from (:argument 1)) ecx)
 711   (:results (result :scs (signed-reg) :from (:argument 0)
 712                     :load-if (not (and (sc-is number signed-stack)
 713                                        (sc-is result signed-stack)
 714                                        (location= number result)))))
 715   (:result-types signed-num)
 716   (:policy :fast-safe)
 717   (:note "inline ASH")
 718   (:generator 4
 719     (move result number)
 720     (move ecx amount)
 721     (inst shl result :cl)))
 722
 723 (define-vop (fast-ash-left/unsigned=>unsigned)
 724   (:translate ash)
 725   (:args (number :scs (unsigned-reg) :target result
 726                  :load-if (not (and (sc-is number unsigned-stack)
 727                                     (sc-is result unsigned-stack)
 728                                     (location= number result))))
 729          (amount :scs (unsigned-reg) :target ecx))
 730   (:arg-types unsigned-num positive-fixnum)
 731   (:temporary (:sc unsigned-reg :offset ecx-offset :from (:argument 1)) ecx)
 732   (:results (result :scs (unsigned-reg) :from (:argument 0)
 733                     :load-if (not (and (sc-is number unsigned-stack)
 734                                        (sc-is result unsigned-stack)
 735                                        (location= number result)))))
 736   (:result-types unsigned-num)
 737   (:policy :fast-safe)
 738   (:note "inline ASH")
 739   (:generator 4
 740     (move result number)
 741     (move ecx amount)
 742     (inst shl result :cl)))
 743
 744 (define-vop (fast-ash/signed=>signed)
 745   (:translate ash)
 746   (:policy :fast-safe)
 747   (:args (number :scs (signed-reg) :target result)
 748          (amount :scs (signed-reg) :target ecx))
 749   (:arg-types signed-num signed-num)
 750   (:results (result :scs (signed-reg) :from (:argument 0)))
 751   (:result-types signed-num)
 752   (:temporary (:sc signed-reg :offset ecx-offset :from (:argument 1)) ecx)
 753   (:note "inline ASH")
 754   (:generator 5
 755     (move result number)
 756     (move ecx amount)
 757     (inst or ecx ecx)
 758     (inst jmp :ns positive)
 759     (inst neg ecx)
 760     (inst cmp ecx 31)
 761     (inst jmp :be okay)
 762     (inst mov ecx 31)
 763     OKAY
 764     (inst sar result :cl)
 765     (inst jmp done)
 766
 767     POSITIVE
 768     ;; The result-type ensures us that this shift will not overflow.
 769     (inst shl result :cl)
 770
 771     DONE))
 772
 773 (define-vop (fast-ash/unsigned=>unsigned)
 774   (:translate ash)
 775   (:policy :fast-safe)
 776   (:args (number :scs (unsigned-reg) :target result)
 777          (amount :scs (signed-reg) :target ecx))
 778   (:arg-types unsigned-num signed-num)
 779   (:results (result :scs (unsigned-reg) :from (:argument 0)))
 780   (:result-types unsigned-num)
 781   (:temporary (:sc signed-reg :offset ecx-offset :from (:argument 1)) ecx)
 782   (:note "inline ASH")
 783   (:generator 5
 784     (move result number)
 785     (move ecx amount)
 786     (inst or ecx ecx)
 787     (inst jmp :ns positive)
 788     (inst neg ecx)
 789     (inst cmp ecx 31)
 790     (inst jmp :be okay)
 791     (inst xor result result)
 792     (inst jmp done)
 793     OKAY
 794     (inst shr result :cl)
 795     (inst jmp done)
 796
 797     POSITIVE
 798     ;; The result-type ensures us that this shift will not overflow.
 799     (inst shl result :cl)
 800
 801     DONE))
 802
 803 (in-package "SB!C")
 804
 805 (defknown %lea (integer integer (member 1 2 4 8) (signed-byte 32))
 806   integer
 807   (foldable flushable movable))
 808
 809 (defoptimizer (%lea derive-type) ((base index scale disp))
 810   (when (and (constant-lvar-p scale)
 811              (constant-lvar-p disp))
 812     (let ((scale (lvar-value scale))
 813           (disp (lvar-value disp))
 814           (base-type (lvar-type base))
 815           (index-type (lvar-type index)))
 816       (when (and (numeric-type-p base-type)
 817                  (numeric-type-p index-type))
 818         (let ((base-lo (numeric-type-low base-type))
 819               (base-hi (numeric-type-high base-type))
 820               (index-lo (numeric-type-low index-type))
 821               (index-hi (numeric-type-high index-type)))
 822           (make-numeric-type :class 'integer
 823                              :complexp :real
 824                              :low (when (and base-lo index-lo)
 825                                     (+ base-lo (* index-lo scale) disp))
 826                              :high (when (and base-hi index-hi)
 827                                      (+ base-hi (* index-hi scale) disp))))))))
 828
 829 (defun %lea (base index scale disp)
 830   (+ base (* index scale) disp))
 831
 832 (in-package "SB!VM")
 833
 834 (define-vop (%lea/unsigned=>unsigned)
 835   (:translate %lea)
 836   (:policy :fast-safe)
 837   (:args (base :scs (unsigned-reg))
 838          (index :scs (unsigned-reg)))
 839   (:info scale disp)
 840   (:arg-types unsigned-num unsigned-num
 841               (:constant (member 1 2 4 8))
 842               (:constant (signed-byte 32)))
 843   (:results (r :scs (unsigned-reg)))
 844   (:result-types unsigned-num)
 845   (:generator 5
 846     (inst lea r (make-ea :dword :base base :index index
 847                          :scale scale :disp disp))))
 848
 849 (define-vop (%lea/signed=>signed)
 850   (:translate %lea)
 851   (:policy :fast-safe)
 852   (:args (base :scs (signed-reg))
 853          (index :scs (signed-reg)))
 854   (:info scale disp)
 855   (:arg-types signed-num signed-num
 856               (:constant (member 1 2 4 8))
 857               (:constant (signed-byte 32)))
 858   (:results (r :scs (signed-reg)))
 859   (:result-types signed-num)
 860   (:generator 4
 861     (inst lea r (make-ea :dword :base base :index index
 862                          :scale scale :disp disp))))
 863
 864 (define-vop (%lea/fixnum=>fixnum)
 865   (:translate %lea)
 866   (:policy :fast-safe)
 867   (:args (base :scs (any-reg))
 868          (index :scs (any-reg)))
 869   (:info scale disp)
 870   (:arg-types tagged-num tagged-num
 871               (:constant (member 1 2 4 8))
 872               (:constant (signed-byte 32)))
 873   (:results (r :scs (any-reg)))
 874   (:result-types tagged-num)
 875   (:generator 3
 876     (inst lea r (make-ea :dword :base base :index index
 877                          :scale scale :disp disp))))
 878
 879 ;;; FIXME: before making knowledge of this too public, it needs to be
 880 ;;; fixed so that it's actually _faster_ than the non-CMOV version; at
 881 ;;; least on my Celeron-XXX laptop, this version is marginally slower
 882 ;;; than the above version with branches.  -- CSR, 2003-09-04
 883 (define-vop (fast-cmov-ash/unsigned=>unsigned)
 884   (:translate ash)
 885   (:policy :fast-safe)
 886   (:args (number :scs (unsigned-reg) :target result)
 887          (amount :scs (signed-reg) :target ecx))
 888   (:arg-types unsigned-num signed-num)
 889   (:results (result :scs (unsigned-reg) :from (:argument 0)))
 890   (:result-types unsigned-num)
 891   (:temporary (:sc signed-reg :offset ecx-offset :from (:argument 1)) ecx)
 892   (:temporary (:sc any-reg :from (:eval 0) :to (:eval 1)) zero)
 893   (:note "inline ASH")
 894   (:guard (member :cmov *backend-subfeatures*))
 895   (:generator 4
 896     (move result number)
 897     (move ecx amount)
 898     (inst or ecx ecx)
 899     (inst jmp :ns positive)
 900     (inst neg ecx)
 901     (inst xor zero zero)
 902     (inst shr result :cl)
 903     (inst cmp ecx 31)
 904     (inst cmov :nbe result zero)
 905     (inst jmp done)
 906
 907     POSITIVE
 908     ;; The result-type ensures us that this shift will not overflow.
 909     (inst shl result :cl)
 910
 911     DONE))
 912 \f
 913 (define-vop (signed-byte-32-len)
 914   (:translate integer-length)
 915   (:note "inline (signed-byte 32) integer-length")
 916   (:policy :fast-safe)
 917   (:args (arg :scs (signed-reg) :target res))
 918   (:arg-types signed-num)
 919   (:results (res :scs (unsigned-reg)))
 920   (:result-types unsigned-num)
 921   (:generator 28
 922     (move res arg)
 923     (inst cmp res 0)
 924     (inst jmp :ge POS)
 925     (inst not res)
 926     POS
 927     (inst bsr res res)
 928     (inst jmp :z zero)
 929     (inst inc res)
 930     (inst jmp done)
 931     ZERO
 932     (inst xor res res)
 933     DONE))
 934
 935 (define-vop (unsigned-byte-32-len)
 936   (:translate integer-length)
 937   (:note "inline (unsigned-byte 32) integer-length")
 938   (:policy :fast-safe)
 939   (:args (arg :scs (unsigned-reg)))
 940   (:arg-types unsigned-num)
 941   (:results (res :scs (unsigned-reg)))
 942   (:result-types unsigned-num)
 943   (:generator 26
 944     (inst bsr res arg)
 945     (inst jmp :z zero)
 946     (inst inc res)
 947     (inst jmp done)
 948     ZERO
 949     (inst xor res res)
 950     DONE))
 951
 952 (define-vop (unsigned-byte-32-count)
 953   (:translate logcount)
 954   (:note "inline (unsigned-byte 32) logcount")
 955   (:policy :fast-safe)
 956   (:args (arg :scs (unsigned-reg)))
 957   (:arg-types unsigned-num)
 958   (:results (result :scs (unsigned-reg)))
 959   (:result-types positive-fixnum)
 960   (:temporary (:sc unsigned-reg :from (:argument 0)) temp)
 961   (:generator 30
 962     (move result arg)
 963
 964     (inst mov temp result)
 965     (inst shr temp 1)
 966     (inst and result #x55555555)
 967     (inst and temp #x55555555)
 968     (inst add result temp)
 969
 970     (inst mov temp result)
 971     (inst shr temp 2)
 972     (inst and result #x33333333)
 973     (inst and temp #x33333333)
 974     (inst add result temp)
 975
 976     (inst mov temp result)
 977     (inst shr temp 4)
 978     (inst and result #x0f0f0f0f)
 979     (inst and temp #x0f0f0f0f)
 980     (inst add result temp)
 981
 982     (inst mov temp result)
 983     (inst shr temp 8)
 984     (inst and result #x00ff00ff)
 985     (inst and temp #x00ff00ff)
 986     (inst add result temp)
 987
 988     (inst mov temp result)
 989     (inst shr temp 16)
 990     (inst and result #x0000ffff)
 991     (inst and temp #x0000ffff)
 992     (inst add result temp)))
 993 \f
 994 ;;;; binary conditional VOPs
 995
 996 (define-vop (fast-conditional)
 997   (:conditional)
 998   (:info target not-p)
 999   (:effects)
1000   (:affected)
1001   (:policy :fast-safe))
1002
1003 (define-vop (fast-conditional/fixnum fast-conditional)
1004   (:args (x :scs (any-reg)
1005             :load-if (not (and (sc-is x control-stack)
1006                                (sc-is y any-reg))))
1007          (y :scs (any-reg control-stack)))
1008   (:arg-types tagged-num tagged-num)
1009   (:note "inline fixnum comparison"))
1010
1011 (define-vop (fast-conditional-c/fixnum fast-conditional/fixnum)
1012   (:args (x :scs (any-reg control-stack)))
1013   (:arg-types tagged-num (:constant (signed-byte 30)))
1014   (:info target not-p y))
1015
1016 (define-vop (fast-conditional/signed fast-conditional)
1017   (:args (x :scs (signed-reg)
1018             :load-if (not (and (sc-is x signed-stack)
1019                                (sc-is y signed-reg))))
1020          (y :scs (signed-reg signed-stack)))
1021   (:arg-types signed-num signed-num)
1022   (:note "inline (signed-byte 32) comparison"))
1023
1024 (define-vop (fast-conditional-c/signed fast-conditional/signed)
1025   (:args (x :scs (signed-reg signed-stack)))
1026   (:arg-types signed-num (:constant (signed-byte 32)))
1027   (:info target not-p y))
1028
1029 (define-vop (fast-conditional/unsigned fast-conditional)
1030   (:args (x :scs (unsigned-reg)
1031             :load-if (not (and (sc-is x unsigned-stack)
1032                                (sc-is y unsigned-reg))))
1033          (y :scs (unsigned-reg unsigned-stack)))
1034   (:arg-types unsigned-num unsigned-num)
1035   (:note "inline (unsigned-byte 32) comparison"))
1036
1037 (define-vop (fast-conditional-c/unsigned fast-conditional/unsigned)
1038   (:args (x :scs (unsigned-reg unsigned-stack)))
1039   (:arg-types unsigned-num (:constant (unsigned-byte 32)))
1040   (:info target not-p y))
1041
1042 (macrolet ((define-logtest-vops ()
1043              `(progn
1044                ,@(loop for suffix in '(/fixnum -c/fixnum
1045                                        /signed -c/signed
1046                                        /unsigned -c/unsigned)
1047                        for cost in '(4 3 6 5 6 5)
1048                        collect
1049                        `(define-vop (,(symbolicate "FAST-LOGTEST" suffix)
1050                                      ,(symbolicate "FAST-CONDITIONAL" suffix))
1051                          (:translate logtest)
1052                          (:generator ,cost
1053                           (inst test x ,(if (eq suffix '-c/fixnum)
1054                                             '(fixnumize y)
1055                                             'y))
1056                           (inst jmp (if not-p :e :ne) target)))))))
1057   (define-logtest-vops))
1058
1059 (defknown %logbitp (integer unsigned-byte) boolean
1060   (movable foldable flushable))
1061
1062 (defun %logbitp (integer index)
1063   (logbitp index integer))
1064
1065 ;;; too much work to do the non-constant case (maybe?)
1066 (define-vop (fast-logbitp-c/fixnum fast-conditional-c/fixnum)
1067   (:translate %logbitp)
1068   (:arg-types tagged-num (:constant (integer 0 29)))
1069   (:generator 4
1070     (inst bt x (+ y n-fixnum-tag-bits))
1071     (inst jmp (if not-p :nc :c) target)))
1072
1073 (define-vop (fast-logbitp/signed fast-conditional/signed)
1074   (:args (x :scs (signed-reg signed-stack))
1075          (y :scs (signed-reg)))
1076   (:translate %logbitp)
1077   (:generator 6
1078     (inst bt x y)
1079     (inst jmp (if not-p :nc :c) target)))
1080
1081 (define-vop (fast-logbitp-c/signed fast-conditional-c/signed)
1082   (:translate %logbitp)
1083   (:arg-types signed-num (:constant (integer 0 31)))
1084   (:generator 5
1085     (inst bt x y)
1086     (inst jmp (if not-p :nc :c) target)))
1087
1088 (define-vop (fast-logbitp/unsigned fast-conditional/unsigned)
1089   (:args (x :scs (unsigned-reg unsigned-stack))
1090          (y :scs (unsigned-reg)))
1091   (:translate %logbitp)
1092   (:generator 6
1093     (inst bt x y)
1094     (inst jmp (if not-p :nc :c) target)))
1095
1096 (define-vop (fast-logbitp-c/unsigned fast-conditional-c/unsigned)
1097   (:translate %logbitp)
1098   (:arg-types unsigned-num (:constant (integer 0 31)))
1099   (:generator 5
1100     (inst bt x y)
1101     (inst jmp (if not-p :nc :c) target)))
1102
1103 (macrolet ((define-conditional-vop (tran cond unsigned not-cond not-unsigned)
1104              `(progn
1105                 ,@(mapcar
1106                    (lambda (suffix cost signed)
1107                      `(define-vop (;; FIXME: These could be done more
1108                                    ;; cleanly with SYMBOLICATE.
1109                                    ,(intern (format nil "~:@(FAST-IF-~A~A~)"
1110                                                     tran suffix))
1111                                    ,(intern
1112                                      (format nil "~:@(FAST-CONDITIONAL~A~)"
1113                                              suffix)))
1114                         (:translate ,tran)
1115                         (:generator ,cost
1116                                     (inst cmp x
1117                                           ,(if (eq suffix '-c/fixnum)
1118                                                '(fixnumize y)
1119                                                'y))
1120                                     (inst jmp (if not-p
1121                                                   ,(if signed
1122                                                        not-cond
1123                                                        not-unsigned)
1124                                                   ,(if signed
1125                                                        cond
1126                                                        unsigned))
1127                                           target))))
1128                    '(/fixnum -c/fixnum /signed -c/signed /unsigned -c/unsigned)
1129                    '(4 3 6 5 6 5)
1130                    '(t t t t nil nil)))))
1131
1132   (define-conditional-vop < :l :b :ge :ae)
1133   (define-conditional-vop > :g :a :le :be))
1134
1135 (define-vop (fast-if-eql/signed fast-conditional/signed)
1136   (:translate eql)
1137   (:generator 6
1138     (inst cmp x y)
1139     (inst jmp (if not-p :ne :e) target)))
1140
1141 (define-vop (fast-if-eql-c/signed fast-conditional-c/signed)
1142   (:translate eql)
1143   (:generator 5
1144     (cond ((and (sc-is x signed-reg) (zerop y))
1145            (inst test x x))  ; smaller instruction
1146           (t
1147            (inst cmp x y)))
1148     (inst jmp (if not-p :ne :e) target)))
1149
1150 (define-vop (fast-if-eql/unsigned fast-conditional/unsigned)
1151   (:translate eql)
1152   (:generator 6
1153     (inst cmp x y)
1154     (inst jmp (if not-p :ne :e) target)))
1155
1156 (define-vop (fast-if-eql-c/unsigned fast-conditional-c/unsigned)
1157   (:translate eql)
1158   (:generator 5
1159     (cond ((and (sc-is x unsigned-reg) (zerop y))
1160            (inst test x x))  ; smaller instruction
1161           (t
1162            (inst cmp x y)))
1163     (inst jmp (if not-p :ne :e) target)))
1164
1165 ;;; EQL/FIXNUM is funny because the first arg can be of any type, not just a
1166 ;;; known fixnum.
1167
1168 ;;; These versions specify a fixnum restriction on their first arg. We have
1169 ;;; also generic-eql/fixnum VOPs which are the same, but have no restriction on
1170 ;;; the first arg and a higher cost. The reason for doing this is to prevent
1171 ;;; fixnum specific operations from being used on word integers, spuriously
1172 ;;; consing the argument.
1173
1174 (define-vop (fast-eql/fixnum fast-conditional)
1175   (:args (x :scs (any-reg)
1176             :load-if (not (and (sc-is x control-stack)
1177                                (sc-is y any-reg))))
1178          (y :scs (any-reg control-stack)))
1179   (:arg-types tagged-num tagged-num)
1180   (:note "inline fixnum comparison")
1181   (:translate eql)
1182   (:generator 4
1183     (inst cmp x y)
1184     (inst jmp (if not-p :ne :e) target)))
1185 (define-vop (generic-eql/fixnum fast-eql/fixnum)
1186   (:args (x :scs (any-reg descriptor-reg)
1187             :load-if (not (and (sc-is x control-stack)
1188                                (sc-is y any-reg))))
1189          (y :scs (any-reg control-stack)))
1190   (:arg-types * tagged-num)
1191   (:variant-cost 7))
1192
1193 (define-vop (fast-eql-c/fixnum fast-conditional/fixnum)
1194   (:args (x :scs (any-reg control-stack)))
1195   (:arg-types tagged-num (:constant (signed-byte 30)))
1196   (:info target not-p y)
1197   (:translate eql)
1198   (:generator 2
1199     (cond ((and (sc-is x any-reg) (zerop y))
1200            (inst test x x))  ; smaller instruction
1201           (t
1202            (inst cmp x (fixnumize y))))
1203     (inst jmp (if not-p :ne :e) target)))
1204 (define-vop (generic-eql-c/fixnum fast-eql-c/fixnum)
1205   (:args (x :scs (any-reg descriptor-reg control-stack)))
1206   (:arg-types * (:constant (signed-byte 30)))
1207   (:variant-cost 6))
1208 \f
1209 ;;;; 32-bit logical operations
1210
1211 (define-vop (merge-bits)
1212   (:translate merge-bits)
1213   (:args (shift :scs (signed-reg unsigned-reg) :target ecx)
1214          (prev :scs (unsigned-reg) :target result)
1215          (next :scs (unsigned-reg)))
1216   (:arg-types tagged-num unsigned-num unsigned-num)
1217   (:temporary (:sc signed-reg :offset ecx-offset :from (:argument 0)) ecx)
1218   (:results (result :scs (unsigned-reg) :from (:argument 1)))
1219   (:result-types unsigned-num)
1220   (:policy :fast-safe)
1221   (:generator 4
1222     (move ecx shift)
1223     (move result prev)
1224     (inst shrd result next :cl)))
1225
1226 ;;; Only the lower 5 bits of the shift amount are significant.
1227 (define-vop (shift-towards-someplace)
1228   (:policy :fast-safe)
1229   (:args (num :scs (unsigned-reg) :target r)
1230          (amount :scs (signed-reg) :target ecx))
1231   (:arg-types unsigned-num tagged-num)
1232   (:temporary (:sc signed-reg :offset ecx-offset :from (:argument 1)) ecx)
1233   (:results (r :scs (unsigned-reg) :from (:argument 0)))
1234   (:result-types unsigned-num))
1235
1236 (define-vop (shift-towards-start shift-towards-someplace)
1237   (:translate shift-towards-start)
1238   (:note "SHIFT-TOWARDS-START")
1239   (:generator 1
1240     (move r num)
1241     (move ecx amount)
1242     (inst shr r :cl)))
1243
1244 (define-vop (shift-towards-end shift-towards-someplace)
1245   (:translate shift-towards-end)
1246   (:note "SHIFT-TOWARDS-END")
1247   (:generator 1
1248     (move r num)
1249     (move ecx amount)
1250     (inst shl r :cl)))
1251 \f
1252 ;;;; Modular functions
1253 (defmacro define-mod-binop ((name prototype) function)
1254   `(define-vop (,name ,prototype)
1255        (:args (x :target r :scs (unsigned-reg signed-reg)
1256                  :load-if (not (and (or (sc-is x unsigned-stack)
1257                                         (sc-is x signed-stack))
1258                                     (or (sc-is y unsigned-reg)
1259                                         (sc-is y signed-reg))
1260                                     (or (sc-is r unsigned-stack)
1261                                         (sc-is r signed-stack))
1262                                     (location= x r))))
1263               (y :scs (unsigned-reg signed-reg unsigned-stack signed-stack)))
1264      (:arg-types untagged-num untagged-num)
1265      (:results (r :scs (unsigned-reg signed-reg) :from (:argument 0)
1266                   :load-if (not (and (or (sc-is x unsigned-stack)
1267                                          (sc-is x signed-stack))
1268                                      (or (sc-is y unsigned-reg)
1269                                          (sc-is y unsigned-reg))
1270                                      (or (sc-is r unsigned-stack)
1271                                          (sc-is r unsigned-stack))
1272                                      (location= x r)))))
1273      (:result-types unsigned-num)
1274      (:translate ,function)))
1275 (defmacro define-mod-binop-c ((name prototype) function)
1276   `(define-vop (,name ,prototype)
1277        (:args (x :target r :scs (unsigned-reg signed-reg)
1278                  :load-if (not (and (or (sc-is x unsigned-stack)
1279                                         (sc-is x signed-stack))
1280                                     (or (sc-is r unsigned-stack)
1281                                         (sc-is r signed-stack))
1282                                     (location= x r)))))
1283      (:info y)
1284      (:arg-types untagged-num (:constant (or (unsigned-byte 32) (signed-byte 32))))
1285      (:results (r :scs (unsigned-reg signed-reg) :from (:argument 0)
1286                   :load-if (not (and (or (sc-is x unsigned-stack)
1287                                          (sc-is x signed-stack))
1288                                      (or (sc-is r unsigned-stack)
1289                                          (sc-is r unsigned-stack))
1290                                      (location= x r)))))
1291      (:result-types unsigned-num)
1292      (:translate ,function)))
1293
1294 (macrolet ((def (name -c-p)
1295              (let ((fun32 (intern (format nil "~S-MOD32" name)))
1296                    (vopu (intern (format nil "FAST-~S/UNSIGNED=>UNSIGNED" name)))
1297                    (vopcu (intern (format nil "FAST-~S-C/UNSIGNED=>UNSIGNED" name)))
1298                    (vopf (intern (format nil "FAST-~S/FIXNUM=>FIXNUM" name)))
1299                    (vopcf (intern (format nil "FAST-~S-C/FIXNUM=>FIXNUM" name)))
1300                    (vop32u (intern (format nil "FAST-~S-MOD32/WORD=>UNSIGNED" name)))
1301                    (vop32f (intern (format nil "FAST-~S-MOD32/FIXNUM=>FIXNUM" name)))
1302                    (vop32cu (intern (format nil "FAST-~S-MOD32-C/WORD=>UNSIGNED" name)))
1303                    (vop32cf (intern (format nil "FAST-~S-MOD32-C/FIXNUM=>FIXNUM" name)))
1304                    (sfun30 (intern (format nil "~S-SMOD30" name)))
1305                    (svop30f (intern (format nil "FAST-~S-SMOD30/FIXNUM=>FIXNUM" name)))
1306                    (svop30cf (intern (format nil "FAST-~S-SMOD30-C/FIXNUM=>FIXNUM" name))))
1307                `(progn
1308                   (define-modular-fun ,fun32 (x y) ,name :unsigned 32)
1309                   (define-modular-fun ,sfun30 (x y) ,name :signed 30)
1310                   (define-mod-binop (,vop32u ,vopu) ,fun32)
1311                   (define-vop (,vop32f ,vopf) (:translate ,fun32))
1312                   (define-vop (,svop30f ,vopf) (:translate ,sfun30))
1313                   ,@(when -c-p
1314                       `((define-mod-binop-c (,vop32cu ,vopcu) ,fun32)
1315                         (define-vop (,svop30cf ,vopcf) (:translate ,sfun30))))))))
1316   (def + t)
1317   (def - t)
1318   ;; (no -C variant as x86 MUL instruction doesn't take an immediate)
1319   (def * nil))
1320
1321
1322 (define-vop (fast-ash-left-mod32-c/unsigned=>unsigned
1323              fast-ash-c/unsigned=>unsigned)
1324   (:translate ash-left-mod32))
1325
1326 (define-vop (fast-ash-left-mod32/unsigned=>unsigned
1327              fast-ash-left/unsigned=>unsigned))
1328 (deftransform ash-left-mod32 ((integer count)
1329                               ((unsigned-byte 32) (unsigned-byte 5)))
1330   (when (sb!c::constant-lvar-p count)
1331     (sb!c::give-up-ir1-transform))
1332   '(%primitive fast-ash-left-mod32/unsigned=>unsigned integer count))
1333
1334 (define-vop (fast-ash-left-smod30-c/fixnum=>fixnum
1335              fast-ash-c/fixnum=>fixnum)
1336   (:translate ash-left-smod30))
1337
1338 (define-vop (fast-ash-left-smod30/fixnum=>fixnum
1339              fast-ash-left/fixnum=>fixnum))
1340 (deftransform ash-left-smod30 ((integer count)
1341                                ((signed-byte 30) (unsigned-byte 5)))
1342   (when (sb!c::constant-lvar-p count)
1343     (sb!c::give-up-ir1-transform))
1344   '(%primitive fast-ash-left-smod30/fixnum=>fixnum integer count))
1345
1346 (in-package "SB!C")
1347
1348 (defknown sb!vm::%lea-mod32 (integer integer (member 1 2 4 8) (signed-byte 32))
1349   (unsigned-byte 32)
1350   (foldable flushable movable))
1351 (defknown sb!vm::%lea-smod30 (integer integer (member 1 2 4 8) (signed-byte 32))
1352   (signed-byte 30)
1353   (foldable flushable movable))
1354
1355 (define-modular-fun-optimizer %lea ((base index scale disp) :unsigned :width width)
1356   (when (and (<= width 32)
1357              (constant-lvar-p scale)
1358              (constant-lvar-p disp))
1359     (cut-to-width base :unsigned width)
1360     (cut-to-width index :unsigned width)
1361     'sb!vm::%lea-mod32))
1362 (define-modular-fun-optimizer %lea ((base index scale disp) :signed :width width)
1363   (when (and (<= width 30)
1364              (constant-lvar-p scale)
1365              (constant-lvar-p disp))
1366     (cut-to-width base :signed width)
1367     (cut-to-width index :signed width)
1368     'sb!vm::%lea-smod30))
1369
1370 #+sb-xc-host
1371 (progn
1372   (defun sb!vm::%lea-mod32 (base index scale disp)
1373     (ldb (byte 32 0) (%lea base index scale disp)))
1374   (defun sb!vm::%lea-smod30 (base index scale disp)
1375     (mask-signed-field 30 (%lea base index scale disp))))
1376 #-sb-xc-host
1377 (progn
1378   (defun sb!vm::%lea-mod32 (base index scale disp)
1379     (let ((base (logand base #xffffffff))
1380           (index (logand index #xffffffff)))
1381       ;; can't use modular version of %LEA, as we only have VOPs for
1382       ;; constant SCALE and DISP.
1383       (ldb (byte 32 0) (+ base (* index scale) disp))))
1384   (defun sb!vm::%lea-smod30 (base index scale disp)
1385     (let ((base (mask-signed-field 30 base))
1386           (index (mask-signed-field 30 index)))
1387       ;; can't use modular version of %LEA, as we only have VOPs for
1388       ;; constant SCALE and DISP.
1389       (mask-signed-field 30 (+ base (* index scale) disp)))))
1390
1391 (in-package "SB!VM")
1392
1393 (define-vop (%lea-mod32/unsigned=>unsigned
1394              %lea/unsigned=>unsigned)
1395   (:translate %lea-mod32))
1396 (define-vop (%lea-smod30/fixnum=>fixnum
1397              %lea/fixnum=>fixnum)
1398   (:translate %lea-smod30))
1399
1400 ;;; logical operations
1401 (define-modular-fun lognot-mod32 (x) lognot :unsigned 32)
1402 (define-vop (lognot-mod32/word=>unsigned)
1403   (:translate lognot-mod32)
1404   (:args (x :scs (unsigned-reg signed-reg unsigned-stack signed-stack) :target r
1405             :load-if (not (and (or (sc-is x unsigned-stack)
1406                                    (sc-is x signed-stack))
1407                                (or (sc-is r unsigned-stack)
1408                                    (sc-is r signed-stack))
1409                                (location= x r)))))
1410   (:arg-types unsigned-num)
1411   (:results (r :scs (unsigned-reg)
1412                :load-if (not (and (or (sc-is x unsigned-stack)
1413                                       (sc-is x signed-stack))
1414                                   (or (sc-is r unsigned-stack)
1415                                       (sc-is r signed-stack))
1416                                   (sc-is r unsigned-stack)
1417                                   (location= x r)))))
1418   (:result-types unsigned-num)
1419   (:policy :fast-safe)
1420   (:generator 1
1421     (move r x)
1422     (inst not r)))
1423
1424 (define-modular-fun logxor-mod32 (x y) logxor :unsigned 32)
1425 (define-mod-binop (fast-logxor-mod32/word=>unsigned
1426                    fast-logxor/unsigned=>unsigned)
1427     logxor-mod32)
1428 (define-mod-binop-c (fast-logxor-mod32-c/word=>unsigned
1429                      fast-logxor-c/unsigned=>unsigned)
1430     logxor-mod32)
1431 (define-vop (fast-logxor-mod32/fixnum=>fixnum
1432              fast-logxor/fixnum=>fixnum)
1433   (:translate logxor-mod32))
1434 (define-vop (fast-logxor-mod32-c/fixnum=>fixnum
1435              fast-logxor-c/fixnum=>fixnum)
1436   (:translate logxor-mod32))
1437
1438 (define-source-transform logeqv (&rest args)
1439   (if (oddp (length args))
1440       `(logxor ,@args)
1441       `(lognot (logxor ,@args))))
1442 (define-source-transform logandc1 (x y)
1443   `(logand (lognot ,x) ,y))
1444 (define-source-transform logandc2 (x y)
1445   `(logand ,x (lognot ,y)))
1446 (define-source-transform logorc1 (x y)
1447   `(logior (lognot ,x) ,y))
1448 (define-source-transform logorc2 (x y)
1449   `(logior ,x (lognot ,y)))
1450 (define-source-transform lognor (x y)
1451   `(lognot (logior ,x ,y)))
1452 (define-source-transform lognand (x y)
1453   `(lognot (logand ,x ,y)))
1454 \f
1455 ;;;; bignum stuff
1456
1457 (define-vop (bignum-length get-header-data)
1458   (:translate sb!bignum:%bignum-length)
1459   (:policy :fast-safe))
1460
1461 (define-vop (bignum-set-length set-header-data)
1462   (:translate sb!bignum:%bignum-set-length)
1463   (:policy :fast-safe))
1464
1465 (define-full-reffer bignum-ref * bignum-digits-offset other-pointer-lowtag
1466   (unsigned-reg) unsigned-num sb!bignum:%bignum-ref)
1467
1468 (define-full-setter bignum-set * bignum-digits-offset other-pointer-lowtag
1469   (unsigned-reg) unsigned-num sb!bignum:%bignum-set)
1470
1471 (define-vop (digit-0-or-plus)
1472   (:translate sb!bignum:%digit-0-or-plusp)
1473   (:policy :fast-safe)
1474   (:args (digit :scs (unsigned-reg)))
1475   (:arg-types unsigned-num)
1476   (:conditional)
1477   (:info target not-p)
1478   (:generator 3
1479     (inst or digit digit)
1480     (inst jmp (if not-p :s :ns) target)))
1481
1482
1483 ;;; For add and sub with carry the sc of carry argument is any-reg so
1484 ;;; that it may be passed as a fixnum or word and thus may be 0, 1, or
1485 ;;; 4. This is easy to deal with and may save a fixnum-word
1486 ;;; conversion.
1487 (define-vop (add-w/carry)
1488   (:translate sb!bignum:%add-with-carry)
1489   (:policy :fast-safe)
1490   (:args (a :scs (unsigned-reg) :target result)
1491          (b :scs (unsigned-reg unsigned-stack) :to :eval)
1492          (c :scs (any-reg) :target temp))
1493   (:arg-types unsigned-num unsigned-num positive-fixnum)
1494   (:temporary (:sc any-reg :from (:argument 2) :to :eval) temp)
1495   (:results (result :scs (unsigned-reg) :from (:argument 0))
1496             (carry :scs (unsigned-reg)))
1497   (:result-types unsigned-num positive-fixnum)
1498   (:generator 4
1499     (move result a)
1500     (move temp c)
1501     (inst neg temp) ; Set the carry flag to 0 if c=0 else to 1
1502     (inst adc result b)
1503     (inst mov carry 0)
1504     (inst adc carry carry)))
1505
1506 ;;; Note: the borrow is 1 for no borrow and 0 for a borrow, the opposite
1507 ;;; of the x86 convention.
1508 (define-vop (sub-w/borrow)
1509   (:translate sb!bignum:%subtract-with-borrow)
1510   (:policy :fast-safe)
1511   (:args (a :scs (unsigned-reg) :to :eval :target result)
1512          (b :scs (unsigned-reg unsigned-stack) :to :result)
1513          (c :scs (any-reg control-stack)))
1514   (:arg-types unsigned-num unsigned-num positive-fixnum)
1515   (:results (result :scs (unsigned-reg) :from :eval)
1516             (borrow :scs (unsigned-reg)))
1517   (:result-types unsigned-num positive-fixnum)
1518   (:generator 5
1519     (inst cmp c 1) ; Set the carry flag to 1 if c=0 else to 0
1520     (move result a)
1521     (inst sbb result b)
1522     (inst mov borrow 1)
1523     (inst sbb borrow 0)))
1524
1525
1526 (define-vop (bignum-mult-and-add-3-arg)
1527   (:translate sb!bignum:%multiply-and-add)
1528   (:policy :fast-safe)
1529   (:args (x :scs (unsigned-reg) :target eax)
1530          (y :scs (unsigned-reg unsigned-stack))
1531          (carry-in :scs (unsigned-reg unsigned-stack)))
1532   (:arg-types unsigned-num unsigned-num unsigned-num)
1533   (:temporary (:sc unsigned-reg :offset eax-offset :from (:argument 0)
1534                    :to (:result 1) :target lo) eax)
1535   (:temporary (:sc unsigned-reg :offset edx-offset :from (:argument 1)
1536                    :to (:result 0) :target hi) edx)
1537   (:results (hi :scs (unsigned-reg))
1538             (lo :scs (unsigned-reg)))
1539   (:result-types unsigned-num unsigned-num)
1540   (:generator 20
1541     (move eax x)
1542     (inst mul eax y)
1543     (inst add eax carry-in)
1544     (inst adc edx 0)
1545     (move hi edx)
1546     (move lo eax)))
1547
1548 (define-vop (bignum-mult-and-add-4-arg)
1549   (:translate sb!bignum:%multiply-and-add)
1550   (:policy :fast-safe)
1551   (:args (x :scs (unsigned-reg) :target eax)
1552          (y :scs (unsigned-reg unsigned-stack))
1553          (prev :scs (unsigned-reg unsigned-stack))
1554          (carry-in :scs (unsigned-reg unsigned-stack)))
1555   (:arg-types unsigned-num unsigned-num unsigned-num unsigned-num)
1556   (:temporary (:sc unsigned-reg :offset eax-offset :from (:argument 0)
1557                    :to (:result 1) :target lo) eax)
1558   (:temporary (:sc unsigned-reg :offset edx-offset :from (:argument 1)
1559                    :to (:result 0) :target hi) edx)
1560   (:results (hi :scs (unsigned-reg))
1561             (lo :scs (unsigned-reg)))
1562   (:result-types unsigned-num unsigned-num)
1563   (:generator 20
1564     (move eax x)
1565     (inst mul eax y)
1566     (inst add eax prev)
1567     (inst adc edx 0)
1568     (inst add eax carry-in)
1569     (inst adc edx 0)
1570     (move hi edx)
1571     (move lo eax)))
1572
1573
1574 (define-vop (bignum-mult)
1575   (:translate sb!bignum:%multiply)
1576   (:policy :fast-safe)
1577   (:args (x :scs (unsigned-reg) :target eax)
1578          (y :scs (unsigned-reg unsigned-stack)))
1579   (:arg-types unsigned-num unsigned-num)
1580   (:temporary (:sc unsigned-reg :offset eax-offset :from (:argument 0)
1581                    :to (:result 1) :target lo) eax)
1582   (:temporary (:sc unsigned-reg :offset edx-offset :from (:argument 1)
1583                    :to (:result 0) :target hi) edx)
1584   (:results (hi :scs (unsigned-reg))
1585             (lo :scs (unsigned-reg)))
1586   (:result-types unsigned-num unsigned-num)
1587   (:generator 20
1588     (move eax x)
1589     (inst mul eax y)
1590     (move hi edx)
1591     (move lo eax)))
1592
1593 (define-vop (bignum-lognot lognot-mod32/word=>unsigned)
1594   (:translate sb!bignum:%lognot))
1595
1596 (define-vop (fixnum-to-digit)
1597   (:translate sb!bignum:%fixnum-to-digit)
1598   (:policy :fast-safe)
1599   (:args (fixnum :scs (any-reg control-stack) :target digit))
1600   (:arg-types tagged-num)
1601   (:results (digit :scs (unsigned-reg)
1602                    :load-if (not (and (sc-is fixnum control-stack)
1603                                       (sc-is digit unsigned-stack)
1604                                       (location= fixnum digit)))))
1605   (:result-types unsigned-num)
1606   (:generator 1
1607     (move digit fixnum)
1608     (inst sar digit 2)))
1609
1610 (define-vop (bignum-floor)
1611   (:translate sb!bignum:%floor)
1612   (:policy :fast-safe)
1613   (:args (div-high :scs (unsigned-reg) :target edx)
1614          (div-low :scs (unsigned-reg) :target eax)
1615          (divisor :scs (unsigned-reg unsigned-stack)))
1616   (:arg-types unsigned-num unsigned-num unsigned-num)
1617   (:temporary (:sc unsigned-reg :offset eax-offset :from (:argument 1)
1618                    :to (:result 0) :target quo) eax)
1619   (:temporary (:sc unsigned-reg :offset edx-offset :from (:argument 0)
1620                    :to (:result 1) :target rem) edx)
1621   (:results (quo :scs (unsigned-reg))
1622             (rem :scs (unsigned-reg)))
1623   (:result-types unsigned-num unsigned-num)
1624   (:generator 300
1625     (move edx div-high)
1626     (move eax div-low)
1627     (inst div eax divisor)
1628     (move quo eax)
1629     (move rem edx)))
1630
1631 (define-vop (signify-digit)
1632   (:translate sb!bignum:%fixnum-digit-with-correct-sign)
1633   (:policy :fast-safe)
1634   (:args (digit :scs (unsigned-reg unsigned-stack) :target res))
1635   (:arg-types unsigned-num)
1636   (:results (res :scs (any-reg signed-reg)
1637                  :load-if (not (and (sc-is digit unsigned-stack)
1638                                     (sc-is res control-stack signed-stack)
1639                                     (location= digit res)))))
1640   (:result-types signed-num)
1641   (:generator 1
1642     (move res digit)
1643     (when (sc-is res any-reg control-stack)
1644       (inst shl res 2))))
1645
1646 (define-vop (digit-ashr)
1647   (:translate sb!bignum:%ashr)
1648   (:policy :fast-safe)
1649   (:args (digit :scs (unsigned-reg unsigned-stack) :target result)
1650          (count :scs (unsigned-reg) :target ecx))
1651   (:arg-types unsigned-num positive-fixnum)
1652   (:temporary (:sc unsigned-reg :offset ecx-offset :from (:argument 1)) ecx)
1653   (:results (result :scs (unsigned-reg) :from (:argument 0)
1654                     :load-if (not (and (sc-is result unsigned-stack)
1655                                        (location= digit result)))))
1656   (:result-types unsigned-num)
1657   (:generator 1
1658     (move result digit)
1659     (move ecx count)
1660     (inst sar result :cl)))
1661
1662 (define-vop (digit-lshr digit-ashr)
1663   (:translate sb!bignum:%digit-logical-shift-right)
1664   (:generator 1
1665     (move result digit)
1666     (move ecx count)
1667     (inst shr result :cl)))
1668
1669 (define-vop (digit-ashl digit-ashr)
1670   (:translate sb!bignum:%ashl)
1671   (:generator 1
1672     (move result digit)
1673     (move ecx count)
1674     (inst shl result :cl)))
1675 \f
1676 ;;;; static functions
1677
1678 (define-static-fun two-arg-/ (x y) :translate /)
1679
1680 (define-static-fun two-arg-gcd (x y) :translate gcd)
1681 (define-static-fun two-arg-lcm (x y) :translate lcm)
1682
1683 (define-static-fun two-arg-and (x y) :translate logand)
1684 (define-static-fun two-arg-ior (x y) :translate logior)
1685 (define-static-fun two-arg-xor (x y) :translate logxor)
1686
1687 \f
1688 ;;; Support for the Mersenne Twister, MT19937, random number generator
1689 ;;; due to Matsumoto and Nishimura.
1690 ;;;
1691 ;;; Makoto Matsumoto and T. Nishimura, "Mersenne twister: A
1692 ;;; 623-dimensionally equidistributed uniform pseudorandom number
1693 ;;; generator.", ACM Transactions on Modeling and Computer Simulation,
1694 ;;; 1997, to appear.
1695 ;;;
1696 ;;; State:
1697 ;;;  0-1:   Constant matrix A. [0, #x9908b0df] (not used here)
1698 ;;;  2:     Index; init. to 1.
1699 ;;;  3-626: State.
1700 (defknown random-mt19937 ((simple-array (unsigned-byte 32) (*)))
1701   (unsigned-byte 32) ())
1702 (define-vop (random-mt19937)
1703   (:policy :fast-safe)
1704   (:translate random-mt19937)
1705   (:args (state :scs (descriptor-reg) :to :result))
1706   (:arg-types simple-array-unsigned-byte-32)
1707   (:temporary (:sc unsigned-reg :from (:eval 0) :to :result) k)
1708   (:temporary (:sc unsigned-reg :offset eax-offset
1709                    :from (:eval 0) :to :result) tmp)
1710   (:results (y :scs (unsigned-reg) :from (:eval 0)))
1711   (:result-types unsigned-num)
1712   (:generator 50
1713     (inst mov k (make-ea :dword :base state
1714                          :disp (- (* (+ 2 vector-data-offset)
1715                                      n-word-bytes)
1716                                   other-pointer-lowtag)))
1717     (inst cmp k 624)
1718     (inst jmp :ne no-update)
1719     (inst mov tmp state)        ; The state is passed in EAX.
1720     (inst call (make-fixup 'random-mt19937-update :assembly-routine))
1721     ;; Restore k, and set to 0.
1722     (inst xor k k)
1723     NO-UPDATE
1724     ;; y = ptgfsr[k++];
1725     (inst mov y (make-ea :dword :base state :index k :scale 4
1726                          :disp (- (* (+ 3 vector-data-offset)
1727                                      n-word-bytes)
1728                                   other-pointer-lowtag)))
1729     ;; y ^= (y >> 11);
1730     (inst shr y 11)
1731     (inst xor y (make-ea :dword :base state :index k :scale 4
1732                          :disp (- (* (+ 3 vector-data-offset)
1733                                      n-word-bytes)
1734                                   other-pointer-lowtag)))
1735     ;; y ^= (y << 7) & #x9d2c5680
1736     (inst mov tmp y)
1737     (inst inc k)
1738     (inst shl tmp 7)
1739     (inst mov (make-ea :dword :base state
1740                        :disp (- (* (+ 2 vector-data-offset)
1741                                    n-word-bytes)
1742                                 other-pointer-lowtag))
1743           k)
1744     (inst and tmp #x9d2c5680)
1745     (inst xor y tmp)
1746     ;; y ^= (y << 15) & #xefc60000
1747     (inst mov tmp y)
1748     (inst shl tmp 15)
1749     (inst and tmp #xefc60000)
1750     (inst xor y tmp)
1751     ;; y ^= (y >> 18);
1752     (inst mov tmp y)
1753     (inst shr tmp 18)
1754     (inst xor y tmp)))
1755
1756 (in-package "SB!C")
1757
1758 (defun mask-result (class width result)
1759   (ecase class
1760     (:unsigned
1761      `(logand ,result ,(1- (ash 1 width))))
1762     (:signed
1763      `(mask-signed-field ,width ,result))))
1764
1765 ;;; This is essentially a straight implementation of the algorithm in
1766 ;;; "Strength Reduction of Multiplications by Integer Constants",
1767 ;;; Youfeng Wu, ACM SIGPLAN Notices, Vol. 30, No.2, February 1995.
1768 (defun basic-decompose-multiplication (class width arg num n-bits condensed)
1769   (case (aref condensed 0)
1770     (0
1771      (let ((tmp (min 3 (aref condensed 1))))
1772        (decf (aref condensed 1) tmp)
1773        (mask-result class width
1774                     `(%lea ,arg
1775                            ,(decompose-multiplication class width
1776                              arg (ash (1- num) (- tmp)) (1- n-bits) (subseq condensed 1))
1777                            ,(ash 1 tmp) 0))))
1778     ((1 2 3)
1779      (let ((r0 (aref condensed 0)))
1780        (incf (aref condensed 1) r0)
1781        (mask-result class width
1782                     `(%lea ,(decompose-multiplication class width
1783                              arg (- num (ash 1 r0)) (1- n-bits) (subseq condensed 1))
1784                            ,arg
1785                            ,(ash 1 r0) 0))))
1786     (t (let ((r0 (aref condensed 0)))
1787          (setf (aref condensed 0) 0)
1788          (mask-result class width
1789                       `(ash ,(decompose-multiplication class width
1790                               arg (ash num (- r0)) n-bits condensed)
1791                             ,r0))))))
1792
1793 (defun decompose-multiplication (class width arg num n-bits condensed)
1794   (cond
1795     ((= n-bits 0) 0)
1796     ((= num 1) arg)
1797     ((= n-bits 1)
1798      (mask-result class width `(ash ,arg ,(1- (integer-length num)))))
1799     ((let ((max 0) (end 0))
1800        (loop for i from 2 to (length condensed)
1801              for j = (reduce #'+ (subseq condensed 0 i))
1802              when (and (> (- (* 2 i) 3 j) max)
1803                        (< (+ (ash 1 (1+ j))
1804                              (ash (ldb (byte (- 32 (1+ j)) (1+ j)) num)
1805                                   (1+ j)))
1806                           (ash 1 32)))
1807                do (setq max (- (* 2 i) 3 j)
1808                         end i))
1809        (when (> max 0)
1810          (let ((j (reduce #'+ (subseq condensed 0 end))))
1811            (let ((n2 (+ (ash 1 (1+ j))
1812                         (ash (ldb (byte (- 32 (1+ j)) (1+ j)) num) (1+ j))))
1813                  (n1 (1+ (ldb (byte (1+ j) 0) (lognot num)))))
1814            (mask-result class width
1815                         `(- ,(optimize-multiply class width arg n2)
1816                             ,(optimize-multiply  class width arg n1))))))))
1817     ((dolist (i '(9 5 3))
1818        (when (integerp (/ num i))
1819          (when (< (logcount (/ num i)) (logcount num))
1820            (let ((x (gensym)))
1821              (return `(let ((,x ,(optimize-multiply class width arg (/ num i))))
1822                        ,(mask-result class width
1823                                      `(%lea ,x ,x (1- ,i) 0)))))))))
1824     (t (basic-decompose-multiplication class width arg num n-bits condensed))))
1825
1826 (defun optimize-multiply (class width arg x)
1827   (let* ((n-bits (logcount x))
1828          (condensed (make-array n-bits)))
1829     (let ((count 0) (bit 0))
1830       (dotimes (i 32)
1831         (cond ((logbitp i x)
1832                (setf (aref condensed bit) count)
1833                (setf count 1)
1834                (incf bit))
1835               (t (incf count)))))
1836     (decompose-multiplication class width arg x n-bits condensed)))
1837
1838 (defun *-transformer (class width y)
1839   (cond
1840     ((= y (ash 1 (integer-length y)))
1841      ;; there's a generic transform for y = 2^k
1842      (give-up-ir1-transform))
1843     ((member y '(3 5 9))
1844      ;; we can do these multiplications directly using LEA
1845      `(%lea x x ,(1- y) 0))
1846     ((member :pentium4 *backend-subfeatures*)
1847      ;; the pentium4's multiply unit is reportedly very good
1848      (give-up-ir1-transform))
1849     ;; FIXME: should make this more fine-grained.  If nothing else,
1850     ;; there should probably be a cutoff of about 9 instructions on
1851     ;; pentium-class machines.
1852     (t (optimize-multiply class width 'x y))))
1853
1854 (deftransform * ((x y)
1855                  ((unsigned-byte 32) (constant-arg (unsigned-byte 32)))
1856                  (unsigned-byte 32))
1857   "recode as leas, shifts and adds"
1858   (let ((y (lvar-value y)))
1859     (*-transformer :unsigned 32 y)))
1860 (deftransform sb!vm::*-mod32
1861     ((x y) ((unsigned-byte 32) (constant-arg (unsigned-byte 32)))
1862      (unsigned-byte 32))
1863   "recode as leas, shifts and adds"
1864   (let ((y (lvar-value y)))
1865     (*-transformer :unsigned 32 y)))
1866
1867 (deftransform * ((x y)
1868                  ((signed-byte 30) (constant-arg (unsigned-byte 32)))
1869                  (signed-byte 30))
1870   "recode as leas, shifts and adds"
1871   (let ((y (lvar-value y)))
1872     (*-transformer :signed 30 y)))
1873 (deftransform sb!vm::*-smod30
1874     ((x y) ((signed-byte 30) (constant-arg (unsigned-byte 32)))
1875      (signed-byte 30))
1876   "recode as leas, shifts and adds"
1877   (let ((y (lvar-value y)))
1878     (*-transformer :signed 30 y)))
1879
1880 ;;; FIXME: we should also be able to write an optimizer or two to
1881 ;;; convert (+ (* x 2) 17), (- (* x 9) 5) to a %LEA.