From f340edfcabcb0de448dbe685f14c0d031c61a674 Mon Sep 17 00:00:00 2001
From: yTb45 <vancurajan84@gmail.com>
Date: Wed, 8 Nov 2023 15:35:39 +0100
Subject: [PATCH] Update interface.cpp

---
 src/interface.cpp | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/src/interface.cpp b/src/interface.cpp
index 0fe1aad..17e7d6d 100644
--- a/src/interface.cpp
+++ b/src/interface.cpp
@@ -47,6 +47,12 @@ LineSegment * find_line_segment_groups(
     // Detect the lines
     auto lines = find_line_segments(im, SEED_DIST, SEED_RATIO, TRACE_TOLERANCE, ctx);
 
+    if (lines.size() < 2)
+    {
+        *n_lines = 0;
+        return nullptr;
+    }
+
     if (refine)
     {
         lines = postprocess_lines_segments(lines, ctx);
@@ -56,6 +62,12 @@ LineSegment * find_line_segment_groups(
     filtered.reserve(lines.size());
     _filter_lines(lines.begin(), lines.end(), back_inserter(filtered), min_length);
 
+    if (filtered.size() == 0)
+    {
+        *n_lines = 0;
+        return nullptr;
+    }
+
     estimate_line_pencils(filtered, MAX_MODELS, ESTIMATOR_INLIER_MAX_ANGLE_DEG, ESTIMATOR_GARBAGE_MAX_ANGLE_DEG, ctx);
     auto & groupped = filtered;