26 #include "imageclasses/ImgWriterGdal.h"
27 #include "imageclasses/ImgReaderGdal.h"
28 #include "imageclasses/ImgReaderOgr.h"
29 #include "base/Optionpk.h"
30 #include "algorithms/Egcs.h"
101 int main(
int argc,
char *argv[])
103 Optionpk<string> input_opt(
"i",
"input",
"Input image file(s). If input contains multiple images, a multi-band output is created");
105 Optionpk<string> projection_opt(
"a_srs",
"a_srs",
"Override the projection for the output file (leave blank to copy from input file, use epsg:3035 to use European projection and force to European grid");
107 Optionpk<string> extent_opt(
"e",
"extent",
"get boundary from extent from polygons in vector file");
108 Optionpk<bool> cut_opt(
"cut",
"crop_to_cutline",
"Crop the extent of the target dataset to the extent of the cutline.",
false);
109 Optionpk<string> mask_opt(
"m",
"mask",
"Use the the specified file as a validity mask (0 is nodata).");
110 Optionpk<float> msknodata_opt(
"msknodata",
"msknodata",
"Mask value not to consider for crop.", 0);
111 Optionpk<short> mskband_opt(
"mskband",
"mskband",
"Mask band to read (0 indexed). Provide band for each mask.", 0);
112 Optionpk<double> ulx_opt(
"ulx",
"ulx",
"Upper left x value bounding box", 0.0);
113 Optionpk<double> uly_opt(
"uly",
"uly",
"Upper left y value bounding box", 0.0);
114 Optionpk<double> lrx_opt(
"lrx",
"lrx",
"Lower right x value bounding box", 0.0);
115 Optionpk<double> lry_opt(
"lry",
"lry",
"Lower right y value bounding box", 0.0);
116 Optionpk<double> dx_opt(
"dx",
"dx",
"Output resolution in x (in meter) (empty: keep original resolution)");
117 Optionpk<double> dy_opt(
"dy",
"dy",
"Output resolution in y (in meter) (empty: keep original resolution)");
118 Optionpk<double> cx_opt(
"x",
"x",
"x-coordinate of image center to crop (in meter)");
119 Optionpk<double> cy_opt(
"y",
"y",
"y-coordinate of image center to crop (in meter)");
122 Optionpk<int> ns_opt(
"ns",
"ns",
"number of samples to crop (in pixels)");
123 Optionpk<int> nl_opt(
"nl",
"nl",
"number of lines to crop (in pixels)");
127 Optionpk<double> autoscale_opt(
"as",
"autoscale",
"scale output to min and max, e.g., --autoscale 0 --autoscale 255");
129 Optionpk<double> offset_opt(
"offset",
"offset",
"output=scale*input+offset");
130 Optionpk<string> otype_opt(
"ot",
"otype",
"Data type for output image ({Byte/Int16/UInt16/UInt32/Int32/Float32/Float64/CInt16/CInt32/CFloat32/CFloat64}). Empty string: inherit type from input image",
"");
131 Optionpk<string> oformat_opt(
"of",
"oformat",
"Output image format (see also gdal_translate).",
"GTiff");
132 Optionpk<string> option_opt(
"co",
"co",
"Creation option for output file. Multiple options can be specified.");
133 Optionpk<string> colorTable_opt(
"ct",
"ct",
"color table (file with 5 columns: id R G B ALFA (0: transparent, 255: solid)");
134 Optionpk<float> nodata_opt(
"nodata",
"nodata",
"Nodata value to put in image if out of bounds.");
135 Optionpk<string> resample_opt(
"r",
"resampling-method",
"Resampling method (near: nearest neighbor, bilinear: bi-linear interpolation).",
"near");
136 Optionpk<string> description_opt(
"d",
"description",
"Set image description");
139 extent_opt.setHide(1);
141 bstart_opt.setHide(1);
144 msknodata_opt.setHide(1);
145 mskband_opt.setHide(1);
146 option_opt.setHide(1);
153 scale_opt.setHide(1);
154 offset_opt.setHide(1);
155 nodata_opt.setHide(1);
156 description_opt.setHide(1);
160 doProcess=input_opt.retrieveOption(argc,argv);
161 output_opt.retrieveOption(argc,argv);
162 projection_opt.retrieveOption(argc,argv);
163 ulx_opt.retrieveOption(argc,argv);
164 uly_opt.retrieveOption(argc,argv);
165 lrx_opt.retrieveOption(argc,argv);
166 lry_opt.retrieveOption(argc,argv);
167 band_opt.retrieveOption(argc,argv);
168 bstart_opt.retrieveOption(argc,argv);
169 bend_opt.retrieveOption(argc,argv);
170 autoscale_opt.retrieveOption(argc,argv);
171 otype_opt.retrieveOption(argc,argv);
172 oformat_opt.retrieveOption(argc,argv);
173 colorTable_opt.retrieveOption(argc,argv);
174 dx_opt.retrieveOption(argc,argv);
175 dy_opt.retrieveOption(argc,argv);
176 resample_opt.retrieveOption(argc,argv);
177 extent_opt.retrieveOption(argc,argv);
178 cut_opt.retrieveOption(argc,argv);
179 mask_opt.retrieveOption(argc,argv);
180 msknodata_opt.retrieveOption(argc,argv);
181 mskband_opt.retrieveOption(argc,argv);
182 option_opt.retrieveOption(argc,argv);
183 cx_opt.retrieveOption(argc,argv);
184 cy_opt.retrieveOption(argc,argv);
185 nx_opt.retrieveOption(argc,argv);
186 ny_opt.retrieveOption(argc,argv);
187 ns_opt.retrieveOption(argc,argv);
188 nl_opt.retrieveOption(argc,argv);
189 scale_opt.retrieveOption(argc,argv);
190 offset_opt.retrieveOption(argc,argv);
191 nodata_opt.retrieveOption(argc,argv);
192 description_opt.retrieveOption(argc,argv);
193 verbose_opt.retrieveOption(argc,argv);
195 catch(
string predefinedString){
196 std::cout << predefinedString << std::endl;
200 cout << setprecision(12) <<
"--ulx=" << ulx_opt[0] <<
" --uly=" << uly_opt[0] <<
" --lrx=" << lrx_opt[0] <<
" --lry=" << lry_opt[0] << endl;
204 cout <<
"Usage: pkcrop -i input -o output" << endl;
206 std::cout <<
"short option -h shows basic options only, use long option --help to show all options" << std::endl;
209 if(input_opt.empty()){
210 std::cerr <<
"No input file provided (use option -i). Use --help for help information" << std::endl;
213 if(output_opt.empty()){
214 std::cerr <<
"No output file provided (use option -o). Use --help for help information" << std::endl;
218 float nodataValue=nodata_opt.size()? nodata_opt[0] : 0;
219 RESAMPLE theResample;
220 if(resample_opt[0]==
"near"){
223 cout <<
"resampling: nearest neighbor" << endl;
225 else if(resample_opt[0]==
"bilinear"){
226 theResample=BILINEAR;
228 cout <<
"resampling: bilinear interpolation" << endl;
231 std::cout <<
"Error: resampling method " << resample_opt[0] <<
" not supported" << std::endl;
235 const char* pszMessage;
236 void* pProgressArg=NULL;
237 GDALProgressFunc pfnProgress=GDALTermProgress;
239 pfnProgress(progress,pszMessage,pProgressArg);
253 if(bstart_opt.size()){
254 if(bend_opt.size()!=bstart_opt.size()){
255 string errorstring=
"Error: options for start and end band indexes must be provided as pairs, missing end band";
259 for(
int ipair=0;ipair<bstart_opt.size();++ipair){
260 if(bend_opt[ipair]<=bstart_opt[ipair]){
261 string errorstring=
"Error: index for end band must be smaller then start band";
264 for(
int iband=bstart_opt[ipair];iband<=bend_opt[ipair];++iband)
265 band_opt.push_back(iband);
270 cerr << error << std::endl;
275 string projectionString;
276 for(
int iimg=0;iimg<input_opt.size();++iimg){
277 imgReader.open(input_opt[iimg]);
279 isGeoRef=imgReader.isGeoRef();
280 if(imgReader.isGeoRef()&&projection_opt.empty())
281 projectionString=imgReader.getProjection();
283 if(!iimg||imgReader.getDeltaX()<dx)
284 dx=imgReader.getDeltaX();
287 if(!iimg||imgReader.getDeltaY()<dy)
288 dy=imgReader.getDeltaY();
291 ncropband+=band_opt.size();
293 ncropband+=imgReader.nrOfBand();
297 GDALDataType theType=GDT_Unknown;
299 cout <<
"possible output data types: ";
300 for(
int iType = 0; iType < GDT_TypeCount; ++iType){
302 cout <<
" " << GDALGetDataTypeName((GDALDataType)iType);
303 if( GDALGetDataTypeName((GDALDataType)iType) != NULL
304 && EQUAL(GDALGetDataTypeName((GDALDataType)iType),
305 otype_opt[0].c_str()))
306 theType=(GDALDataType) iType;
310 if(theType==GDT_Unknown)
311 cout <<
"Unknown output pixel type: " << otype_opt[0] << endl;
313 cout <<
"Output pixel type: " << GDALGetDataTypeName(theType) << endl;
316 double cropulx=ulx_opt[0];
317 double cropuly=uly_opt[0];
318 double croplrx=lrx_opt[0];
319 double croplry=lry_opt[0];
323 if(extent_opt.size()){
328 for(
int iextent=0;iextent<extent_opt.size();++iextent){
329 extentReader.open(extent_opt[iextent]);
330 if(!(extentReader.getExtent(e_ulx,e_uly,e_lrx,e_lry))){
331 cerr <<
"Error: could not get extent from " << extent_opt[0] << endl;
350 extentReader.close();
353 extentReader.open(extent_opt[0]);
355 else if(cx_opt.size()&&cy_opt.size()&&nx_opt.size()&&ny_opt.size()){
356 ulx_opt[0]=cx_opt[0]-nx_opt[0]/2.0;
357 uly_opt[0]=(isGeoRef) ? cy_opt[0]+ny_opt[0]/2.0 : cy_opt[0]-ny_opt[0]/2.0;
358 lrx_opt[0]=cx_opt[0]+nx_opt[0]/2.0;
359 lry_opt[0]=(isGeoRef) ? cy_opt[0]-ny_opt[0]/2.0 : cy_opt[0]+ny_opt[0]/2.0;
369 else if(cx_opt.size()&&cy_opt.size()&&ns_opt.size()&&nl_opt.size()){
370 ulx_opt[0]=cx_opt[0]-ns_opt[0]*dx/2.0;
371 uly_opt[0]=(isGeoRef) ? cy_opt[0]+nl_opt[0]*dy/2.0 : cy_opt[0]-nl_opt[0]*dy/2.0;
372 lrx_opt[0]=cx_opt[0]+ns_opt[0]*dx/2.0;
373 lry_opt[0]=(isGeoRef) ? cy_opt[0]-nl_opt[0]*dy/2.0 : cy_opt[0]+nl_opt[0]*dy/2.0;
385 cout <<
"--ulx=" << ulx_opt[0] <<
" --uly=" << uly_opt[0] <<
" --lrx=" << lrx_opt[0] <<
" --lry=" << lry_opt[0] << endl;
391 if(extent_opt.size()&&cut_opt[0]){
393 ncropcol=abs(static_cast<int>(ceil((lrx_opt[0]-ulx_opt[0])/dx)));
394 ncroprow=abs(static_cast<int>(ceil((uly_opt[0]-lry_opt[0])/dy)));
395 maskWriter.open(
"/vsimem/mask.tif",ncropcol,ncroprow,1,GDT_Float32,
"GTiff",option_opt);
403 maskWriter.setGeoTransform(gt);
404 if(projection_opt.size())
405 maskWriter.setProjectionProj4(projection_opt[0]);
406 else if(projectionString.size())
407 maskWriter.setProjection(projectionString);
410 vector<double> burnValues(1,1);
411 maskWriter.rasterizeOgr(extentReader,burnValues);
415 cerr << error << std::endl;
419 cerr <<
"error catched" << std::endl;
424 mask_opt.push_back(
"/vsimem/mask.tif");
429 if(verbose_opt[0]>=1)
430 std::cout <<
"opening mask image file " << mask_opt[0] << std::endl;
431 maskReader.open(mask_opt[0]);
432 if(mskband_opt[0]>=maskReader.nrOfBand()){
433 string errorString=
"Error: illegal mask band";
438 cerr << error << std::endl;
442 cerr <<
"error catched" << std::endl;
450 if(scale_opt.size()){
451 while(scale_opt.size()<band_opt.size())
452 scale_opt.push_back(scale_opt[0]);
454 if(offset_opt.size()){
455 while(offset_opt.size()<band_opt.size())
456 offset_opt.push_back(offset_opt[0]);
458 if(autoscale_opt.size()){
459 assert(autoscale_opt.size()%2==0);
466 for(
int iimg=0;iimg<input_opt.size();++iimg){
468 cout <<
"opening image " << input_opt[iimg] << endl;
469 imgReader.open(input_opt[iimg]);
471 if(theType==GDT_Unknown){
472 theType=imgReader.getDataType();
474 cout <<
"Using data type from input image: " << GDALGetDataTypeName(theType) << endl;
476 if(option_opt.findSubstring(
"INTERLEAVE=")==option_opt.end()){
477 string theInterleave=
"INTERLEAVE=";
478 theInterleave+=imgReader.getInterleave();
479 option_opt.push_back(theInterleave);
481 int nrow=imgReader.nrOfRow();
482 int ncol=imgReader.nrOfCol();
488 cout <<
"size of " << input_opt[iimg] <<
": " << ncol <<
" cols, "<< nrow <<
" rows" << endl;
489 double uli,ulj,lri,lrj;
490 bool forceEUgrid=
false;
491 if(projection_opt.size())
492 forceEUgrid=(!(projection_opt[0].compare(
"EPSG:3035"))||!(projection_opt[0].compare(
"EPSG:3035"))||projection_opt[0].find(
"ETRS-LAEA")!=string::npos);
493 if(ulx_opt[0]>=lrx_opt[0]){
495 lri=imgReader.nrOfCol()-1;
497 lrj=imgReader.nrOfRow()-1;
498 ncropcol=imgReader.nrOfCol();
499 ncroprow=imgReader.nrOfRow();
500 imgReader.getBoundingBox(cropulx,cropuly,croplrx,croplry);
501 double magicX=1,magicY=1;
506 egcs.setLevel(egcs.res2level(dx));
507 egcs.force2grid(cropulx,cropuly,croplrx,croplry);
508 imgReader.geo2image(cropulx+(magicX-1.0)*imgReader.getDeltaX(),cropuly-(magicY-1.0)*imgReader.getDeltaY(),uli,ulj);
509 imgReader.geo2image(croplrx+(magicX-2.0)*imgReader.getDeltaX(),croplry-(magicY-2.0)*imgReader.getDeltaY(),lri,lrj);
511 imgReader.geo2image(cropulx+(magicX-1.0)*imgReader.getDeltaX(),cropuly-(magicY-1.0)*imgReader.getDeltaY(),uli,ulj);
512 imgReader.geo2image(croplrx+(magicX-2.0)*imgReader.getDeltaX(),croplry-(magicY-2.0)*imgReader.getDeltaY(),lri,lrj);
514 ncropcol=abs(static_cast<int>(ceil((croplrx-cropulx)/dx)));
515 ncroprow=abs(static_cast<int>(ceil((cropuly-croplry)/dy)));
518 double magicX=1,magicY=1;
527 egcs.setLevel(egcs.res2level(dx));
528 egcs.force2grid(cropulx,cropuly,croplrx,croplry);
529 imgReader.geo2image(cropulx+(magicX-1.0)*imgReader.getDeltaX(),cropuly-(magicY-1.0)*imgReader.getDeltaY(),uli,ulj);
530 imgReader.geo2image(croplrx+(magicX-2.0)*imgReader.getDeltaX(),croplry-(magicY-2.0)*imgReader.getDeltaY(),lri,lrj);
532 imgReader.geo2image(cropulx+(magicX-1.0)*imgReader.getDeltaX(),cropuly-(magicY-1.0)*imgReader.getDeltaY(),uli,ulj);
533 imgReader.geo2image(croplrx+(magicX-2.0)*imgReader.getDeltaX(),croplry-(magicY-2.0)*imgReader.getDeltaY(),lri,lrj);
535 ncropcol=abs(static_cast<int>(ceil((croplrx-cropulx)/dx)));
536 ncroprow=abs(static_cast<int>(ceil((cropuly-croplry)/dy)));
545 double deltaX=imgReader.getDeltaX();
546 double deltaY=imgReader.getDeltaY();
547 dcropcol=(lri-uli+1)/(dx/deltaX);
548 dcroprow=(lrj-ulj+1)/(dy/deltaY);
549 if(!imgWriter.nrOfBand()){
551 cout <<
"cropulx: " << cropulx << endl;
552 cout <<
"cropuly: " << cropuly << endl;
553 cout <<
"croplrx: " << croplrx << endl;
554 cout <<
"croplry: " << croplry << endl;
555 cout <<
"ncropcol: " << ncropcol << endl;
556 cout <<
"ncroprow: " << ncroprow << endl;
557 cout <<
"cropulx+ncropcol*dx: " << cropulx+ncropcol*dx << endl;
558 cout <<
"cropuly-ncroprow*dy: " << cropuly-ncroprow*dy << endl;
559 cout <<
"upper left column of input image: " << uli << endl;
560 cout <<
"upper left row of input image: " << ulj << endl;
561 cout <<
"lower right column of input image: " << lri << endl;
562 cout <<
"lower right row of input image: " << lrj << endl;
563 cout <<
"new number of cols: " << ncropcol << endl;
564 cout <<
"new number of rows: " << ncroprow << endl;
565 cout <<
"new number of bands: " << ncropband << endl;
578 if(oformat_opt.size())
579 imageType=oformat_opt[0];
581 imgWriter.open(output_opt[0],ncropcol,ncroprow,ncropband,theType,imageType,option_opt);
582 if(nodata_opt.size()){
583 for(
int iband=0;iband<ncropband;++iband)
584 imgWriter.GDALSetNoDataValue(nodata_opt[0],iband);
587 catch(
string errorstring){
588 cout << errorstring << endl;
591 if(description_opt.size())
592 imgWriter.setImageDescription(description_opt[0]);
599 gt[5]=(imgReader.isGeoRef())? -dy : dy;
600 imgWriter.setGeoTransform(gt);
601 if(projection_opt.size()){
603 cout <<
"projection: " << projection_opt[0] << endl;
604 imgWriter.setProjectionProj4(projection_opt[0]);
607 imgWriter.setProjection(imgReader.getProjection());
608 if(imgWriter.getDataType()==GDT_Byte){
609 if(colorTable_opt.size()){
610 if(colorTable_opt[0]!=
"none")
611 imgWriter.setColorTable(colorTable_opt[0]);
613 else if (imgReader.getColorTable()!=NULL)
614 imgWriter.setColorTable(imgReader.getColorTable());
622 else if(uli>=imgReader.nrOfCol())
623 startCol=imgReader.nrOfCol()-1;
626 else if(lri>=imgReader.nrOfCol())
627 endCol=imgReader.nrOfCol()-1;
632 else if(ulj>=imgReader.nrOfRow())
633 startRow=imgReader.nrOfRow()-1;
636 else if(lrj>=imgReader.nrOfRow())
637 endRow=imgReader.nrOfRow()-1;
641 int readncol=endCol-startCol+1;
642 vector<double> readBuffer(readncol+1);
643 int nband=(band_opt.size())?band_opt.size() : imgReader.nrOfBand();
644 for(
int iband=0;iband<nband;++iband){
645 int readBand=(band_opt.size()>iband)?band_opt[iband]:iband;
647 cout <<
"extracting band " << readBand << endl;
648 pfnProgress(progress,pszMessage,pProgressArg);
652 if(autoscale_opt.size()){
654 imgReader.getMinMax(static_cast<int>(startCol),static_cast<int>(endCol),static_cast<int>(startRow),static_cast<int>(endRow),readBand,theMin,theMax);
656 catch(
string errorString){
657 cout << errorString << endl;
660 cout <<
"minmax: " << theMin <<
", " << theMax << endl;
661 double theScale=(autoscale_opt[1]-autoscale_opt[0])/(theMax-theMin);
662 double theOffset=autoscale_opt[0]-theScale*theMin;
663 imgReader.setScale(theScale,readBand);
664 imgReader.setOffset(theOffset,readBand);
667 if(scale_opt.size()){
668 if(scale_opt.size()>iband)
669 imgReader.setScale(scale_opt[iband],readBand);
671 imgReader.setScale(scale_opt[0],readBand);
673 if(offset_opt.size()){
674 if(offset_opt.size()>iband)
675 imgReader.setOffset(offset_opt[iband],readBand);
677 imgReader.setOffset(offset_opt[0],readBand);
685 for(
int irow=0;irow<imgWriter.nrOfRow();++irow){
686 vector<float> lineMask;
690 imgWriter.image2geo(0,irow,x,y);
692 imgReader.geo2image(x,y,readCol,readRow);
693 vector<double> writeBuffer;
694 if(readRow<0||readRow>=imgReader.nrOfRow()){
699 for(
int icol=0;icol<imgWriter.nrOfCol();++icol)
700 writeBuffer.push_back(nodataValue);
704 cout <<
"reading row: " << readRow << endl;
706 if(endCol<imgReader.nrOfCol()-1)
707 imgReader.readData(readBuffer,GDT_Float64,startCol,endCol+1,readRow,readBand,theResample);
709 imgReader.readData(readBuffer,GDT_Float64,startCol,endCol,readRow,readBand,theResample);
711 double oldRowMask=-1;
712 for(
int icol=0;icol<imgWriter.nrOfCol();++icol){
713 imgWriter.image2geo(icol,irow,x,y);
715 imgReader.geo2image(x,y,readCol,readRow);
716 if(readCol<0||readCol>=imgReader.nrOfCol()){
722 writeBuffer.push_back(nodataValue);
733 imgWriter.image2geo(icol,irow,geox,geoy);
734 maskReader.geo2image(geox,geoy,colMask,rowMask);
735 colMask=
static_cast<int>(colMask);
736 rowMask=
static_cast<int>(rowMask);
737 if(rowMask>=0&&rowMask<maskReader.nrOfRow()&&colMask>=0&&colMask<maskReader.nrOfCol()){
738 if(static_cast<int>(rowMask)!=
static_cast<int>(oldRowMask)){
740 assert(rowMask>=0&&rowMask<maskReader.nrOfRow());
742 maskReader.readData(lineMask,GDT_Float32,static_cast<int>(rowMask),mskband_opt[0]);
744 catch(
string errorstring){
745 cerr << errorstring << endl;
749 cerr <<
"error catched" << std::endl;
754 if(lineMask[colMask]==msknodata_opt[0])
760 writeBuffer.push_back(nodataValue);
764 lowerCol=readCol-0.5;
765 lowerCol=
static_cast<int>(lowerCol);
766 upperCol=readCol+0.5;
767 upperCol=
static_cast<int>(upperCol);
770 if(upperCol>=imgReader.nrOfCol())
771 upperCol=imgReader.nrOfCol()-1;
773 writeBuffer.push_back((readCol-0.5-lowerCol)*readBuffer[upperCol-startCol]+(1-readCol+0.5+lowerCol)*readBuffer[lowerCol-startCol]);
776 readCol=
static_cast<int>(readCol);
779 writeBuffer.push_back(readBuffer[readCol]);
786 catch(
string errorstring){
787 cout << errorstring << endl;
791 if(writeBuffer.size()!=imgWriter.nrOfCol())
792 cout <<
"writeBuffer.size()=" << writeBuffer.size() <<
", imgWriter.nrOfCol()=" << imgWriter.nrOfCol() << endl;
793 assert(writeBuffer.size()==imgWriter.nrOfCol());
795 imgWriter.writeData(writeBuffer,GDT_Float64,irow,writeBand);
797 catch(
string errorstring){
798 cout << errorstring << endl;
803 progress/=imgWriter.nrOfRow();
804 pfnProgress(progress,pszMessage,pProgressArg);
808 progress+=(imgWriter.nrOfRow()*writeBand);
809 progress/=imgWriter.nrOfBand()*imgWriter.nrOfRow();
812 pfnProgress(progress,pszMessage,pProgressArg);
819 if(extent_opt.size()&&cut_opt.size()){
820 extentReader.close();