Update deploying_3_3.py

deploying_3_3.py

@@ -41,6 +41,13 @@ Original file is located at
 # pip install ezdxf scipy
 
 """## Imports"""
+import xml.etree.ElementTree as ET
+from PyPDF2 import PdfReader, PdfWriter
+from PyPDF2.generic import TextStringObject, NameObject, ArrayObject, FloatObject
+from PyPDF2.generic import NameObject, TextStringObject, DictionaryObject, FloatObject, ArrayObject
+
+from typing import NewType
+from ctypes import sizeof
 
 import numpy as np
 import cv2
@@ -65,6 +72,8 @@ import tsadropboxretrieval
 from ezdxf import bbox
 from math import sin, cos, radians
 import google_sheet_Legend
+from PyPDF2 import PdfReader
+from io import BytesIO
 
 """## Notes"""
 
@@ -876,7 +885,8 @@ def rotate_polygon(polygon, angle, pdfrotation,width,height,center_point=(0, 0))
 
 
 def Create_DF(dxfpath,datadoc,hatched_areas):
-
+    
+ 
   FinalRatio= RetriveRatio(datadoc,dxfpath)
 
   # hatched_areas = get_hatched_areas(dxfpath,FinalRatio)
@@ -923,7 +933,159 @@ def Create_DF(dxfpath,datadoc,hatched_areas):
   return SimilarAreaDictionary
 """### Draw on Image and PDF"""
 
+def adjustannotations(OutputPdfStage1):
+  input_pdf_path = OutputPdfStage1
+  output_pdf_path = "AnnotationAdjusted.pdf"
+
+  # Load the input PDF
+  pdf_bytes_io = BytesIO(OutputPdfStage1)
+
+  reader = PdfReader(pdf_bytes_io)
+  writer = PdfWriter()
+
+  # Append all pages to the writer
+  writer.append_pages_from_reader(reader)
+
+  # Add metadata (optional)
+  metadata = reader.metadata
+  writer.add_metadata(metadata)
+
+  # Iterate over pages
+  for page_index, page in enumerate(writer.pages):
+      
+      # page.update({
+      #           NameObject("/UserUnit"): FloatObject(1.0),  # 1 unit = 1 real-world unit (e.g., 1 meter)
+      #           NameObject("/VP"): ArrayObject([
+      #               DictionaryObject({
+      #                   NameObject("/Type"): NameObject("/Viewport"),
+      #                   NameObject("/BBox"): ArrayObject([
+      #                   FloatObject(0), FloatObject(0), FloatObject(1000), FloatObject(1000)
+      #                   ]),  # Bounding box for the viewport
+      #                   NameObject("/Measure"): DictionaryObject({
+      #                       NameObject("/Type"): NameObject("/Measure"),
+      #                       NameObject("/Subtype"): NameObject("/RL"),
+      #                       NameObject("/X"): FloatObject(1),
+      #                       NameObject("/Y"): FloatObject(1),
+      #                       NameObject("/U"): TextStringObject("m"),  # Units (meters)
+      #                   }),
+      #               })
+      #           ])
+      #       })
+      if "/Annots" in page:
+          annotations = page["/Annots"]
+          for annot_index, annot in enumerate(annotations):
+              obj = annot.get_object()
+
+              print("obj", obj)
+              # print(obj.get("/IT"))
+
+              if obj.get("/Subtype") == "/Polygon":
+                print("AWL ANNOT IF")
+                # Check the /IT value to differentiate annotations
+                if "/Contents" in obj and "sq m" in obj["/Contents"]:
+                    print("Tany IF")
+                    obj.update({
+                        NameObject("/Measure"): DictionaryObject({
+                            NameObject("/Type"): NameObject("/Measure"),
+                            NameObject("/Area"): DictionaryObject({
+                                NameObject("/G"): FloatObject(1),
+                                NameObject("/U"): TextStringObject("sq m"),  # Unit of measurement for area
+                            }),
+                            
+                            NameObject("/X"): FloatObject(1),  # Horizontal scale (e.g., 1 unit = 1 meter)
+                            NameObject("/Y"): FloatObject(1),  # Vertical scale
+
+                        }),
+                        NameObject("/IT"): NameObject("/Area_Annotation"),  # Use more distinctive name
+                        NameObject("/Subj"): TextStringObject("Area Measurement"),   # Intent explicitly for Area
+                    })
+                  
+
+                print("After Update:", obj)
+  
+  # # Save the modified PDF
+
+  output_pdf_io = BytesIO()
+  writer.write(output_pdf_io)
+  output_pdf_io.seek(0)
+
+  return output_pdf_io.read()
+
+  # writer.write(new_bytes_object)  # This writes the modified PDF data to new_bytes_object
+  # new_bytes_object.seek(0)
+  # return new_bytes_object.read()
+
+# def adjustannotations(OutputPdfStage1):
+#     """
+#     Adjusts annotations in the PDF to include measurement and scale information.
+
+#     Parameters:
+#         OutputPdfStage1 (str): Path to the input PDF file.
+
+#     Returns:
+#         bytes: The adjusted PDF data as bytes.
+#     """
+#     with open(OutputPdfStage1, "rb") as pdf_file:
+#         reader = PdfReader(pdf_file)
+#         writer = PdfWriter()
+
+#         # Append all pages from reader to writer
+#         writer.append_pages_from_reader(reader)
+
+#         # Iterate over pages and add measurement details
+#         for page_index, page in enumerate(writer.pages):
+#             # Add scale settings at the page level
+#             # page.update({
+#             #     NameObject("/UserUnit"): FloatObject(1.0),  # 1 unit = 1 real-world unit
+#             #     NameObject("/VP"): ArrayObject([
+#             #         DictionaryObject({
+#             #             NameObject("/Type"): NameObject("/Viewport"),
+#             #             NameObject("/BBox"): ArrayObject([
+#             #                 FloatObject(0), FloatObject(0), FloatObject(100), FloatObject(100)
+#             #             ]),  # Bounding box for the viewport
+#             #             NameObject("/Measure"): DictionaryObject({
+#             #                 NameObject("/Type"): NameObject("/Measure"),
+#             #                 NameObject("/Subtype"): NameObject("/RL"),
+#             #                 NameObject("/X"): FloatObject(1),  # Horizontal scale
+#             #                 NameObject("/Y"): FloatObject(1),  # Vertical scale
+#             #                 NameObject("/U"): TextStringObject("m"),  # Units (meters)
+#             #             }),
+#             #         })
+#             #     ])
+#             # })
+
+#             # Process annotations
+#             if "/Annots" in page:
+#                 annotations = page["/Annots"]
+#                 for annot in annotations:
+#                     obj = annot.get_object()
+
+#                     # Adjust polygon annotations with area measurements
+#                     if obj.get("/Subtype") == "/Polygon" and "/Contents" in obj and "sq m" in obj["/Contents"]:
+#                         obj.update({
+#                             NameObject("/Measure"): DictionaryObject({
+#                                 NameObject("/Type"): NameObject("/Measure"),
+#                                 NameObject("/Area"): DictionaryObject({
+#                                     NameObject("/G"): FloatObject(1),
+#                                     NameObject("/U"): TextStringObject("sq m"),  # Area unit
+#                                 }),
+#                                 NameObject("/X"): FloatObject(1),  # Horizontal scale
+#                                 NameObject("/Y"): FloatObject(1),  # Vertical scale
+#                                 NameObject("/U"): TextStringObject("m"),  # Units (meters)
+#                             }),
+#                             NameObject("/IT"): NameObject("/Area_Annotation"),
+#                             NameObject("/Subj"): TextStringObject("Area Measurement"),
+#                         })
+
+#         print(obj)
+#         output_pdf_io = BytesIO()
+#         writer.write(output_pdf_io)
+#         output_pdf_io.seek(0)  # Ensure buffer is at the start
+#         return output_pdf_io
+
+    
 def mainFunctionDrawImgPdf(datadoc,dxfpath, dxfratio,pdfpath=0,pdfname=0):
+  OutputPdfStage1='BB Trial.pdf'
   FinalRatio= RetriveRatio(datadoc,dxfpath)
 
   # hatched_areas = get_hatched_areas(dxfpath,FinalRatio)
@@ -965,7 +1127,8 @@ def mainFunctionDrawImgPdf(datadoc,dxfpath, dxfratio,pdfpath=0,pdfname=0):
   for polygon in hatched_areas:
       cntPoints = []
       cntPoints1 = []
-      shapee = []
+      shapeePerimeter = []
+      shapeeArea = []
 
       blackImgShapes = np.zeros(imgg.shape[:2], dtype="uint8")
       blackImgShapes= cv2.cvtColor(blackImgShapes, cv2.COLOR_GRAY2BGR)
@@ -983,16 +1146,40 @@ def mainFunctionDrawImgPdf(datadoc,dxfpath, dxfratio,pdfpath=0,pdfname=0):
 
       cv2.drawContours(blackImgShapes, [np.array(cntPoints)], -1, ([255,255,255]), thickness=-1)
       x, y, w, h = cv2.boundingRect(np.array(cntPoints))
+      firstpoint = 0
+      for poi in np.array(cntPoints1):
+          if firstpoint == 0:
+            x2, y2 = poi
+            p2 = fitz.Point(x2,y2)
+            # p1 = fitz.Point(x1,y1)
+            p2=p2*derotationMatrix
+            shapeePerimeter.append([p2[0],p2[1]])
+            firstpoint = 1
+          else:
+            x1, y1 = poi
+            p1 = fitz.Point(x1,y1)
+            # p1 = fitz.Point(x1,y1)
+            p1=p1*derotationMatrix
+            print("P1 = ",p1)
+            shapeePerimeter.append([p1[0],p1[1]])
+
+      shapeePerimeter.append([p2[0],p2[1]])
+      shapeePerimeter=np.flip(shapeePerimeter,1)
+      shapeePerimeter=rotate_polygon(shapeePerimeter,rotationangle,rotationOld,width,height)
 
       for poi in np.array(cntPoints1):
-          x1, y1 = poi
-          p1 = fitz.Point(x1,y1)
-          # p1 = fitz.Point(x1,y1)
-          p1=p1*derotationMatrix
-          shapee.append([p1[0],p1[1]])
-
-      shapee=np.flip(shapee,1)
-      shapee=rotate_polygon(shapee,rotationangle,rotationOld,width,height)
+            x1, y1 = poi
+            p1 = fitz.Point(x1,y1)
+            # p1 = fitz.Point(x1,y1)
+            p1=p1*derotationMatrix
+            print("P1 = ",p1)
+            shapeeArea.append([p1[0],p1[1]])
+
+      shapeeArea.append([p2[0],p2[1]])
+      shapeeArea=np.flip(shapeeArea,1)
+      shapeeArea=rotate_polygon(shapeeArea,rotationangle,rotationOld,width,height)
+
+      
       tol=0
       condition1 = (SimilarAreaDictionary['Area'] >= polygon[1] - tol) & (SimilarAreaDictionary['Area'] <= polygon[1] +tol)
       condition2 = (SimilarAreaDictionary['Perimeter'] >= polygon[2] -tol) & (SimilarAreaDictionary['Perimeter'] <= polygon[2] +tol)
@@ -1080,20 +1267,20 @@ def mainFunctionDrawImgPdf(datadoc,dxfpath, dxfratio,pdfpath=0,pdfname=0):
             
       cv2.drawContours(imgg, [np.array(cntPoints)], -1, ([NewColors[2],NewColors[1],NewColors[0]]), thickness=-1)
                     
-      annot11 = page2.add_polygon_annot( points=shapee)  # 'Polygon'
+      annot11 = page2.add_polygon_annot( points=shapeeArea)  # 'Polygon'
       annot11.set_border(width=0.2)
       annot11.set_colors(stroke=(int(NewColors[0])/255,int(NewColors[1])/255,int(NewColors[2])/255), fill= (int(NewColors[0])/255,int(NewColors[1])/255,int(NewColors[2])/255) )
-      annot11.set_info(content='Area='+str(polygon[1])+' m^2',subject='ADR Team')
+      annot11.set_info(content=str(polygon[1])+' sq m',subject='Area Measurement', title="ADR Team")
       annot11.set_opacity(0.8)
       # annot.set_line_ends(fitz.PDF_ANNOT_LE_DIAMOND, fitz.PDF_ANNOT_LE_CIRCLE)
       annot11.update()
 
 
 
-      annot12 = page2.add_polygon_annot( points=shapee)  # 'Polygon'
-      annot12.set_border(width=0.2)
+      annot12 = page2.add_polyline_annot( points=shapeePerimeter )  # 'Polygon'
+      annot12.set_border(width=0.8)
       annot12.set_colors(stroke=(int(NewColors[0])/255,int(NewColors[1])/255,int(NewColors[2])/255))
-      annot12.set_info(content='Perimeter='+str(polygon[2])+' m',subject='ADR Team')
+      annot12.set_info(content=str(polygon[2])+' m',subject='Perimeter Measurement', title="ADR Team")
       annot12.set_opacity(0.8)
       # annot.set_line_ends(fitz.PDF_ANNOT_LE_DIAMOND, fitz.PDF_ANNOT_LE_CIRCLE)
       annot12.update()
@@ -1130,6 +1317,17 @@ def mainFunctionDrawImgPdf(datadoc,dxfpath, dxfratio,pdfpath=0,pdfname=0):
         }).reset_index()
     
   SimilarAreaDictionary = grouped_df
+  # doc.save(OutputPdfStage1)
+  modified_pdf_data = doc.tobytes()
+  OutputPdfStage2=adjustannotations(modified_pdf_data)
+
+  # with open("Adjusted_PDF.pdf", "wb") as f:
+  #   f.write(OutputPdfStage2)
+      
+  # doc2 = fitz.open(stream=OutputPdfStage2, filetype="pdf")
+  # doc2 = fitz.open(stream=OutputPdfStage2, filetype="pdf")
+      
+  doc2 =fitz.open('pdf',OutputPdfStage2)
     
   gc,spreadsheet_service,spreadsheetId, spreadsheet_url , namepathArr=google_sheet_Legend.legendGoogleSheets(SimilarAreaDictionary , pdfname,pdfpath)
   # dbxTeam=tsadropboxretrieval.ADR_Access_DropboxTeam('user')
@@ -1139,7 +1337,8 @@ def mainFunctionDrawImgPdf(datadoc,dxfpath, dxfratio,pdfpath=0,pdfname=0):
   # list1=pd.DataFrame(columns=['content', 'creationDate', 'id', 'modDate', 'name', 'subject', 'title'])
   list1=pd.DataFrame(columns=['content',  'id',  'subject','color'])
 
-  for page in doc:    
+  # for page in doc:
+  for page in doc2:  
     # Iterate through annotations on the page
     for annot in page.annots():
         # Get the color of the annotation
@@ -1155,7 +1354,8 @@ def mainFunctionDrawImgPdf(datadoc,dxfpath, dxfratio,pdfpath=0,pdfname=0):
               v='stroke'
             x,y,z=int(annot_color.get(v)[0]*255),int(annot_color.get(v)[1]*255),int(annot_color.get(v)[2]*255)
             list1.loc[len(list1)] =[annot.info['content'],annot.info['id'],annot.info['subject'],[x,y,z]]
-  return doc,image_new1, SimilarAreaDictionary ,spreadsheetId, spreadsheet_url , namepathArr , list1,hatched_areas
+  print('LISTTT',list1)
+  return doc2,image_new1, SimilarAreaDictionary ,spreadsheetId, spreadsheet_url , namepathArr , list1,hatched_areas
 
   # doc.save('Testing(2.7).pdf')
   # return doc,image_new1#, SimilarAreaDictionary ,spreadsheetId, spreadsheet_url , namepathArr , list1,hatched_areas
@@ -1166,3 +1366,4 @@ def mainFunctionDrawImgPdf(datadoc,dxfpath, dxfratio,pdfpath=0,pdfname=0):
 # doc,image_new1=mainFunctionDrawImgPdf(datadoc,dxfpath, dxfratio)
 # cv2_imshow(image_new1)
 
+