Skip to content

Commit 9467c75

Browse files
committed
cleanup
1 parent 4b8add2 commit 9467c75

File tree

5 files changed

+6
-87
lines changed

5 files changed

+6
-87
lines changed

opencv.go

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -836,9 +836,7 @@ func (e *openCVEncoder) Encode(f *Framebuffer, opt map[int]int) ([]byte, error)
836836
firstOpt = (*C.int)(unsafe.Pointer(&optList[0]))
837837
}
838838

839-
success := bool(C.opencv_encoder_write(e.encoder, f.mat, firstOpt, C.size_t(len(optList))))
840-
841-
if !success {
839+
if !C.opencv_encoder_write(e.encoder, f.mat, firstOpt, C.size_t(len(optList))) {
842840
return nil, ErrInvalidImage
843841
}
844842

opencv_test.go

Lines changed: 1 addition & 74 deletions
Original file line numberDiff line numberDiff line change
@@ -269,7 +269,7 @@ func TestICC(t *testing.T) {
269269

270270
// try encoding a WebP image, including ICC profile data when available
271271
dstBuf := make([]byte, destinationBufferSize)
272-
encoder, err := newWebpEncoder(decoder, dstBuf, false)
272+
encoder, err := newWebpEncoder(decoder, dstBuf)
273273
if err != nil {
274274
t.Fatalf("Failed to create a new webp encoder: %v", err)
275275
}
@@ -292,76 +292,3 @@ func TestICC(t *testing.T) {
292292
}
293293
}
294294

295-
// TestStandaloneToneMapping tests the standalone tone mapping API
296-
func TestStandaloneToneMapping(t *testing.T) {
297-
// Test that tone mapping can be applied independently of encoding
298-
hdrPngData, err := ioutil.ReadFile("data/hdr-ohmama.png")
299-
if err != nil {
300-
t.Skipf("HDR test image not found: %v", err)
301-
}
302-
303-
decoder, err := NewDecoder(hdrPngData)
304-
if err != nil {
305-
t.Fatalf("Failed to create decoder: %v", err)
306-
}
307-
defer decoder.Close()
308-
309-
header, err := decoder.Header()
310-
if err != nil {
311-
t.Fatalf("Failed to read header: %v", err)
312-
}
313-
314-
// Create a framebuffer and decode the image
315-
fb := NewFramebuffer(header.Width(), header.Height())
316-
defer fb.Close()
317-
318-
if err := decoder.DecodeTo(fb); err != nil {
319-
t.Fatalf("Failed to decode: %v", err)
320-
}
321-
322-
// Get the ICC profile
323-
icc := decoder.ICC()
324-
if len(icc) == 0 {
325-
t.Skip("No ICC profile found in test image")
326-
}
327-
328-
// Apply tone mapping as a standalone operation
329-
if err := fb.ApplyToneMapping(icc); err != nil {
330-
t.Fatalf("Failed to apply tone mapping: %v", err)
331-
}
332-
333-
// Verify the framebuffer is still valid
334-
if fb.Width() != header.Width() || fb.Height() != header.Height() {
335-
t.Fatalf("Framebuffer dimensions changed after tone mapping: got %dx%d, want %dx%d",
336-
fb.Width(), fb.Height(), header.Width(), header.Height())
337-
}
338-
339-
// Now encode to multiple formats to demonstrate independence
340-
outputBuf := make([]byte, 50*1024*1024)
341-
342-
for _, format := range []string{".png", ".webp"} {
343-
encodeOptions := map[int]int{}
344-
if format == ".png" {
345-
encodeOptions[PngCompression] = 7
346-
} else if format == ".webp" {
347-
encodeOptions[WebpQuality] = 85
348-
}
349-
350-
// Create encoder without ForceSdr since we already tone-mapped
351-
encoder, err := NewEncoder(format, decoder, outputBuf)
352-
if err != nil {
353-
t.Fatalf("Failed to create %s encoder: %v", format, err)
354-
}
355-
356-
encoded, err := encoder.Encode(fb, encodeOptions)
357-
if err != nil {
358-
t.Fatalf("Failed to encode to %s: %v", format, err)
359-
}
360-
361-
if len(encoded) == 0 {
362-
t.Fatalf("Encoded %s data is empty", format)
363-
}
364-
365-
encoder.Close()
366-
}
367-
}

webp.cpp

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,14 +1,11 @@
11
#include "webp.hpp"
2-
#include "tone_mapping.hpp"
32
#include <opencv2/imgproc.hpp>
43
#include <webp/decode.h>
54
#include <webp/encode.h>
65
#include <webp/mux.h>
76
#include <webp/mux_types.h>
87
#include <webp/demux.h>
98
#include <stdbool.h>
10-
#include <lcms2.h>
11-
#include <memory>
129

1310
struct webp_decoder_struct {
1411
WebPMux* mux;

webp.go

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -228,10 +228,7 @@ func (e *webpEncoder) Encode(f *Framebuffer, opt map[int]int) ([]byte, error) {
228228

229229
// Encode the current frame
230230
frameDelay := int(f.duration.Milliseconds())
231-
232-
length := C.webp_encoder_write(e.encoder, f.mat, firstOpt, C.size_t(len(optList)),
233-
C.int(frameDelay), C.int(f.blend), C.int(f.dispose), 0, 0)
234-
231+
length := C.webp_encoder_write(e.encoder, f.mat, firstOpt, C.size_t(len(optList)), C.int(frameDelay), C.int(f.blend), C.int(f.dispose), 0, 0)
235232
if length == 0 {
236233
return nil, ErrInvalidImage
237234
}

webp_test.go

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -75,7 +75,7 @@ func testNewWebpEncoder(t *testing.T) {
7575
defer decoder.Close()
7676

7777
dstBuf := make([]byte, destinationBufferSize)
78-
encoder, err := newWebpEncoder(decoder, dstBuf, false)
78+
encoder, err := newWebpEncoder(decoder, dstBuf)
7979
if err != nil {
8080
t.Fatalf("Unexpected error: %v", err)
8181
}
@@ -161,7 +161,7 @@ func testWebpEncoderEncode(t *testing.T) {
161161
}
162162

163163
dstBuf := make([]byte, destinationBufferSize)
164-
encoder, err := newWebpEncoder(decoder, dstBuf, false)
164+
encoder, err := newWebpEncoder(decoder, dstBuf)
165165
if err != nil {
166166
t.Fatalf("Failed to create a new webp encoder: %v", err)
167167
}
@@ -195,7 +195,7 @@ func testWebpEncoderEncode(t *testing.T) {
195195
defer decoder.Close()
196196

197197
dstBuf := make([]byte, destinationBufferSize)
198-
encoder, err := newWebpEncoder(decoder, dstBuf, false)
198+
encoder, err := newWebpEncoder(decoder, dstBuf)
199199
if err != nil {
200200
t.Fatalf("Failed to create a new webp encoder: %v", err)
201201
}

0 commit comments

Comments
 (0)