factor/extra/math/splines/viewer/viewer.factor

50 lines
1.4 KiB
Factor

! Copyright (C) 2010 Erik Charlebois.
! See http://factorcode.org/license.txt for BSD license.
USING: accessors kernel locals math math.order math.polynomials
math.splines opengl.gl sequences ui.gadgets ui.gadgets.panes ui.render
arrays ;
IN: math.splines.viewer
<PRIVATE
: eval-polynomials ( polynomials-seq n -- xy-sequence )
[
[ 1 + iota ] keep [
/f swap [ polyval ] with map
] curry with map
] curry map concat ;
PRIVATE>
TUPLE: spline-gadget < gadget polynomials steps spline-dim ;
M: spline-gadget pref-dim* spline-dim>> ;
M:: spline-gadget draw-gadget* ( gadget -- )
0 0 0 glColor3f
gadget [ polynomials>> ] [ steps>> ] bi eval-polynomials :> pts
pts [ first ] [ max ] map-reduce :> x-max
pts [ first ] [ min ] map-reduce :> x-min
pts [ second ] [ max ] map-reduce :> y-max
pts [ second ] [ min ] map-reduce :> y-min
pts [
[ first x-min - x-max x-min - / gadget spline-dim>> first * ]
[ second y-min - y-max y-min - / gadget spline-dim>> second * ] bi 2array
] map :> pts
GL_LINE_STRIP glBegin
pts [
first2 neg gadget spline-dim>> second + glVertex2f
] each
glEnd ;
:: <spline-gadget> ( polynomials dim steps -- gadget )
spline-gadget new
dim >>spline-dim
polynomials >>polynomials
steps >>steps ;
: spline. ( curve dim steps -- )
<spline-gadget> gadget. ;