diff options
Diffstat (limited to 'src/toys/levelsets-test.cpp')
| -rw-r--r-- | src/toys/levelsets-test.cpp | 155 |
1 files changed, 155 insertions, 0 deletions
diff --git a/src/toys/levelsets-test.cpp b/src/toys/levelsets-test.cpp new file mode 100644 index 0000000..cd5874e --- /dev/null +++ b/src/toys/levelsets-test.cpp @@ -0,0 +1,155 @@ +#include <2geom/d2.h> +#include <2geom/sbasis.h> +#include <2geom/bezier-to-sbasis.h> + +#include <toys/path-cairo.h> +#include <toys/toy-framework-2.h> + +using std::vector; +using namespace std; +using namespace Geom; + +static double exp_rescale(double x){ return std::pow(10, x);} +std::string exp_formatter(double x){ return default_formatter(exp_rescale(x));} + + +static void plot(cairo_t* cr, SBasis const &B,double vscale=1,double a=0,double b=1){ + D2<SBasis> plot; + plot[0]=SBasis(Linear(150+a*300,150+b*300)); + plot[1]=B*(-vscale); + plot[1]+=300; + cairo_d2_sb(cr, plot); + cairo_stroke(cr); +} +static void plot_bar(cairo_t* cr, double height, double vscale=1,double a=0,double b=1){ + cairo_move_to(cr, Geom::Point(150+300*a,-height*vscale+300)); + cairo_line_to(cr, Geom::Point(150+300*b,-height*vscale+300)); +} +static void plot_bar(cairo_t* cr, Interval height, double vscale=1,double a=0,double b=1){ + Point A(150+300*a,-height.max()*vscale+300); + Point B(150+300*b,-height.min()*vscale+300); + cairo_rectangle(cr, Rect(A,B) ); +} + +class BoundsTester: public Toy { + + void draw(cairo_t *cr, std::ostringstream *notify, int width, int height, bool save, std::ostringstream *timer_stream) override { + + for (unsigned i=0;i<size;i++){ + hand.pts[i ][0]=150+15*(i-size); + hand.pts[i+size][0]=450+15*(i+1); + cairo_move_to(cr, Geom::Point(hand.pts[i ][0],150)); + cairo_line_to(cr, Geom::Point(hand.pts[i ][0],450)); + cairo_move_to(cr, Geom::Point(hand.pts[i+size][0],150)); + cairo_line_to(cr, Geom::Point(hand.pts[i+size][0],450)); + } + cairo_move_to(cr, Geom::Point(0,300)); + cairo_line_to(cr, Geom::Point(600,300)); + + cairo_set_line_width (cr, .3); + cairo_set_source_rgba (cr, 0.2, 0.2, 0.2, 1); + cairo_stroke(cr); + + SBasis B(size, Linear()); + for (unsigned i=0;i<size;i++){ + B[i] = Linear(-(hand.pts[i ][1]-300)*std::pow(4.,(int)i), + -(hand.pts[i+size][1]-300)*std::pow(4.,(int)i) ); + } + B.normalize(); + plot(cr,B,1); + cairo_set_source_rgba (cr, 0., 0., 0.8, 1); + cairo_stroke(cr); + + double vtol = exp_rescale(slider.value()); + if (vtol<1e-4) vtol=0; + + hand.pts[2*size ][X]=150; + hand.pts[2*size+1][X]=150; + hand.pts[2*size+2][X]=150; + hand.pts[2*size+1][Y]=std::max(hand.pts[2*size+1][Y],hand.pts[2*size+2][Y]+2*vtol); + hand.pts[2*size ][Y]=std::max(hand.pts[2*size ][Y],hand.pts[2*size+1][Y]+2*vtol); + + vector<Interval> levels; + levels.emplace_back(300-(hand.pts[2*size ][Y]-vtol), 300-(hand.pts[2*size ][Y]+vtol) ); + levels.emplace_back(300-(hand.pts[2*size+1][Y]-vtol), 300-(hand.pts[2*size+1][Y]+vtol) ); + levels.emplace_back(300-(hand.pts[2*size+2][Y]-vtol), 300-(hand.pts[2*size+2][Y]+vtol) ); + + for (auto & level : levels) plot_bar(cr,level.middle()); + cairo_set_source_rgba( cr, 1., 0., 0., 1); + cairo_stroke(cr); + for (auto & level : levels) plot_bar(cr,level); + cairo_set_source_rgba( cr, 1., 0., 0., .2); + cairo_fill(cr); + + cairo_set_source_rgba (cr, 0., 0.5, 0., 1); + + *notify<<"Use hand.pts to set the coefficients of the s-basis."<<std::endl; + + vector<vector<Interval> > sols=level_sets(B,levels,0,1); + for (unsigned i=0;i<sols.size();i++){ + for (unsigned j=0;j<sols[i].size();j++){ + Interval ys = levels[i]; + ys.expandTo(0.); + cairo_set_line_width (cr, .3); + plot_bar(cr,ys, 1., sols[i][j].min(), sols[i][j].max()); + cairo_set_source_rgba( cr, 0., 0., 1., .3); + cairo_fill(cr); + plot_bar(cr,ys, 1., sols[i][j].min(), sols[i][j].max()); + cairo_set_source_rgba( cr, 0., 0., 1., .3); + cairo_stroke(cr); + cairo_set_line_width (cr, 1.6); + cairo_set_source_rgba( cr, 0., 0., 1, 1); + plot_bar(cr,0., 1., sols[i][j].min(), sols[i][j].max()); + Point sol ( 150 + 300 * sols[i][j].middle(), 300); + draw_cross(cr, sol); + cairo_stroke(cr); + } + } + + cairo_set_source_rgba( cr, 0., 0., 1., .5); + cairo_fill(cr); + + Toy::draw(cr, notify, width, height, save,timer_stream); + } + + +public: + BoundsTester(){ + size=5; + if(hand.pts.empty()) { + for(unsigned i = 0; i < 2*size; i++) + hand.pts.emplace_back(0,150+150+uniform()*300*0); + } + hand.pts.emplace_back(150,300+ 50+uniform()*100); + hand.pts.emplace_back(150,300- 50+uniform()*100); + hand.pts.emplace_back(150,300-150+uniform()*100); + handles.push_back(&hand); + slider = Slider(-5, 2, 0, 0.5, "tolerance"); + slider.geometry(Point(50, 20), 250); + slider.formatter(&exp_formatter); + handles.push_back(&slider); + } + + +private: + unsigned size; + PointSetHandle hand; + Slider slider; + +}; + +int main(int argc, char **argv) { + init(argc, argv, new BoundsTester); + return 0; +} + +/* + Local Variables: + mode:c++ + c-file-style:"stroustrup" + c-file-offsets:((innamespace . 0)(inline-open . 0)(case-label . +)) + indent-tabs-mode:nil + fill-column:99 + End: +*/ +// vim: filetype = cpp:expandtab:shiftwidth = 4:tabstop = 8:softtabstop = 4:encoding = utf-8:textwidth = 99 : |