rightLogShift.js 3.5 KB

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