diff options
Diffstat (limited to 'src/toys/circle-tangent-fitting.cpp')
| -rw-r--r-- | src/toys/circle-tangent-fitting.cpp | 224 |
1 files changed, 224 insertions, 0 deletions
diff --git a/src/toys/circle-tangent-fitting.cpp b/src/toys/circle-tangent-fitting.cpp new file mode 100644 index 0000000..f0d74ee --- /dev/null +++ b/src/toys/circle-tangent-fitting.cpp @@ -0,0 +1,224 @@ +/* + * Circle and Elliptical Arc Fitting Example + * + * Authors: + * Marco Cecchetti <mrcekets at gmail.com> + * + * Copyright 2008 authors + * + * This library is free software; you can redistribute it and/or + * modify it either under the terms of the GNU Lesser General Public + * License version 2.1 as published by the Free Software Foundation + * (the "LGPL") or, at your option, under the terms of the Mozilla + * Public License Version 1.1 (the "MPL"). If you do not alter this + * notice, a recipient may use your version of this file under either + * the MPL or the LGPL. + * + * You should have received a copy of the LGPL along with this library + * in the file COPYING-LGPL-2.1; if not, write to the Free Software + * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA + * You should have received a copy of the MPL along with this library + * in the file COPYING-MPL-1.1 + * + * The contents of this file are subject to the Mozilla Public License + * Version 1.1 (the "License"); you may not use this file except in + * compliance with the License. You may obtain a copy of the License at + * http://www.mozilla.org/MPL/ + * + * This software is distributed on an "AS IS" basis, WITHOUT WARRANTY + * OF ANY KIND, either express or implied. See the LGPL or the MPL for + * the specific language governing rights and limitations. + */ + +#include <memory> +#include <2geom/numeric/fitting-tool.h> +#include <2geom/numeric/fitting-model.h> + +#include <2geom/circle.h> +#include <2geom/elliptical-arc.h> + +#include <toys/path-cairo.h> +#include <toys/toy-framework-2.h> + + +using namespace Geom; + + + +class LFMCircleTangentEquation + : public NL::LinearFittingModelWithFixedTerms<Point, double, Circle> +{ + public: + mutable unsigned count; // sigh + mutable Point pb; + vector<Point> bases; + void feed( NL::VectorView & coeff, double & fixed_term, Point const& p ) const + { + if (count >= bases.size()) { + coeff[0] = p[X]; + coeff[1] = p[Y]; + coeff[2] = 1; + fixed_term = p[X] * p[X] + p[Y] * p[Y]; + pb = p; + } else { + Point pb = bases[count]; + coeff[0] = p[X]; + coeff[1] = p[Y]; + coeff[2] = 0; + fixed_term = 2*p[X]*pb[X] + 2*p[Y]*pb[Y]; + count ++; + } + } + + size_t size() const + { + return 3; + } + void instance(Circle & c, NL::Vector const& coeff) const + { + c.setCoefficients(1, coeff[0], coeff[1], coeff[2]); + } +}; + + +class CircleFitting : public Toy +{ + private: + void draw( cairo_t *cr, std::ostringstream *notify, + int width, int height, bool save, std::ostringstream *timer_stream) override + { + if (first_time) + { + first_time = false; + Point toggle_sp( 300, height - 50); + toggles[0].bounds = Rect( toggle_sp, toggle_sp + Point(120,25) ); + sliders[0].geometry(Point(50, height - 50), 100); + } + + try + { + vector<Point> points; + points.push_back(psh.pts[1]-psh.pts[0]); // tangents + //points.push_back(psh.pts[2]-psh.pts[3]); + points.push_back(psh.pts[3]); + points.push_back(psh.pts[0]); + size_t sz = points.size(); + if (sz < 3) + { + THROW_RANGEERROR("fitting error: too few points passed"); + } + LFMCircleTangentEquation model; + model.count = 0; + model.bases.push_back(psh.pts[0]); + //model.bases.push_back(psh.pts[3]); + NL::least_squeares_fitter<LFMCircleTangentEquation> fitter(model, sz); + + for (size_t i = 0; i < sz; ++i) + { + fitter.append(points[i]); + } + fitter.update(); + + NL::Vector z(sz, 0.0); + model.instance(c, fitter.result(z)); + //c.set(psh.pts); + } + catch(RangeError exc) + { + std::cerr << exc.what() << std::endl; + Toy::draw(cr, notify, width, height, save,timer_stream); + return; + } + + if (toggles[0].on) + { + try + { + std::unique_ptr<EllipticalArc> eap( c.arc(psh.pts[0], psh.pts[1], psh.pts[3]) ); + ea = *eap; + } + catch(RangeError exc) + { + std::cerr << exc.what() << std::endl; + Toy::draw(cr, notify, width, height, save,timer_stream); + return; + } + } + cairo_set_source_rgba(cr, 0.3, 0.3, 0.3, 1.0); + cairo_set_line_width (cr, 0.3); + cairo_move_to(cr, psh.pts[1]); + cairo_line_to(cr, 2*psh.pts[0] - psh.pts[1]); + cairo_move_to(cr, psh.pts[2]); + cairo_line_to(cr, 2*psh.pts[3] - psh.pts[2]); + cairo_stroke(cr); + + cairo_set_source_rgba(cr, 0.3, 0.3, 0.3, 1.0); + cairo_set_line_width (cr, 0.3); + if (!toggles[0].on) + { + cairo_arc(cr, c.center(X), c.center(Y), c.radius(), 0, 2*M_PI); + } + else + { + draw_text(cr, psh.pts[0], "initial"); + draw_text(cr, psh.pts[1], "inner"); + draw_text(cr, psh.pts[2], "inner"); + draw_text(cr, psh.pts[3], "final"); + + D2<SBasis> easb = ea.toSBasis(); + cairo_d2_sb(cr, easb); + } + cairo_stroke(cr); + + Toy::draw(cr, notify, width, height, save,timer_stream); + } + + public: + CircleFitting() + { + first_time = true; + + psh.pts.resize(4); + psh.pts[0] = Point(450, 250); + psh.pts[1] = Point(450+100, 250+100); + psh.pts[2] = Point(250+100, 400+100); + psh.pts[3] = Point(250, 400); + + + + toggles.emplace_back(" arc / circle ", false); + sliders.emplace_back(0, 5, 1, 0, "more handles"); + + handles.push_back(&psh); + handles.push_back(&(toggles[0])); + handles.push_back(&(sliders[0])); + } + + private: + Circle c; + EllipticalArc ea; + bool first_time; + PointSetHandle psh; + std::vector<Toggle> toggles; + std::vector<Slider> sliders; +}; + + + +int main(int argc, char **argv) +{ + init( argc, argv, new CircleFitting(), 600, 600 ); + 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:fileencoding=utf-8:textwidth=99 : |