/* Emacs style mode select   -*- C++ -*-
 *-----------------------------------------------------------------------------
 *
 *
 *  PrBoom a Doom port merged with LxDoom and LSDLDoom
 *  based on BOOM, a modified and improved DOOM engine
 *  Copyright (C) 1999 by
 *  id Software, Chi Hoang, Lee Killough, Jim Flynn, Rand Phares, Ty Halderman
 *  Copyright (C) 1999-2000 by
 *  Jess Haas, Nicolas Kalkhof, Colin Phipps, Florian Schulze
 *
 *  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., 59 Temple Place - Suite 330, Boston, MA
 *  02111-1307, USA.
 *
 * DESCRIPTION:
 *      Handles WAD file header, directory, lump I/O.
 *
 *-----------------------------------------------------------------------------
 */

#include "doomstat.h"
#include "doomtype.h"

#ifdef __GNUG__
#pragma implementation "w_wad.h"
#endif
#include "w_wad.h"
#include "m_swap.h"
#include "i_system.h"

#include "rockmacros.h"
//
// GLOBALS
//

// Location of each lump on disk.
lumpinfo_t *lumpinfo IBSS_ATTR;
int        numlumps IBSS_ATTR;         // killough
void       **lumpcache IBSS_ATTR;      // killough
#ifdef TIMEDIAG
static int *locktic; // cph

static void W_ReportLocks(void)
{
   int i;
   lprintf(LO_DEBUG, "W_ReportLocks:\nLump     Size   Locks  Tics\n");
   for (i=0; i<numlumps; i++) {
      if (lumpinfo[i].locks)
         printf("%8.8s %6u %2d   %6d\n", lumpinfo[i].name,
                 W_LumpLength(i), lumpinfo[i].locks, gametic - locktic[i]);
   }
}
#endif

#ifdef HEAPDUMP
void W_PrintLump(FILE* fp, void* p) {
   int i;
   for (i=0; i<numlumps; i++)
      if (lumpcache[i] == p) {
         fdprintf(fp, " %8.8s %6u %2d %6d", lumpinfo[i].name,
                 W_LumpLength(i), lumpinfo[i].locks, gametic - locktic[i]);
         return;
      }
   fprintf(fp, " not found");
}
#endif



static int W_Filelength(int handle)
{
   /*
     struct stat   fileinfo;
     if (fstat(handle,&fileinfo) == -1)
       I_Error("W_Filelength: Error fstating");
     return fileinfo.st_size;
   */
   return filesize(handle);
}

void ExtractFileBase (const char *path, char *dest)
{
   const char *src = path + strlen(path) - 1;
   int length;

   // back up until a \ or the start
   while (src != path && src[-1] != ':' // killough 3/22/98: allow c:filename
          && *(src-1) != '\\'
          && *(src-1) != '/')
      src--;

   // copy up to eight characters
   memset(dest,0,8);
   length = 0;

   while (*src && *src != '.' && ++length<9)
      *dest++ = toupper(*src++);
   /* cph - length check removed, just truncate at 8 chars.
    * If there are 8 or more chars, we'll copy 8, and no zero termination
    */
}

//
// 1/18/98 killough: adds a default extension to a path
// Note: Backslashes are treated specially, for MS-DOS.
//

char *AddDefaultExtension(char *path, const char *ext)
{
   char *p = path;
   while (*p++);
   while (p-->path && *p!='/' && *p!='\\')
      if (*p=='.')
         return path;
   if (*ext!='.')
      strcat(path,".");
   return strcat(path,ext);
}

//
// LUMP BASED ROUTINES.
//

//
// W_AddFile
// All files are optional, but at least one file must be
//  found (PWAD, if all required lumps are present).
// Files with a .wad extension are wadlink files
//  with multiple lumps.
// Other files are single lumps with the base filename
//  for the lump name.
//
// Reload hack removed by Lee Killough
// CPhipps - source is an enum
//

static void W_AddFile(const char *filename, wad_source_t source)
// killough 1/31/98: static, const
{
   wadinfo_t   header;
   lumpinfo_t* lump_p;
   unsigned    i;
   int         handle;
   int         length;
   int         startlump;
   filelump_t  *fileinfo, *fileinfo2free=NULL; //killough
   filelump_t  singleinfo;

   // open the file and add to directory

   handle = open(filename,O_RDONLY);

#ifdef HAVE_NET
   if (handle == -1 && D_NetGetWad(filename)) // CPhipps
      handle = open(filename,O_RDONLY);
#endif

   if (handle == -1)
   {
      if (  strlen(filename)<=4 ||      // add error check -- killough
            (strcasecmp(filename+strlen(filename)-4 , ".lmp" ) &&
             strcasecmp(filename+strlen(filename)-4 , ".gwa" ) )
         )
         I_Error("W_AddFile: couldn't open %s",filename);
      return;
   }

   //jff 8/3/98 use logical output routine
   printf (" adding %s\n",filename);
   startlump = numlumps;

   if (  strlen(filename)<=4 ||
         (
            strcasecmp(filename+strlen(filename)-4,".wad") &&
            strcasecmp(filename+strlen(filename)-4,".gwa")
         )
      )
   {
      // single lump file
      fileinfo = &singleinfo;
      singleinfo.filepos = 0;
      singleinfo.size = LONG(W_Filelength(handle));
      ExtractFileBase(filename, singleinfo.name);
      numlumps++;
   }
   else
   {
      // WAD file
      read(handle, &header, sizeof(header));
      if (strncmp(header.identification,"IWAD",4) &&
            strncmp(header.identification,"PWAD",4))
         I_Error("W_AddFile: Wad file %s doesn't have IWAD or PWAD id", filename);
      header.numlumps = LONG(header.numlumps);
      header.infotableofs = LONG(header.infotableofs);
      length = header.numlumps*sizeof(filelump_t);
      fileinfo2free = fileinfo = malloc(length);    // killough
      lseek(handle, header.infotableofs, SEEK_SET);
      read(handle, fileinfo, length);
      numlumps += header.numlumps;
   }

   // Fill in lumpinfo
   lumpinfo = realloc(lumpinfo, numlumps*sizeof(lumpinfo_t));

   lump_p = &lumpinfo[startlump];
   for (i=startlump ; (int)i<numlumps ; i++,lump_p++, fileinfo++)
   {
      lump_p->handle = handle;                    //  killough 4/25/98
      lump_p->position = LONG(fileinfo->filepos);
      lump_p->size = LONG(fileinfo->size);
#ifndef NO_PREDEFINED_LUMPS
      lump_p->data = NULL;                        // killough 1/31/98
#endif
      lump_p->namespace = ns_global;              // killough 4/17/98
      strncpy (lump_p->name, fileinfo->name, 8);
      lump_p->source = source;                    // Ty 08/29/98
      lump_p->locks = 0;                   // CPhipps - initialise locks
   }

   free(fileinfo2free);      // killough
}

// jff 1/23/98 Create routines to reorder the master directory
// putting all flats into one marked block, and all sprites into another.
// This will allow loading of sprites and flats from a PWAD with no
// other changes to code, particularly fast hashes of the lumps.
//
// killough 1/24/98 modified routines to be a little faster and smaller

static int IsMarker(const char *marker, const char *name)
{
   return !strncasecmp(name, marker, 8) ||
          (*name == *marker && !strncasecmp(name+1, marker, 7));
}

// killough 4/17/98: add namespace tags

static void W_CoalesceMarkedResource(const char *start_marker,
                                     const char *end_marker, int namespace)
{
   lumpinfo_t *marked = malloc(sizeof(*marked) * numlumps);
   size_t i, num_marked = 0, num_unmarked = 0;
   int is_marked = 0, mark_end = 0;
   lumpinfo_t *lump = lumpinfo;

   for (i=numlumps; i--; lump++)
      if (IsMarker(start_marker, lump->name))       // start marker found
      { // If this is the first start marker, add start marker to marked lumps
         if (!num_marked)
         {
            strncpy(marked->name, start_marker, 8);
            marked->size = 0;  // killough 3/20/98: force size to be 0
            marked->namespace = ns_global;        // killough 4/17/98
            num_marked = 1;
         }
         is_marked = 1;                            // start marking lumps
      }
      else
         if (IsMarker(end_marker, lump->name))       // end marker found
         {
            mark_end = 1;                           // add end marker below
            is_marked = 0;                          // stop marking lumps
         }
         else
            if (is_marked)                            // if we are marking lumps,
            {                                       // move lump to marked list
               marked[num_marked] = *lump;
               marked[num_marked++].namespace = namespace;  // killough 4/17/98
            }
            else
               lumpinfo[num_unmarked++] = *lump;       // else move down THIS list

   // Append marked list to end of unmarked list
   memcpy(lumpinfo + num_unmarked, marked, num_marked * sizeof(*marked));

   free(marked);                                   // free marked list

   numlumps = num_unmarked + num_marked;           // new total number of lumps

   if (mark_end)                                   // add end marker
   {
      lumpinfo[numlumps].size = 0;  // killough 3/20/98: force size to be 0
      lumpinfo[numlumps].namespace = ns_global;   // killough 4/17/98
      strncpy(lumpinfo[numlumps++].name, end_marker, 8);
   }
}

// Hash function used for lump names.
// Must be mod'ed with table size.
// Can be used for any 8-character names.
// by Lee Killough

unsigned W_LumpNameHash(const char *s)
{
   unsigned hash;
   (void) ((hash =        toupper(s[0]), s[1]) &&
           (hash = hash*3+toupper(s[1]), s[2]) &&
           (hash = hash*2+toupper(s[2]), s[3]) &&
           (hash = hash*2+toupper(s[3]), s[4]) &&
           (hash = hash*2+toupper(s[4]), s[5]) &&
           (hash = hash*2+toupper(s[5]), s[6]) &&
           (hash = hash*2+toupper(s[6]),
            hash = hash*2+toupper(s[7]))
          );
   return hash;
}

//
// W_CheckNumForName
// Returns -1 if name not found.
//
// Rewritten by Lee Killough to use hash table for performance. Significantly
// cuts down on time -- increases Doom performance over 300%. This is the
// single most important optimization of the original Doom sources, because
// lump name lookup is used so often, and the original Doom used a sequential
// search. For large wads with > 1000 lumps this meant an average of over
// 500 were probed during every search. Now the average is under 2 probes per
// search. There is no significant benefit to packing the names into longwords
// with this new hashing algorithm, because the work to do the packing is
// just as much work as simply doing the string comparisons with the new
// algorithm, which minimizes the expected number of comparisons to under 2.
//
// killough 4/17/98: add namespace parameter to prevent collisions
// between different resources such as flats, sprites, colormaps
//

int (W_CheckNumForName)(register const char *name, register int namespace)
{
   // Hash function maps the name to one of possibly numlump chains.
   // It has been tuned so that the average chain length never exceeds 2.

   int i;

   if (!lumpinfo)
      return -1;

   i = lumpinfo[W_LumpNameHash(name) % (unsigned) numlumps].index;

   // We search along the chain until end, looking for case-insensitive
   // matches which also match a namespace tag. Separate hash tables are
   // not used for each namespace, because the performance benefit is not
   // worth the overhead, considering namespace collisions are rare in
   // Doom wads.

   while (i >= 0 && (strncasecmp(lumpinfo[i].name, name, 8) ||
                     lumpinfo[i].namespace != (unsigned)namespace))
      i = lumpinfo[i].next;

   // Return the matching lump, or -1 if none found.

   return i;
}

//
// killough 1/31/98: Initialize lump hash table
//

static void W_InitLumpHash(void)
{
   int i;

   for (i=0; i<numlumps; i++)
      lumpinfo[i].index = -1;                     // mark slots empty

   // Insert nodes to the beginning of each chain, in first-to-last
   // lump order, so that the last lump of a given name appears first
   // in any chain, observing pwad ordering rules. killough

   for (i=0; i<numlumps; i++)
   {                                           // hash function:
      int j = W_LumpNameHash(lumpinfo[i].name) % (unsigned) numlumps;
      lumpinfo[i].next = lumpinfo[j].index;     // Prepend to list
      lumpinfo[j].index = i;
   }
}

// End of lump hashing -- killough 1/31/98



// W_GetNumForName
// Calls W_CheckNumForName, but bombs out if not found.
//
int W_GetNumForName (const char* name)     // killough -- const added
{
   int i = W_CheckNumForName (name);
   if (i == -1)
      I_Error("W_GetNumForName: %s not found", name);
   return i;
}

// W_Init
// Loads each of the files in the wadfiles array.
// All files are optional, but at least one file
//  must be found.
// Files with a .wad extension are idlink files
//  with multiple lumps.
// Other files are single lumps with the base filename
//  for the lump name.
// Lump names can appear multiple times.
// The name searcher looks backwards, so a later file
//  does override all earlier ones.
//
// CPhipps - modified to use the new wadfiles array
//
struct wadfile_info *wadfiles=NULL;

size_t numwadfiles = 0; // CPhipps - size of the wadfiles array (dynamic, no limit)

void W_Init(void)
{
#ifndef NO_PREDEFINED_LUMPS
   // killough 1/31/98: add predefined lumps first

   numlumps = num_predefined_lumps;

   // lumpinfo will be realloced as lumps are added
   lumpinfo = malloc(numlumps*sizeof(*lumpinfo));

   memcpy(lumpinfo, predefined_lumps, numlumps*sizeof(*lumpinfo));
   // Ty 08/29/98 - add source flag to the predefined lumps
   {
      int i;
      for (i=0;i<numlumps;i++)
         lumpinfo[i].source = source_pre;
   }
#else
   // CPhipps - start with nothing

   numlumps = 0; lumpinfo = malloc(0);
#endif

   { // CPhipps - new wadfiles array used
      // open all the files, load headers, and count lumps
      int i;
      for (i=0; (size_t)i<numwadfiles; i++)
         W_AddFile(wadfiles[i].name, wadfiles[i].src);
   }
   if (!numlumps)
      I_Error ("W_Init: No files found");

   //jff 1/23/98
   // get all the sprites and flats into one marked block each
   // killough 1/24/98: change interface to use M_START/M_END explicitly
   // killough 4/17/98: Add namespace tags to each entry

   W_CoalesceMarkedResource("S_START", "S_END", ns_sprites);
   W_CoalesceMarkedResource("F_START", "F_END", ns_flats);

   // killough 4/4/98: add colormap markers
   W_CoalesceMarkedResource("C_START", "C_END", ns_colormaps);

   // set up caching
   lumpcache = calloc(sizeof *lumpcache, numlumps); // killough

   if (!lumpcache)
      I_Error ("W_Init: Couldn't allocate lumpcache");

   // killough 1/31/98: initialize lump hash table
   W_InitLumpHash();

#ifdef TIMEDIAG
   // cph - allocate space for lock time diagnostics
   locktic = malloc(sizeof(*locktic)*numlumps);
   atexit(W_ReportLocks);
#endif
}

//
// W_LumpLength
// Returns the buffer size needed to load the given lump.
//
int W_LumpLength (int lump)
{
   if (lump >= numlumps)
      I_Error ("W_LumpLength: %d >= numlumps",lump);
   return lumpinfo[lump].size;
}

//
// W_ReadLump
// Loads the lump into the given buffer,
//  which must be >= W_LumpLength().
//
#undef DEBUGCACHE
void W_ReadLump(int lump, void *dest)
{
   lumpinfo_t *l = lumpinfo + lump;

#ifdef RANGECHECK
   if (lump >= numlumps)
      I_Error ("W_ReadLump: %d >= numlumps",lump);
#endif

#ifndef NO_PREDEFINED_LUMPS
   if (l->data)     // killough 1/31/98: predefined lump data
      memcpy(dest, l->data, l->size);
   else
#endif
   {
      int c;

#ifdef DEBUGCACHE
      if(gamestate==GS_LEVEL)
         printf("Loading %s\n", lumpinfo[lump].name);
#endif

      // killough 1/31/98: Reload hack (-wart) removed

      lseek(l->handle, l->position, SEEK_SET);
      c = read(l->handle, dest, l->size);
      if (c < l->size)
         I_Error("W_ReadLump: only read %d of %d on lump %d", c, l->size, lump);
   }
}

//
// W_CacheLumpNum
/*
 * killough 4/25/98: simplified
 * CPhipps - modified for new lump locking scheme
 *           returns a const*
 */

void * (W_CacheLumpNum)(int lump, unsigned short locks)
{
#ifdef RANGECHECK
   if ((unsigned)lump >= (unsigned)numlumps)
      I_Error ("W_CacheLumpNum: %d >= numlumps",lump);
#endif

   if (!lumpcache[lump])      // read the lump in
      W_ReadLump(lump, Z_Malloc(W_LumpLength(lump), PU_CACHE, &lumpcache[lump]));

   /* cph - if wasn't locked but now is, tell z_zone to hold it */
   if (!lumpinfo[lump].locks && locks) {
      Z_ChangeTag(lumpcache[lump],PU_STATIC);
#ifdef TIMEDIAG
      locktic[lump] = gametic;
#endif
   }
   lumpinfo[lump].locks += locks;

#ifdef SIMPLECHECKS
   if (!((lumpinfo[lump].locks+1) & 0xf))
      printf("W_CacheLumpNum: High lock on %s (%d)\n",
              lumpinfo[lump].name, lumpinfo[lump].locks);
#endif

   // CPhipps - if not locked, can't give you a pointer
   return (locks ? lumpcache[lump] : NULL);
}

/* cph -
 * W_CacheLumpNumPadded
 *
 * Caches a lump and pads the memory following it.
 * The thing returned is *only* guaranteed to be padded if
 *  the lump isn't already cached (otherwise, you get whatever is
 *  currently cached, which if it was cached by a previous call
 *  to this will also be padded)
 */

void * W_CacheLumpNumPadded(int lump, size_t len, unsigned char pad)
{
   const int locks = 1;
#ifdef RANGECHECK
   if ((unsigned)lump >= (unsigned)numlumps)
      I_Error ("W_CacheLumpNum: %d >= numlumps",lump);
#endif

   if (!lumpcache[lump]) {     /* read the lump in */
      size_t lumplen = W_LumpLength(lump);
      unsigned char* p;
      W_ReadLump(lump, p = Z_Malloc(len, PU_CACHE, &lumpcache[lump]));
      memset(p+lumplen, pad, len-lumplen);
   }

   /* cph - if wasn't locked but now is, tell z_zone to hold it */
   if (!lumpinfo[lump].locks && locks) {
      Z_ChangeTag(lumpcache[lump],PU_STATIC);
#ifdef TIMEDIAG
      locktic[lump] = gametic;
#endif
   }
   lumpinfo[lump].locks += locks;

#ifdef SIMPLECHECKS
   if (!((lumpinfo[lump].locks+1) & 0xf))
      printf("W_CacheLumpNum: High lock on %s (%d)\n",
              lumpinfo[lump].name, lumpinfo[lump].locks);
#endif

   return lumpcache[lump];
}

//
// W_UnlockLumpNum
//
// CPhipps - this changes (should reduce) the number of locks on a lump

void (W_UnlockLumpNum)(int lump, signed short unlocks)
{
#ifdef SIMPLECHECKS
   if ((signed short)lumpinfo[lump].locks < unlocks)
      printf("W_UnlockLumpNum: Excess unlocks on %s (%d-%d)\n",
              lumpinfo[lump].name, lumpinfo[lump].locks, unlocks);
#endif
   lumpinfo[lump].locks -= unlocks;
   // cph - Note: must only tell z_zone to make purgeable if currently locked,
   // else it might already have been purged
   if (unlocks && !lumpinfo[lump].locks)
      Z_ChangeTag(lumpcache[lump], PU_CACHE);
}

// W_CacheLumpName macroized in w_wad.h -- killough

#ifndef NO_PREDEFINED_LUMPS
// WritePredefinedLumpWad
// Args: Filename - string with filename to write to
// Returns: void
//
// If the user puts a -dumplumps switch on the command line, we will
// write all those predefined lumps above out into a pwad.  User
// supplies the pwad name.
//
// killough 4/22/98: make endian-independent, remove tab chars
void WritePredefinedLumpWad(const char *filename)
{
   int handle;         // for file open
   char filenam[256];  // we may have to add ".wad" to the name they pass

   if (!filename || !*filename)  // check for null pointer or empty name
      return;  // early return

   AddDefaultExtension(strcpy(filenam, filename), ".wad");

   // The following code writes a PWAD from the predefined lumps array
   // How to write a PWAD will not be explained here.
#ifdef _MSC_VER // proff: In Visual C open is defined a bit different
   if ( (handle = open (filenam, O_RDWR | O_CREAT | O_BINARY, _S_IWRITE|_S_IREAD)) != -1)
#else
   if ( (handle = open (filenam, O_RDWR | O_CREAT | O_BINARY, S_IWUSR|S_IRUSR)) != -1)
#endif
   {
      wadinfo_t header = {"PWAD"};
      size_t filepos = sizeof(wadinfo_t) + num_predefined_lumps * sizeof(filelump_t);
      int i;

      header.numlumps = LONG(num_predefined_lumps);
      header.infotableofs = LONG(sizeof(header));

      // write header
      write(handle, &header, sizeof(header));

      // write directory
      for (i=0;(size_t)i<num_predefined_lumps;i++)
      {
         filelump_t fileinfo = {0};
         fileinfo.filepos = LONG(filepos);
         fileinfo.size = LONG(predefined_lumps[i].size);
         strncpy(fileinfo.name, predefined_lumps[i].name, 8);
         write(handle, &fileinfo, sizeof(fileinfo));
         filepos += predefined_lumps[i].size;
      }

      // write lumps
      for (i=0;(size_t)i<num_predefined_lumps;i++)
         write(handle, predefined_lumps[i].data, predefined_lumps[i].size);

      close(handle);
      I_Error("WritePredefinedLumpWad: Predefined lumps wad, %s written", filename);
   }
   I_Error("WritePredefinedLumpWad: Can't open predefined lumps wad %s for output", filename);
}
#endif