Initial commit

generators/modernportuguese.go

@@ -0,0 +1,75 @@
+package generators
+
+import (
+	"encoding/json"
+	"math/rand"
+	"os"
+)
+
+type ModernPortugueseEUArrays struct {
+	Nm1 []string `json:"nm1"`
+	Nm2 []string `json:"nm2"`
+	Nm3 []string `json:"nm3"`
+	Nm4 []string `json:"nm4"`
+	Nm5 []string `json:"nm5"`
+	Nm6 []string `json:"nm6"`
+	Nm7 []string `json:"nm7"`
+}
+
+func LoadModernPortugueseEUArrays(path string) (ModernPortugueseEUArrays, error) {
+	var arr ModernPortugueseEUArrays
+	f, err := os.Open(path)
+	if err != nil {
+		return arr, err
+	}
+	defer f.Close()
+	dec := json.NewDecoder(f)
+	if err := dec.Decode(&arr); err != nil {
+		return arr, err
+	}
+	return arr, nil
+}
+
+// type: 0=male, 1=female, 2=unisex
+func GenerateModernPortugueseEUNames(arr *ModernPortugueseEUArrays, swear func(string) bool, t int) []string {
+	var names []string
+	for i := 0; i < 10; i++ {
+		var name string
+		switch t {
+		case 1:
+			if i < 5 {
+				rnd := rand.Intn(len(arr.Nm2))
+				rnd2 := rand.Intn(len(arr.Nm3))
+				name = arr.Nm2[rnd] + " " + arr.Nm3[rnd2]
+			} else {
+				rnd := rand.Intn(len(arr.Nm5))
+				rnd2 := rand.Intn(len(arr.Nm6))
+				name = arr.Nm5[rnd] + " " + arr.Nm6[rnd2]
+			}
+		case 2:
+			if i < 5 {
+				rnd := rand.Intn(len(arr.Nm7))
+				rnd2 := rand.Intn(len(arr.Nm3))
+				name = arr.Nm7[rnd] + " " + arr.Nm3[rnd2]
+			} else {
+				rnd := rand.Intn(len(arr.Nm7))
+				rnd2 := rand.Intn(len(arr.Nm6))
+				name = arr.Nm7[rnd] + " " + arr.Nm6[rnd2]
+			}
+		default:
+			if i < 5 {
+				rnd := rand.Intn(len(arr.Nm1))
+				rnd2 := rand.Intn(len(arr.Nm3))
+				name = arr.Nm1[rnd] + " " + arr.Nm3[rnd2]
+			} else {
+				rnd := rand.Intn(len(arr.Nm4))
+				rnd2 := rand.Intn(len(arr.Nm6))
+				name = arr.Nm4[rnd] + " " + arr.Nm6[rnd2]
+			}
+		}
+		if !swear(name) {
+			names = append(names, name)
+		}
+	}
+	return names
+}