diff --git a/Classes/BEMLine.m b/Classes/BEMLine.m index c408620..e45aaa7 100644 --- a/Classes/BEMLine.m +++ b/Classes/BEMLine.m @@ -154,11 +154,14 @@ - (void)drawRect:(CGRect)rect { } } - if (!self.disableMainLine && self.bezierCurveIsEnabled) { + BOOL bezierStatus = self.bezierCurveIsEnabled; + if (self.arrayOfPoints.count <= 2 && self.bezierCurveIsEnabled == YES) bezierStatus = NO; + + if (!self.disableMainLine && bezierStatus) { line = [BEMLine quadCurvedPathWithPoints:self.points]; fillBottom = [BEMLine quadCurvedPathWithPoints:self.bottomPointsArray]; fillTop = [BEMLine quadCurvedPathWithPoints:self.topPointsArray]; - } else if (!self.disableMainLine && !self.bezierCurveIsEnabled) { + } else if (!self.disableMainLine && !bezierStatus) { line = [BEMLine linesToPoints:self.points]; fillBottom = [BEMLine linesToPoints:self.bottomPointsArray]; fillTop = [BEMLine linesToPoints:self.topPointsArray];