save

.gitignore

@@ -0,0 +1 @@
+img/

db_util.py

@@ -17,25 +17,19 @@ conn = psycopg2.connect(
 )
 
 
-def getDiffEwkt(code, ewkt):
+def getDiffEwkt(ewkt, tbName):
     with conn.cursor(cursor_factory=DictCursor) as cur:
-        # sql = """ SELECT
-        #                 st_asewkt ( st_difference ( st_geomfromewkt ( %s ), tb.geom ) )
-        #             FROM
-        #                 tb
-        #             WHERE
-        #                 tb."编号" = %s;"""
-        sql="""SELECT
+        sql = f"""SELECT
 	st_asewkt ( st_difference ( st_geomfromewkt ( %s ), st_union ( tb.geom ) ) )
 FROM
-	tb
+	"{tbName}" as tb
 WHERE
 	st_contains ( st_geomfromewkt ( %s ), tb.geom );"""
-        # complete_sql = cur.mogrify(sql, (ewkt,ewkt,)).decode('utf-8')
-        # logger.info(f"Executing SQL: {complete_sql}")
-        cur.execute(sql, (ewkt,ewkt,))
+        complete_sql = cur.mogrify(sql, (ewkt,  ewkt,)).decode('utf-8')
+        logger.info(f"Executing SQL: {complete_sql}")
+        cur.execute(sql, (ewkt,  ewkt,))
         res = cur.fetchone()
     return res[0]
 # ewkt='SRID=4490;POLYGON ((106.67844989540063 38.207187208754306, 106.67844989540063 38.212680372816806, 106.67295673133813 38.212680372816806, 106.67295673133813 38.207187208754306, 106.67844989540063 38.207187208754306))'
 # getEwkt=getDiffEwkt('6401812024070108270001',ewkt)
-# print(getEwkt)
+# print(getEwkt)

file_util.py

@@ -13,12 +13,27 @@ def getAllFiles(filePath):
     for root, dirs, files in os.walk(folder_path):
         for file in files:
             file_path = os.path.join(root, file)
-            fileList.append(file_path)
+            if file_path.endswith('.tif'):
+                fileList.append(file_path)
     return fileList
 
 # allPath = getAllFiles('img/DDOM')
 
 
+# 获取当前目录下的文件夹
+def getDirList(filePath,startName):
+    dirList = []
+    # 设置要遍历的文件夹路径
+    folder_path = os.path.join(os.path.dirname(__file__), filePath)
+    # 遍历文件夹
+    for root, dirs, files in os.walk(folder_path):
+        for dir in dirs:
+            if dir.startswith(startName):
+                dirList.append(filePath+'/'+dir)
+    return dirList
+# getDirList('img')
+
+
 def getTifEwkt(filePath):
     # 读取 tif 文件
     with rasterio.open(filePath) as src:

img_util.py

@@ -2,55 +2,26 @@ import geopandas as gpd
 import matplotlib.pyplot as plt
 from shapely.wkt import loads
 from PIL import Image
+import os
 # 调整尺寸
 
 
-def resizeImage(filePath):
+def resizeImage(dir,fileName):
     # 打开原始图片
-    original_image = Image.open(filePath)
+    original_image = Image.open(dir+"/"+fileName)
     # 调整图片尺寸到 256x256
     resized_image = original_image.resize((256, 256))
-    filePath = filePath.replace('tb', 'tb_256')
+    dir = dir.replace('tb', 'tb_256')
+    if not os.path.exists(dir):
+        os.makedirs(dir)
     # 保存调整后的图片
-    resized_image.save(filePath)
+    resized_image.save(dir+"/"+fileName)
 
 
 # resizeImage('img/tb/6401812024070108270001change.png')
 
-# # 矢量保存为 PNG 图片
-# def savePng(fileName, ewkt):
-#     # 新的 WKT 数据字符串
-
-#     # 创建一个 GeoDataFrame
-#     polygon = loads(ewkt.split(';')[1])  # 解析 WKT
-#     gdf = gpd.GeoDataFrame(geometry=[polygon])
-
-#     # 绘制
-#     fig, ax = plt.subplots(figsize=(10, 10))
-
-#     # 绘制多边形，外部填充为黑色，内部（洞）为白色
-#     gdf.plot(ax=ax, edgecolor='black', facecolor='black')
-
-#     # 设定中间洞为白色
-#     ax.fill(*polygon.exterior.xy, color='black')  # 填充外部为黑色
-#     for interior in polygon.interiors:
-#         ax.fill(*interior.xy, color='white')  # 填充内部（洞）为白色
-
-#     # 设置坐标轴范围
-#     ax.set_xlim(gdf.total_bounds[[0, 2]])
-#     ax.set_ylim(gdf.total_bounds[[1, 3]])
-
-#     # 去掉坐标轴
-#     ax.axis('off')
-
-#     # 保存为 PNG 图片
-#     filePath = 'img/tb/'+fileName+'.png'
-#     plt.savefig(filePath, dpi=300,
-#                 bbox_inches='tight', pad_inches=0)
-#     plt.close()
-#     resizeImage(filePath)
-
-def savePng(fileName, ewkt):
+# 保存为 PNG 图片
+def savePng(fileName, ewkt,dir):
     # 解析 WKT 数据字符串
     geometry = loads(ewkt.split(';')[1])  # 解析 WKT
     gdf = gpd.GeoDataFrame(geometry=[geometry])
@@ -78,12 +49,16 @@ def savePng(fileName, ewkt):
     ax.axis('off')
 
     # 保存为 PNG 图片
-    filePath = 'img/tb/' + fileName + '.png'
-    plt.savefig(filePath, dpi=300, bbox_inches='tight', pad_inches=0)
+    dir=dir+'/tb'
+    if not os.path.exists(dir):
+        os.makedirs(dir)
+
+    fileName = fileName + '.png'
+    plt.savefig(dir+"/"+fileName, dpi=300, bbox_inches='tight', pad_inches=0)
     plt.close()
 
     # 调用 resizeImage 函数来调整图片大小
-    resizeImage(filePath)
+    resizeImage(dir,fileName)
 
 
 

main.py

@@ -4,22 +4,22 @@ from db_util import getDiffEwkt
 from img_util import savePng
 import re
 
+# 1.获取图斑的根目录
+dirs = getDirList('img', '2024')
+for dir in dirs:
+    # 2.获取tif路径
+    allPathList = getAllFiles(dir+'/DDOM')
 
-# 1.获取tif路径
-allPathList = getAllFiles('img/DDOM')
+    for filePath in allPathList:
 
-for filePath in allPathList:
+        print('tif路径：'+filePath)
 
-    filePath='img/DDOM/6401812024073108270308DDOM.tif'
-    print('tif路径：'+filePath)
-
-    # 2.获取tif的边界矢量范围
-    ewkt = getTifEwkt(filePath)
-    # 3.tif边界-变化矢量（生成图片的黑色部分）
-    match = re.search(r'/DDOM/(.*)DDOM\.tif', filePath)
-    code= match.group(1) if match else ''
-    ewkt=getDiffEwkt(code,ewkt)
-    # 4.生成变化的图片
-    savePng(code+"_CHANGE",ewkt)
-    
-    break
+        # 3.获取tif的边界矢量范围
+        ewkt = getTifEwkt(filePath)
+        # 4.tif边界-变化矢量（生成图片的黑色部分）
+        match = re.search(r'/DDOM/(.*)DDOM\.tif', filePath)
+        code = match.group(1) if match else ''
+        tbName=dir.replace('img/','')
+        ewkt = getDiffEwkt(ewkt, tbName)
+        # 5.生成变化的图片
+        savePng(code+"_CHANGE", ewkt, dir)