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