OpenGTA/gfx_extract.cpp

/************************************************************************
* Copyright (c) 2005-2007 tok@openlinux.org.uk                          *
*                                                                       *
* This software is provided as-is, without any express or implied       *
* warranty. In no event will the authors be held liable for any         *
* damages arising from the use of this software.                        *
*                                                                       *
* Permission is granted to anyone to use this software for any purpose, *
* including commercial applications, and to alter it and redistribute   *
* it freely, subject to the following restrictions:                     *
*                                                                       *
* 1. The origin of this software must not be misrepresented; you must   *
* not claim that you wrote the original software. If you use this       *
* software in a product, an acknowledgment in the product documentation *
* would be appreciated but is not required.                             *
*                                                                       *
* 2. Altered source versions must be plainly marked as such, and must   *
* not be misrepresented as being the original software.                 *
*                                                                       *
* 3. This notice may not be removed or altered from any source          *
* distribution.                                                         *
************************************************************************/
#include <iostream>
#include <getopt.h>
#include <stdlib.h>
#include <SDL.h>

#ifdef DUMP_DELTA_DEBUG
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#endif

#include "opengta.h"
#include "dataholder.h"
#include "m_exceptions.h"
#include "set.h"

SDL_Surface* image = NULL;

void at_exit() {
  if (image)
    SDL_FreeSurface(image);
  PHYSFS_deinit();
  SDL_Quit();
}

void display_image(SDL_Surface* s) {
  SDL_Surface *screen = SDL_SetVideoMode(640, 480, 32, SDL_DOUBLEBUF);
  SDL_Event event;
  SDL_BlitSurface(s, NULL, screen, NULL);
  SDL_Flip(screen);
  while (1) {
    while (SDL_PollEvent(&event)) {
      switch(event.type) {
        case SDL_QUIT:
          return;
        case SDL_KEYDOWN:
          if (event.key.keysym.sym == SDLK_ESCAPE)
            return;
        default:
          break;
      }
    }
    SDL_Delay(100);
  }
}

SDL_Surface* get_image(unsigned char* rp, unsigned int w, unsigned int h) {
#if SDL_BYTEORDER == SDL_BIG_ENDIAN 
#define rmask 0xff000000 
#define gmask 0x00ff0000 
#define bmask 0x0000ff00 
#define amask 0x000000ff
#else 
#define rmask 0x000000ff 
#define gmask 0x0000ff00
#define bmask 0x00ff0000 
#define amask 0xff000000 
#endif 
  SDL_Surface* s = SDL_CreateRGBSurface(SDL_SWSURFACE|SDL_SRCALPHA, 
      w, h, 32, rmask, gmask, bmask, amask);
  SDL_LockSurface(s);
  unsigned char* dst = static_cast<unsigned char*>(s->pixels);
  for (unsigned int i=0; i<w*h; i++) {
    *dst = *rp; ++dst;++rp;
    *dst = *rp; ++dst;++rp;
    *dst = *rp; ++dst;++rp;
    *dst = *rp; ++dst;++rp;
    //*dst = 0xff; ++dst;
  }
  SDL_UnlockSurface(s);
  return s;
}

void usage(const char* a0) {
  std::cout << "USAGE: " << a0 << " --file $STYLE --extract|--display [--section N] --index N" << 
    std::endl; 
  std::cout << "Where section 0 are 'side blocks', 1 'lid blocks', 2 'aux blocks' and 3 are 'sprites'" << std::endl;
  std::cout << "You can apply remaps (--remap N) and deltas (--delta N) but the relative indices are" << std::endl;
  std::cout << "not always correct (== experimental)." << std::endl;
  return;
}

int main(int argc, char* argv[]) {
  atexit(at_exit);
  PHYSFS_init(argv[0]);
  // add pwd to search path
  PHYSFS_addToSearchPath(PHYSFS_getBaseDir(), 1);
  PHYSFS_addToSearchPath("gtadata.zip", 1);
  char* file = NULL;
  SDL_Init(SDL_INIT_VIDEO);

  int c = 0;
  int mode = 0;
  int remap = -1;
  int delta = 0;
  Util::Set delta_as_set(32, (unsigned char*)&delta);
  bool delta_set = false;
  bool rgba = true;

  unsigned int idx = 0;
  unsigned int section = 0;
  while (1) {
    int option_index = 0;
    static struct option long_options[] = {
      {"info", 0, 0, 'i'},
      {"display", 0, 0, 'd'},
      {"extract", 0, 0, 'e'},
      {"section", 1, 0, 's'},
      {"index", 1, 0, 'x'},
      {"file", 1, 0, 'f'},
      {"remap", 1, 0, 'r'},
      {"delta", 1, 0, 'a'},
      {"delta-set", 1, 0, 'A'},
      {0, 0, 0, 0}
    };

    c = getopt_long (argc, argv, "h",
        long_options, &option_index);
    if (c == -1)
      break;
    switch (c) {
      case 'i':
        mode = 0;
        break;
      case 'd':
        mode |= 1;
        break;
      case 'e':
        mode |= 2;
        break;
      case 'f':
        file = optarg;
        break;
      case 'x':
        idx = atoi(optarg);
        break;
      case 's':
        section = atoi(optarg);
        break;
      case 'r':
        remap = atoi(optarg);
        break;
      case 'a':
        delta = atoi(optarg);
        break;
      case 'A':
        delta_set = true;
        delta_as_set.set_item(atoi(optarg), true);
        break;
      case 'h':
      default:
        usage(argv[0]);
        return(0);
    }
  }
  
  if (!file) {
    std::cerr << "Error: no data file selected" << std::endl;
    usage(argv[0]);
    return 1;
  }
  if (!PHYSFS_exists(file)) {
    std::cerr << "File does not exist in searchpath: " << file << std::endl;
    return 1;
  }
  try {
    // exception handling of the constructor doesn't work here; urgh...
    OpenGTA::StyleHolder::Instance().load(file);
    OpenGTA::GraphicsBase & graphics = OpenGTA::StyleHolder::Instance().get();
    if (delta_set)
      graphics.setDeltaHandling(true);

    if (!mode)
      return 0;

    switch(section) {
      case 0:
        graphics.getSide(idx, 0, rgba);
        image = get_image(graphics.getTmpBuffer(rgba), 64,64);
        break;
      case 1:
        graphics.getLid(idx, 0, rgba);
        image = get_image(graphics.getTmpBuffer(rgba), 64, 64);
        break;
      case 2:
        graphics.getAux(idx, 0, rgba);
        image = get_image(graphics.getTmpBuffer(rgba), 64, 64);
        break;
      case 3:
        OpenGTA::GraphicsBase::SpriteInfo *sprite = graphics.getSprite(idx);
        std::cout << "Sprite is " << int(sprite->w) << "x" << int(sprite->h) <<
          " with " << int(sprite->deltaCount) << " deltas" << std::endl;
        image = get_image(graphics.getSpriteBitmap(idx, remap, delta), sprite->w, sprite->h);
#ifdef DUMP_DELTA_DEBUG
        if (delta && !delta_set) {
          std::cout << "dumping delta" << std::endl;
          OpenGTA::GraphicsBase::DeltaInfo & dinfo = sprite->delta[delta-1];
          int dump_fd = open("delta.raw", O_WRONLY | O_CREAT | O_TRUNC, 
            S_IRUSR | S_IWUSR);
          write(dump_fd, dinfo.ptr, dinfo.size);
          close(dump_fd);
        }
#endif
        break; 
    }

    if (!image) {
      std::cerr << "Error: image pointer is NULL; aborting" << std::endl;
      return 1;
    }
    if (mode & 1) {
      display_image(image);
    }
    if (mode & 2) {
      SDL_SaveBMP(image, "out.bmp");
    }   
  }
  catch (Exception & e) {
    std::cerr << "Exception occured: " << e.what() << std::endl;
    return 1;
  }

  
  return 0;
}