Skip to content

Commit fd6e3d6

Browse files
Thibault-Pelletierjourdain
authored andcommitted
fix(rca_encoder): Fix turbo JPEG encode color
1 parent 4ddcf08 commit fd6e3d6

File tree

2 files changed

+3
-2
lines changed

2 files changed

+3
-2
lines changed

examples/vtk_cone_simple.py

+1
Original file line numberDiff line numberDiff line change
@@ -72,6 +72,7 @@ def setup_vtk(self):
7272
mapper.SetInputConnection(cone_source.GetOutputPort())
7373
actor = vtkActor()
7474
actor.SetMapper(mapper)
75+
actor.GetProperty().SetColor(1, 0.8, 0.8)
7576

7677
renderer.AddActor(actor)
7778
renderer.ResetCamera()

trame_rca/encoders/turbo_jpeg.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
from numpy.typing import NDArray
2-
from turbojpeg import TurboJPEG
2+
from turbojpeg import TurboJPEG, TJPF_RGB
33
from trame_rca.encoders.img import TO_IMAGE_TYPE
44
# import time
55

@@ -43,7 +43,7 @@ def encode_np_img_to_bytes(
4343
# t0 = time.time()
4444
image = image.reshape((cols, rows, -1))
4545
image = image[::-1, :, :]
46-
result = jpeg.encode(image, quality=quality)
46+
result = jpeg.encode(image, quality=quality, pixel_format=TJPF_RGB)
4747
# t1 = time.time()
4848
# print(f"tubo-jpeg encode {t1-t0:.04f}s")
4949

0 commit comments

Comments
 (0)