Greg Kroah-Hartman | b244131 | 2017-11-01 15:07:57 +0100 | [diff] [blame] | 1 | // SPDX-License-Identifier: GPL-2.0 |
Davidlohr Bueso | 30493cc | 2013-04-29 16:18:09 -0700 | [diff] [blame] | 2 | /* |
| 3 | * Copyright (C) 2013 Davidlohr Bueso <davidlohr.bueso@hp.com> |
| 4 | * |
| 5 | * Based on the shift-and-subtract algorithm for computing integer |
| 6 | * square root from Guy L. Steele. |
| 7 | */ |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 8 | |
| 9 | #include <linux/kernel.h> |
Paul Gortmaker | 8bc3bcc | 2011-11-16 21:29:17 -0500 | [diff] [blame] | 10 | #include <linux/export.h> |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 11 | |
| 12 | /** |
| 13 | * int_sqrt - rough approximation to sqrt |
| 14 | * @x: integer of which to calculate the sqrt |
| 15 | * |
| 16 | * A very rough approximation to the sqrt() function. |
| 17 | */ |
| 18 | unsigned long int_sqrt(unsigned long x) |
| 19 | { |
Davidlohr Bueso | 30493cc | 2013-04-29 16:18:09 -0700 | [diff] [blame] | 20 | unsigned long b, m, y = 0; |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 21 | |
Davidlohr Bueso | 30493cc | 2013-04-29 16:18:09 -0700 | [diff] [blame] | 22 | if (x <= 1) |
| 23 | return x; |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 24 | |
Davidlohr Bueso | 30493cc | 2013-04-29 16:18:09 -0700 | [diff] [blame] | 25 | m = 1UL << (BITS_PER_LONG - 2); |
| 26 | while (m != 0) { |
| 27 | b = y + m; |
| 28 | y >>= 1; |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 29 | |
Davidlohr Bueso | 30493cc | 2013-04-29 16:18:09 -0700 | [diff] [blame] | 30 | if (x >= b) { |
| 31 | x -= b; |
| 32 | y += m; |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 33 | } |
Davidlohr Bueso | 30493cc | 2013-04-29 16:18:09 -0700 | [diff] [blame] | 34 | m >>= 2; |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 35 | } |
Davidlohr Bueso | 30493cc | 2013-04-29 16:18:09 -0700 | [diff] [blame] | 36 | |
| 37 | return y; |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 38 | } |
| 39 | EXPORT_SYMBOL(int_sqrt); |