mirror of
https://github.com/unidoc/unipdf.git
synced 2025-05-02 22:17:06 +08:00
24648f4481
* Fixing platform indepenedent integer size * Cleared test logs. * Cleared unnecessary int32 * Defined precise integer size for jbig2 segments.
123 lines
3.4 KiB
Go
123 lines
3.4 KiB
Go
/*
|
|
* This file is subject to the terms and conditions defined in
|
|
* file 'LICENSE.md', which is part of this source code package.
|
|
*/
|
|
|
|
package segments
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/unidoc/unipdf/v3/common"
|
|
|
|
"github.com/unidoc/unipdf/v3/internal/jbig2/bitmap"
|
|
"github.com/unidoc/unipdf/v3/internal/jbig2/reader"
|
|
)
|
|
|
|
// TestDecodeGenericRegion tests the decode process of the jbig2 Generic Region.
|
|
func TestDecodeGenericRegion(t *testing.T) {
|
|
if testing.Verbose() {
|
|
common.SetLogger(common.NewConsoleLogger(common.LogLevelDebug))
|
|
}
|
|
|
|
t.Run("AnnexH", func(t *testing.T) {
|
|
t.Run("S-12th", func(t *testing.T) {
|
|
data := []byte{
|
|
// header
|
|
0x00, 0x00, 0x00, 0x0B, 0x27, 0x00, 0x02, 0x00, 0x00, 0x00, 0x23,
|
|
|
|
// data
|
|
0x00, 0x00, 0x00, 0x36, 0x00, 0x00, 0x00, 0x2C, 0x00, 0x00, 0x00, 0x04,
|
|
0x00, 0x00, 0x00, 0x0B, 0x00, 0x08, 0x03, 0xFF, 0xFD, 0xFF, 0x02, 0xFE,
|
|
0xFE, 0xFE, 0x04, 0xEE, 0xED, 0x87, 0xFB, 0xCB, 0x2B, 0xFF, 0xAC,
|
|
}
|
|
|
|
r := reader.New(data)
|
|
d := &document{}
|
|
h, err := NewHeader(d, r, 0, OSequential)
|
|
require.NoError(t, err)
|
|
|
|
assert.Equal(t, uint32(11), h.SegmentNumber)
|
|
assert.Equal(t, uint64(35), h.SegmentDataLength)
|
|
assert.Equal(t, 2, h.PageAssociation)
|
|
|
|
sg, err := h.GetSegmentData()
|
|
require.NoError(t, err)
|
|
|
|
s, ok := sg.(*GenericRegion)
|
|
require.True(t, ok)
|
|
|
|
assert.Equal(t, uint32(44), s.RegionSegment.BitmapHeight)
|
|
assert.Equal(t, uint32(54), s.RegionSegment.BitmapWidth)
|
|
assert.Equal(t, bitmap.CmbOpOr, s.RegionSegment.CombinaionOperator)
|
|
assert.Equal(t, true, s.IsTPGDon)
|
|
assert.Equal(t, byte(0), s.GBTemplate)
|
|
|
|
bm, err := s.GetRegionBitmap()
|
|
require.NoError(t, err)
|
|
|
|
isTestingFrame(t, bm)
|
|
})
|
|
|
|
t.Run("S-5th", func(t *testing.T) {
|
|
data := []byte{
|
|
// Header
|
|
0x00, 0x00, 0x00, 0x04, 0x27, 0x00, 0x01, 0x00, 0x00, 0x00, 0x2C,
|
|
|
|
// Data part
|
|
0x00, 0x00, 0x00, 0x36, 0x00, 0x00, 0x00, 0x2C, 0x00, 0x00, 0x00,
|
|
0x04, 0x00, 0x00, 0x00, 0x0B, 0x00, 0x01, 0x26, 0xA0, 0x71, 0xCE,
|
|
0xA7, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF,
|
|
0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xF8, 0xF0,
|
|
}
|
|
r := reader.New(data)
|
|
d := &document{}
|
|
h, err := NewHeader(d, r, 0, OSequential)
|
|
require.NoError(t, err)
|
|
|
|
assert.Equal(t, uint32(4), h.SegmentNumber)
|
|
assert.Equal(t, uint64(44), h.SegmentDataLength)
|
|
assert.Equal(t, 1, h.PageAssociation)
|
|
|
|
gs, err := h.GetSegmentData()
|
|
require.NoError(t, err)
|
|
|
|
s, ok := gs.(*GenericRegion)
|
|
require.True(t, ok)
|
|
|
|
assert.Equal(t, uint32(44), s.RegionSegment.BitmapHeight)
|
|
assert.Equal(t, uint32(54), s.RegionSegment.BitmapWidth)
|
|
assert.Equal(t, bitmap.CmbOpOr, s.RegionSegment.CombinaionOperator)
|
|
assert.Equal(t, true, s.IsMMREncoded)
|
|
|
|
b, err := s.GetRegionBitmap()
|
|
require.NoError(t, err)
|
|
|
|
isTestingFrame(t, b)
|
|
})
|
|
})
|
|
}
|
|
|
|
func isTestingFrame(t *testing.T, b *bitmap.Bitmap) {
|
|
assert.Equal(t, 44, b.Height)
|
|
assert.Equal(t, 54, b.Width)
|
|
|
|
for y := 0; y < b.Height; y++ {
|
|
for x := 0; x < b.Width; x++ {
|
|
pix := b.GetPixel(x, y)
|
|
|
|
// first two and last two rows are set to '1'
|
|
if y == 0 || y == 1 || y == b.Height-1 || y == b.Height-2 {
|
|
assert.Equal(t, true, pix)
|
|
} else if x == 0 || x == 1 || x == b.Width-1 || x == b.Width-2 {
|
|
assert.Equal(t, true, pix)
|
|
} else {
|
|
assert.Equal(t, false, pix)
|
|
}
|
|
}
|
|
}
|
|
}
|