blob: 2ddcb72e18fdf4aba433eae7dc8be3692062b170 [file] [log] [blame]
/*****************************************************************************
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 */