modules/base/math/fixed_point/f16_sqrt.c

   1 /*
   2  *  Copyright Droids Corporation, Microb Technology, Eirbot (2005)
   3  *
   4  *  This program is free software; you can redistribute it and/or modify
   5  *  it under the terms of the GNU General Public License as published by
   6  *  the Free Software Foundation; either version 2 of the License, or
   7  *  (at your option) any later version.
   8  *
   9  *  This program is distributed in the hope that it will be useful,
  10  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12  *  GNU General Public License for more details.
  13  *
  14  *  You should have received a copy of the GNU General Public License
  15  *  along with this program; if not, write to the Free Software
  16  *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  17  *
  18  *  Revision : $Id: f16_sqrt.c,v 1.5.4.4 2009-02-02 22:21:20 zer0 Exp $
  19  *
  20  */
  21
  22 #include <stdio.h>
  23 #include <f16.h>
  24 #include <s16_to_f16.h>
  25
  26
  27 #define NEXT(n, i)  (((n) + (i)/(n)) >> 1)
  28
  29 static uint16_t u16_sqrt(uint16_t number) {
  30         uint16_t n  = 1;
  31         uint16_t n1 = NEXT(n, number);
  32
  33         if (number == 0)
  34                 return 0;
  35
  36         while(ABS(n1 - n) > 1) {
  37                 n  = n1;
  38                 n1 = NEXT(n, number);
  39         }
  40         while((n1*n1) > number) {
  41                 n1 -= 1;
  42         }
  43         return n1;
  44 }
  45
  46
  47 f16 f16_sqrt(f16 f)
  48 {
  49         uint16_t a,b,c,d;
  50
  51         if (F16_IS_NEG(f))
  52                 return F16_NAN;
  53
  54         if(f.f16_integer) {
  55                 /* sqrt(a+b) = sqrt(a)*sqrt(1+b/a) */
  56                 a=(uint16_t)(f.f16_integer) << 8 ;
  57                 b=(uint16_t)(f.f16_decimal) << 8 ;
  58                 c=u16_sqrt(a);
  59                 d=u16_sqrt(0x100 + (b/a));
  60                 return f16_mul(s16_to_f16( c<<4 ), s16_to_f16( d<<4 ));
  61         }
  62         else {
  63                 b=(uint16_t)(f.f16_decimal) << 8 ;
  64                 return s16_to_f16(u16_sqrt(b));
  65         }
  66 }