Compare commits

..

2 Commits

Author SHA1 Message Date
72e7a8171b Update .gitignore 2024-05-25 19:45:16 +01:00
fb972c4737 Load the image in a separate thread 2024-05-12 20:00:09 +01:00
3 changed files with 47 additions and 149 deletions

39
compile
View File

@@ -1,41 +1,20 @@
#!/bin/bash #!/bin/bash
BUILD_TYPE="debug"
while [[ $# > 0 ]];do
case $1 in
--release)
BUILD_TYPE="release"
shift
;;
*|-*|--*)
echo "Unknown option $1"
exit 1
;;
esac
done
CC=clang CC=clang
CFLAGS="-g -Wall -Werror -pedantic -fsanitize=address -fsanitize=undefined -DDEBUG"
if [[ $BUILD_TYPE == "release" ]]; then
CFLAGS="-O3 -Wall -Werror -pedantic"
else
CFLAGS="-g -Wall -Werror -pedantic -fsanitize=address -fsanitize=undefined -DDEBUG"
fi
INCLUDES="\ INCLUDES="\
-I$(find ./src -type d | xargs -I{} echo -n "-I{} ") \ -I$(find ./src -type d | xargs -I{} echo -n "-I{} ") \
$(find intern/wizapp/src -type d | xargs -I{} echo -n "-I{} ") \ $(find intern/wizapp/src -type d | xargs -I{} echo -n "-I{} ") \
$(pkg-config --cflags sdl2) \ $(pkg-config --cflags sdl2) \
" "
LIBS="\ LIBS="\
-lm \ -lm \
-pthread \ -pthread \
$(pkg-config --libs sdl2) \ $(pkg-config --libs sdl2) \
" "
SRC="\ SRC="\
$(find ./src -name "*.c" | xargs -I{} echo -n "{} ") \ $(find ./src -name *.c | xargs -I{} echo -n "{} ") \
$(find intern/wizapp/src -type f -name "*.c" | xargs -I{} echo -n "{} ") \ $(find intern/wizapp/src -type f -name *.c | xargs -I{} echo -n "{} ") \
" "
OUT=tiffread OUT=tiffread

View File

@@ -4,7 +4,6 @@
#include "tiffread.h" #include "tiffread.h"
#include <SDL2/SDL.h> #include <SDL2/SDL.h>
#include <SDL2/SDL_events.h> #include <SDL2/SDL_events.h>
#include <SDL2/SDL_rect.h>
#include <SDL2/SDL_render.h> #include <SDL2/SDL_render.h>
#include <SDL2/SDL_surface.h> #include <SDL2/SDL_surface.h>
#include <SDL2/SDL_video.h> #include <SDL2/SDL_video.h>
@@ -19,8 +18,6 @@
#define WINDOW_WIDTH 800 #define WINDOW_WIDTH 800
#define WINDOW_HEIGHT 600 #define WINDOW_HEIGHT 600
#define VIEW_AREA_WIDTH 700
#define VIEW_AREA_HEIGHT 500
#define AMASK 0xff000000 #define AMASK 0xff000000
#define BMASK 0x00ff0000 #define BMASK 0x00ff0000
@@ -37,7 +34,6 @@ struct point {
internal void write_debug_ppm6(const Image *img); internal void write_debug_ppm6(const Image *img);
internal void *load_tiff_image(void *args); internal void *load_tiff_image(void *args);
internal SDL_Rect get_dest_rect(SDL_Surface *surface);
typedef struct img_thread_args ImgThreadArgs; typedef struct img_thread_args ImgThreadArgs;
struct img_thread_args { struct img_thread_args {
@@ -94,7 +90,12 @@ int main(int argc, char *argv[]) {
goto MAIN_DESTROY_SURFACE; goto MAIN_DESTROY_SURFACE;
} }
SDL_Rect dest = get_dest_rect(surface); SDL_Rect dest = {
.h = surface->h,
.w = surface->w,
.x = (WINDOW_WIDTH - surface->w) / 2,
.y = (WINDOW_HEIGHT - surface->h) / 2,
};
bool running = true; bool running = true;
@@ -187,16 +188,3 @@ internal void *load_tiff_image(void *args) {
return surface; return surface;
} }
internal SDL_Rect get_dest_rect(SDL_Surface *surface) {
f64 ratio = (f64)(surface->h) / (f64)(surface->w);
u64 width = surface->w <= VIEW_AREA_WIDTH ? surface->w : VIEW_AREA_WIDTH;
u64 height = width * ratio;
return (SDL_Rect){
.w = width,
.h = height,
.x = (WINDOW_WIDTH - width) / 2,
.y = (WINDOW_HEIGHT - height) / 2,
};
}

View File

@@ -4,10 +4,8 @@
#include "image.h" #include "image.h"
#include "mem_arena.h" #include "mem_arena.h"
#include <assert.h> #include <assert.h>
#include <bits/pthreadtypes.h>
#include <math.h> #include <math.h>
#include <netinet/in.h> #include <netinet/in.h>
#include <pthread.h>
#include <stdbool.h> #include <stdbool.h>
#include <stddef.h> #include <stddef.h>
#include <stdint.h> #include <stdint.h>
@@ -202,7 +200,6 @@ struct tiff_image {
typedef struct tiff_reader TiffReader; typedef struct tiff_reader TiffReader;
struct tiff_reader { struct tiff_reader {
const char *filename;
FILE *fp; FILE *fp;
TiffHdr header; TiffHdr header;
TiffIFD ifd; TiffIFD ifd;
@@ -210,17 +207,6 @@ struct tiff_reader {
Pixel *pixels; Pixel *pixels;
}; };
typedef struct strip_thread_args StripThreadArgs;
struct strip_thread_args {
FILE *fp;
const TiffStrip *strip;
u64 pixel_count;
u64 samples_per_pixel;
u64 main_samples;
TiffAlpha alpha;
Pixel *pixel_write_start;
};
internal bool read_tiff_header(TiffReader *reader); internal bool read_tiff_header(TiffReader *reader);
internal bool read_ifd(TiffReader *reader, Arena *arena); internal bool read_ifd(TiffReader *reader, Arena *arena);
internal bool read_ifd_fields(TiffReader *reader); internal bool read_ifd_fields(TiffReader *reader);
@@ -228,8 +214,7 @@ internal void read_alpha(TiffReader *reader);
internal void read_bits_per_sample(TiffReader *reader); internal void read_bits_per_sample(TiffReader *reader);
internal bool read_image_pixels(TiffReader *reader, Arena *arena); internal bool read_image_pixels(TiffReader *reader, Arena *arena);
internal bool read_strip_data(TiffReader *reader, Arena *arena); internal bool read_strip_data(TiffReader *reader, Arena *arena);
internal bool read_strips(TiffReader *reader); internal void read_strips(TiffReader *reader);
internal void *read_strip(void *arg);
internal void read_strip_data_field(const TiffReader *reader, internal void read_strip_data_field(const TiffReader *reader,
StripDataField *field); StripDataField *field);
internal bool read_field(const TiffField *field, TiffImage *img); internal bool read_field(const TiffField *field, TiffImage *img);
@@ -246,7 +231,7 @@ Image *read_baseline_tiff(const char *file, Arena *arena) {
goto READ_BASELINE_RETURN_IMG; goto READ_BASELINE_RETURN_IMG;
} }
TiffReader reader = {.filename = file, .fp = fopen(file, "rb")}; TiffReader reader = {.fp = fopen(file, "rb")};
if (!reader.fp) { if (!reader.fp) {
goto READ_BASELINE_RETURN_IMG; goto READ_BASELINE_RETURN_IMG;
} }
@@ -549,7 +534,9 @@ internal bool read_image_pixels(TiffReader *reader, Arena *arena) {
return false; return false;
} }
return read_strips(reader); read_strips(reader);
return true;
} }
internal bool read_strip_data(TiffReader *reader, Arena *arena) { internal bool read_strip_data(TiffReader *reader, Arena *arena) {
@@ -598,98 +585,42 @@ internal bool read_strip_data(TiffReader *reader, Arena *arena) {
return true; return true;
} }
internal bool read_strips(TiffReader *reader) { internal void read_strips(TiffReader *reader) {
u64 position = 0; u64 position = 0;
StripThreadArgs args[reader->img.strip_count]; u64 main_samples = reader->img.sample_count - reader->img.extra_samples_count;
pthread_t threads[reader->img.strip_count]; TiffAlpha alpha = reader->img.alpha;
Pixel *p;
u64 start_offset;
u64 alpha_offset;
for (u64 i = 0; i < reader->img.strip_count; ++i) { for (u64 i = 0; i < reader->img.strip_count; ++i) {
const TiffStrip *strip = &(reader->img.strips[i]); const TiffStrip *strip = &(reader->img.strips[i]);
u64 pixel_count = strip->byte_count / reader->img.sample_count;
args[i] = (StripThreadArgs){
.fp = fopen(reader->filename, "rb"),
.strip = strip,
.pixel_count = pixel_count,
.samples_per_pixel = reader->img.sample_count,
.main_samples =
reader->img.sample_count - reader->img.extra_samples_count,
.alpha = reader->img.alpha,
.pixel_write_start = &(reader->pixels[position]),
};
if (!args[i].fp) { for (u64 j = 0; j < strip->byte_count / reader->img.sample_count; ++j) {
for (u64 j = 0; j < i; ++j) { p = &(reader->pixels[position]);
fclose(args[j].fp); start_offset = strip->offset + j * reader->img.sample_count;
alpha_offset = start_offset + main_samples + alpha.sample_offset;
fread_with_offset(reader->fp, p, main_samples, start_offset);
if (alpha.type == ALPHA_TYPE_UNDEFINED) {
p->a = 255;
} else {
fread_with_offset(reader->fp, &(p->a), 1, alpha_offset);
if (alpha.type == ALPHA_TYPE_UNASSOCIATED) {
f32 a_norm = u8_normalise(p->a);
p->r = u8_denormalise(u8_normalise(p->r) * a_norm);
p->g = u8_denormalise(u8_normalise(p->g) * a_norm);
p->b = u8_denormalise(u8_normalise(p->b) * a_norm);
}
} }
return false; ++position;
}
position += pixel_count;
}
for (u64 i = 0; i < reader->img.strip_count; ++i) {
if (pthread_create(&(threads[i]), NULL, read_strip, &(args[i])) != 0) {
for (u64 j = 0; j < i; ++j) {
pthread_cancel(threads[i]);
fclose(args[j].fp);
}
// Close the remaining files
for (u64 j = i; j < reader->img.strip_count; ++j) {
fclose(args[j].fp);
}
return false;
} }
} }
bool image_loaded = true;
for (u64 i = 0; i < reader->img.strip_count; ++i) {
if (pthread_join(threads[i], NULL) != 0) {
image_loaded = false;
}
fclose(args[i].fp);
}
return image_loaded;
}
internal void *read_strip(void *arg) {
StripThreadArgs *args = (StripThreadArgs *)arg;
Pixel *p;
u64 write_offset = 0;
u64 read_offset;
u64 alpha_offset;
for (u64 j = 0; j < args->pixel_count; ++j) {
p = args->pixel_write_start + write_offset;
read_offset = args->strip->offset + j * args->samples_per_pixel;
alpha_offset = read_offset + args->main_samples + args->alpha.sample_offset;
fread_with_offset(args->fp, p, args->main_samples, read_offset);
if (args->alpha.type == ALPHA_TYPE_UNDEFINED) {
p->a = 255;
} else {
fread_with_offset(args->fp, &(p->a), 1, alpha_offset);
if (args->alpha.type == ALPHA_TYPE_UNASSOCIATED) {
f32 a_norm = u8_normalise(p->a);
p->r = u8_denormalise(u8_normalise(p->r) * a_norm);
p->g = u8_denormalise(u8_normalise(p->g) * a_norm);
p->b = u8_denormalise(u8_normalise(p->b) * a_norm);
}
}
++write_offset;
}
return NULL;
} }
internal void read_strip_data_field(const TiffReader *reader, internal void read_strip_data_field(const TiffReader *reader,