Cram/tools/cli/main.c

118 lines
1.7 KiB
C

#include "cram.h"
#include "dirent.h"
#define STB_IMAGE_WRITE_IMPLEMENTATION
#include "stb_image_write.h"
#define MAX_DIR_LENGTH 2048
static Cram_Context *context;
#ifdef _WIN32
#define SEPARATOR "\\"
#endif
#ifdef __unix__
#define SEPARATOR "/"
#endif
static const char* GetFilenameExtension(const char *filename)
{
const char *dot = strrchr(filename, '.');
if (!dot || dot == filename)
{
return "";
}
return dot + 1;
}
/* Mostly taken from K&R C 2nd edition page 182 */
static void dirwalk(char *dir)
{
dirent *dp;
DIR *dfd;
char subname[2048];
if ((dfd = opendir(dir)) == NULL)
{
fprintf(stderr, "Can't open %s\n", dir);
return;
}
while ((dp = readdir(dfd)) != NULL)
{
if ( strcmp(dp->d_name, ".") == 0 ||
strcmp(dp->d_name, "..") == 0
)
{
continue;
}
sprintf(subname, "%s%s%s", dir, SEPARATOR, dp->d_name);
if (dp->d_type == DT_DIR)
{
dirwalk(subname);
}
else
{
if (strcmp(GetFilenameExtension(subname), "png") == 0)
{
Cram_AddFile(context, subname);
}
else
{
fprintf(stdout, "skipping %s\n", subname);
}
}
}
closedir(dfd);
}
/* TODO: command line options */
int main(int argc, char *argv[])
{
Cram_ContextCreateInfo createInfo;
uint8_t *pixelData;
uint32_t width;
uint32_t height;
if (argc < 2)
{
fprintf(stderr, "Must provide directory!\n");
return 1;
}
createInfo.padding = 0;
createInfo.trim = 1;
createInfo.maxDimension = 2048;
createInfo.name = "test";
context = Cram_Init(&createInfo);
dirwalk(argv[1]);
Cram_Pack(context);
Cram_GetPixelData(context, &pixelData, &width, &height);
stbi_write_png(
"output.png",
width,
height,
4,
pixelData,
width * 4
);
Cram_Destroy(context);
return 0;
}