Update Counting_Columns_2_1.py

Counting_Columns_2_1.py

@@ -135,6 +135,7 @@ def distance(point1, point2):
 
 def getNearestText(point_list, p):
   nearbyy = []
+  selected_clm_point = [] #save the clmn for drawing cirlce on it 
   dis = []
   for i in range(len(p)):
     nearest_point = min(point_list, key=lambda point: distance(point, p[i]))
@@ -142,7 +143,8 @@ def getNearestText(point_list, p):
     dis.append(dist)
     if dist < 44:
       nearbyy.append(nearest_point)
-  return nearbyy
+      selected_clm_point.append(p[i])
+  return nearbyy, selected_clm_point
 
 
 def getColumnsTypes(nearbyy, x):
@@ -184,7 +186,7 @@ def mainfun(plan):
 
   if len(column_points) > 10:
       # BROWN COLUMNS
-      nearby = getNearestText(text_points, column_points)
+      nearby, slctd_clm = getNearestText(text_points, column_points)
       columns_types_v = getColumnsTypesKeyValue(nearby, txtpts_ky_vlu)
       legend = generate_legend(columns_types_v)
 
@@ -194,10 +196,10 @@ def mainfun(plan):
       imgResult = segment_blue(img_blue)
       outsu = threshold(imgResult)
       column_points,mask_clmns, mask_walls = get_columns_info(outsu, img)
-      nearby = getNearestText(text_points, column_points)
+      nearby, slctd_clm = getNearestText(text_points, column_points)
       columns_types_v = getColumnsTypesKeyValue(nearby, txtpts_ky_vlu)
       legend = generate_legend(columns_types_v)
-  return legend
+  return legend, slctd_clm
 
 '''def mainfun(plan):
   texts_from_pdf = get_text_from_pdf(plan)