Update pixelconversion.py

pixelconversion.py

@@ -24,34 +24,48 @@ from io import BytesIO
 ################################################
 ### Check if page is visually rotated or not 
 
-def is_content_rotated(mask):
-      contours, _ = cv2.findContours(mask, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
-      # Check the orientation of the purple shape 
+def is_content_rotated(docpage):
+      docpage.draw_rect([250,250,docpage.mediabox.width-500,docpage.mediabox.height-500],  color = (0,0,0), width = 1,fill=(0,0,0),fill_opacity=1)
+      pix = docpage.get_pixmap()
+      image = Image.frombytes("RGB", [pix.width, pix.height], pix.samples)
+      img = cv2.cvtColor(np.array(image), cv2.COLOR_RGB2GRAY)
+      kernel = np.ones((3,3),np.uint8)
+      img = cv2.dilate(img,kernel, iterations=6)
+      img=cv2.threshold(img, 0, 255, cv2.THRESH_BINARY_INV)[1]
+      # cv2_imshow(img)
+      contours, _ = cv2.findContours(img, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
+      # Check the orientation of page through large shape drawn
       for contour in contours:
           rect = cv2.minAreaRect(contour)
           angle = rect[-1]
-          print(angle)
-          if angle != 0.0 and angle != 90.0 and angle != 180.0:
+          # print(angle)
+          if angle != 0.0 and angle != 90.0 and angle != 180.0 and angle!=-90.0 and angle!=-180.0:
               print("Page content appears visually rotated.")
               return True
       print("Page content does not appear visually rotated.")
       return False
 
-def drawisrotated(data):
+def drawisrotated(data,dpi=0):
   docPath = fitz.open("pdf",data) #dropbox path
   pageDocPath=docPath[0]
-  if pageDocPath.rotation !=0:
-   pageDocPath.draw_rect([250,250,pageDocPath.mediabox.width-500,pageDocPath.mediabox.height-500],  color = (75/255,0,130/255), width = 1,fill=(75/255,0,130/255),fill_opacity=0.9)
-  else:
-   pageDocPath.draw_rect([0+250,0+250,pageDocPath.mediabox.width-500,pageDocPath.mediabox.height-500],  color =(75/255,0,130/255), width = 1,fill=(75/255,0,130/255),fill_opacity=0.9)
   
-  pix = pageDocPath.get_pixmap()
-  image = Image.frombytes("RGB", [pix.width, pix.height], pix.samples)
-  img = cv2.cvtColor(np.array(image), cv2.COLOR_RGB2BGR)
-  mask=DetectColor(img,color=(75,0,130))
-  bool_T_F=is_content_rotated(mask)
-  print(bool_T_F)
+  bool_T_F=is_content_rotated(pageDocPath)
+
   if bool_T_F==True:
+    docPath = fitz.open("pdf",data) #dropbox path
+    pageDocPath=docPath[0]
+    pageDocPath.draw_rect([250,250,pageDocPath.mediabox.width-500,pageDocPath.mediabox.height-500],  color = (75/255,0,130/255), width = 1,fill=(75/255,0,130/255),fill_opacity=0.9)
+    if dpi:
+      pix=pageDocPath.get_pixmap(dpi=dpi)
+
+      newwidth_ratio=pix.width/pageDocPath.mediabox.width
+      newheight_ratio=pix.height/pageDocPath.mediabox.height
+    else:
+      newwidth_ratio=1
+      newheight_ratio=1
+    area=((pageDocPath.mediabox.height-500-250)*newheight_ratio)  *  ((pageDocPath.mediabox.width-500-250)*newwidth_ratio)
+    perimeter=(((pageDocPath.mediabox.width-500-250)*newwidth_ratio) *2)  + ( ((pageDocPath.mediabox.height-500-250)*newheight_ratio)  *2)
+
     if pageDocPath.mediabox.height >pageDocPath.mediabox.width:
       rectText=fitz.Rect(300, 200, (pageDocPath.mediabox.width-80),( pageDocPath.mediabox.width+80) )
     else:
@@ -60,81 +74,58 @@ def drawisrotated(data):
     text = """Scale Document"""
     annot1=pageDocPath.add_freetext_annot(rectText, text, fontsize=45, fontname='helv', border_color=(1,1,1), text_color=(1,1,1), rotate= pageDocPath.rotation, align=1)
     annot1.update()
-    return docPath
+    # docPath.save('drawnOn.pdf')
+    return docPath,area,perimeter
   else:
-    doc=openDrawPDF(data)
-    return doc
+    if dpi:
+      doc,area,perimeter=openDrawPDF(data,dpi=dpi)
+      return doc,area,perimeter 
+    else:
+      doc,area,perimeter=openDrawPDF(data)
+      return doc,area,perimeter,docPath
 
 
 """### Open PDF and draw a rectangle on it using Fitz"""
-def openDrawPDF(data):
+def openDrawPDF(data,dpi=0):
   
   doc=fitz.open("pdf", data)
   page = doc[0]
-  
-  if page.rect.height > page.rect.width:
-    rectText=fitz.Rect(300, 200, (page.rect.width-80),( page.rect.width+80) )
+
+
+  if page.rect.height >page.rect.width:
+    rectText=fitz.Rect(300, 200, (page.mediabox.width-80),( page.mediabox.width+80) )
   else:
-    rectText=fitz.Rect(500, 200, (page.rect.height-80),( page.rect.height+80) )
+    rectText=fitz.Rect(500, 200, (page.mediabox.height-80),( page.mediabox.height+80) )
   if page.rotation !=0:
-    page.draw_rect([0+10,0+10,page.rect.height-10,page.rect.width-10],  color = (75/255,0,130/255), width = 1,fill=(75/255,0,130/255),fill_opacity=0.9 )
+    # print('if')
+    page.draw_rect([250,250,page.mediabox.width-500,page.mediabox.height-500],  color = (75/255,0,130/255), width = 1,fill=(75/255,0,130/255),fill_opacity=0.9 )
+    if dpi:
+      pix=page.get_pixmap(dpi=dpi)
+      newwidth_ratio=pix.width/page.mediabox.width
+      newheight_ratio=pix.height/page.mediabox.height
+    else:
+      newwidth_ratio=1
+      newheight_ratio=1
+    area=((page.mediabox.height-500-250)*newheight_ratio) * ((page.mediabox.width-500-250)*newwidth_ratio)
+    perimeter=(((page.mediabox.width-500-250)*newwidth_ratio)*2) + (((page.mediabox.height-500-250)*newheight_ratio)*2)
+
   else:
-    page.draw_rect([0+10,0+10,page.rect.width-10, page.rect.height-10],  color = (75/255,0,130/255), width = 1,fill=(75/255,0,130/255),fill_opacity=0.9 )
- 
+    page.draw_rect([0+10,0+10,page.mediabox.width-10,page.mediabox.height-10],  color = (75/255,0,130/255), width = 1,fill=(75/255,0,130/255),fill_opacity=0.9 )
+    if dpi:
+      pix=page.get_pixmap(dpi=dpi)
+      newwidth_ratio=pix.width/page.mediabox.width
+      newheight_ratio=pix.height/page.mediabox.height
+    else:
+      newwidth_ratio=1
+      newheight_ratio=1
+    area=((page.mediabox.width-10-10)*newwidth_ratio)*((page.mediabox.height-10-10)*newheight_ratio)
+    perimeter=(((page.mediabox.width-10-10)*2)*newwidth_ratio)+(((page.mediabox.height-10-10)*2)*newheight_ratio)
+
   text = """Scale Document"""
   annot1=page.add_freetext_annot(rectText, text, fontsize=45, fontname='helv', border_color=(1,1,1), text_color=(1,1,1), rotate= page.rotation, align=1)
   annot1.update()
 
-  return doc#,encrypt,perm
+  # doc.save('drawnOn.pdf')#, encryption=encrypt, permissions=perm)
+  return  doc,area,perimeter
 """### Extract color"""
 
-def DetectColor(img,color=0):
-
-  imgCopy=img.copy()
-  imgCopy=cv2.cvtColor(imgCopy,cv2.COLOR_BGR2HSV)
-  tol=5 #tolerance
-  # color=hexRGB(color)
-  h,s,v = cv2.cvtColor(np.uint8([[[color[2],color[1],color[0]]]]),cv2.COLOR_BGR2HSV)[0][0]
-
-  lower =np.array( [h- tol, 100, 100 ], dtype='uint8')
-  upper = np.array( [h + tol, 255, 255],dtype='uint8')
-
-  mask = cv2.inRange(imgCopy, lower , upper)
-
-  detectedColors = cv2.bitwise_and(imgCopy,imgCopy, mask= mask)   # Bitwise-AND mask and original image
-
-  kernel=np.ones((3,3),np.uint8)
-  mask=cv2.dilate(mask,kernel, iterations=5)
-  mask=cv2.erode(mask,kernel, iterations=4)
-
-  detectedColors=cv2.dilate(detectedColors,kernel, iterations=5)
-  detectedColors=cv2.erode(detectedColors,kernel, iterations=4)
-
-  detectedColors=cv2.cvtColor(detectedColors,cv2.COLOR_HSV2BGR)
-  detectedColors=cv2.medianBlur(detectedColors,7)
-  # cv2_imshow(detectedColors)
-
-  return mask
-
-"""### For backend - calc area and perim"""
-
-def getAreaPerimeter(path,name):
-  dbxTeam=tsadropboxretrieval.ADR_Access_DropboxTeam('user')
-  md, res =dbxTeam.files_download(path= path+name)
-  data = res.content
-  doc=fitz.open("pdf", data)
-  area=0
-  perimeter=0
-  for page in doc:
-      pix = page.get_pixmap(dpi=300)  # render page to an image
-      pl=Image.frombytes('RGB', [pix.width,pix.height],pix.samples)
-      img=np.array(pl)
-      print(img.shape)
-      img = cv2.cvtColor(img, cv2.COLOR_RGB2BGR)
-  mask=DetectColor(img,color=(73,0,130)) #detect colored rect drawn on the pdf
-  # cv2.imwrite('maskk.png',mask)
-  contours, hierarchy = cv2.findContours(mask, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
-  for contour in contours:
-      area = cv2.contourArea(contour)
-      perimeter = cv2.arcLength(contour, True)
-  return area,perimeter