416 lines
12 KiB
C
416 lines
12 KiB
C
/*****************************************************************************
|
|
|
|
gifclrmap - extract colormaps from GIF images
|
|
|
|
SPDX-License-Identifier: MIT
|
|
|
|
*****************************************************************************/
|
|
|
|
#include <assert.h>
|
|
#include <ctype.h>
|
|
#include <math.h>
|
|
#include <stdbool.h>
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
|
|
#include "getarg.h"
|
|
#include "gif_lib.h"
|
|
|
|
#define PROGRAM_NAME "gifclrmp"
|
|
|
|
static char *VersionStr = PROGRAM_NAME VERSION_COOKIE
|
|
" Gershon Elber, " __DATE__ ", " __TIME__ "\n"
|
|
"(C) Copyright 1989 Gershon Elber.\n";
|
|
static char *CtrlStr =
|
|
PROGRAM_NAME " v%- s%- t%-TranslationFile!s l%-ColorMapFile!s g%-Gamma!F "
|
|
"i%-Image#!d h%- GifFile!*s";
|
|
|
|
static bool SaveFlag = false, TranslateFlag = false, LoadFlag = false,
|
|
GammaFlag = false;
|
|
static double Gamma = 1.0;
|
|
static FILE *ColorFile = NULL;
|
|
FILE *TranslateFile = NULL;
|
|
static GifPixelType Translation[256];
|
|
|
|
static ColorMapObject *ModifyColorMap(ColorMapObject *ColorMap);
|
|
static void QuitGifError(GifFileType *GifFileIn, GifFileType *GifFileOut);
|
|
|
|
/******************************************************************************
|
|
Interpret the command line and scan the given GIF file.
|
|
******************************************************************************/
|
|
int main(int argc, char **argv) {
|
|
int NumFiles, ExtCode, CodeSize, ImageNum = 0, ImageN, HasGIFOutput,
|
|
ErrorCode;
|
|
bool Error, ImageNFlag = false, HelpFlag = false, GifNoisyPrint = false;
|
|
GifRecordType RecordType;
|
|
GifByteType *Extension, *CodeBlock;
|
|
char **FileName = NULL, *ColorFileName, *TranslateFileName;
|
|
GifFileType *GifFileIn = NULL, *GifFileOut = NULL;
|
|
|
|
if ((Error = GAGetArgs(argc, argv, CtrlStr, &GifNoisyPrint, &SaveFlag,
|
|
&TranslateFlag, &TranslateFileName, &LoadFlag,
|
|
&ColorFileName, &GammaFlag, &Gamma, &ImageNFlag,
|
|
&ImageN, &HelpFlag, &NumFiles, &FileName)) !=
|
|
false ||
|
|
(NumFiles > 1 && !HelpFlag)) {
|
|
if (Error) {
|
|
GAPrintErrMsg(Error);
|
|
} else if (NumFiles > 1) {
|
|
GIF_MESSAGE("Error in command line parsing - one GIF "
|
|
"file please.");
|
|
}
|
|
GAPrintHowTo(CtrlStr);
|
|
exit(EXIT_FAILURE);
|
|
}
|
|
|
|
if (HelpFlag) {
|
|
(void)fprintf(stderr, VersionStr, GIFLIB_MAJOR, GIFLIB_MINOR);
|
|
GAPrintHowTo(CtrlStr);
|
|
exit(EXIT_SUCCESS);
|
|
}
|
|
|
|
if (SaveFlag + LoadFlag + GammaFlag + TranslateFlag > 1) {
|
|
GIF_EXIT("Can not handle more than one of -s -l, -t, or -g at "
|
|
"the same time.");
|
|
}
|
|
|
|
/* Default action is to dump colormaps */
|
|
if (!SaveFlag && !LoadFlag && !GammaFlag && !TranslateFlag) {
|
|
SaveFlag = true;
|
|
}
|
|
|
|
if (NumFiles == 1) {
|
|
if ((GifFileIn = DGifOpenFileName(*FileName, &ErrorCode)) ==
|
|
NULL) {
|
|
PrintGifError(ErrorCode);
|
|
exit(EXIT_FAILURE);
|
|
}
|
|
} else {
|
|
/* Use stdin instead: */
|
|
if ((GifFileIn = DGifOpenFileHandle(0, &ErrorCode)) == NULL) {
|
|
PrintGifError(ErrorCode);
|
|
exit(EXIT_FAILURE);
|
|
}
|
|
}
|
|
|
|
if (SaveFlag) {
|
|
/* We are dumping out the color map as text file to stdout: */
|
|
ColorFile = stdout;
|
|
} else {
|
|
if (TranslateFlag) {
|
|
/* We are loading new color map from specified file: */
|
|
if ((TranslateFile = fopen(TranslateFileName, "rt")) ==
|
|
NULL) {
|
|
GIF_EXIT("Failed to open specified color "
|
|
"translation file.");
|
|
}
|
|
}
|
|
|
|
if (LoadFlag) {
|
|
/* We are loading new color map from specified file: */
|
|
if ((ColorFile = fopen(ColorFileName, "rt")) == NULL) {
|
|
GIF_EXIT(
|
|
"Failed to open specified color map file.");
|
|
}
|
|
}
|
|
}
|
|
|
|
if ((HasGIFOutput = (LoadFlag || TranslateFlag || GammaFlag)) != 0) {
|
|
/* Open stdout for GIF output file: */
|
|
if ((GifFileOut = EGifOpenFileHandle(1, &ErrorCode)) == NULL) {
|
|
PrintGifError(ErrorCode);
|
|
exit(EXIT_FAILURE);
|
|
}
|
|
}
|
|
|
|
if (!ImageNFlag) {
|
|
/* We are supposed to modify the screen color map, so do it: */
|
|
if (!GifFileIn->SColorMap) {
|
|
GIF_EXIT("No colormap to modify");
|
|
}
|
|
GifFileIn->SColorMap = ModifyColorMap(GifFileIn->SColorMap);
|
|
if (!HasGIFOutput) {
|
|
/* We can quit here, as we have the color map: */
|
|
DGifCloseFile(GifFileIn, NULL);
|
|
fclose(ColorFile);
|
|
exit(EXIT_SUCCESS);
|
|
}
|
|
}
|
|
/* And dump out its new possible repositioned screen information: */
|
|
if (HasGIFOutput) {
|
|
if (EGifPutScreenDesc(GifFileOut, GifFileIn->SWidth,
|
|
GifFileIn->SHeight,
|
|
GifFileIn->SColorResolution,
|
|
GifFileIn->SBackGroundColor,
|
|
GifFileIn->SColorMap) == GIF_ERROR) {
|
|
QuitGifError(GifFileIn, GifFileOut);
|
|
}
|
|
}
|
|
|
|
/* Scan the content of the GIF file and load the image(s) in: */
|
|
do {
|
|
if (DGifGetRecordType(GifFileIn, &RecordType) == GIF_ERROR) {
|
|
QuitGifError(GifFileIn, GifFileOut);
|
|
}
|
|
|
|
switch (RecordType) {
|
|
case IMAGE_DESC_RECORD_TYPE:
|
|
if (DGifGetImageDesc(GifFileIn) == GIF_ERROR) {
|
|
QuitGifError(GifFileIn, GifFileOut);
|
|
}
|
|
if ((++ImageNum == ImageN) && ImageNFlag) {
|
|
/* We are suppose to modify this image color
|
|
* map, do it: */
|
|
GifFileIn->SColorMap =
|
|
ModifyColorMap(GifFileIn->SColorMap);
|
|
if (!HasGIFOutput) {
|
|
/* We can quit here, as we have the
|
|
* color map: */
|
|
DGifCloseFile(GifFileIn, NULL);
|
|
fclose(ColorFile);
|
|
exit(EXIT_SUCCESS);
|
|
}
|
|
}
|
|
if (HasGIFOutput) {
|
|
if (EGifPutImageDesc(
|
|
GifFileOut, GifFileIn->Image.Left,
|
|
GifFileIn->Image.Top,
|
|
GifFileIn->Image.Width,
|
|
GifFileIn->Image.Height,
|
|
GifFileIn->Image.Interlace,
|
|
GifFileIn->Image.ColorMap) ==
|
|
GIF_ERROR) {
|
|
QuitGifError(GifFileIn, GifFileOut);
|
|
}
|
|
}
|
|
|
|
if (!TranslateFlag ||
|
|
(ImageNFlag && (ImageN != ImageNum))) {
|
|
/* Now read image itself in decoded form as we
|
|
* don't */
|
|
/* really care what we have there, and this is
|
|
* much */
|
|
/* faster.
|
|
*/
|
|
if (DGifGetCode(GifFileIn, &CodeSize,
|
|
&CodeBlock) == GIF_ERROR) {
|
|
QuitGifError(GifFileIn, GifFileOut);
|
|
}
|
|
if (HasGIFOutput) {
|
|
if (EGifPutCode(GifFileOut, CodeSize,
|
|
CodeBlock) ==
|
|
GIF_ERROR) {
|
|
QuitGifError(GifFileIn,
|
|
GifFileOut);
|
|
}
|
|
}
|
|
while (CodeBlock != NULL) {
|
|
if (DGifGetCodeNext(GifFileIn,
|
|
&CodeBlock) ==
|
|
GIF_ERROR) {
|
|
QuitGifError(GifFileIn,
|
|
GifFileOut);
|
|
}
|
|
if (HasGIFOutput) {
|
|
if (EGifPutCodeNext(
|
|
GifFileOut,
|
|
CodeBlock) ==
|
|
GIF_ERROR) {
|
|
QuitGifError(
|
|
GifFileIn,
|
|
GifFileOut);
|
|
}
|
|
}
|
|
}
|
|
} else /* we need to mung pixels intices */
|
|
{
|
|
int i;
|
|
register GifPixelType *cp;
|
|
|
|
GifPixelType *Line = (GifPixelType *)malloc(
|
|
GifFileIn->Image.Width *
|
|
sizeof(GifPixelType));
|
|
for (i = 0; i < GifFileIn->Image.Height; i++) {
|
|
if (DGifGetLine(
|
|
GifFileIn, Line,
|
|
GifFileIn->Image.Width) ==
|
|
GIF_ERROR) {
|
|
QuitGifError(GifFileIn,
|
|
GifFileOut);
|
|
}
|
|
|
|
/* translation step goes here */
|
|
for (cp = Line;
|
|
cp < Line + GifFileIn->Image.Width;
|
|
cp++) {
|
|
*cp = Translation[*cp];
|
|
}
|
|
|
|
if (EGifPutLine(
|
|
GifFileOut, Line,
|
|
GifFileIn->Image.Width) ==
|
|
GIF_ERROR) {
|
|
QuitGifError(GifFileIn,
|
|
GifFileOut);
|
|
}
|
|
}
|
|
free((char *)Line);
|
|
}
|
|
break;
|
|
case EXTENSION_RECORD_TYPE:
|
|
assert(GifFileOut != NULL); /* might pacify Coverity */
|
|
/* pass through extension records */
|
|
if (DGifGetExtension(GifFileIn, &ExtCode, &Extension) ==
|
|
GIF_ERROR) {
|
|
QuitGifError(GifFileIn, GifFileOut);
|
|
}
|
|
if (Extension == NULL) {
|
|
break;
|
|
}
|
|
if (EGifPutExtensionLeader(GifFileOut, ExtCode) ==
|
|
GIF_ERROR) {
|
|
QuitGifError(GifFileIn, GifFileOut);
|
|
}
|
|
if (EGifPutExtensionBlock(GifFileOut, Extension[0],
|
|
Extension + 1) == GIF_ERROR) {
|
|
QuitGifError(GifFileIn, GifFileOut);
|
|
}
|
|
while (Extension != NULL) {
|
|
if (DGifGetExtensionNext(
|
|
GifFileIn, &Extension) == GIF_ERROR) {
|
|
QuitGifError(GifFileIn, GifFileOut);
|
|
}
|
|
if (Extension != NULL) {
|
|
if (EGifPutExtensionBlock(
|
|
GifFileOut, Extension[0],
|
|
Extension + 1) == GIF_ERROR) {
|
|
QuitGifError(GifFileIn,
|
|
GifFileOut);
|
|
}
|
|
}
|
|
}
|
|
if (EGifPutExtensionTrailer(GifFileOut) == GIF_ERROR) {
|
|
QuitGifError(GifFileIn, GifFileOut);
|
|
}
|
|
break;
|
|
case TERMINATE_RECORD_TYPE:
|
|
break;
|
|
default: /* Should be trapped by DGifGetRecordType. */
|
|
break;
|
|
}
|
|
} while (RecordType != TERMINATE_RECORD_TYPE);
|
|
|
|
if (DGifCloseFile(GifFileIn, &ErrorCode) == GIF_ERROR) {
|
|
PrintGifError(ErrorCode);
|
|
exit(EXIT_FAILURE);
|
|
}
|
|
if (HasGIFOutput) {
|
|
if (EGifCloseFile(GifFileOut, &ErrorCode) == GIF_ERROR) {
|
|
PrintGifError(ErrorCode);
|
|
exit(EXIT_FAILURE);
|
|
}
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
/******************************************************************************
|
|
Modify the given colormap according to global variables setting.
|
|
******************************************************************************/
|
|
static ColorMapObject *ModifyColorMap(ColorMapObject *ColorMap) {
|
|
int i, Dummy, Red, Green, Blue;
|
|
|
|
if (SaveFlag) {
|
|
/* Save this color map to ColorFile: */
|
|
for (i = 0; i < ColorMap->ColorCount; i++) {
|
|
fprintf(ColorFile, "%3d %3d %3d %3d\n", i,
|
|
ColorMap->Colors[i].Red,
|
|
ColorMap->Colors[i].Green,
|
|
ColorMap->Colors[i].Blue);
|
|
}
|
|
return (ColorMap);
|
|
} else if (LoadFlag) {
|
|
/* Read the color map in ColorFile into this color map: */
|
|
for (i = 0; i < ColorMap->ColorCount; i++) {
|
|
if (feof(ColorFile)) {
|
|
GIF_EXIT("Color file to load color map from, "
|
|
"too small.");
|
|
}
|
|
if (fscanf(ColorFile, "%3d %3d %3d %3d\n", &Dummy, &Red,
|
|
&Green, &Blue) == 4) {
|
|
ColorMap->Colors[i].Red = Red;
|
|
ColorMap->Colors[i].Green = Green;
|
|
ColorMap->Colors[i].Blue = Blue;
|
|
}
|
|
}
|
|
return (ColorMap);
|
|
} else if (GammaFlag) {
|
|
/* Apply gamma correction to this color map: */
|
|
double Gamma1 = 1.0 / Gamma;
|
|
for (i = 0; i < ColorMap->ColorCount; i++) {
|
|
ColorMap->Colors[i].Red =
|
|
((int)(255 * pow(ColorMap->Colors[i].Red / 255.0,
|
|
Gamma1)));
|
|
ColorMap->Colors[i].Green =
|
|
((int)(255 * pow(ColorMap->Colors[i].Green / 255.0,
|
|
Gamma1)));
|
|
ColorMap->Colors[i].Blue =
|
|
((int)(255 * pow(ColorMap->Colors[i].Blue / 255.0,
|
|
Gamma1)));
|
|
}
|
|
return (ColorMap);
|
|
} else if (TranslateFlag) {
|
|
ColorMapObject *NewMap;
|
|
int Max = 0;
|
|
|
|
/* Read the translation table in TranslateFile: */
|
|
for (i = 0; i < ColorMap->ColorCount; i++) {
|
|
int tmp;
|
|
if (feof(TranslateFile)) {
|
|
GIF_EXIT("Color file to load color map from, "
|
|
"too small.");
|
|
}
|
|
if (fscanf(TranslateFile, "%3d %3d\n", &Dummy, &tmp) ==
|
|
2) {
|
|
Translation[i] = tmp & 0xff;
|
|
if (Translation[i] > Max) {
|
|
Max = Translation[i];
|
|
}
|
|
}
|
|
}
|
|
|
|
if ((NewMap = GifMakeMapObject(1 << GifBitSize(Max + 1),
|
|
NULL)) == NULL) {
|
|
GIF_EXIT("Out of memory while allocating color map!");
|
|
}
|
|
|
|
/* Apply the translation; we'll do it to the pixels, too */
|
|
for (i = 0; i < ColorMap->ColorCount; i++) {
|
|
NewMap->Colors[i] = ColorMap->Colors[Translation[i]];
|
|
}
|
|
|
|
return (NewMap);
|
|
} else {
|
|
GIF_EXIT("Nothing to do!");
|
|
return (ColorMap);
|
|
}
|
|
}
|
|
|
|
/******************************************************************************
|
|
Close both input and output file (if open), and exit.
|
|
******************************************************************************/
|
|
static void QuitGifError(GifFileType *GifFileIn, GifFileType *GifFileOut) {
|
|
if (GifFileIn != NULL) {
|
|
PrintGifError(GifFileIn->Error);
|
|
EGifCloseFile(GifFileIn, NULL);
|
|
}
|
|
if (GifFileOut != NULL) {
|
|
PrintGifError(GifFileOut->Error);
|
|
EGifCloseFile(GifFileOut, NULL);
|
|
}
|
|
exit(EXIT_FAILURE);
|
|
}
|
|
|
|
/* end */
|