agsalphahack.c

   1 #define _GNU_SOURCE
   2 #include "DataFile.h"
   3 #include <stdio.h>
   4 #include <stdlib.h>
   5 #include <unistd.h>
   6 #include "version.h"
   7 #define ADS ":::AGSalphahack " VERSION " by rofl0r:::"
   8
   9 static int usage(char *argv0) {
  10         fprintf(stderr, ADS "\nusage:\n%s [-s spriteno] DIR\n"
  11                    "removes alphachannel flag from specified sprite in game file.\n"
  12                    "if spriteno is -1 (default), all flags are removed.\n"
  13                    "gamefile is typically ac2game.dta or game28.dta inside DIR.\n"
  14                    "this is useful when sprites have been hacked/converted to 16bit.\n"
  15                    "you should make a backup of the file before doing this.\n"
  16                    , argv0);
  17         return 1;
  18 }
  19
  20 #define SPF_ALPHACHANNEL 0x10
  21
  22 int main(int argc, char**argv) {
  23         int c, flags = 0, spriteno = -1;
  24         while ((c = getopt(argc, argv, "s:")) != EOF) switch(c) {
  25                 case 's': spriteno = atoi(optarg); break;
  26                 default: return usage(argv[0]);
  27         }
  28         if(!argv[optind]) return usage(argv[0]);
  29         char *dir = argv[optind];
  30
  31         ADF a_b, *a = &a_b;
  32         char fnbuf[512];
  33         if(!ADF_find_datafile(dir, fnbuf, sizeof(fnbuf)))
  34                 return 1;
  35
  36         enum ADF_open_error aoe = ADF_open(a, fnbuf);
  37         if(aoe != AOE_success && aoe <= AOE_gamebase) {
  38                 fprintf(stderr, "failed to open/process data file: %s\n", AOE2str(aoe));
  39                 return 1;
  40         } else if (aoe != AOE_success) {
  41                 fprintf(stderr, "warning: failed to process some non-essential parts (%s) of gamefile, probably from a newer game format\n", AOE2str(aoe));
  42         }
  43
  44         off_t off = ADF_get_spriteflagsstart(a);
  45         unsigned nsprites = ADF_get_spritecount(a);
  46         ADF_close(a);
  47
  48         printf("removing alpha of %u out of %u spriteflags.\n", spriteno==-1?nsprites:1, nsprites);
  49
  50         FILE *f = fopen(fnbuf, "r+b");
  51         if(!f) return 1;
  52         fseeko(f, off, SEEK_SET);
  53         unsigned char *buf = malloc(nsprites);
  54         fread(buf, 1, nsprites, f);
  55
  56         size_t i;
  57         for(i=0;i<nsprites;++i) {
  58                 if(spriteno == -1 || spriteno == i)
  59                         buf[i] = buf[i] & (~SPF_ALPHACHANNEL);
  60         }
  61         fseeko(f, off, SEEK_SET);
  62         fwrite(buf, 1, nsprites, f);
  63         fclose(f);
  64         free(buf);
  65         return 0;
  66 }