po file almost done

Author: youthlin

plurals/exp.go

@@ -11,7 +11,7 @@ import (
 )
 
 // Eval 传入复数表达式，返回 n 应该使用哪种复数形式
-func Eval(ctx context.Context, exp string, n int) (result int, err error) {
+func Eval(ctx context.Context, exp string, n int64) (result int64, err error) {
 	defer func() {
 		if e := recover(); e != nil {
 			err = errors.Errorf("unexpected error: %v", e)
@@ -43,17 +43,17 @@ const (
 type myListener struct {
 	// 实现这个父类上感兴趣的方法(相当于一个 adaptor)
 	*parser.BasepluralListener
-	stack  antlr.IntStack  // 操作数栈
+	stack  Int64Stack      // 操作数栈
 	ctx    context.Context // 外部传入的 ctx， 用于 debug
-	n      int             // 参数 n
-	result int             // 结果
+	n      int64           // 参数 n
+	result int64           // 结果
 }
 
-func newListener(ctx context.Context, n int) *myListener {
+func newListener(ctx context.Context, n int64) *myListener {
 	return &myListener{
 		ctx:   ctx,
 		n:     n,
-		stack: make(antlr.IntStack, 0),
+		stack: make(Int64Stack, 0),
 	}
 }
 
@@ -291,7 +291,7 @@ func (s *myListener) ExitPrimary(ctx *parser.PrimaryContext) {
 		if err != nil {
 			panic("assert error: not a number: " + num)
 		}
-		s.stack.Push(int(iNum))
+		s.stack.Push(iNum)
 	}
 	s.debug("primary: %v stack=%v", ctx.GetText(), s.stack)
 }

plurals/exp_test.go

@@ -13,12 +13,12 @@ var ctx = context.Background()
 type args struct {
 	ctx context.Context
 	exp string
-	n   int
+	n   int64
 }
 type tCase struct {
 	name    string
 	args    args
-	want    int
+	want    int64
 	wantErr bool
 }
 
@@ -46,11 +46,11 @@ func okCase() []tCase {
 	}
 
 	// 拉脱维亚语 Latvian 三种复数，0特殊
-	for i := 0; i < 215; i++ {
+	for i := int64(0); i < 215; i++ {
 		tests = append(tests,
 			tCase{fmt.Sprintf("test-case-3-#%v", i),
 				args{ctx, "n%10==1 && n%100!=11 ? 0 : n != 0 ? 1 : 2", i},
-				func(n int) int {
+				func(n int64) int64 {
 					// n%10==1 && n%100!=11 ? 0 : n != 0 ? 1 : 2
 					if n%10 == 1 && n%100 != 11 {
 						return 0
@@ -63,11 +63,11 @@ func okCase() []tCase {
 		)
 	}
 	// 三种复数，00 [2-9][0-9] Romanian 罗马尼亚
-	for i := 0; i < 215; i++ {
+	for i := int64(0); i < 215; i++ {
 		tests = append(tests,
 			tCase{fmt.Sprintf("test-case-4-#%v", i),
 				args{ctx, "n==1 ? 0 : (n==0 || (n%100 > 0 && n%100 < 20)) ? 1 : 2", i},
-				func(n int) int {
+				func(n int64) int64 {
 					// n==1 ? 0 : (n==0 || (n%100 > 0 && n%100 < 20)) ? 1 : 2
 					if n == 1 {
 						return 0
@@ -80,11 +80,11 @@ func okCase() []tCase {
 		)
 	}
 	// 三种复数，1[2-9] 立陶宛 Lithuanian
-	for i := 0; i < 215; i++ {
+	for i := int64(0); i < 215; i++ {
 		tests = append(tests,
 			tCase{fmt.Sprintf("test-case-5-#%v", i),
 				args{ctx, "n%10==1 && n%100!=11 ? 0 : n%10>=2 && (n%100<10 || n%100>=20) ? 1 : 2", i},
-				func(n int) int {
+				func(n int64) int64 {
 					// n%10==1 && n%100!=11 ? 0 : n%10>=2 && (n%100<10 || n%100>=20) ? 1 : 2
 					if n%10 == 1 && n%100 != 11 {
 						return 0
@@ -97,11 +97,11 @@ func okCase() []tCase {
 		)
 	}
 	// 三种复数，1234结尾的特殊，但不是 1[1-4] Russian, Ukrainian, Belarusian, Serbian, Croatian
-	for i := 0; i < 215; i++ {
+	for i := int64(0); i < 215; i++ {
 		tests = append(tests,
 			tCase{fmt.Sprintf("test-case-6-#%v", i),
 				args{ctx, "n%10==1 && n%100!=11 ? 0 : n%10>=2 && n%10<=4 && (n%100<10 || n%100>=20) ? 1 : 2", i},
-				func(n int) int {
+				func(n int64) int64 {
 					// n%10==1 && n%100!=11 ? 0 : n%10>=2 && n%10<=4 && (n%100<10 || n%100>=20) ? 1 : 2
 					if n%10 == 1 && n%100 != 11 {
 						return 0
@@ -114,11 +114,11 @@ func okCase() []tCase {
 		)
 	}
 	// 三种复数，1234结尾的特殊 Czech, Slovak
-	for i := 0; i < 215; i++ {
+	for i := int64(0); i < 215; i++ {
 		tests = append(tests,
 			tCase{fmt.Sprintf("test-case-7-#%v", i),
 				args{ctx, "(n==1) ? 0 : (n>=2 && n<=4) ? 1 : 2", i},
-				func(n int) int {
+				func(n int64) int64 {
 					// (n==1) ? 0 : (n>=2 && n<=4) ? 1 : 2
 					if n == 1 {
 						return 0
@@ -132,11 +132,11 @@ func okCase() []tCase {
 	}
 	// 三种复数，Three forms, special case for one and some numbers ending in 2, 3, or 4
 	// Polish
-	for i := 0; i < 215; i++ {
+	for i := int64(0); i < 215; i++ {
 		tests = append(tests,
 			tCase{fmt.Sprintf("test-case-8-#%v", i),
 				args{ctx, "n==1 ? 0 :  n%10>=2 && n%10<=4 && (n%100<10 || n%100>=20) ? 1 : 2", i},
-				func(n int) int {
+				func(n int64) int64 {
 					// n==1 ? 0 : n%10>=2 && n%10<=4 && (n%100<10 || n%100>=20) ? 1 : 2
 					if n == 1 {
 						return 0
@@ -150,11 +150,11 @@ func okCase() []tCase {
 	}
 	// Four forms, special case for one and all numbers ending in 02, 03, or 04
 	// Slovenian
-	for i := 0; i < 215; i++ {
+	for i := int64(0); i < 215; i++ {
 		tests = append(tests,
 			tCase{fmt.Sprintf("test-case-9-#%v", i),
 				args{ctx, "n%100==1 ? 0 : n%100==2 ? 1 : n%100==3 || n%100==4 ? 2 : 3", i},
-				func(n int) int {
+				func(n int64) int64 {
 					//n%100==1 ? 0 : n%100==2 ? 1 : n%100==3 || n%100==4 ? 2 : 3
 					if n%100 == 1 {
 						return 0
@@ -171,11 +171,11 @@ func okCase() []tCase {
 	}
 	// Six forms, special cases for one, two, all numbers ending in 02, 03, … 10, all numbers ending in 11 … 99, and others
 	// Arabic
-	for i := 0; i < 215; i++ {
+	for i := int64(0); i < 215; i++ {
 		tests = append(tests,
 			tCase{fmt.Sprintf("test-case-10-#%v", i),
 				args{ctx, " n==0 ? 0 : n==1 ? 1 : n==2 ? 2 : n%100>=3 && n%100<=10 ? 3  : n%100>=11 ? 4 : 5", i},
-				func(n int) int {
+				func(n int64) int64 {
 					if n == 0 {
 						return 0
 					}

plurals/stack.go

@@ -0,0 +1,24 @@
+package plurals
+
+import "errors"
+
+// A simple integer stack
+
+// see antlr.IntStack
+type Int64Stack []int64
+
+var ErrEmptyStack = errors.New("Stack is empty")
+
+func (s *Int64Stack) Pop() (int64, error) {
+	l := len(*s) - 1
+	if l < 0 {
+		return 0, ErrEmptyStack
+	}
+	v := (*s)[l]
+	*s = (*s)[0:l]
+	return v, nil
+}
+
+func (s *Int64Stack) Push(e int64) {
+	*s = append(*s, e)
+}