X-Git-Url: https://git.sesse.net/?a=blobdiff_plain;f=nageru%2Fmjpeg_encoder.h;fp=nageru%2Fmjpeg_encoder.h;h=87bf3b5836509429b67195733d567c77ae0550e4;hb=29f8890fa9150f4077c8a15bd60365f0c5c98a3b;hp=eff361ea4e4cb1ebada2236daef46321e690a942;hpb=1a23c2c0194b81c3c73620a26ac93bf62f3e5ee0;p=nageru

diff --git a/nageru/mjpeg_encoder.h b/nageru/mjpeg_encoder.h
index eff361e..87bf3b5 100644
--- a/nageru/mjpeg_encoder.h
+++ b/nageru/mjpeg_encoder.h
@@ -154,6 +154,22 @@ private:
 	std::unique_ptr<VADisplayWithCleanup> va_dpy;
 	VAConfigID config_id;
 
+	struct VAKey {
+		unsigned width, height;
+		movit::RGBTriplet white_balance;
+
+		bool operator< (const VAKey &other) const {
+			if (width != other.width)
+				return width < other.width;
+			if (height != other.height)
+				return height < other.height;
+			if (white_balance.r != other.white_balance.r)
+				return white_balance.r < other.white_balance.r;
+			if (white_balance.g != other.white_balance.g)
+				return white_balance.g < other.white_balance.g;
+			return white_balance.b < other.white_balance.b;
+		}
+	};
 	struct VAData {
 		std::vector<uint8_t> jpeg_header;
 		VAEncPictureParameterBufferJPEG pic_param;
@@ -161,8 +177,8 @@ private:
 		VAHuffmanTableBufferJPEGBaseline huff;
 		VAEncSliceParameterBufferJPEG parms;
 	};
-	std::map<std::pair<unsigned, unsigned>, VAData> va_data_for_resolution;
-	VAData get_va_data_for_resolution(unsigned width, unsigned height, const movit::RGBTriplet &white_balance);
+	std::map<VAKey, VAData> va_data_for_parameters;
+	VAData get_va_data_for_parameters(unsigned width, unsigned height, const movit::RGBTriplet &white_balance);
 
 	std::list<VAResources> va_resources_freelist;
 	std::mutex va_resources_mutex;