From ad11276c4f06d0cd5523fc2808c6c2c639eb7265 Mon Sep 17 00:00:00 2001
From: Andri Yngvason <andri@yngvason.is>
Date: Wed, 8 Jul 2020 18:21:25 +0000
Subject: [PATCH] Use tight encoder v2

---
 include/common.h |  4 ++--
 meson.build      |  2 +-
 src/server.c     | 15 +++++++++------
 3 files changed, 12 insertions(+), 9 deletions(-)

diff --git a/include/common.h b/include/common.h
index 0d7754f..bd85a51 100644
--- a/include/common.h
+++ b/include/common.h
@@ -24,7 +24,7 @@
 #include "sys/queue.h"
 
 #include "neatvnc.h"
-#include "tight.h"
+#include "tight-encoder-v2.h"
 #include "config.h"
 
 #ifdef ENABLE_TLS
@@ -74,7 +74,7 @@ struct nvnc_client {
 	bool is_updating;
 	nvnc_client_fn cleanup_fn;
 	z_stream z_stream;
-	struct tight_encoder tight_encoder;
+	struct tight_encoder_v2 tight_encoder;
 	size_t buffer_index;
 	size_t buffer_len;
 	uint8_t msg_buffer[MSG_BUFFER_SIZE];
diff --git a/meson.build b/meson.build
index 72bc7b1..e4c14c6 100644
--- a/meson.build
+++ b/meson.build
@@ -73,7 +73,7 @@ config = configuration_data()
 
 if libturbojpeg.found()
 	dependencies += libturbojpeg
-	sources += 'src/tight.c'
+	sources += 'src/tight-encoder-v2.c'
 	config.set('ENABLE_TIGHT', true)
 endif
 
diff --git a/src/server.c b/src/server.c
index f4e9416..2c987ef 100644
--- a/src/server.c
+++ b/src/server.c
@@ -16,7 +16,7 @@
 
 #include "rfb-proto.h"
 #include "zrle.h"
-#include "tight.h"
+#include "tight-encoder-v2.h"
 #include "raw-encoding.h"
 #include "vec.h"
 #include "type-macros.h"
@@ -82,7 +82,7 @@ static void client_close(struct nvnc_client* client)
 	LIST_REMOVE(client, link);
 	stream_destroy(client->net_stream);
 #ifdef ENABLE_TIGHT
-	tight_encoder_destroy(&client->tight_encoder);
+	tight_encoder_v2_destroy(&client->tight_encoder);
 #endif
 	deflateEnd(&client->z_stream);
 	pixman_region_fini(&client->damage);
@@ -731,7 +731,9 @@ static void on_connection(void* obj)
 		goto deflate_failure;
 
 #ifdef ENABLE_TIGHT
-	if (tight_encoder_init(&client->tight_encoder) < 0)
+	int width = server->display->buffer->width;
+	int height = server->display->buffer->height;
+	if (tight_encoder_v2_init(&client->tight_encoder, width, height) < 0)
 		goto tight_failure;
 #endif
 
@@ -753,7 +755,7 @@ static void on_connection(void* obj)
 
 payload_failure:
 #ifdef ENABLE_TIGHT
-	tight_encoder_destroy(&client->tight_encoder);
+	tight_encoder_v2_destroy(&client->tight_encoder);
 #endif
 	pixman_region_fini(&client->damage);
 #ifdef ENABLE_TIGHT
@@ -961,8 +963,9 @@ static void do_client_update_fb(void* work)
 		break;
 #ifdef ENABLE_TIGHT
 	case RFB_ENCODING_TIGHT:
-		tight_encode_frame(&client->tight_encoder, &update->frame, fb,
-		                   &update->server_fmt, &update->region);
+		tight_encode_frame_v2(&client->tight_encoder, &update->frame,
+				&client->pixfmt, fb, &update->server_fmt,
+				&update->region);
 		break;
 #endif
 	case RFB_ENCODING_ZRLE: