mirror of
https://codeberg.org/forgejo/forgejo.git
synced 2024-11-23 08:47:42 -05:00
311 lines
9.2 KiB
Go
311 lines
9.2 KiB
Go
|
package qr
|
||
|
|
||
|
import "math"
|
||
|
|
||
|
// ErrorCorrectionLevel indicates the amount of "backup data" stored in the QR code
|
||
|
type ErrorCorrectionLevel byte
|
||
|
|
||
|
const (
|
||
|
// L recovers 7% of data
|
||
|
L ErrorCorrectionLevel = iota
|
||
|
// M recovers 15% of data
|
||
|
M
|
||
|
// Q recovers 25% of data
|
||
|
Q
|
||
|
// H recovers 30% of data
|
||
|
H
|
||
|
)
|
||
|
|
||
|
func (ecl ErrorCorrectionLevel) String() string {
|
||
|
switch ecl {
|
||
|
case L:
|
||
|
return "L"
|
||
|
case M:
|
||
|
return "M"
|
||
|
case Q:
|
||
|
return "Q"
|
||
|
case H:
|
||
|
return "H"
|
||
|
}
|
||
|
return "unknown"
|
||
|
}
|
||
|
|
||
|
type encodingMode byte
|
||
|
|
||
|
const (
|
||
|
numericMode encodingMode = 1
|
||
|
alphaNumericMode encodingMode = 2
|
||
|
byteMode encodingMode = 4
|
||
|
kanjiMode encodingMode = 8
|
||
|
)
|
||
|
|
||
|
type versionInfo struct {
|
||
|
Version byte
|
||
|
Level ErrorCorrectionLevel
|
||
|
ErrorCorrectionCodewordsPerBlock byte
|
||
|
NumberOfBlocksInGroup1 byte
|
||
|
DataCodeWordsPerBlockInGroup1 byte
|
||
|
NumberOfBlocksInGroup2 byte
|
||
|
DataCodeWordsPerBlockInGroup2 byte
|
||
|
}
|
||
|
|
||
|
var versionInfos = []*versionInfo{
|
||
|
&versionInfo{1, L, 7, 1, 19, 0, 0},
|
||
|
&versionInfo{1, M, 10, 1, 16, 0, 0},
|
||
|
&versionInfo{1, Q, 13, 1, 13, 0, 0},
|
||
|
&versionInfo{1, H, 17, 1, 9, 0, 0},
|
||
|
&versionInfo{2, L, 10, 1, 34, 0, 0},
|
||
|
&versionInfo{2, M, 16, 1, 28, 0, 0},
|
||
|
&versionInfo{2, Q, 22, 1, 22, 0, 0},
|
||
|
&versionInfo{2, H, 28, 1, 16, 0, 0},
|
||
|
&versionInfo{3, L, 15, 1, 55, 0, 0},
|
||
|
&versionInfo{3, M, 26, 1, 44, 0, 0},
|
||
|
&versionInfo{3, Q, 18, 2, 17, 0, 0},
|
||
|
&versionInfo{3, H, 22, 2, 13, 0, 0},
|
||
|
&versionInfo{4, L, 20, 1, 80, 0, 0},
|
||
|
&versionInfo{4, M, 18, 2, 32, 0, 0},
|
||
|
&versionInfo{4, Q, 26, 2, 24, 0, 0},
|
||
|
&versionInfo{4, H, 16, 4, 9, 0, 0},
|
||
|
&versionInfo{5, L, 26, 1, 108, 0, 0},
|
||
|
&versionInfo{5, M, 24, 2, 43, 0, 0},
|
||
|
&versionInfo{5, Q, 18, 2, 15, 2, 16},
|
||
|
&versionInfo{5, H, 22, 2, 11, 2, 12},
|
||
|
&versionInfo{6, L, 18, 2, 68, 0, 0},
|
||
|
&versionInfo{6, M, 16, 4, 27, 0, 0},
|
||
|
&versionInfo{6, Q, 24, 4, 19, 0, 0},
|
||
|
&versionInfo{6, H, 28, 4, 15, 0, 0},
|
||
|
&versionInfo{7, L, 20, 2, 78, 0, 0},
|
||
|
&versionInfo{7, M, 18, 4, 31, 0, 0},
|
||
|
&versionInfo{7, Q, 18, 2, 14, 4, 15},
|
||
|
&versionInfo{7, H, 26, 4, 13, 1, 14},
|
||
|
&versionInfo{8, L, 24, 2, 97, 0, 0},
|
||
|
&versionInfo{8, M, 22, 2, 38, 2, 39},
|
||
|
&versionInfo{8, Q, 22, 4, 18, 2, 19},
|
||
|
&versionInfo{8, H, 26, 4, 14, 2, 15},
|
||
|
&versionInfo{9, L, 30, 2, 116, 0, 0},
|
||
|
&versionInfo{9, M, 22, 3, 36, 2, 37},
|
||
|
&versionInfo{9, Q, 20, 4, 16, 4, 17},
|
||
|
&versionInfo{9, H, 24, 4, 12, 4, 13},
|
||
|
&versionInfo{10, L, 18, 2, 68, 2, 69},
|
||
|
&versionInfo{10, M, 26, 4, 43, 1, 44},
|
||
|
&versionInfo{10, Q, 24, 6, 19, 2, 20},
|
||
|
&versionInfo{10, H, 28, 6, 15, 2, 16},
|
||
|
&versionInfo{11, L, 20, 4, 81, 0, 0},
|
||
|
&versionInfo{11, M, 30, 1, 50, 4, 51},
|
||
|
&versionInfo{11, Q, 28, 4, 22, 4, 23},
|
||
|
&versionInfo{11, H, 24, 3, 12, 8, 13},
|
||
|
&versionInfo{12, L, 24, 2, 92, 2, 93},
|
||
|
&versionInfo{12, M, 22, 6, 36, 2, 37},
|
||
|
&versionInfo{12, Q, 26, 4, 20, 6, 21},
|
||
|
&versionInfo{12, H, 28, 7, 14, 4, 15},
|
||
|
&versionInfo{13, L, 26, 4, 107, 0, 0},
|
||
|
&versionInfo{13, M, 22, 8, 37, 1, 38},
|
||
|
&versionInfo{13, Q, 24, 8, 20, 4, 21},
|
||
|
&versionInfo{13, H, 22, 12, 11, 4, 12},
|
||
|
&versionInfo{14, L, 30, 3, 115, 1, 116},
|
||
|
&versionInfo{14, M, 24, 4, 40, 5, 41},
|
||
|
&versionInfo{14, Q, 20, 11, 16, 5, 17},
|
||
|
&versionInfo{14, H, 24, 11, 12, 5, 13},
|
||
|
&versionInfo{15, L, 22, 5, 87, 1, 88},
|
||
|
&versionInfo{15, M, 24, 5, 41, 5, 42},
|
||
|
&versionInfo{15, Q, 30, 5, 24, 7, 25},
|
||
|
&versionInfo{15, H, 24, 11, 12, 7, 13},
|
||
|
&versionInfo{16, L, 24, 5, 98, 1, 99},
|
||
|
&versionInfo{16, M, 28, 7, 45, 3, 46},
|
||
|
&versionInfo{16, Q, 24, 15, 19, 2, 20},
|
||
|
&versionInfo{16, H, 30, 3, 15, 13, 16},
|
||
|
&versionInfo{17, L, 28, 1, 107, 5, 108},
|
||
|
&versionInfo{17, M, 28, 10, 46, 1, 47},
|
||
|
&versionInfo{17, Q, 28, 1, 22, 15, 23},
|
||
|
&versionInfo{17, H, 28, 2, 14, 17, 15},
|
||
|
&versionInfo{18, L, 30, 5, 120, 1, 121},
|
||
|
&versionInfo{18, M, 26, 9, 43, 4, 44},
|
||
|
&versionInfo{18, Q, 28, 17, 22, 1, 23},
|
||
|
&versionInfo{18, H, 28, 2, 14, 19, 15},
|
||
|
&versionInfo{19, L, 28, 3, 113, 4, 114},
|
||
|
&versionInfo{19, M, 26, 3, 44, 11, 45},
|
||
|
&versionInfo{19, Q, 26, 17, 21, 4, 22},
|
||
|
&versionInfo{19, H, 26, 9, 13, 16, 14},
|
||
|
&versionInfo{20, L, 28, 3, 107, 5, 108},
|
||
|
&versionInfo{20, M, 26, 3, 41, 13, 42},
|
||
|
&versionInfo{20, Q, 30, 15, 24, 5, 25},
|
||
|
&versionInfo{20, H, 28, 15, 15, 10, 16},
|
||
|
&versionInfo{21, L, 28, 4, 116, 4, 117},
|
||
|
&versionInfo{21, M, 26, 17, 42, 0, 0},
|
||
|
&versionInfo{21, Q, 28, 17, 22, 6, 23},
|
||
|
&versionInfo{21, H, 30, 19, 16, 6, 17},
|
||
|
&versionInfo{22, L, 28, 2, 111, 7, 112},
|
||
|
&versionInfo{22, M, 28, 17, 46, 0, 0},
|
||
|
&versionInfo{22, Q, 30, 7, 24, 16, 25},
|
||
|
&versionInfo{22, H, 24, 34, 13, 0, 0},
|
||
|
&versionInfo{23, L, 30, 4, 121, 5, 122},
|
||
|
&versionInfo{23, M, 28, 4, 47, 14, 48},
|
||
|
&versionInfo{23, Q, 30, 11, 24, 14, 25},
|
||
|
&versionInfo{23, H, 30, 16, 15, 14, 16},
|
||
|
&versionInfo{24, L, 30, 6, 117, 4, 118},
|
||
|
&versionInfo{24, M, 28, 6, 45, 14, 46},
|
||
|
&versionInfo{24, Q, 30, 11, 24, 16, 25},
|
||
|
&versionInfo{24, H, 30, 30, 16, 2, 17},
|
||
|
&versionInfo{25, L, 26, 8, 106, 4, 107},
|
||
|
&versionInfo{25, M, 28, 8, 47, 13, 48},
|
||
|
&versionInfo{25, Q, 30, 7, 24, 22, 25},
|
||
|
&versionInfo{25, H, 30, 22, 15, 13, 16},
|
||
|
&versionInfo{26, L, 28, 10, 114, 2, 115},
|
||
|
&versionInfo{26, M, 28, 19, 46, 4, 47},
|
||
|
&versionInfo{26, Q, 28, 28, 22, 6, 23},
|
||
|
&versionInfo{26, H, 30, 33, 16, 4, 17},
|
||
|
&versionInfo{27, L, 30, 8, 122, 4, 123},
|
||
|
&versionInfo{27, M, 28, 22, 45, 3, 46},
|
||
|
&versionInfo{27, Q, 30, 8, 23, 26, 24},
|
||
|
&versionInfo{27, H, 30, 12, 15, 28, 16},
|
||
|
&versionInfo{28, L, 30, 3, 117, 10, 118},
|
||
|
&versionInfo{28, M, 28, 3, 45, 23, 46},
|
||
|
&versionInfo{28, Q, 30, 4, 24, 31, 25},
|
||
|
&versionInfo{28, H, 30, 11, 15, 31, 16},
|
||
|
&versionInfo{29, L, 30, 7, 116, 7, 117},
|
||
|
&versionInfo{29, M, 28, 21, 45, 7, 46},
|
||
|
&versionInfo{29, Q, 30, 1, 23, 37, 24},
|
||
|
&versionInfo{29, H, 30, 19, 15, 26, 16},
|
||
|
&versionInfo{30, L, 30, 5, 115, 10, 116},
|
||
|
&versionInfo{30, M, 28, 19, 47, 10, 48},
|
||
|
&versionInfo{30, Q, 30, 15, 24, 25, 25},
|
||
|
&versionInfo{30, H, 30, 23, 15, 25, 16},
|
||
|
&versionInfo{31, L, 30, 13, 115, 3, 116},
|
||
|
&versionInfo{31, M, 28, 2, 46, 29, 47},
|
||
|
&versionInfo{31, Q, 30, 42, 24, 1, 25},
|
||
|
&versionInfo{31, H, 30, 23, 15, 28, 16},
|
||
|
&versionInfo{32, L, 30, 17, 115, 0, 0},
|
||
|
&versionInfo{32, M, 28, 10, 46, 23, 47},
|
||
|
&versionInfo{32, Q, 30, 10, 24, 35, 25},
|
||
|
&versionInfo{32, H, 30, 19, 15, 35, 16},
|
||
|
&versionInfo{33, L, 30, 17, 115, 1, 116},
|
||
|
&versionInfo{33, M, 28, 14, 46, 21, 47},
|
||
|
&versionInfo{33, Q, 30, 29, 24, 19, 25},
|
||
|
&versionInfo{33, H, 30, 11, 15, 46, 16},
|
||
|
&versionInfo{34, L, 30, 13, 115, 6, 116},
|
||
|
&versionInfo{34, M, 28, 14, 46, 23, 47},
|
||
|
&versionInfo{34, Q, 30, 44, 24, 7, 25},
|
||
|
&versionInfo{34, H, 30, 59, 16, 1, 17},
|
||
|
&versionInfo{35, L, 30, 12, 121, 7, 122},
|
||
|
&versionInfo{35, M, 28, 12, 47, 26, 48},
|
||
|
&versionInfo{35, Q, 30, 39, 24, 14, 25},
|
||
|
&versionInfo{35, H, 30, 22, 15, 41, 16},
|
||
|
&versionInfo{36, L, 30, 6, 121, 14, 122},
|
||
|
&versionInfo{36, M, 28, 6, 47, 34, 48},
|
||
|
&versionInfo{36, Q, 30, 46, 24, 10, 25},
|
||
|
&versionInfo{36, H, 30, 2, 15, 64, 16},
|
||
|
&versionInfo{37, L, 30, 17, 122, 4, 123},
|
||
|
&versionInfo{37, M, 28, 29, 46, 14, 47},
|
||
|
&versionInfo{37, Q, 30, 49, 24, 10, 25},
|
||
|
&versionInfo{37, H, 30, 24, 15, 46, 16},
|
||
|
&versionInfo{38, L, 30, 4, 122, 18, 123},
|
||
|
&versionInfo{38, M, 28, 13, 46, 32, 47},
|
||
|
&versionInfo{38, Q, 30, 48, 24, 14, 25},
|
||
|
&versionInfo{38, H, 30, 42, 15, 32, 16},
|
||
|
&versionInfo{39, L, 30, 20, 117, 4, 118},
|
||
|
&versionInfo{39, M, 28, 40, 47, 7, 48},
|
||
|
&versionInfo{39, Q, 30, 43, 24, 22, 25},
|
||
|
&versionInfo{39, H, 30, 10, 15, 67, 16},
|
||
|
&versionInfo{40, L, 30, 19, 118, 6, 119},
|
||
|
&versionInfo{40, M, 28, 18, 47, 31, 48},
|
||
|
&versionInfo{40, Q, 30, 34, 24, 34, 25},
|
||
|
&versionInfo{40, H, 30, 20, 15, 61, 16},
|
||
|
}
|
||
|
|
||
|
func (vi *versionInfo) totalDataBytes() int {
|
||
|
g1Data := int(vi.NumberOfBlocksInGroup1) * int(vi.DataCodeWordsPerBlockInGroup1)
|
||
|
g2Data := int(vi.NumberOfBlocksInGroup2) * int(vi.DataCodeWordsPerBlockInGroup2)
|
||
|
return (g1Data + g2Data)
|
||
|
}
|
||
|
|
||
|
func (vi *versionInfo) charCountBits(m encodingMode) byte {
|
||
|
switch m {
|
||
|
case numericMode:
|
||
|
if vi.Version < 10 {
|
||
|
return 10
|
||
|
} else if vi.Version < 27 {
|
||
|
return 12
|
||
|
}
|
||
|
return 14
|
||
|
|
||
|
case alphaNumericMode:
|
||
|
if vi.Version < 10 {
|
||
|
return 9
|
||
|
} else if vi.Version < 27 {
|
||
|
return 11
|
||
|
}
|
||
|
return 13
|
||
|
|
||
|
case byteMode:
|
||
|
if vi.Version < 10 {
|
||
|
return 8
|
||
|
}
|
||
|
return 16
|
||
|
|
||
|
case kanjiMode:
|
||
|
if vi.Version < 10 {
|
||
|
return 8
|
||
|
} else if vi.Version < 27 {
|
||
|
return 10
|
||
|
}
|
||
|
return 12
|
||
|
default:
|
||
|
return 0
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func (vi *versionInfo) modulWidth() int {
|
||
|
return ((int(vi.Version) - 1) * 4) + 21
|
||
|
}
|
||
|
|
||
|
func (vi *versionInfo) alignmentPatternPlacements() []int {
|
||
|
if vi.Version == 1 {
|
||
|
return make([]int, 0)
|
||
|
}
|
||
|
|
||
|
first := 6
|
||
|
last := vi.modulWidth() - 7
|
||
|
space := float64(last - first)
|
||
|
count := int(math.Ceil(space/28)) + 1
|
||
|
|
||
|
result := make([]int, count)
|
||
|
result[0] = first
|
||
|
result[len(result)-1] = last
|
||
|
if count > 2 {
|
||
|
step := int(math.Ceil(float64(last-first) / float64(count-1)))
|
||
|
if step%2 == 1 {
|
||
|
frac := float64(last-first) / float64(count-1)
|
||
|
_, x := math.Modf(frac)
|
||
|
if x >= 0.5 {
|
||
|
frac = math.Ceil(frac)
|
||
|
} else {
|
||
|
frac = math.Floor(frac)
|
||
|
}
|
||
|
|
||
|
if int(frac)%2 == 0 {
|
||
|
step--
|
||
|
} else {
|
||
|
step++
|
||
|
}
|
||
|
}
|
||
|
|
||
|
for i := 1; i <= count-2; i++ {
|
||
|
result[i] = last - (step * (count - 1 - i))
|
||
|
}
|
||
|
}
|
||
|
|
||
|
return result
|
||
|
}
|
||
|
|
||
|
func findSmallestVersionInfo(ecl ErrorCorrectionLevel, mode encodingMode, dataBits int) *versionInfo {
|
||
|
dataBits = dataBits + 4 // mode indicator
|
||
|
for _, vi := range versionInfos {
|
||
|
if vi.Level == ecl {
|
||
|
if (vi.totalDataBytes() * 8) >= (dataBits + int(vi.charCountBits(mode))) {
|
||
|
return vi
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
return nil
|
||
|
}
|