factor/extra/math/quadratic/quadratic.factor

! Copyright (C) 2007 Slava Pestov.
! See http://factorcode.org/license.txt for BSD license.
USING: kernel math math.functions ;
IN: math.quadratic

: monic ( c b a -- c' b' ) tuck [ / ] 2bi@ ;

: discriminant ( c b -- b d ) tuck sq 4 / swap - sqrt ;

: critical ( b d -- -b/2 d ) [ -2 / ] dip ;

: +- ( x y -- x+y x-y ) [ + ] [ - ] 2bi ;

: quadratic ( c b a -- alpha beta )
    #! Solve a quadratic equation ax^2 + bx + c = 0
    monic discriminant critical +- ;

: qeval ( x c b a -- y )
    #! Evaluate ax^2 + bx + c
    [ pick * ] dip roll sq * + + ;