Davidlohr Bueso | 30493cc | 2013-04-29 16:18:09 -0700 | [diff] [blame] | 1 | /* |
2 | * Copyright (C) 2013 Davidlohr Bueso <davidlohr.bueso@hp.com> | ||||
3 | * | ||||
4 | * Based on the shift-and-subtract algorithm for computing integer | ||||
5 | * square root from Guy L. Steele. | ||||
6 | */ | ||||
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 7 | |
8 | #include <linux/kernel.h> | ||||
Paul Gortmaker | 8bc3bcc | 2011-11-16 21:29:17 -0500 | [diff] [blame] | 9 | #include <linux/export.h> |
Peter Zijlstra | 627f9c3 | 2017-11-17 15:28:08 -0800 | [diff] [blame] | 10 | #include <linux/bitops.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 | |
Peter Zijlstra | 627f9c3 | 2017-11-17 15:28:08 -0800 | [diff] [blame] | 25 | m = 1UL << (__fls(x) & ~1UL); |
Davidlohr Bueso | 30493cc | 2013-04-29 16:18:09 -0700 | [diff] [blame] | 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); |