-
Notifications
You must be signed in to change notification settings - Fork 22
/
augment.go
103 lines (95 loc) · 3.19 KB
/
augment.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
92
93
94
95
96
97
98
99
100
101
102
103
//
// Copyright © 2018 Aljabr, Inc.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
//
package symbol
import (
"github.com/kocircuit/kocircuit/lang/circuit/eval"
"github.com/kocircuit/kocircuit/lang/circuit/model"
"github.com/kocircuit/kocircuit/lang/go/kit/tree"
)
func (vty *VarietySymbol) Augment(span *model.Span, fields eval.Fields) (eval.Shape, eval.Effect, error) {
augmented, err := GroupFieldsToSymbols(span, fields)
if err != nil {
return nil, nil, err
}
aggregate := append(append(FieldSymbols{}, vty.Arg...), augmented...)
if err = VerifyNoDuplicateFieldSymbol(span, aggregate); err != nil {
return nil, nil, span.Errorf(err, "augmenting %s", tree.Sprint(vty))
}
return MakeVarietySymbol(vty.Macro, aggregate), nil, nil
}
func GroupFieldsToSymbols(span *model.Span, fields eval.Fields) (FieldSymbols, error) {
evalFields := FieldSymbols{}
fields = FilterUnderscoreFields(fields)
for _, fieldGroup := range fields.FieldGroup() {
groupName := fieldGroup[0].Name
fieldGroup = FilterEmptyEvalFields(fieldGroup)
switch len(fieldGroup) {
case 0: // add a field with an empty symbol value (useful to All macro)
evalFields = append(evalFields,
&FieldSymbol{Name: groupName, Monadic: groupName == "", Value: EmptySymbol{}},
)
case 1:
y := fieldGroup[0].Shape.(Symbol)
evalFields = append(evalFields,
&FieldSymbol{Name: groupName, Monadic: groupName == "", Value: y},
)
default:
repeatedSymbols := make(Symbols, len(fieldGroup))
for i, f := range fieldGroup {
repeatedSymbols[i] = f.Shape.(Symbol)
}
if repeated, err := MakeSeriesSymbol(span, repeatedSymbols); err != nil {
return nil, span.Errorf(err, "repeated field group %s", groupName)
} else {
evalFields = append(evalFields,
&FieldSymbol{Name: groupName, Monadic: groupName == "", Value: repeated},
)
}
}
}
return evalFields, nil
}
// FilterUnderscoreFields filters out all fields whose name begins with underscore.
func FilterUnderscoreFields(fields []eval.Field) (filtered []eval.Field) {
for _, field := range fields {
if !beginsWithUnderscore(field.Name) {
filtered = append(filtered, field)
}
}
return
}
func beginsWithUnderscore(n string) bool {
return len(n) > 0 && n[0] == '_'
}
func FilterEmptyEvalFields(group []eval.Field) (filtered []eval.Field) {
for _, field := range group {
if !IsEmptySymbol(field.Shape.(Symbol)) {
filtered = append(filtered, field)
}
}
return
}
func VerifyNoDuplicateFieldSymbol(span *model.Span, fieldSymbols FieldSymbols) error {
seen := map[string]bool{}
for _, f := range fieldSymbols {
if seen[f.Name] {
return span.Errorf(nil, "augmenting duplicate field %s", f.Name)
} else {
seen[f.Name] = true
}
}
return nil
}