Spaces:
Sleeping
Sleeping
| # -*- 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 | |
| """### Open PDF and draw a rectangle on it using Fitz""" | |
| def openDrawPDF(path): | |
| doc = fitz.open(path) | |
| page = doc[0] | |
| pix = page.get_pixmap(dpi=200) # render page to an image | |
| rotate=0 | |
| if page.rect.height > page.rect.width: | |
| rect = fitz.Rect(80,80, (page.rect.width-80),( page.rect.width+80)) | |
| rectText=fitz.Rect(300, 200, (page.rect.width-80),( page.rect.width+80) ) | |
| rotate=0 | |
| else: | |
| rect = fitz.Rect(80,80, (page.rect.height-80),( page.rect.height+80)) | |
| rectText=fitz.Rect(500, 200, (page.rect.height-80),( page.rect.height+80) ) | |
| rotate=90 | |
| annot = page.add_rect_annot(rect) | |
| annot.set_colors( fill=(75/255,0,130/255) ,stroke=(75/255,0,130/255) ) | |
| annot.set_opacity(0.9) | |
| annot.set_border(border=None, width=0) | |
| annot.update() | |
| 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=rotate, align=1) | |
| annot1.update() | |
| perm = int( # permissions bit flags | |
| fitz.PDF_PERM_ACCESSIBILITY #always use this | |
| | fitz.PDF_PERM_PRINT # permits printing | |
| ) | |
| encrypt = fitz.PDF_ENCRYPT_AES_256 # strongest algorithm | |
| # Ndoc=doc.write(encryption=encrypt,permissions=perm) | |
| return doc | |
| """### 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(pdfpath, plan): | |
| dbx=db.dropbox_connect() | |
| md, res =dbx.files_download(path= pdfpath+plan) | |
| data = res.content | |
| doc=fitz.open("pdf", data) | |
| for page in doc: | |
| pix = page.get_pixmap(dpi=200) # 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 | |
| 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 |