Blob Blame Raw
#pragma once

#ifndef T_SEGMENTADJUSTER_H
#define T_SEGMENTADJUSTER_H

#include "tgeometry.h"
#include "tstroke.h"

#undef DVAPI
#undef DVVAR
#ifdef TVECTORIMAGE_EXPORTS
#define DVAPI DV_EXPORT_API
#define DVVAR DV_EXPORT_VAR
#else
#define DVAPI DV_IMPORT_API
#define DVVAR DV_IMPORT_VAR
#endif

#ifdef _WIN32
#pragma warning(disable : 4251)
#endif

/*!
  The TSegmentAdjuster minimizes the distance between

     stroke0->getPoint(w0) and stroke1->getPoint(w1)

  changing the values of w0 and w1 (in a defined range).

  Usage:

  TSegmentAdjuster::End p0(stroke0, w0, 0, 1);
  TSegmentAdjuster::End p1(stroke1, w1, 0, 1);
  TSegmentAdjuster adjuster;
  adjuster.compute(p0,p1);

  // the adjusted segment is p0.getPoint(), p1.getPoint()

  if you want to adjust the segment only in the w-growing-direction:

  TSegmentAdjuster::End p0(stroke0, w0, w0, 1);
  TSegmentAdjuster::End p1(stroke1, w1, w1, 1);
  TSegmentAdjuster adjuster;
  adjuster.compute(p0,p1);


 */

class DVAPI TSegmentAdjuster {
public:
  class End {
  public:
    TStroke *m_stroke;
    double m_w;
    double m_wmin, m_wmax;
    End() : m_stroke(0), m_w(0), m_wmin(0), m_wmax(1) {}
    End(TStroke *stroke, double w)
        : m_stroke(stroke), m_w(w), m_wmin(0), m_wmax(1) {}
    End(TStroke *stroke, double w, double wmin, double wmax)
        : m_stroke(stroke), m_w(w), m_wmin(wmin), m_wmax(wmax) {}
    TPointD getPoint() const { return m_stroke->getPoint(m_w); }
    TPointD getSpeed() const { return m_stroke->getSpeed(m_w); }
  };

  TSegmentAdjuster();

  // debug
  void enableTrace(bool enabled) { m_traceEnabled = enabled; }
  void draw();
  void clear() { m_links.clear(); }

  void compute(End &a, End &b);

private:
  End m_a, m_b, m_c, m_d;
  std::vector<std::pair<TPointD, TPointD>> m_links;
  bool m_traceEnabled;

  inline double fun(double u, double v) const {
    return norm2(m_a.m_stroke->getPoint(u) - m_b.m_stroke->getPoint(v));
  }

  inline void gradient(double &dfdu, double &dfdv, double u, double v) {
    const double h = 0.0001;
    dfdu           = (fun(u + h, v) - fun(u - h, v)) / (2 * h);
    dfdv           = (fun(u, v + h) - fun(u, v - h)) / (2 * h);
  }

  inline static double dsda(TStroke *stroke, double w, double dwda) {
    const double h = 0.0001;
    return norm(stroke->getPoint(w + dwda * h) -
                stroke->getPoint(w - dwda * h)) /
           (2 * h);
  }
};
//---------------------------------------------------------------------------------------

#endif  // T_SEGMENTADJUSTER_H