summaryrefslogtreecommitdiff
path: root/indexing/zhmakeindex/stroke_collator.go
blob: 43f87e47d9b04b485be1112e3a6aaca4985dc397 (plain)
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
package main

import (
	"strconv"
	"unicode"
	"unicode/utf8"

	"github.com/leo-liu/zhmakeindex/CJK"
)

// 汉字按笔画排序,汉字按笔画分组排在英文字母组后面
type StrokeIndexCollator struct{}

func (_ StrokeIndexCollator) InitGroups(style *OutputStyle) []IndexGroup {
	// 分组:符号、数字、字母 A..Z、笔划 1..MAX_STROKE
	groups := make([]IndexGroup, 2+26+CJK.MAX_STROKE)
	if style.headings_flag > 0 {
		groups[0].name = style.symhead_positive
		groups[1].name = style.numhead_positive
		for alph, i := 'A', 2; alph <= 'Z'; alph++ {
			groups[i].name = string(alph)
			i++
		}
	} else if style.headings_flag < 0 {
		groups[0].name = style.symhead_negative
		groups[1].name = style.numhead_negative
		for alph, i := 'a', 2; alph <= 'z'; alph++ {
			groups[i].name = string(alph)
			i++
		}
	}
	for stroke, i := 1, 2+26; stroke <= CJK.MAX_STROKE; stroke++ {
		groups[i].name = style.stroke_prefix + strconv.Itoa(stroke) + style.stroke_suffix
		i++
	}
	return groups
}

// 取得分组
func (_ StrokeIndexCollator) Group(entry *IndexEntry) int {
	first, _ := utf8.DecodeRuneInString(entry.level[0].key)
	first = unicode.ToLower(first)
	switch {
	case IsNumString(entry.level[0].key):
		return 1
	case 'a' <= first && first <= 'z':
		return 2 + int(first) - 'a'
	case len(CJK.Strokes[first]) > 0:
		return 2 + 26 + (len(CJK.Strokes[first]) - 1)
	default:
		// 符号组
		return 0
	}
}

// 按汉字笔画、笔顺序比较两个字符大小
// 笔画数不同的,短的在前;笔画数相同的,笔顺字典序;笔顺相同的,内码序
func (_ StrokeIndexCollator) RuneCmp(a, b rune) int {
	a_strokes, b_strokes := len(CJK.Strokes[a]), len(CJK.Strokes[b])
	switch {
	case a_strokes == 0 && b_strokes == 0:
		return RuneCmpIgnoreCases(a, b)
	case a_strokes == 0 && b_strokes != 0:
		return -1
	case a_strokes != 0 && b_strokes == 0:
		return 1
	case a_strokes != b_strokes:
		return a_strokes - b_strokes
	case CJK.Strokes[a] < CJK.Strokes[b]:
		return -1
	case CJK.Strokes[a] > CJK.Strokes[b]:
		return 1
	default:
		return int(a - b)
	}
}

// 判断是否字母或汉字
func (_ StrokeIndexCollator) IsLetter(r rune) bool {
	r = unicode.ToLower(r)
	switch {
	case 'a' <= r && r <= 'z':
		return true
	case CJK.Strokes[r] != "":
		return true
	default:
		return false
	}
}