/*-------------------------------------------------------------------------
  Program for creating maps out of Flight Gear scenery data

  Written by Per Liedman, started February 2000.
  Based on a perl-script written by Alexei Novikov (anovikov@heron.itep.ru)

  Copyright (C) 2000 Per Liedman, liedman@home.se

  This program is free software; you can redistribute it and/or modify
  it under the terms of the GNU General Public License as published by
  the Free Software Foundation; either version 2 of the License, or
  (at your option) any later version.

  This program is distributed in the hope that it will be useful,
  but WITHOUT ANY WARRANTY; without even the implied warranty of
  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  GNU General Public License for more details.

  You should have received a copy of the GNU General Public License
  along with this program; if not, write to the Free Software
  Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
---------------------------------------------------------------------------
  CHANGES
  2000-02-20        Included some compatibility changes submitted by
                    Christian Mayer.
  2000-02-26        Major code reorganisation, made class MapMaker, etc.
  2000-03-06        Following Norman Vine's advice, map now uses OpenGL
                    for output
  2000-04-29        New, cuter, airports
  2004-12-22        DCL: Now reads FG_SCENERY when present.
  2005-01-16        DCL: Capable of off-screen rendering on modern GLX machines.
  2005-01-30        Switched to cross-platform render-texture code for off-screen
                    rendering, plus jpg support (submitted by Fred Bouvier).
  2005-02-26        FB: Arbitrary size by tiling.
                    Move fg_set_scenery to a specific file.
---------------------------------------------------------------------------*/

// Needs to be included *before* the UL_GLX check!
#include <plib/ul.h>

#include <simgear/compiler.h>
#include SG_GL_H
#ifdef UL_GLX
#  define GLX_GLXEXT_PROTOTYPES
#  ifdef __APPLE__
#    include <OpenGL/glx.h>
#  else
#    include <GL/glx.h>
#  endif
#elif defined UL_WIN32
#  include <windows.h>
#endif
#include SG_GLUT_H
#include <stdlib.h>
#include <stdio.h>
#include <sys/stat.h>
#ifndef _MSC_VER
#  include <unistd.h>
#endif
#include "MapMaker.hxx"
#include "OutputGL.hxx"
#include "fg_mkdir.hxx"
#include <simgear/misc/sg_path.hxx>
#include "extensions.hxx"
#include "RenderTexture.h"
#include "Scenery.hxx"
#include <vector>
#include STL_STRING

SG_USING_STD(vector);
SG_USING_STD(string);

typedef vector<string> string_list;

float clat = -100.0f, clon = -100.0f;   // initialize to unreasonable values
float autoscale = 0.0f;                 // 0.0f == no autoscale
char *outp = 0;                         // output file name
char *outpp = "map.png";                // output file name
char *outpj = "map.jpg";                // output file name
bool global = false;
bool doublebuffer = true, headless = false, create_jpeg = false;
bool smooth_shade = true, textured_fonts = true;
int features = MapMaker::DO_SHADE;
int jpeg_quality = 75;
int rescale_factor = 1;
MapMaker mapobj;

char outname[512], *scenerypath, *palette;
string raw_scenery_path = "";
string_list fg_scenery;
unsigned int scenery_pos;
unsigned int max_path_length = 0;
int opathl, spathl;
ulDir *dir1 = NULL, *dir2 = NULL;
ulDirEnt *ent = NULL;
RenderTexture *rt2 = 0;

/*****************************************************************************/
void reshapeMap( int width, int height ) {
  glViewport( 0, 0, width, height );
  glMatrixMode( GL_PROJECTION );
  glLoadIdentity();
  glOrtho( 0.0f, (GLfloat)width, (GLfloat)height, 0.0f, -1.0f, 1.0f );
  glMatrixMode( GL_MODELVIEW );
  glLoadIdentity();
}

void redrawMap() {
  char title_buffer[256];
  char font_name[512];

  if (textured_fonts) {
    strcpy( font_name, mapobj.getFGRoot() );
    strcat( font_name, "/Fonts/helvetica_medium.txf");
  }

  if (!global) {
    sprintf(title_buffer, "%c%.1f %c%.1f",
	    (clat<0.0f)?'S':'N', fabs(clat * SG_RADIANS_TO_DEGREES),
	    (clon<0.0f)?'W':'E', fabs(clon * SG_RADIANS_TO_DEGREES));
    if(!headless) glutSetWindowTitle(title_buffer);

    if ( outp == 0 && create_jpeg ) {
      outp = outpj;
    } else if ( outp == 0 ) {
      outp = outpp;
    }
    OutputGL output( outp, mapobj.getSize(), smooth_shade, 
		     textured_fonts, font_name, create_jpeg, jpeg_quality, rescale_factor );
    mapobj.createMap( &output, clat, clon, scenerypath, autoscale );
    output.closeOutput();
    exit(0);
  } else {
    char ns, ew;
    int lat, lon;
    int s = mapobj.getSize();
  
    do {
      while (dir2 == NULL) {
	while(dir1 == NULL) {
	  ++scenery_pos;
	  if(scenery_pos >= fg_scenery.size()) {
	    delete[] scenerypath;
	    exit(0);  // Ought to flag whether at least one path read OK before exiting OK.
	  }
	  dir1 = ulOpenDir(fg_scenery[scenery_pos].c_str());
	  strcpy(scenerypath, fg_scenery[scenery_pos].c_str());
	  NormalisePath(scenerypath);
	  spathl = strlen(scenerypath);
	  if(dir1 == NULL) {
	    printf("Unable to open directory \"%s\"\\n", scenerypath);
	  }
	}
	do {
	  ent = ulReadDir(dir1);
	} while (ent != NULL && ent->d_name[0] == '.');
	if (ent != NULL) {
	  strcpy( scenerypath+spathl, ent->d_name );
	  dir2 = ulOpenDir(scenerypath);
	} else {
	  ulCloseDir(dir1);
	  dir1 = NULL;
	}
      }
      
      bool exit = false;
      do {
	ent = ulReadDir(dir2);

	if (ent != NULL) {
	  exit = (ent->d_name[0] != '.' &&
		  sscanf(ent->d_name, "%c%d%c%d",
			 &ew, &lon, &ns, &lat) == 4 &&
                  strlen(ent->d_name) == 7);
	} else {
	  exit = true;
	}
      } while (!exit);

      if (ent == NULL) {
	ulCloseDir(dir2);
	dir2 = NULL;
      } else {
	// we have found a scenery directory - let's check if we already
	// have an image for it!
	lat *= (ns=='n')?1:-1;
	lon *= (ew=='e')?1:-1;
	clat = ((float)lat + 0.5f) * SG_DEGREES_TO_RADIANS;
	clon = ((float)lon + 0.5f) * SG_DEGREES_TO_RADIANS;
	
	sprintf( outname+opathl, "/%c%03d%c%02d.%s", 
		 (lon<0)?'w':'e', abs(lon), 
		 (lat<0)?'s':'n', abs(lat),
		 create_jpeg ? "jpg" : "png" );

	struct stat filestat;
	if ( stat(outname, &filestat) == 0 ) {
	  // the file already exists, so skip it!
	  ent = NULL;
	}  
      }
    } while (ent == NULL);
    
    if (!headless) {
      sprintf(title_buffer, "%c%.1f %c%.1f",
	      (clat<0.0f)?'S':'N', clat * SG_RADIANS_TO_DEGREES,
	      (clon<0.0f)?'W':'E', clon * SG_RADIANS_TO_DEGREES);
      glutSetWindowTitle(title_buffer);
    }

    OutputGL output(outname, s, smooth_shade, textured_fonts, font_name,
                    create_jpeg, jpeg_quality, rescale_factor);
    mapobj.createMap( &output, clat, clon, fg_scenery[scenery_pos], 1.0f );
    output.closeOutput();
    if (doublebuffer && !headless) {
      glutSwapBuffers();
    }

    if(!headless) glutPostRedisplay();
  }
}

/*****************************************************************************/

void print_help() {
  printf("MAP - FlightGear mapping utility\n\n");
  printf("Usage:\n");
  printf("  --lat=xx.xx             Start at latitude xx.xx (deg., south is neg.)\n");
  printf("  --lon=xx.xx             Start at longitude xx.xx (deg., west is neg.)\n");
  printf("  --size=pixels           Create map of size pixels*pixels (default 256)\n");
  printf("  --scale=x               Kilometers from top to bottom of map (default 100)\n"); 
  printf("  --autoscale             Automatically set scale to 1x1 degree tile\n");
  printf("  --light=x, y, z         Set light vector for shading\n");
  printf("  --airport-filter=string Display only airports with id beginning 'string'\n");
  printf("  --output=name           Write output to given file name (default 'map.png')\n");
  printf("  --fg-root=path          Overrides FG_ROOT environment variable\n");
  printf("  --fg-scenery=path       Overrides FG_SCENERY environment variable\n");
  printf("  --enable-airports       Show airports\n");
  printf("  --enable-navaids        Show navaids\n");
  printf("  --flat-shading          Don't do nice shading of the terrain\n");
  printf("  --atlas=path            Create maps of all scenery, and store them in path\n");
  printf("  --verbose               Display information during processing\n");
  printf("  --singlebuffer          Use single buffered display\n");
  printf("  --headless              Don't display output (render into an off-screen buffer)\n");
  printf("  --glutfonts             Use GLUT built-in fonts\n");
  printf("  --palette=path          Set the palette file to use\n");
  printf("  --smooth-color          Make smooth color heights\n");
  printf("  --jpeg                  Create JPEG images with default quality (75)\n");
  printf("  --jpeg=integer          Create JPEG images with specified quality\n");
  printf("  --aafactor=integer      Do antialiasing on image ( factor must be a power of two )\n");
}

bool parse_arg(char* arg) {
  int param;
  char cparam[128];
  float x, y, z;

  if ( sscanf(arg, "--lat=%f", &clat) == 1) {
    // do nothing
  } else if ( sscanf(arg, "--lon=%f", &clon) == 1 ) {
    // do nothing
  } else if ( sscanf(arg, "--size=%d", &param) == 1 ) {
    mapobj.setSize( param );
  } else if ( sscanf(arg, "--scale=%f", &x) == 1 ) {
    mapobj.setScale( (int)(x * 1000.0f) );
  } else if ( sscanf(arg, "--light=%f, %f, %f", &x, &y, &z) == 3 ) {
    mapobj.setLight( x, y, z );
  } else if ( sscanf(arg, "--airport-filter=%s", cparam) == 1 ) {
    mapobj.setAPFilter( strdup(cparam) );
  } else if ( sscanf(arg, "--output=%s", cparam) == 1 ) {
    outp = strdup(cparam);
  } else if ( sscanf(arg, "--fg-root=%s", cparam) == 1 ) {
    mapobj.setFGRoot( cparam );
  } else if ( strncmp(arg, "--fg-scenery=", 13 ) == 0 ) {
    raw_scenery_path = arg + 13;
  } else if ( strcmp(arg, "--enable-airports" ) == 0 ) {
    features |= MapMaker::DO_AIRPORTS;
  } else if ( strcmp(arg, "--enable-navaids" ) == 0 ) {
    features |= MapMaker::DO_NAVAIDS;
  } else if ( strcmp(arg, "--flat-shading" ) == 0 ) {
    smooth_shade = false;
  } else if ( strcmp(arg, "--autoscale") == 0 ) {
    autoscale = 1.0f;
  } else if ( sscanf(arg, "--autoscale=%f", &autoscale) == 1 ) {
    // do nothing
  } else if ( strcmp(arg, "--singlebuffer") == 0 ) {
    doublebuffer = false;
  } else if ( sscanf(arg, "--aafactor=%d", &rescale_factor) == 1 ) {
    // do nothing
  } else if ( sscanf(arg, "--jpeg=%d", &jpeg_quality) == 1 ) {
    create_jpeg = true;
  } else if ( strcmp(arg, "--jpeg") == 0 ) {
    create_jpeg = true;
  } else if ( strcmp(arg, "--headless") == 0 ) {
    headless = true;
  } else if ( sscanf(arg, "--atlas=%s", cparam) == 1 ) {
    global = true;
    outp = strdup( cparam );
  } else if ( sscanf(arg, "--palette=%s", cparam) == 1 ) {
    mapobj.setPalette(cparam);
  } else if ( strcmp(arg, "--glutfonts") == 0 ) {
    textured_fonts = false;
  } else if ( strcmp(arg, "--verbose") == 0 ) {
    features |= MapMaker::DO_VERBOSE;
  } else if ( strcmp(arg, "--smooth-color") == 0 ) {
    features |= MapMaker::DO_SMOOTH_COLOR;
  } else if ( strcmp(arg, "--help") == 0 ) {
    print_help();
    exit(0);
  } else if ( strcmp(arg, "--disable-airports" ) == 0 ) {
    // Do nothing - only for backwards compatibility
  } else if ( strcmp(arg, "--disable-navaids" ) == 0 ) {
    // Do nothing - only for backwards compatibility
  } else {
    return false;
  }

  return true;
}

bool ContinueIfNoHeadless() {
  cout << "Unable to continue in headless mode - revert to doublebuffer mode? [Y/n] ";
  char c;
  cin >> c;
  return((c == 'n' || c == 'N') ? false : true);
}

#ifdef UL_GLX
int mapXerrorHandler( Display *d, XErrorEvent * ) {
  return 0;
}
#endif

bool InitPbuffer( int &tex_size ) {
    rt2 = new RenderTexture(); 
    rt2->Reset("rgb tex2D");
    tex_size = mapobj.getSize();
    bool cur_ok;
#ifdef UL_GLX
    XSetErrorHandler( mapXerrorHandler );
#endif
    fprintf(stderr, "Trying size %d : ", tex_size );
    while (!(cur_ok = rt2->Initialize(tex_size, tex_size)) && tex_size > 1)
    {
#ifdef UL_GLX
	fprintf( stderr, "Error\n" );
#endif
        tex_size >>= 1;
        fprintf(stderr, "Trying size %d : ", tex_size );
    }
#ifdef UL_GLX
    XSetErrorHandler( NULL );
#endif
    if ( !cur_ok )
    {
        fprintf(stderr, "RenderTexture Initialization failed!\n");
    }
    else
    {
        fprintf(stderr, "Ok\n");
    }

    // for shadow mapping we still have to bind it and set the correct 
    // texture parameters using the SGI_shadow or ARB_shadow extension
    // setup the rendering context for the RenderTexture
    cur_ok = cur_ok && rt2->BeginCapture();
    if ( !cur_ok )
    {
        delete rt2;
	if ( !ContinueIfNoHeadless() )
	    exit(-1);
    }
    return cur_ok;
}

int main( int argc, char **argv ) {
  if (argc == 0)
    print_help();
  
  // Read the FG_SCENERY env variable before processing .atlasmaprc and command args,
  // so that we can override it if necessary.
  char* scenedirs = getenv("FG_SCENERY");
  if(scenedirs != NULL) raw_scenery_path = scenedirs;

  // process ~/.atlasmaprc
  char* homedir = getenv("HOME"), *rcpath;
  const char atlasmaprc[] = ".atlasmaprc";
  if (homedir != NULL) {
    rcpath = new char[strlen(homedir) + strlen(atlasmaprc) + 2];
    strcpy(rcpath, homedir);
    strcat(rcpath, "/");
    strcat(rcpath, atlasmaprc);
  } else {
    rcpath = new char[strlen(atlasmaprc) + 1];
    strcpy(rcpath, atlasmaprc);
  }

  FILE* rc = fopen(rcpath, "r");

  if (rc != NULL) {
    char line[256];
    fgets(line, 256, rc);
    while (!feof(rc)) {
      line[strlen(line) - 1] = '\0';
      if (!parse_arg(line)) {
	fprintf(stderr, "%s: unknown argument \"%s\" in file \"%s\"\n",
		argv[0], line, rcpath);
      }
      fgets(line, 256,rc);
    }
    fclose(rc);
  }

  delete[] rcpath;

  // process command line arguments
  for (int arg = 1; arg < argc; arg++) {
    if (!parse_arg(argv[arg])) {
      fprintf(stderr, "%s: unknown argument '%s'.\n", argv[0], argv[arg]);
      print_help();
      exit(1);
    }
  }

  if ( rescale_factor & ( rescale_factor-1 ) ) {
    fprintf(stderr, "%s: --aafactor should be a power of 2.\n", argv[0]);
    exit(1);
  }
  
  if (!global && (fabs(clat) > 90.0f || fabs(clon) > 180.0f)) {
    fprintf(stderr, 
	    "%s: Invalid position. Check latitude and longitude.\n", argv[0]);
    print_help();
    exit(1);
  }
  
  set_fg_scenery(raw_scenery_path);
  if(fg_scenery.size()) {
    scenerypath = new char[max_path_length + 256];
    scenery_pos = 0;
  } else {
    cout << "No scenery paths could be found.  You need to set either a valid FG_ROOT and/or FG_SCENERY variable, or specify a valid --fg-root and/or --fg-scenery on the command line.\n";
    exit(-1);
  }

  mapobj.setSize( mapobj.getSize() * rescale_factor );
  
  mapobj.setFeatures(features);

  // convert lat & lon to radians
  clat *= SG_DEGREES_TO_RADIANS;
  clon *= SG_DEGREES_TO_RADIANS;

  if (global) {
    int s = mapobj.getSize();

    if ( (s & (s-1)) != 0 ) {           // Thanks for this cutie, Steve
      fprintf(stderr, "%s: WARNING! Size is not a power of two - you will " \
	      "not be able to use\n" \
	      "these maps with the Atlas program!\n", argv[0] );
    }
    
    // Check that the path to store the images exists
    dir1 = ulOpenDir(outp);
    if(NULL == dir1) {
      fg_mkdir((const char*)outp);
    }
    dir1 = ulOpenDir(outp);
    if(NULL == dir1) {
      cout << "Unable to create requested Atlas map directory " << outp << "... exiting :-(\n";
      exit(-1);
    }

    dir1 = ulOpenDir(fg_scenery[0].c_str());
    strcpy(scenerypath, fg_scenery[0].c_str());
    NormalisePath(scenerypath);
    spathl = strlen(scenerypath);
    if(dir1 == NULL) {
      printf("Unable to open directory \"%s\"\\n", scenerypath);
    }
   
    strcpy(outname, outp);
    opathl = strlen(outname);
  } else {
    // Need to set the scenery path to point to the required location
    float dlat = clat * SG_RADIANS_TO_DEGREES, dlon = clon * SG_RADIANS_TO_DEGREES;
    int lat2 = (int)floor(dlat), lon2 = (int)floor(dlon);
    int lat1 = (int)(floor((float)lat2 / 10.0f) * 10.0), lon1 = (int)(floor((float)lon2 / 10.0f) * 10.0);
    char* dpath1 = new char[32];
    char* dpath2 = new char[32];
    sprintf(dpath1, "%c%03d%c%02d", (lon1 < 0 ? 'w' : 'e'), abs(lon1), (lat1 < 0 ? 's' : 'n'), abs(lat1));
    sprintf(dpath2, "%c%03d%c%02d", (lon2 < 0 ? 'w' : 'e'), abs(lon2), (lat2 < 0 ? 's' : 'n'), abs(lat2));
    bool path_found = false;
    for(unsigned int i = 0; i < fg_scenery.size(); ++i) {
      strcpy(scenerypath, fg_scenery[i].c_str());
      NormalisePath(scenerypath);
      int sz = strlen(scenerypath);
      strcat(scenerypath, dpath1);
      NormalisePath(scenerypath);
      strcat(scenerypath, dpath2);
      dir1 = ulOpenDir(scenerypath);
      if(dir1 != NULL) {
	path_found = true;
	scenerypath[sz] = '\0';
	//cout << "Scenerypath found,  = " << scenerypath << '\n';
	break;
      }
      //cout << scenerypath << (dir1 == NULL ? " does not exist..." : " exists!") << '\n';
    }
    if(!path_found) {
      cout << "Unable to find required subdirectory " << dpath1 << '/' << dpath2 << " on the available scenery paths:\n";
      for(unsigned int i = 0; i < fg_scenery.size(); ++i) {
	cout << fg_scenery[i] << '\n';
      }
      cout << "... unable to continue - exiting!\n";
      exit(-1);
    }
  }

  // now initialize GLUT
  glutInit( &argc, argv );
  if (doublebuffer) {
    glutInitDisplayMode( GLUT_DOUBLE | GLUT_RGBA | GLUT_DEPTH );
  } else {
    glutInitDisplayMode( GLUT_SINGLE | GLUT_RGBA | GLUT_DEPTH );
  }
  glutInitWindowSize( 400, 1 );
  glutCreateWindow( "MAP - Please wait while drawing" );
  
  int tex_size = 0;
  if(headless) {
    headless = InitPbuffer( tex_size );
    mapobj.setDeviceSize( tex_size );
    if(headless) {
      while(1) {
	redrawMap();
      }
      exit(0);
    }
  }
  if ( mapobj.getSize() > 1024 )
      mapobj.setDeviceSize( 1024 );
  glutReshapeWindow( mapobj.getDeviceSize(), mapobj.getDeviceSize() );

  glutReshapeFunc( reshapeMap );
  glutDisplayFunc( redrawMap );

  glutMainLoop();

  return 0;
}