# -*- coding:utf-8 -*-
"""
實現(xiàn)arcpy按照圖斑自動截圖,要使用python2.7，需要提前設(shè)置好mxd工程，并進行保存，如果進行前后時相截圖，需要設(shè)置好label與所需季度影像
"""
import arcpy
#import os

#截圖地圖文件路徑
input = "D:\\data\\tb_1.shp"
mxd_path = u"D:\\data\\tb_1.mxd"
mxd = arcpy.mapping.MapDocument(mxd_path)#通過MapDocument函數(shù)來引用地圖文檔
df = arcpy.mapping.ListDataFrames(mxd)[0]#返回名為Layers的數(shù)據(jù)框?qū)ο?
for lyr in arcpy.mapping.ListLayers(mxd): #遍歷地圖文檔的圖層
    if lyr.name.startswith("tb_"):#指定圖層文件，搜索為tb開頭的文件夾，startwith函數(shù)用于檢查字符串是否已指定字符串為開頭，正確返回Trur,錯誤返回False。
        layermain = lyr#主要圖層指定為layermain


#輸出lyr.name10(輸出指定圖層的名稱（需要截圖圖層）)
print "Layer name: " + layermain.name

#循環(huán)
fields = ['TBBH']#根據(jù)field這個字段進行截圖，field必須為唯一字段
values1 = [row[0] for row in arcpy.da.SearchCursor(input, fields)]#使用游標訪問其中的字段，返回已fields字段的列表
#print values1
#清空選擇
#arcpy.SelectLayerByAttribute_management(layermain, "CLEAR_SELECTION")#按屬性選擇圖層工具，選擇要素與行，并移除所有選擇內(nèi)容
#arcpy.RefreshActiveView()#刷新當前的地圖文檔的活動視圖和內(nèi)容列表

delimfield = arcpy.AddFieldDelimiters(input, 'TBBH')#生成相應(yīng)的sql語句
#print delimfield

#循環(huán)
for val in values1:#遍歷字段列表
    try:
       # 狀態(tài)是0的圖斑
        whereClause0 = delimfield+" = '"+str(val)+"'"#whereClause為"TBBH=val"字符串
        print(whereClause0)
        arcpy.SelectLayerByAttribute_management(layermain, "NEW_SELECTION", whereClause0)#按屬性選擇圖層，創(chuàng)建一個新的選擇內(nèi)容，默認選擇方式為后邊字符串
        df.zoomToSelectedFeatures()#縮放至圖層，按照當前圖斑縮放
        df.scale = df.scale*2.5#圖版按位置縮放距離圖片邊界3倍位置，根據(jù)圖斑大小進行調(diào)節(jié)
        save_path = "D:\\data\\pic\\"+str(val)+".jpg"#保存圖片為JPEG格式路徑
        arcpy.mapping.ExportToJPEG(mxd,save_path)#按照val為文件名稱進行保存，保存格式為JPEG格式
#        arcpy.mapping.ExportToJPEG(mxd,save_path,resolution = 200)#按照val為文件名稱進行保存，保存格式為PNG格式
        arcpy.SelectLayerByAttribute_management(layermain, "CLEAR_SELECTION")#按照屬性選擇圖層，并移除所有選擇內(nèi)容
        arcpy.RefreshActiveView()#刷新當前活動視圖和內(nèi)容列表
    except:
        print 'wrong'
del mxd
print "Beautiful!"

  	@Test
    void pic() {
        String srcDirStr = "D:\\data\\output";
        String destDirStr = "D:\\data\\res";
        File srcDir = new File(srcDirStr);
        String[] srcList = srcDir.list();
        String year = "2020年影像";
        for (String srcFileStr : srcList) {
            File srcFile = new File(srcFileStr);
            String fileName = getFileNameNoEx(srcFile.getName());
            List<String> markList = new ArrayList<String>();
            markList.add(fileName);
            markList.add(year);
            String srcFilePath = srcDirStr + File.separator + srcFile.getName();
            String destFilePath = destDirStr + File.separator + srcFile.getName();
            pressText(markList,
                    srcFilePath, destFilePath,
                    "黑體",
                    Font.TYPE1_FONT, new Color(0, 255, 255), 0.9f);
        }
    }


    /**
     * 獲取不帶擴展名的文件名
     *
     * @param filename 文件名
     * @return 不帶擴展名的文件名
     */
    public String getFileNameNoEx(String filename) {
        if ((filename != null) && (filename.length() > 0)) {
            int dot = filename.lastIndexOf('.');
            if ((dot > -1) && (dot < (filename.length()))) {
                return filename.substring(0, dot);
            }
        }
        return filename;
    }

    /**
     * 給圖片添加文字水印
     *
     * @param pressTextList 水印文字(一個集合一個值對應(yīng)一條水印)
     * @param srcImageFile  源圖像地址
     * @param destImageFile 目標圖像地址
     * @param fontName      水印的字體名稱
     * @param fontStyle     水印的字體樣式
     * @param color         水印的字體顏色
     * @param alpha         透明度：alpha 必須是范圍 [0.0, 1.0] 之內(nèi)（包含邊界值）的一個浮點數(shù)字
     */
    public void pressText(List<String> pressTextList,
                          String srcImageFile, String destImageFile, String fontName,
                          int fontStyle, Color color, float alpha) {
        try {
            File img = new File(srcImageFile);
            Image src = ImageIO.read(img);
            int width = src.getWidth(null);
            int height = src.getHeight(null);
            BufferedImage image = new BufferedImage(width, height,
                    BufferedImage.TYPE_INT_RGB);
            Graphics2D g = image.createGraphics();
            // 計算水印字體大小 首先找出最長的水印文字
            //如果需要加水印的文字沒有超過18個就按18個計算
            int maxLength = 18;
            for (String pressText : pressTextList) {
                if (getLength(pressText) > maxLength) {
                    maxLength = getLength(pressText);
                }
            }
            //根據(jù)圖片寬度計算水字體大小
            int fontSize = (int) (((width) / (maxLength)) * 0.8);
            g.drawImage(src, 0, 0, width, height, null);
            g.setColor(color);
            g.setFont(new Font(fontName, fontStyle, fontSize));
//            g.setComposite(AlphaComposite.getInstance(AlphaComposite.SRC_ATOP,
//                    alpha));
            // 循環(huán)在指定坐標繪制水印文字
            int count = pressTextList.size();
            for (int i = 0; i < pressTextList.size(); i++) {
                if (i == 0) {
                    g.drawString(pressTextList.get(0), (int) ((width * 0.43) - (2 * fontSize)), (int) ((height * 0.05) + fontSize * count));
//                    g.drawString(pressTextList.get(0), (int) ((width * 0.35) - (2 * fontSize)), (int) ((height * 0.05) + fontSize * count));
                }
                if (i == 1) {
                    g.drawString(pressTextList.get(1), (int) ((width * 0.48) - (2 * fontSize)), (int) ((height * 0.90) - fontSize * count));
                }
                count--;
            }
            g.dispose();
            ImageIO.write(image, "JPEG", new File(destImageFile));// 輸出到文件流
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

    /**
     * 計算text的長度（一個中文算兩個字符）
     *
     * @param text 文本
     * @return int
     */
    public int getLength(String text) {
        int length = 0;
        for (int i = 0; i < text.length(); i++) {
            if (new String(text.charAt(i) + "").getBytes().length > 1) {
                length += 2;
            } else {
                length += 1;
            }
        }
        return length / 2;
    }