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); |
Peter Zijlstra | 3f32957 | 2017-11-17 15:28:04 -0800 | [diff] [blame^] | 26 | while (m > x) |
27 | m >>= 2; | ||||
28 | |||||
Davidlohr Bueso | 30493cc | 2013-04-29 16:18:09 -0700 | [diff] [blame] | 29 | while (m != 0) { |
30 | b = y + m; | ||||
31 | y >>= 1; | ||||
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 32 | |
Davidlohr Bueso | 30493cc | 2013-04-29 16:18:09 -0700 | [diff] [blame] | 33 | if (x >= b) { |
34 | x -= b; | ||||
35 | y += m; | ||||
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 36 | } |
Davidlohr Bueso | 30493cc | 2013-04-29 16:18:09 -0700 | [diff] [blame] | 37 | m >>= 2; |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 38 | } |
Davidlohr Bueso | 30493cc | 2013-04-29 16:18:09 -0700 | [diff] [blame] | 39 | |
40 | return y; | ||||
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 41 | } |
42 | EXPORT_SYMBOL(int_sqrt); |