mirror of
https://github.com/unidoc/unipdf.git
synced 2025-05-01 22:17:29 +08:00
24648f4481
* Fixing platform indepenedent integer size * Cleared test logs. * Cleared unnecessary int32 * Defined precise integer size for jbig2 segments.
117 lines
3.1 KiB
Go
117 lines
3.1 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 (
|
|
"fmt"
|
|
"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"
|
|
)
|
|
|
|
// TestDecodePatternDictionary tests the decode process of the pattern dictionary segment.
|
|
func TestDecodePatternDictionary(t *testing.T) {
|
|
t.Run("AnnexH", func(t *testing.T) {
|
|
if testing.Verbose() {
|
|
common.SetLogger(common.NewConsoleLogger(common.LogLevelDebug))
|
|
}
|
|
|
|
t.Run("S-6th", func(t *testing.T) {
|
|
data := []byte{
|
|
// Header
|
|
0x00, 0x00, 0x00, 0x05, 0x10, 0x01, 0x01, 0x00, 0x00, 0x00, 0x2D,
|
|
|
|
// Data part
|
|
0x01, 0x04, 0x04, 0x00, 0x00, 0x00, 0x0F, 0x20, 0xD1, 0x84,
|
|
0x61, 0x18, 0x45, 0xF2, 0xF9, 0x7C, 0x8F, 0x11, 0xC3, 0x9E,
|
|
0x45, 0xF2, 0xF9, 0x7D, 0x42, 0x85, 0x0A, 0xAA, 0x84, 0x62,
|
|
0x2F, 0xEE, 0xEC, 0x44, 0x62, 0x22, 0x35, 0x2A, 0x0A, 0x83,
|
|
0xB9, 0xDC, 0xEE, 0x77, 0x80,
|
|
}
|
|
r := reader.New(data)
|
|
d := &document{}
|
|
h, err := NewHeader(d, r, 0, OSequential)
|
|
require.NoError(t, err)
|
|
|
|
ps, err := h.GetSegmentData()
|
|
require.NoError(t, err)
|
|
|
|
p, ok := ps.(*PatternDictionary)
|
|
require.True(t, ok)
|
|
|
|
assert.Equal(t, uint32(5), h.SegmentNumber)
|
|
assert.Equal(t, uint64(45), h.SegmentDataLength)
|
|
assert.Equal(t, 1, h.PageAssociation)
|
|
|
|
require.Equal(t, true, p.IsMMREncoded)
|
|
require.Equal(t, byte(4), p.HdpWidth)
|
|
require.Equal(t, byte(4), p.HdpHeight)
|
|
require.Equal(t, uint32(15), p.GrayMax)
|
|
|
|
dict, err := p.GetDictionary()
|
|
require.NoError(t, err)
|
|
|
|
for i, s := range dict {
|
|
toCompare := bitmap.New(4, 4)
|
|
switch i {
|
|
case 15:
|
|
require.NoError(t, toCompare.SetPixel(0, 3, 1))
|
|
fallthrough
|
|
case 14:
|
|
require.NoError(t, toCompare.SetPixel(3, 3, 1))
|
|
fallthrough
|
|
case 13:
|
|
require.NoError(t, toCompare.SetPixel(3, 0, 1))
|
|
fallthrough
|
|
case 12:
|
|
require.NoError(t, toCompare.SetPixel(0, 0, 1))
|
|
fallthrough
|
|
case 11:
|
|
require.NoError(t, toCompare.SetPixel(3, 1, 1))
|
|
fallthrough
|
|
case 10:
|
|
require.NoError(t, toCompare.SetPixel(2, 3, 1))
|
|
fallthrough
|
|
case 9:
|
|
require.NoError(t, toCompare.SetPixel(1, 0, 1))
|
|
require.NoError(t, toCompare.SetPixel(0, 2, 1))
|
|
fallthrough
|
|
case 8:
|
|
require.NoError(t, toCompare.SetPixel(3, 2, 1))
|
|
fallthrough
|
|
case 7:
|
|
require.NoError(t, toCompare.SetPixel(1, 3, 1))
|
|
fallthrough
|
|
case 6:
|
|
require.NoError(t, toCompare.SetPixel(0, 1, 1))
|
|
fallthrough
|
|
case 5:
|
|
require.NoError(t, toCompare.SetPixel(2, 0, 1))
|
|
fallthrough
|
|
case 4:
|
|
require.NoError(t, toCompare.SetPixel(1, 2, 1))
|
|
fallthrough
|
|
case 3:
|
|
require.NoError(t, toCompare.SetPixel(2, 2, 1))
|
|
fallthrough
|
|
case 2:
|
|
require.NoError(t, toCompare.SetPixel(1, 1, 1))
|
|
fallthrough
|
|
case 1:
|
|
require.NoError(t, toCompare.SetPixel(2, 1, 1))
|
|
}
|
|
assert.True(t, toCompare.Equals(s), fmt.Sprintf("i: %d, %v, %v", i, s.String(), toCompare.String()))
|
|
}
|
|
})
|
|
})
|
|
}
|