Update pixelconversion.py

pixelconversion.py

@@ -1,239 +1,240 @@
-# -*- coding: utf-8 -*-
-"""pixelconversion.ipynb
-
-Automatically generated by Colaboratory.
-
-Original file is located at
-    https://colab.research.google.com/drive/1hfdgkYOw8w6DdJqsZx8txmw8INUGxesl
-"""
-
-
-# pip install pymupdf -q
-
-"""### Imports"""
-
-import fitz
-from PIL import Image
-import numpy as np
-import cv2
-import db
-import tsadropboxretrieval
-import fitz
-import math
-from io import BytesIO
-
-################################################
-### Check if page is visually rotated or not 
-
-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 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=0,pdf_content=0,dpi=0):
-#   if pdf_content:
-#     docPath = fitz.open(stream=pdf_content, filetype="pdf")
-#   else:
-#     docPath = fitz.open("pdf",data) #dropbox path
-#   pageDocPath=docPath[0]
-  
-#   bool_T_F=is_content_rotated(pageDocPath)
-
-#   if bool_T_F==True:
-#     if pdf_content:
-#       docPath = fitz.open(stream=pdf_content, filetype="pdf")
-#     else:
-#       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:
-#       rectText=fitz.Rect(500, 200, (pageDocPath.mediabox.height-80),( pageDocPath.mediabox.height+80) )
-
-#     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()
-#     # docPath.save('drawnOn.pdf')
-#     return docPath,area,perimeter
-#   else:
-#     if pdf_content:
-#       if dpi:
-#         doc,area,perimeter=openDrawPDF(pdf_content=pdf_content,dpi=dpi)
-#         return doc,area,perimeter 
-#       else:
-#         doc,area,perimeter=openDrawPDF(pdf_content=pdf_content)
-#         return doc,area,perimeter
-#     else:
-#       if dpi:
-#         doc,area,perimeter=openDrawPDF(data,dpi=dpi)
-#         return doc,area,perimeter 
-#       else:
-#         doc,area,perimeter=openDrawPDF(data)
-#         return doc,area,perimeter
-    
-
-    
-def drawisrotated(data=0,pdf_content=0,dpi=0):
-  if pdf_content:
-    docPath = fitz.open(stream=pdf_content, filetype="pdf")
-  else:
-    docPath = fitz.open("pdf",data) #dropbox path
-  if pdf_content:
-    if dpi:
-      doc,area,perimeter=openDrawPDF(pdf_content=pdf_content,dpi=dpi)
-      return doc,area,perimeter 
-    else:
-      doc,area,perimeter=openDrawPDF(pdf_content=pdf_content)
-      return doc,area,perimeter
-  else:
-    if dpi:
-      doc,area,perimeter=openDrawPDF(data,dpi=dpi)
-      return doc,area,perimeter 
-    else:
-      doc,area,perimeter=openDrawPDF(data)
-      return doc,area,perimeter
-
-
-
-# def openDrawPDF(data=0,pdf_content=0,dpi=0):
-  
-#     if pdf_content:
-#       doc = fitz.open(stream=pdf_content, filetype="pdf")
-#     else:
-#       doc = fitz.open("pdf",data) #dropbox path
-#     page = doc[0]
-#     print(page.rotation)
-
-#     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.mediabox.height - 80), (page.mediabox.height + 80))
-#     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:
-#         print('No DPI specified')
-#         newwidth_ratio = 1
-#         newheight_ratio = 1
-
-#     # Apply the transformation to the rectangle's area and perimeter
-#     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)
-
-#     # Apply the transformation to the text annotation (shift and scale its position)
-#     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()
-#     # print('fitzzzzzzzzzzzz',fitz.__version__)
-#     # doc.save('drawnOn_transformed.pdf')
-#     return doc, area, perimeter
-
-def openDrawPDF(data=0,pdf_content=0,dpi=0):
-    # Open the document
-    # doc = fitz.open(pdfOutput)  # Dropbox 
-    if pdf_content:
-      doc = fitz.open(stream=pdf_content, filetype="pdf")
-    else:
-      doc = fitz.open("pdf",data) #dropbox path
-    page = doc[0]
-    derotationMatrix=page.derotation_matrix
-    print(page.rotation)
-
-    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.mediabox.height - 80), (page.mediabox.height + 80))
-
-    if dpi:
-        pix = page.get_pixmap(dpi=dpi)
-        newwidth_ratio = pix.width / page.mediabox.width
-        newheight_ratio = pix.height / page.mediabox.height
-    else:
-        print('No DPI specified')
-        newwidth_ratio = 1
-        newheight_ratio = 1
-
-    # Apply the transformation to the rectangle's area and perimeter
-    area = ((page.mediabox.width) * newwidth_ratio) * ((page.mediabox.height ) * newheight_ratio)
-    perimeter = (((page.mediabox.width ) * 2) * newwidth_ratio) + (((page.mediabox.height) * 2) * newheight_ratio)
-    # full page rectangle
-
-    # Original diagonal points
-    startRect = fitz.Point(0, 0)
-    endRect = fitz.Point(page.mediabox.width, page.mediabox.height)
-
-    # Apply derotation if needed
-    # startRect = startRect * derotationMatrix
-    # endRect = endRect * derotationMatrix
-
-    # Create rectangle corners from diagonal points
-    top_left = fitz.Point(startRect.x, startRect.y)
-    top_right = fitz.Point(endRect.x, startRect.y)
-    bottom_right = fitz.Point(endRect.x, endRect.y)
-    bottom_left = fitz.Point(startRect.x, endRect.y)
-
-    # List of points for polyline (closing the rectangle)
-    rect_points = [top_left, top_right, bottom_right, bottom_left, top_left]
-
-    # Add polyline annotation
-    annot = page.add_polyline_annot(rect_points)
-    annot.set_colors(stroke=(75/255, 0, 130/255), fill=(75/255, 0, 130/255))
-    annot.set_opacity(0.9)
-    annot.set_border(width=20)
-    annot.set_info(content=str(perimeter)+' m', subject='Perimeter Measurement', title="ADR Team")
-    annot.update()
-
-    page_width  = page.rect.width
-    page_height = page.rect.height
-    startLength = fitz.Point(0, page_height / 2)   *derotationMatrix     # left side
-    endLength = fitz.Point(page_width, page_height / 2)  *derotationMatrix  # right side
-
-    print(f"Line length: {length} points")
-    # startLength = startLength * derotationMatrix
-    # endLength = endLength * derotationMatrix
-
-    shapeLength=[startLength, endLength]
-    line_annot = page.add_polyline_annot(shapeLength)
-    line_annot.set_colors(stroke=(75/255, 0, 130/255))  # red line
-    line_annot.set_border(width=10)
-    line_annot.set_info(content=str(length)+' m',subject='Perimeter Measurement', title="ADR Team")
-    line_annot.update()
-
-    # Apply the transformation to the text annotation (shift and scale its position)
-    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()
-
-    # doc.save('drawnOn_transformed.pdf')
+# -*- coding: utf-8 -*-
+"""pixelconversion.ipynb
+
+Automatically generated by Colaboratory.
+
+Original file is located at
+    https://colab.research.google.com/drive/1hfdgkYOw8w6DdJqsZx8txmw8INUGxesl
+"""
+
+
+# pip install pymupdf -q
+
+"""### Imports"""
+
+import fitz
+from PIL import Image
+import numpy as np
+import cv2
+import db
+import tsadropboxretrieval
+import fitz
+import math
+from io import BytesIO
+
+################################################
+### Check if page is visually rotated or not 
+
+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 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=0,pdf_content=0,dpi=0):
+#   if pdf_content:
+#     docPath = fitz.open(stream=pdf_content, filetype="pdf")
+#   else:
+#     docPath = fitz.open("pdf",data) #dropbox path
+#   pageDocPath=docPath[0]
+  
+#   bool_T_F=is_content_rotated(pageDocPath)
+
+#   if bool_T_F==True:
+#     if pdf_content:
+#       docPath = fitz.open(stream=pdf_content, filetype="pdf")
+#     else:
+#       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:
+#       rectText=fitz.Rect(500, 200, (pageDocPath.mediabox.height-80),( pageDocPath.mediabox.height+80) )
+
+#     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()
+#     # docPath.save('drawnOn.pdf')
+#     return docPath,area,perimeter
+#   else:
+#     if pdf_content:
+#       if dpi:
+#         doc,area,perimeter=openDrawPDF(pdf_content=pdf_content,dpi=dpi)
+#         return doc,area,perimeter 
+#       else:
+#         doc,area,perimeter=openDrawPDF(pdf_content=pdf_content)
+#         return doc,area,perimeter
+#     else:
+#       if dpi:
+#         doc,area,perimeter=openDrawPDF(data,dpi=dpi)
+#         return doc,area,perimeter 
+#       else:
+#         doc,area,perimeter=openDrawPDF(data)
+#         return doc,area,perimeter
+    
+
+    
+def drawisrotated(data=0,pdf_content=0,dpi=0):
+  if pdf_content:
+    docPath = fitz.open(stream=pdf_content, filetype="pdf")
+  else:
+    docPath = fitz.open("pdf",data) #dropbox path
+  if pdf_content:
+    if dpi:
+      doc,area,perimeter=openDrawPDF(pdf_content=pdf_content,dpi=dpi)
+      return doc,area,perimeter 
+    else:
+      doc,area,perimeter=openDrawPDF(pdf_content=pdf_content)
+      return doc,area,perimeter
+  else:
+    if dpi:
+      doc,area,perimeter=openDrawPDF(data,dpi=dpi)
+      return doc,area,perimeter 
+    else:
+      doc,area,perimeter=openDrawPDF(data)
+      return doc,area,perimeter
+
+
+
+# def openDrawPDF(data=0,pdf_content=0,dpi=0):
+  
+#     if pdf_content:
+#       doc = fitz.open(stream=pdf_content, filetype="pdf")
+#     else:
+#       doc = fitz.open("pdf",data) #dropbox path
+#     page = doc[0]
+#     print(page.rotation)
+
+#     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.mediabox.height - 80), (page.mediabox.height + 80))
+#     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:
+#         print('No DPI specified')
+#         newwidth_ratio = 1
+#         newheight_ratio = 1
+
+#     # Apply the transformation to the rectangle's area and perimeter
+#     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)
+
+#     # Apply the transformation to the text annotation (shift and scale its position)
+#     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()
+#     # print('fitzzzzzzzzzzzz',fitz.__version__)
+#     # doc.save('drawnOn_transformed.pdf')
+#     return doc, area, perimeter
+
+def openDrawPDF(data=0,pdf_content=0,dpi=0):
+    # Open the document
+    # doc = fitz.open(pdfOutput)  # Dropbox 
+    if pdf_content:
+      doc = fitz.open(stream=pdf_content, filetype="pdf")
+    else:
+      doc = fitz.open("pdf",data) #dropbox path
+    page = doc[0]
+    derotationMatrix=page.derotation_matrix
+    print(page.rotation)
+
+    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.mediabox.height - 80), (page.mediabox.height + 80))
+
+    if dpi:
+        pix = page.get_pixmap(dpi=dpi)
+        newwidth_ratio = pix.width / page.mediabox.width
+        newheight_ratio = pix.height / page.mediabox.height
+    else:
+        print('No DPI specified')
+        newwidth_ratio = 1
+        newheight_ratio = 1
+
+    # Apply the transformation to the rectangle's area and perimeter
+    area = ((page.mediabox.width) * newwidth_ratio) * ((page.mediabox.height ) * newheight_ratio)
+    perimeter = (((page.mediabox.width ) * 2) * newwidth_ratio) + (((page.mediabox.height) * 2) * newheight_ratio)
+    # full page rectangle
+
+    # Original diagonal points
+    startRect = fitz.Point(0, 0)
+    endRect = fitz.Point(page.mediabox.width, page.mediabox.height)
+
+    # Apply derotation if needed
+    # startRect = startRect * derotationMatrix
+    # endRect = endRect * derotationMatrix
+
+    # Create rectangle corners from diagonal points
+    top_left = fitz.Point(startRect.x, startRect.y)
+    top_right = fitz.Point(endRect.x, startRect.y)
+    bottom_right = fitz.Point(endRect.x, endRect.y)
+    bottom_left = fitz.Point(startRect.x, endRect.y)
+
+    # List of points for polyline (closing the rectangle)
+    rect_points = [top_left, top_right, bottom_right, bottom_left, top_left]
+
+    # Add polyline annotation
+    annot = page.add_polyline_annot(rect_points)
+    annot.set_colors(stroke=(75/255, 0, 130/255), fill=(75/255, 0, 130/255))
+    annot.set_opacity(0.9)
+    annot.set_border(width=20)
+    annot.set_info(content=str(perimeter)+' m', subject='Perimeter Measurement', title="ADR Team")
+    annot.update()
+
+    page_width  = page.rect.width
+    page_height = page.rect.height
+    startLength = fitz.Point(0, page_height / 2)   *derotationMatrix     # left side
+    endLength = fitz.Point(page_width, page_height / 2)  *derotationMatrix  # right side
+
+    length = math.sqrt((endLength.x - startLength.x)**2 + (endLength.y - startLength.y)**2)
+    print(f"Line length: {length} points")
+    # startLength = startLength * derotationMatrix
+    # endLength = endLength * derotationMatrix
+
+    shapeLength=[startLength, endLength]
+    line_annot = page.add_polyline_annot(shapeLength)
+    line_annot.set_colors(stroke=(75/255, 0, 130/255))  # red line
+    line_annot.set_border(width=10)
+    line_annot.set_info(content=str(length)+' m',subject='Perimeter Measurement', title="ADR Team")
+    line_annot.update()
+
+    # Apply the transformation to the text annotation (shift and scale its position)
+    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()
+
+    # doc.save('drawnOn_transformed.pdf')
     return doc, area, perimeter