-
Notifications
You must be signed in to change notification settings - Fork 0
/
settings.go
91 lines (85 loc) · 2.61 KB
/
settings.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
package classer
import (
"github.com/dlocmelis/pdfreader/internal/jbig2/bitmap"
"github.com/dlocmelis/pdfreader/internal/jbig2/errors"
)
// Settings keeps the settings for the classer.
type Settings struct {
// MaxCompWidth is max component width allowed.
MaxCompWidth int
// MaxCompHeight is max component height allowed.
MaxCompHeight int
// SizeHaus is the size of square struct elem for hausdorf method.
SizeHaus int
// Rank val of hausdorf method match.
RankHaus float64
// Thresh is the threshold value for the correlation score.
Thresh float64
// Corrects thresh value for heavier components; 0 for no correction.
WeightFactor float64
// KeepClassInstances is a flag that defines if the class instances should be stored
// in the 'ClassInstances' BitmapsArray.
KeepClassInstances bool
// Components is the setting the classification.
Components bitmap.Component
// Method is the encoding method.
Method Method
}
// DefaultSettings returns default settings struct.
func DefaultSettings() Settings {
s := &Settings{}
s.SetDefault()
return *s
}
// SetDefault sets the default value for the settings.
func (s *Settings) SetDefault() {
// if max width is not defined get the value from the constants.
if s.MaxCompWidth == 0 {
switch s.Components {
case bitmap.ComponentConn:
s.MaxCompWidth = MaxConnCompWidth
case bitmap.ComponentCharacters:
s.MaxCompWidth = MaxCharCompWidth
case bitmap.ComponentWords:
s.MaxCompWidth = MaxWordCompWidth
}
}
// if max height is not defined take the 'MaxCompHeight' value.
if s.MaxCompHeight == 0 {
s.MaxCompHeight = MaxCompHeight
}
if s.Thresh == 0.0 {
s.Thresh = 0.9
}
if s.WeightFactor == 0.0 {
s.WeightFactor = 0.75
}
if s.RankHaus == 0.0 {
s.RankHaus = 0.97
}
if s.SizeHaus == 0 {
s.SizeHaus = 2
}
}
// Validate validates the settings input.
func (s Settings) Validate() error {
const processName = "Settings.Validate"
if s.Thresh < 0.4 || s.Thresh > 0.98 {
return errors.Error(processName, "jbig2 encoder thresh not in range [0.4 - 0.98]")
}
if s.WeightFactor < 0.0 || s.WeightFactor > 1.0 {
return errors.Error(processName, "jbig2 encoder weight factor not in range [0.0 - 1.0]")
}
if s.RankHaus < 0.5 || s.RankHaus > 1.0 {
return errors.Error(processName, "jbig2 encoder rank haus value not in range [0.5 - 1.0]")
}
if s.SizeHaus < 1 || s.SizeHaus > 10 {
return errors.Error(processName, "jbig2 encoder size haus value not in range [1 - 10]")
}
switch s.Components {
case bitmap.ComponentConn, bitmap.ComponentCharacters, bitmap.ComponentWords:
default:
return errors.Error(processName, "invalid classer component")
}
return nil
}