#pragma once

#include <QtNodes/internal/AbstractConnectionPainter.hpp>

class SquareConnectionPainter : public QtNodes::AbstractConnectionPainter {
public:
  void paint(QPainter *painter,
             QtNodes::ConnectionGraphicsObject const &cgo) const override;
  QPainterPath
  getPainterStroke(QtNodes::ConnectionGraphicsObject const &cgo) const override;

private:
  QPainterPath
  orthogonalPath(QtNodes::ConnectionGraphicsObject const &cgo) const;
};