Update pilecaps_adr.py

pilecaps_adr.py

@@ -18,6 +18,7 @@ import cv2
 from matplotlib import pyplot as plt 
 #from pdf2image import convert_from_path
 import math
+from math import sin, cos, radians
 import pandas as pd
 import random
 # import imutils
@@ -292,8 +293,7 @@ def DetectColor(img,color=0):
 
   detectedColors=cv2.cvtColor(detectedColors,cv2.COLOR_HSV2BGR)
   detectedColors=cv2.medianBlur(detectedColors,7)
-  # cv2_imshow(detectedColors)
-  
+  cv2.imwrite('det.png',detectedColors)
   return mask, detectedColors, color
 
     
@@ -487,7 +487,7 @@ def findContoursFullImage(green2,img,number,finalColorArray,ratioarea,ratioperim
     return contourss
   else:
     mask, detectedColors, rgbcolor =DetectColor(img,color)
-    print(rgbcolor)
+    # print(rgbcolor)
     
     pil_image=Image.fromarray(mask)
     
@@ -547,7 +547,6 @@ def getAreasPerimeter(green2,img,number,ratioarea,ratioperim,flag,finalColorArra
         perimeter1 = cv2.arcLength(approx, True)
         # approx = cv2.approxPolyDP(contourss[contour][0], 0.01* perimeter, True)
         # area1 = cv2.contourArea(approx)
-          
         x, y , width, height = cv2.boundingRect(contourss[contour][0])
 
         angleR,widthR ,heightR= StraightenImage(contourss[contour][0],img)
@@ -561,7 +560,7 @@ def getAreasPerimeter(green2,img,number,ratioarea,ratioperim,flag,finalColorArra
             # print('numss')
             # if flag=='area':
               # addedMargin=area1+perimeter1*2
-              areaa=round(area1* ratioarea , 3) # true value of area of any shape/ area px value of same shape
+              areaa=round(area1* ratioarea , 2) # true value of area of any shape/ area px value of same shape
               appended.append([areaa,width,height])
 
         #else:
@@ -573,13 +572,14 @@ def getAreasPerimeter(green2,img,number,ratioarea,ratioperim,flag,finalColorArra
 
 def FillDictionary(green2,SimilarAreaDictionary,img,number,ratioarea,ratioperim,flag,finalColorArray,rgbcolor=[0,0,0],color=[0,0,0]):
   #fills dictionary with key areas and number of occurences
-  print('wttttt')
   areas_Perimeters=sorted(getAreasPerimeter(green2,img,number,ratioarea,ratioperim,flag,finalColorArray,color) )
   
   indices=[]
-  colorRanges=[[255,153,153],[51,255,51],[201,56,147],[255,0,0],[255,0,255],[0,102,204],[102,0,102],[153,0,76],[200,92,135],[52,161,99],[235,250,24],[40,30,170],[98,149,63],[100,30,179],[200,55,67],[150,80,200],[0,102,102],[250,28,191],[101,27,101],[230,150,76],[3,65,127],[114,39,39],[250,36,100],[180,30,40],[10,250,60],[140,30,253],[114,58,245],[47,255,255],[18,236,206],[225,105,29],[189,65,121],[206,204,48],[126,7,247],[3,168,251]]
-  print(colorRanges[0])
-  print(colorRanges[0][0],colorRanges[0][1], colorRanges[0][2])
+  # colorRanges=[[255,153,153],[51,255,51],[201,56,147],[255,0,0],[255,0,255],[0,102,204],[102,0,102],[153,0,76],[200,92,135],[52,161,99],[235,250,24],[40,30,170],[98,149,63],[100,30,179],[200,55,67],[150,80,200],[0,102,102],[250,28,191],[101,27,101],[230,150,76],[3,65,127],[114,39,39],[250,36,100],[180,30,40],[10,250,60],[140,30,253],[114,58,245],[47,255,255],[18,236,206],[225,105,29],[189,65,121],[206,204,48],[126,7,247],[3,168,251]]
+  
+  colorRanges=[[255,0,0],[0,0,255],[0,255,255],[0,64,0],[255,204,0],[255,128,64],[255,0,128],[255,128,192],[128,128,255],[128,64,0],[0,255,0],[179,106,179],[115,52,179],[0,128,192],[128,0,128],[128,0,0],[0,128,255],[255,182,128],[255,0,255],[0,0,128],[0,128,64],[255,255,0],[128,0,64],[203,203,106],[128,255,166],[255,128,0],[255,98,98],[90,105,138],[114,10,72],[36,82,78],[225,105,29],[108,62,40],[11,35,75],[42,176,203],[255,153,153],[129,74,138],[99,123,137],[159,179,30]]
+  # print(colorRanges[0])
+  # print(colorRanges[0][0],colorRanges[0][1], colorRanges[0][2])
   colorsUsed=[]
   for i in range(len(areas_Perimeters)):
      
@@ -641,21 +641,54 @@ def FillDictionary(green2,SimilarAreaDictionary,img,number,ratioarea,ratioperim,
           SimilarAreaDictionary.loc[len(SimilarAreaDictionary)] =mydata
   # s= SimilarAreaDictionary
   for i in range(len(SimilarAreaDictionary)):
-      SimilarAreaDictionary['R'].loc[i] =colorRanges[i][0]
+      SimilarAreaDictionary['B'].loc[i] =colorRanges[i][0]
       SimilarAreaDictionary['G'].loc[i] =colorRanges[i][1]
-      SimilarAreaDictionary['B'].loc[i] = colorRanges[i][2]
+      SimilarAreaDictionary['R'].loc[i] = colorRanges[i][2]
       colorsUsed.append(colorRanges[i])
 
 
   return SimilarAreaDictionary, colorsUsed , areas_Perimeters
+
+
+
+def rotate_point(point, angle, center_point=(0, 0)):
+    """Rotates a point around center_point(origin by default)
+    Angle is in degrees.
+    Rotation is counter-clockwise
+    """
+    angle_rad = radians(angle % 360)
+    # Shift the point so that center_point becomes the origin
+    new_point = (point[0] - center_point[0], point[1] - center_point[1])
+    new_point = (new_point[0] * cos(angle_rad) - new_point[1] * sin(angle_rad),
+                 new_point[0] * sin(angle_rad) + new_point[1] * cos(angle_rad))
+    # Reverse the shifting we have done
+    new_point = (new_point[0] + center_point[0], new_point[1] + center_point[1])
+    return new_point
+
+def rotate_polygon(polygon, angle, center_point=(0, 0)):
+    """Rotates the given polygon which consists of corners represented as (x,y)
+    around center_point (origin by default)
+    Rotation is counter-clockwise
+    Angle is in degrees
+    """
+    rotated_polygon = []
+    for corner in polygon:
+        rotated_corner = rotate_point(corner, angle, center_point)
+        rotated_polygon.append(rotated_corner)
+    return rotated_polygon
+
 def drawAllContours(img,number,finalColorArray,ratioarea,ratioperim,flag , path,pdfpath):
   green2=allpreSteps(img)
   doc = fitz.open('dropbox_plans/1.0/'+path)
-  page = doc[0]    
-  page.set_rotation(0)
+  page = doc[0]
+  # page.set_rotation(270)
   pix=page.get_pixmap()
-  ratio =  pix.width/ img.shape[1] 
-  
+  print('SHAPPEE',img.shape)
+  # if img.shape[0] > img.shape[1]:
+  ratio =  pix.width / img.shape[1]  
+  # ratio0 = img.shape[1]  * pix.width
+  #   ratio=    pix.height /img.shape[0] 
+ 
   areasinImage=[]
   totaldf=pd.DataFrame()
   imgArea1= img.copy()
@@ -709,7 +742,7 @@ def drawAllContours(img,number,finalColorArray,ratioarea,ratioperim,flag , path,
 
             box = cv2.boxPoints(rect)
             box = box.astype('int')
-            print(box)
+            # print(box)
 
             x, y , width, height = cv2.boundingRect(contourss[contour][0])
             perimeter = cv2.arcLength(contourss[contour][0], True)
@@ -728,7 +761,7 @@ def drawAllContours(img,number,finalColorArray,ratioarea,ratioperim,flag , path,
             allpoints.append(shape)
             # print(x,y,width,height)
             # print(allpoints)
-            print(shape)
+            # print(shape)
             if (angleR != 90.0 and angleR != -90.0 and angleR != 0.0 and angleR != -0.0 ): #inclined b ay degree 
               width=widthR
               height=heightR
@@ -778,6 +811,7 @@ def drawAllContours(img,number,finalColorArray,ratioarea,ratioperim,flag , path,
                           # print(index)
             # cv2.drawContours(imgArea1, [contour], 0, (int(rgbcolor[2]), int(rgbcolor[1]), int(rgbcolor[0])), -1)
                         cv2.drawContours(imgArea1, [contourss[contour][0]], 0, ( int(SimilarAreaDictionary['B'].loc[i]),  int(SimilarAreaDictionary['G'].loc[i]), int(SimilarAreaDictionary['R'].loc[i])), -1)
+                       
                         annot = page.add_polygon_annot( points=shape )  # 'Polygon'
                         annot.set_border(width=0.3, dashes=[2])
                         annot.set_colors( fill=(  int(SimilarAreaDictionary['R'].loc[i])/255 ,  int(SimilarAreaDictionary['G'].loc[i])/255 ,  int(SimilarAreaDictionary['B'].loc[i])/255 ) ) 
@@ -786,6 +820,7 @@ def drawAllContours(img,number,finalColorArray,ratioarea,ratioperim,flag , path,
                         annot.set_info(content='Area='+str(areaa)+' m2' +'\n \nPerimeter='+str(perimeterr)+' m',subject='ADR Team')#,title='uuum')
                         # annot.set_line_ends(fitz.PDF_ANNOT_LE_DIAMOND, fitz.PDF_ANNOT_LE_CIRCLE)
                         annot.update()
+           
             cv2.putText(imgtransparent1,'Area= '+str(area1) , (x+50,y-10)  ,cv2.FONT_HERSHEY_SIMPLEX, 0.6, (50, 50, 255), 2)
             # cv2.putText(imgtransparent1,'Width= '+str(width) , (x+50,y-10)  ,cv2.FONT_HERSHEY_SIMPLEX, 0.6, (50, 50, 255), 2)
             # cv2.putText(imgtransparent1,'Length= '+str(height) , (x+50,y-20)  ,cv2.FONT_HERSHEY_SIMPLEX, 0.6, (50, 50, 255), 2)
@@ -795,10 +830,10 @@ def drawAllContours(img,number,finalColorArray,ratioarea,ratioperim,flag , path,
             cv2.putText(imgPerimeter1,'Area: '+str(areaa)+' m2', (x+50,y-40)  ,cv2.FONT_HERSHEY_SIMPLEX, 0.6, (50, 50, 255), 2)
 
   
-    # for i,row in SimilarAreaDictionary.iterrows():
-    #   # print(row)
-    #   if row[5] not in areasinImage: # column of area 
-    #     SimilarAreaDictionary = SimilarAreaDictionary.drop(SimilarAreaDictionary.loc[SimilarAreaDictionary.index==i].index)
+    for i,row in SimilarAreaDictionary.iterrows():
+      # print(row)
+      if row[5] not in areasinImage: # column of area 
+        SimilarAreaDictionary = SimilarAreaDictionary.drop(SimilarAreaDictionary.loc[SimilarAreaDictionary.index==i].index)
 
     print(SimilarAreaDictionary)
     # display(totaldf)
@@ -821,14 +856,20 @@ def drawAllContours(img,number,finalColorArray,ratioarea,ratioperim,flag , path,
         x, y , width, height = cv2.boundingRect(contourss[contour][0])
         # approxo = cv2.approxPolyDP(contourss[contour],0.1*cv2.arcLength(contourss[contour],True),True)
         # approx = cv2.approxPolyDP(contourss[contour][0], 0.005 * perimeter, True)
-        approx = cv2.approxPolyDP(contourssA[contour][0], 0.0002* perimeter, True)
-        area1 = cv2.contourArea(approx)
+        approxa = cv2.approxPolyDP(contourssA[contour][0], 0.0002* perimeter, True)
+        area1 = cv2.contourArea(approxa)
         approx = cv2.approxPolyDP(contourss[contour][0], 0.01 * perimeter, True) #0.0009
         perimeter1 = cv2.arcLength(approx, True)
-        for point in approx:
+        M = cv2.moments(contourssA[contour][0])
+        if M['m00'] != 0:
+          cx = int(M['m10']/M['m00'])
+          cy = int(M['m01']/M['m00'])
+  
+        for point in approxa:
             x1, y1 = point[0]
+
             shape.append([int(x1*ratio),int(y1*ratio)])
-        allpoints.append(shape)
+        # shape=rotate_polygon(shape, 90)
         if (angleR != 90.0 and angleR != -90.0 and angleR != 0.0 and angleR != -0.0 ): #inclined b ay degree
             width=widthR
             height=heightR
@@ -867,17 +908,19 @@ def drawAllContours(img,number,finalColorArray,ratioarea,ratioperim,flag , path,
 
                     passed=1
                     cv2.drawContours(imgArea1, [contourss[contour][0]], 0, ( int(SimilarAreaDictionary['B'].loc[i]),  int(SimilarAreaDictionary['G'].loc[i]), int(SimilarAreaDictionary['R'].loc[i])), -1)
-
+        
                     annot = page.add_polygon_annot( points=shape )  # 'Polygon'
+                    
                     annot.set_border(width=0.3, dashes=[2])
-                    annot.set_colors( fill=(  int(SimilarAreaDictionary['B'].loc[i])/255 ,  int(SimilarAreaDictionary['G'].loc[i])/255 ,  int(SimilarAreaDictionary['R'].loc[i])/255 ) )
+                    annot.set_colors( fill=(  int(SimilarAreaDictionary['R'].loc[i])/255 ,  int(SimilarAreaDictionary['G'].loc[i])/255 ,  int(SimilarAreaDictionary['B'].loc[i])/255 ) )
                     # annot.set_colors( fill=(1,0,1) )
                     annot.set_opacity(0.5)
                     annot.set_info(content='Area='+str(areaa)+' m2' +'\n \nPerimeter='+str(perimeterr)+' m',subject='ADR Team')#,title='uuum')
                     # annot.set_line_ends(fitz.PDF_ANNOT_LE_DIAMOND, fitz.PDF_ANNOT_LE_CIRCLE)
                     annot.update()
-        cv2.putText(imgtransparent1,'area2= '+str(areaa) , (x+50,y-10)  ,cv2.FONT_HERSHEY_SIMPLEX, 0.6, (50, 50, 255), 2)
-        cv2.putText(imgtransparent1,'perim= '+str(perimeterr)+','+str(len(approx)) , (x+50,y-40)  ,cv2.FONT_HERSHEY_SIMPLEX, 0.6, (50, 50, 255), 2)
+         
+        cv2.putText(imgtransparent1,'Area: '+str(areaa)+' m2' , (x+50,y-10)  ,cv2.FONT_HERSHEY_SIMPLEX, 0.6, (50, 50, 255), 2)
+        cv2.putText(imgtransparent1,'Perimeter: '+str(perimeterr)+' m' , (x+50,y-40)  ,cv2.FONT_HERSHEY_SIMPLEX, 0.6, (50, 50, 255), 2)
 
   alpha = 0.4  # Transparency factor.
   image_new1 = cv2.addWeighted(imgArea1, alpha,  imgtransparent1, 1 - alpha, 0)
@@ -886,6 +929,7 @@ def drawAllContours(img,number,finalColorArray,ratioarea,ratioperim,flag , path,
   print(SimilarAreaDictionary)
  
   # annotationsSave
+
   pdflink= db.dropbox_upload_file(doc=doc,pdfname=path,pdfpath=pdfpath)
   dbx=db.dropbox_connect()
   md, res =dbx.files_download(path= pdfpath+path)
@@ -1375,6 +1419,7 @@ def mapnametoLegend(McTName):
   row0=['MC_T Name','Qty','Unit']
   worksheetS.update_row(1,row0)
   
+  
   for i in range(len(McTName)):
     allgbnames=''
     item=''
@@ -1418,21 +1463,21 @@ def mapnametoLegend(McTName):
       # guessednames=guessednames[3:]
       print('gg',guessednames)
       for m in McTName[i][1]:
-      
-        if m.startswith('text1'):
-          name=m.removeprefix('text1')
-          allgbnames+= name +' +'
-          indices = [o for o, x in enumerate(guessednames) if x == name]
-          print(indices)
-          # roww=worksheetw.find(str(name))
-          # print(roww)
-          for j in range(len(indices)):
-            # print('kjjjj',roww[j])
-            ar+=float(worksheetw.cell((indices[j]+1 ,rowvalue)).value) 
-        else:
-            item+=m + ' ,'
-            print(item)
-        n= McTName[i][0] + ' ( '+ allgbnames[:-2] +' , ' + item[:-1]  + ' )  ' 
+        if m:
+          if m.startswith('text1'):
+            name=m.removeprefix('text1')
+            allgbnames+= name +' +'
+            indices = [o for o, x in enumerate(guessednames) if x == name]
+            print(indices)
+            # roww=worksheetw.find(str(name))
+            # print(roww)
+            for j in range(len(indices)):
+              # print('kjjjj',roww[j])
+              ar+=float(worksheetw.cell((indices[j]+1 ,rowvalue)).value) 
+          else:
+              item+=m + ' ,'
+              print(item)
+          n= McTName[i][0] + ' ( '+ allgbnames[:-2] +' , ' + item[:-1]  + ' )  ' 
     else:
       if McTName[i][1].startswith('text1'):
         name=McTName[i][1].removeprefix('text1')
@@ -1535,7 +1580,6 @@ def mapnametoLegend(McTName):
   spreadsheet_service.spreadsheets().batchUpdate(spreadsheetId=spreadsheet_key, body=body2).execute()
   return summaryId,guessednamesfinal   
 
-
   # print(x,xarea)
 def getguessnames(gc,ws):
   guessednamesfinal=[]