rightArithShift.js 4.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125
  1. "use strict";
  2. Object.defineProperty(exports, "__esModule", {
  3. value: true
  4. });
  5. exports.createRightArithShift = void 0;
  6. var _bitwise = require("../../utils/bignumber/bitwise.js");
  7. var _matAlgo02xDS = require("../../type/matrix/utils/matAlgo02xDS0.js");
  8. var _matAlgo11xS0s = require("../../type/matrix/utils/matAlgo11xS0s.js");
  9. var _matAlgo14xDs = require("../../type/matrix/utils/matAlgo14xDs.js");
  10. var _matAlgo01xDSid = require("../../type/matrix/utils/matAlgo01xDSid.js");
  11. var _matAlgo10xSids = require("../../type/matrix/utils/matAlgo10xSids.js");
  12. var _matAlgo08xS0Sid = require("../../type/matrix/utils/matAlgo08xS0Sid.js");
  13. var _factory = require("../../utils/factory.js");
  14. var _matrixAlgorithmSuite = require("../../type/matrix/utils/matrixAlgorithmSuite.js");
  15. var _useMatrixForArrayScalar = require("./useMatrixForArrayScalar.js");
  16. var _index = require("../../plain/number/index.js");
  17. var name = 'rightArithShift';
  18. var dependencies = ['typed', 'matrix', 'equalScalar', 'zeros', 'DenseMatrix'];
  19. var createRightArithShift = /* #__PURE__ */(0, _factory.factory)(name, dependencies, function (_ref) {
  20. var typed = _ref.typed,
  21. matrix = _ref.matrix,
  22. equalScalar = _ref.equalScalar,
  23. zeros = _ref.zeros,
  24. DenseMatrix = _ref.DenseMatrix;
  25. var matAlgo01xDSid = (0, _matAlgo01xDSid.createMatAlgo01xDSid)({
  26. typed: typed
  27. });
  28. var matAlgo02xDS0 = (0, _matAlgo02xDS.createMatAlgo02xDS0)({
  29. typed: typed,
  30. equalScalar: equalScalar
  31. });
  32. var matAlgo08xS0Sid = (0, _matAlgo08xS0Sid.createMatAlgo08xS0Sid)({
  33. typed: typed,
  34. equalScalar: equalScalar
  35. });
  36. var matAlgo10xSids = (0, _matAlgo10xSids.createMatAlgo10xSids)({
  37. typed: typed,
  38. DenseMatrix: DenseMatrix
  39. });
  40. var matAlgo11xS0s = (0, _matAlgo11xS0s.createMatAlgo11xS0s)({
  41. typed: typed,
  42. equalScalar: equalScalar
  43. });
  44. var matAlgo14xDs = (0, _matAlgo14xDs.createMatAlgo14xDs)({
  45. typed: typed
  46. });
  47. var matrixAlgorithmSuite = (0, _matrixAlgorithmSuite.createMatrixAlgorithmSuite)({
  48. typed: typed,
  49. matrix: matrix
  50. });
  51. var useMatrixForArrayScalar = (0, _useMatrixForArrayScalar.createUseMatrixForArrayScalar)({
  52. typed: typed,
  53. matrix: matrix
  54. });
  55. /**
  56. * Bitwise right arithmetic shift of a value x by y number of bits, `x >> y`.
  57. * For matrices, the function is evaluated element wise.
  58. * For units, the function is evaluated on the best prefix base.
  59. *
  60. * Syntax:
  61. *
  62. * math.rightArithShift(x, y)
  63. *
  64. * Examples:
  65. *
  66. * math.rightArithShift(4, 2) // returns number 1
  67. *
  68. * math.rightArithShift([16, -32, 64], 4) // returns Array [1, -2, 4]
  69. *
  70. * See also:
  71. *
  72. * bitAnd, bitNot, bitOr, bitXor, rightArithShift, rightLogShift
  73. *
  74. * @param {number | BigNumber | Array | Matrix} x Value to be shifted
  75. * @param {number | BigNumber} y Amount of shifts
  76. * @return {number | BigNumber | Array | Matrix} `x` sign-filled shifted right `y` times
  77. */
  78. return typed(name, {
  79. 'number, number': _index.rightArithShiftNumber,
  80. 'BigNumber, BigNumber': _bitwise.rightArithShiftBigNumber,
  81. 'SparseMatrix, number | BigNumber': typed.referToSelf(function (self) {
  82. return function (x, y) {
  83. // check scalar
  84. if (equalScalar(y, 0)) {
  85. return x.clone();
  86. }
  87. return matAlgo11xS0s(x, y, self, false);
  88. };
  89. }),
  90. 'DenseMatrix, number | BigNumber': typed.referToSelf(function (self) {
  91. return function (x, y) {
  92. // check scalar
  93. if (equalScalar(y, 0)) {
  94. return x.clone();
  95. }
  96. return matAlgo14xDs(x, y, self, false);
  97. };
  98. }),
  99. 'number | BigNumber, SparseMatrix': typed.referToSelf(function (self) {
  100. return function (x, y) {
  101. // check scalar
  102. if (equalScalar(x, 0)) {
  103. return zeros(y.size(), y.storage());
  104. }
  105. return matAlgo10xSids(y, x, self, true);
  106. };
  107. }),
  108. 'number | BigNumber, DenseMatrix': typed.referToSelf(function (self) {
  109. return function (x, y) {
  110. // check scalar
  111. if (equalScalar(x, 0)) {
  112. return zeros(y.size(), y.storage());
  113. }
  114. return matAlgo14xDs(y, x, self, true);
  115. };
  116. })
  117. }, useMatrixForArrayScalar, matrixAlgorithmSuite({
  118. SS: matAlgo08xS0Sid,
  119. DS: matAlgo01xDSid,
  120. SD: matAlgo02xDS0
  121. }));
  122. });
  123. exports.createRightArithShift = createRightArithShift;