PRO Subset_via_shp_update
COMPILE_OPT idl2
ENVI,/restore_base_save_files
envi_batch_init,LOG_FILE='batch.log'
;打開要裁剪的圖像
image_dir='F:\PDF\' ;根據文件存放的目錄進行相應修改
image_files=file_search(image_dir,'*.tif',count=numfiles) ;根據相應的文件格式修改過濾條件
for i=0,numfiles-1 do begin
image_file=image_files[i]
print,image_file
if strlen(image_file) eq 0 then return
ENVI_OPEN_FILE, image_file, r_fid=fid, /no_interactive_query, /no_realize
IF fid EQ -1 THEN RETURN
ENVI_FILE_QUERY, fid, file_type=file_type, nl=nl, ns=ns,dims=dims,nb=nb
;打開shape文件
;shapefile = DIALOG_PICKFILE(title='choose the SHP file:',filter='*.shp')
shapefile=file_search(image_dir,'*.shp')
if strlen(shapefile) eq 0 then return
oshp = OBJ_NEW('IDLffshape',shapefile)
oshp->Getproperty,n_entities=n_ent,Attribute_info=attr_info,$
n_attributes=n_attr,Entity_type=ent_type
roi_shp = LONARR(n_ent)
FOR ishp = 0,n_ent-1 DO BEGIN
entitie = oshp->Getentity(ishp)
IF entitie.shape_type EQ 5 THEN BEGIN
record = *(entitie.vertices)
;轉換文件坐標
ENVI_CONVERT_FILE_COORDINATES,fid,xmap,ymap,record[0,*],record[1,*]
;創建ROI
roi_shp[ishp] = ENVI_CREATE_ROI(ns=ns,nl=nl)
ENVI_DEFINE_ROI,roi_shp[ishp],/polygon,xpts=REFORM(xmap),ypts=REFORM(ymap)
;記錄X,Y的區間，裁剪用
IF ishp EQ 0 THEN BEGIN
xMin = ROUND(MIN(xMap,max = xMax))
yMin = ROUND(MIN(yMap,max = yMax))
ENDIF ELSE BEGIN
xMin = xMin < ROUND(MIN(xMap))
xMax = xMax > ROUND(MAX(xMap))
yMin = yMin < ROUND(MIN(yMap))
yMax = yMax > ROUND(MAX(yMap))
ENDELSE
ENDIF
oshp->Destroyentity,entitie
ENDFOR;ishp
xMin = xMin >0
xMax = xMax < ns-1
yMin = yMin >0
yMax = yMax < nl-1
;判斷輸出文件路徑，在原文件名基礎上輸出
outfiledir=file_dirname(image_file,/MARK_DIRECTORY)
out_name = outfiledir +'\' +file_baseName(image_file,'.tif')+'_roi.img
out_dims = [-1,xMin,xMax,yMin,yMax]
pos = INDGEN(nb)
ENVI_DOIT,'ENVI_SUBSET_VIA_ROI_DOIT',background=0,fid=fid,dims=out_dims,out_name=out_name,$
ns = ns, nl = nl,pos=pos,roi_ids=roi_shp
endfor
tmp = DIALOG_MESSAGE('裁切結束!',/info)
envi_batch_exit
END
======================
另一個代碼
;+
;:Description:
; Mask Raster data by Evf file.
;
;:Author: tiands@esrichina.com.cn
;
;:Date: 2013-9-5 10:59:58
;-
PRO maskimg_byevf
COMPILE_OPT idl2
ENVI,/RESTORE_BASE_SAVE_FILES
;initialize ENVI in batch mode:
ENVI_BATCH_INIT
fileextension='img'
;選擇數據文件夾
folder=ENVI_PICKFILE(TITLE='Select image Data Folder',/DIRECTORY)
;選擇evf
evfname=DIALOG_PICKFILE(title='Select Evf ')
IF (folder EQ '') THEN RETURN
;查找數據
imagefiles=FILE_SEARCH(folder,'*.'+fileextension,/FOLD_CASE,COUNT=count);這里的fold_case需要查詢
IF (count EQ 0) THEN BEGIN
void=DIALOG_MESSAGE(['Unable to locate image tile datasets in the selected folder :',$
'',folder])
!QUIET=quietInit
RETURN
ENDIF
;輸出文件夾
outFolder=ENVI_PICKFILE(TITLE='Select Folder For Output',/DIRECTORY)
IF (outFolder EQ '') THEN BEGIN
!QUIET=quietInit
RETURN
ENDIF
outFolder = outFolder + PATH_SEP()
CD, outFolder, CURRENT=current
FOR i = 0 ,(count-1) DO BEGIN
;open dataset:
ENVI_OPEN_FILE, imagefiles[i], R_FID=fid
IF (fid EQ -1) THEN BEGIN
CONTINUE
ENDIF
;輸出文件
out_name=outFolder+FILE_BASENAME(imagefiles[i],fileExtension)+'bil'
;掩膜功能
SPATIALSUBSET,fid,evfName,out_name
ENDFOR
END
;基於EVF文件的掩膜
PRO SPATIALSUBSET,data_fid,evfName,out_name
COMPILE_OPT IDL2
ENVI_FILE_QUERY,data_fid,BNAMES= BNAMES,ns=ns,nl=nl,nb=nb
;打開矢量文件
evf_id = ENVI_EVF_OPEN(evfName)
;獲取相關信息
ENVI_EVF_INFO, evf_id, num_recs=num_recs, $
data_type=data_type, projection=projection, $
layer_name=layer_name
roi_ids = LONARR(num_recs)
;輸出的坐標系統
; oproj=envi_get_projection(fid=data_fid)
;讀取各個記錄的點數
FOR i=0,num_recs-1 DO BEGIN
record = ENVI_EVF_READ_RECORD(evf_id, i)
; ;轉換投影坐標點
; envi_convert_projection_coordinates, $
; record[0,*],record[1,*], projection, $
; oxmap, oymap, oproj
;轉換為文件坐標;這里更改了oxmap, oymap
;原來的是 ENVI_CONVERT_FILE_COORDINATES,data_fid,xmap,ymap,record[0,*],record[1,*]
ENVI_CONVERT_FILE_COORDINATES,data_fid,xmap,ymap,record[0,*],record[1,*]
;創建ROI
roi_id = ENVI_CREATE_ROI(color=4,ns = ns , nl = nl)
ENVI_DEFINE_ROI, roi_id, /polygon, xpts=REFORM(xMap), ypts=REFORM(yMap)
roi_ids[i] = roi_id
;記錄XY的區間，裁剪用
IF i EQ 0 THEN BEGIN
xmin = ROUND(MIN(xMap,max = xMax))
yMin = ROUND(MIN(yMap,max = yMax))
ENDIF ELSE BEGIN
xmin = xMin < ROUND(MIN(xMap))
xMax = xMax > ROUND(MAX(xMap))
yMin = yMin < ROUND(MIN(yMap))
yMax = yMax > ROUND(MAX(yMap))
ENDELSE
ENDFOR
xMin = xMin >0
xmax = xMax < ns-1
yMin = yMin >0
ymax = yMax < nl-1
;創建掩膜，裁剪后掩
ENVI_MASK_DOIT,$
AND_OR =1, $
/IN_MEMORY, $
ROI_IDS= roi_ids, $ ;ROI的ID
ns = ns, nl = nl, $
/inside, $ ;區域內或外
r_fid = m_fid
out_dims = [-1,xMin,xMax,yMin,yMax]
ENVI_MASK_APPLY_DOIT, FID = data_fid, POS = INDGEN(nb), DIMS = out_dims, $
M_FID = m_fid, M_POS = [0], VALUE = 0, $
OUT_BNAME= BNAMES+' mask',IN_MEMORY=0,out_name=out_name,r_fid=r_fid
;掩膜文件ID移除
ENVI_FILE_MNG, id =m_fid,/remove
;ENVI_FILE_MNG, id =data_fid,/remove
END