Merge pull request #60 from yangzq50/fix-create-pgm-index-for-only-one-element

gvinciguerra · web-flow · commit c6fcf3d34e55 · 2024-11-28T12:22:31.000+01:00
Support special case of creating PGM index for only one input element
diff --git a/include/pgm/piecewise_linear_model.hpp b/include/pgm/piecewise_linear_model.hpp
@@ -303,7 +303,7 @@ size_t make_segmentation(size_t n, size_t start, size_t end, size_t epsilon, Fin
             add_point(in(i), i);
         }
     }
-    if (in(end - 1) != in(end - 2))
+    if (end >= start + 2 && in(end - 1) != in(end - 2))
         add_point(in(end - 1), end - 1);
 
     if (end == n) {

Original file line number	Diff line number	Diff line change
`@@ -303,7 +303,7 @@ size_t make_segmentation(size_t n, size_t start, size_t end, size_t epsilon, Fin`
`303`	`303`	`add_point(in(i), i);`
`304`	`304`	`}`
`305`	`305`	`}`
`306`		`- if (in(end - 1) != in(end - 2))`
	`306`	`+ if (end >= start + 2 && in(end - 1) != in(end - 2))`
`307`	`307`	`add_point(in(end - 1), end - 1);`
`308`	`308`
`309`	`309`	`if (end == n) {`