// vim: set tabstop=4 paste nocindent noautoindent ff=unix: 

/*
*/



#include "read_srtm.h"




int main(int argc, char** argv)
{
	int   i, wsize, hsize, oclv;
	float lat, lng, rate;

	sWord* hgt = NULL;
	sWord* map_height;


	oclv = SRTM_OCLEVEL;
	lat = lng = 0.0;
	rate  = 1.0;
	wsize = 256;

	for (i=1; i<argc; i++) {
		if (!strcmp(argv[i],"-lati")) {if (i!=argc-1) lat   = atof(argv[i+1]);}
		if (!strcmp(argv[i],"-n"))    {if (i!=argc-1) lat   = atof(argv[i+1]);}
		if (!strcmp(argv[i],"-long")) {if (i!=argc-1) lng   = atof(argv[i+1]);}
		if (!strcmp(argv[i],"-e"))    {if (i!=argc-1) lng   = atof(argv[i+1]);}
		if (!strcmp(argv[i],"-size")) {if (i!=argc-1) wsize = atoi(argv[i+1]);}
		if (!strcmp(argv[i],"-s"))    {if (i!=argc-1) wsize = atoi(argv[i+1]);}
		if (!strcmp(argv[i],"-rate")) {if (i!=argc-1) rate  = atof(argv[i+1]);}
		if (!strcmp(argv[i],"-r"))    {if (i!=argc-1) rate  = atof(argv[i+1]);}
		if (!strcmp(argv[i],"-oclv")) {if (i!=argc-1) oclv  = atoi(argv[i+1]);}
		if (!strcmp(argv[i],"-o"))    {if (i!=argc-1) oclv  = atoi(argv[i+1]);}
	}
	if (wsize<=0.0) wsize = 256;
	hsize = (int)(wsize*rate) + 1;
	if (wsize>=SRTM_SIZE) wsize = SRTM_SIZE - 1;
	if (hsize>=SRTM_SIZE) hsize = SRTM_SIZE - 1;

	map_height = (sWord*)malloc(wsize*hsize*sizeof(sWord));
	if (map_height==NULL) {
		print_message("No more memory!! (%d)\n", wsize*hsize*sizeof(sWord));
		exit(1);
	}
	for (i=0; i<wsize*hsize; i++) map_height[i] = 0;


	set_height(map_height, wsize, hsize, lat, lng);

	int chk;
	do {
		map_height = median (map_height, wsize, hsize, &chk);
		//map_height = average(map_height, wsize, hsize, &chk);
	} while (chk==ON);

	print_height(map_height, wsize, hsize, oclv);

	return 0;
}



sWord*  median(sWord* map_height, int wsize, int hsize, int* chk)
{
	int i, j, mapj, hh, nn;
	int median[9];

	if (chk!=NULL) *chk = OFF;

	sWord* hgt = (sWord*)malloc(wsize*hsize*sizeof(sWord));
	if (hgt==NULL) return map_height;


	for (j=0; j<hsize; j++) {
		mapj = j*wsize;
		for (i=0; i<wsize; i++) {
			if (map_height[mapj+i]==SRTM_NODATA) {
				hh = nn = 0;
				if (i!=0) { 
					int gg = map_height[mapj+i-1];	
					if (gg!=SRTM_NODATA) {
						median[nn] = gg;
						nn++;
					}
				}
				if (i!=wsize-1) { 
					int gg = map_height[mapj+i+1];	
					if (gg!=SRTM_NODATA) {
						median[nn] = gg;
						nn++;
					}
				}
				if (j!=0) { 
					int gg = map_height[mapj-wsize+i];	
					if (gg!=SRTM_NODATA) {
						median[nn] = gg;
						nn++;
					}
				}
				if (j!=hsize-1) { 
					int gg = map_height[mapj+wsize+i];	
					if (gg!=SRTM_NODATA) {
						median[nn] = gg;
						nn++;
					}
				}

				if (i!=0 && j!=0) { 
					int gg = map_height[mapj-wsize+i-1];	
					if (gg!=SRTM_NODATA) {
						median[nn] = gg;
						nn++;
					}
				}
				if (i!=0 && j!=hsize-1) { 
					int gg = map_height[mapj+wsize+i-1];	
					if (gg!=SRTM_NODATA) {
						median[nn] = gg;
						nn++;
					}
				}
				if (i!=wsize-1 && j!=0) { 
					int gg = map_height[mapj-wsize+i+1];	
					if (gg!=SRTM_NODATA) {
						median[nn] = gg;
						nn++;
					}
				}
				if (i!=wsize-1 && j!=hsize-1) { 
					int gg = map_height[mapj+wsize+i+1];	
					if (gg!=SRTM_NODATA) {
						median[nn] = gg;
						nn++;
					}
				}

				if (nn!=0) {
					int k, l;
					for (k=0; k<nn-1; k++) {
						for (l=k+1; l<nn; l++) {
							if (median[k]>median[l]) {
								int swp = median[k];
								median[k] = median[l];
								median[l] = swp;
							}
						}
					}
					hgt[mapj+i] = median[nn/2];
				}
				else {
					hgt[mapj+i] = SRTM_NODATA;
					if (chk!=NULL) *chk = ON;
				}
			}
			else {
				hgt[mapj+i] = map_height[mapj+i];
			}
		}
	}
	free(map_height);

	return hgt;
}



sWord*  average(sWord* map_height, int wsize, int hsize, int* chk)
{
	int i, j, mapj, hh, nn;

	if (chk!=NULL) *chk = OFF;

	sWord* hgt = (sWord*)malloc(wsize*hsize*sizeof(sWord));
	if (hgt==NULL) return map_height;

	for (j=0; j<hsize; j++) {
		mapj = j*wsize;
		for (i=0; i<wsize; i++) {
			if (map_height[mapj+i]==SRTM_NODATA) {
				hh = nn = 0;
				if (i!=0) { 
					int gg = map_height[mapj+i-1];	
					if (gg!=SRTM_NODATA) {
						hh += 2*gg;
						nn += 2;
					}
				}
				if (i!=wsize-1) { 
					int gg = map_height[mapj+i+1];	
					if (gg!=SRTM_NODATA) {
						hh += 2*gg;
						nn += 2;
					}
				}
				if (j!=0) { 
					int gg = map_height[mapj-wsize+i];	
					if (gg!=SRTM_NODATA) {
						hh += 2*gg;
						nn += 2;
					}
				}
				if (j!=hsize-1) { 
					int gg = map_height[mapj+wsize+i];	
					if (gg!=SRTM_NODATA) {
						hh += 2*gg;
						nn += 2;
					}
				}

				if (i!=0 && j!=0) { 
					int gg = map_height[mapj-wsize+i-1];	
					if (gg!=SRTM_NODATA) {
						hh += gg;
						nn++;
					}
				}
				if (i!=0 && j!=hsize-1) { 
					int gg = map_height[mapj+wsize+i-1];	
					if (gg!=SRTM_NODATA) {
						hh += gg;
						nn++;
					}
				}
				if (i!=wsize-1 && j!=0) { 
					int gg = map_height[mapj-wsize+i+1];	
					if (gg!=SRTM_NODATA) {
						hh += gg;
						nn++;
					}
				}
				if (i!=wsize-1 && j!=hsize-1) { 
					int gg = map_height[mapj+wsize+i+1];	
					if (gg!=SRTM_NODATA) {
						hh += gg;
						nn++;
					}
				}

				if (nn!=0) hgt[mapj+i] = (int)((float)hh/nn+0.5);
				else {
					hgt[mapj+i] = SRTM_NODATA;
					if (chk!=NULL) *chk = ON;
				}
			}
			else {
				hgt[mapj+i] = map_height[mapj+i];
			}
		}
	}
	free(map_height);

	return hgt;
}




void  set_height(sWord* map_height, int wsize, int hsize, float lat, float lng)
{
	int i, j;
	sWord* hgt;

	int ndg = (int)lat;
	int edg = (int)lng;

	lat = lat - ndg;
	lng = lng - edg;
	int sx = (int)(lng*(SRTM_SIZE-1)  + 0.5);					// 0 - 1200
	int sy = (SRTM_SIZE-1) - (int)(lat*(SRTM_SIZE-1) + 0.5);	// 0 - 1200
	int ex = sx + wsize;
	int ey = sy + hsize;
	int mx = Min(ex, SRTM_SIZE-1);
	int my = Min(ey, SRTM_SIZE-1);


	hgt = read_hgt_file(ndg, edg);
	if (hgt!=NULL) {
		for (j=sy; j<my; j++) {
			int mapj = (j-sy)*wsize;
			int hgtj = j*SRTM_SIZE;
			for (i=sx; i<mx; i++) {
				map_height[mapj + i - sx] = hgt[hgtj + i];
			}
		}
		free(hgt);
	}

	if (ex>=SRTM_SIZE) {
		hgt = read_hgt_file(ndg, edg+1);
		if (hgt!=NULL) {
			for (j=sy; j<my; j++) {
				int mapj = (j-sy)*wsize;
				int hgtj = j*SRTM_SIZE;
				for (i=0; i<ex-SRTM_SIZE+1; i++) {
					map_height[mapj + i + mx - sx] = hgt[hgtj + i];
				}
			}
			free(hgt);
		}
	}

	if (ey>=SRTM_SIZE) {
		hgt = read_hgt_file(ndg-1, edg);
		if (hgt!=NULL) {
			for (j=0; j<ey-SRTM_SIZE+1; j++) {
				int mapj = (j+my-sy)*wsize;
				int hgtj = j*SRTM_SIZE;
				for (i=sx; i<mx; i++) {
					map_height[mapj + i - sx] = hgt[hgtj + i];
				}
			}
			free(hgt);
		}
	}

	if (ex>=SRTM_SIZE && ey>=SRTM_SIZE) {
		hgt = read_hgt_file(ndg-1, edg+1);
		if (hgt!=NULL) {
			for (j=0; j<ey-SRTM_SIZE+1; j++) {
				int mapj = (j+my-sy)*wsize;
				int hgtj = j*SRTM_SIZE;
				for (i=0; i<ex-SRTM_SIZE+1; i++) {
					map_height[mapj + i + mx - sx] = hgt[hgtj + i];
				}
			}
			free(hgt);
		}
	}
}





void  print_height(sWord* map_height, int wsize, int hsize, int oclv)
{
	int  i, j;

	for (j=0; j<hsize; j++) {
		int mapj = j*wsize;
		for (i=0; i<wsize; i++) {
			int height = map_height[mapj + i];
			if (height<=0) height += oclv;
			printf("%06d ", height);
		}
		printf("\n");
	}
}







/**
  ËÌ°Þ nlat, Åì·Ð elng¤Î¥Ç¡¼¥¿¤òÆɤ߽Ф¹¡¥

*/
sWord* read_hgt_file(int nlat, int elng)
{
	int  i;
	char fname[256];

	int dtsz = sizeof(sWord);

	sWord* hgt = (sWord*)malloc(SRTM_SIZE_SQR*dtsz);
	if (hgt==NULL) return NULL;
	for (i=0; i<SRTM_SIZE_SQR; i++) hgt[i] = 0;

	sprintf(fname, SRTM_FNAME_FORMAT, SRTM_DATA_DIR, nlat, elng);
	FILE* fp = fopen(fname, "rb");
	if (fp==NULL) return hgt;
	fread((void*)hgt, SRTM_SIZE_SQR*dtsz, 1, fp);
	fclose(fp);

	ntoh_ar(hgt, SRTM_SIZE_SQR*dtsz);
	return hgt;
}