File: generate_gen.go

package info (click to toggle)
golang-github-kshedden-dstream 0.0~git20190512.c4c4106-2
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 596 kB
  • sloc: makefile: 30
file content (130 lines) | stat: -rw-r--r-- 2,397 bytes parent folder | download
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
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
// GENERATED CODE, DO NOT EDIT

package dstream

import (
	"fmt"
	"time"
)

func (gen *generate) init() {

	for _, na := range gen.source.Names() {
		if gen.newvarname == na {
			msg := fmt.Sprintf("Generate: variable '%s' already exists.\n", gen.newvarname)
			panic(msg)
		}
	}

	gen.names = append(gen.source.Names(), gen.newvarname)
	gen.bdata = make([]interface{}, len(gen.names))

	switch gen.dtype {
	case String:
		gen.bdata[len(gen.bdata)-1] = make([]string, 0)

	case Time:
		gen.bdata[len(gen.bdata)-1] = make([]time.Time, 0)

	case Uint8:
		gen.bdata[len(gen.bdata)-1] = make([]uint8, 0)

	case Uint16:
		gen.bdata[len(gen.bdata)-1] = make([]uint16, 0)

	case Uint32:
		gen.bdata[len(gen.bdata)-1] = make([]uint32, 0)

	case Uint64:
		gen.bdata[len(gen.bdata)-1] = make([]uint64, 0)

	case Int8:
		gen.bdata[len(gen.bdata)-1] = make([]int8, 0)

	case Int16:
		gen.bdata[len(gen.bdata)-1] = make([]int16, 0)

	case Int32:
		gen.bdata[len(gen.bdata)-1] = make([]int32, 0)

	case Int64:
		gen.bdata[len(gen.bdata)-1] = make([]int64, 0)

	case Float32:
		gen.bdata[len(gen.bdata)-1] = make([]float32, 0)

	case Float64:
		gen.bdata[len(gen.bdata)-1] = make([]float64, 0)

	default:
		panic("Generate: unknown dtype")
	}
}

func (gen *generate) Next() bool {

	if !gen.source.Next() {
		return false
	}

	// All but new variable
	for j := 0; j < gen.source.NumVar(); j++ {
		gen.bdata[j] = gen.source.GetPos(j)
	}

	n := ilen(gen.GetPos(0))

	// The new variable goes in the last position
	q := len(gen.names) - 1

	switch x := gen.bdata[q].(type) {
	case []string:
		gen.bdata[q] = resizeString(x, n)

	case []time.Time:
		gen.bdata[q] = resizeTime(x, n)

	case []uint8:
		gen.bdata[q] = resizeUint8(x, n)

	case []uint16:
		gen.bdata[q] = resizeUint16(x, n)

	case []uint32:
		gen.bdata[q] = resizeUint32(x, n)

	case []uint64:
		gen.bdata[q] = resizeUint64(x, n)

	case []int8:
		gen.bdata[q] = resizeInt8(x, n)

	case []int16:
		gen.bdata[q] = resizeInt16(x, n)

	case []int32:
		gen.bdata[q] = resizeInt32(x, n)

	case []int64:
		gen.bdata[q] = resizeInt64(x, n)

	case []float32:
		gen.bdata[q] = resizeFloat32(x, n)

	case []float64:
		gen.bdata[q] = resizeFloat64(x, n)

	default:
		panic("unknown type")
	}

	mp := make(map[string]interface{})
	for k, na := range gen.names {
		if na != gen.newvarname {
			mp[na] = gen.bdata[k]
		}
	}

	gen.gfunc(mp, gen.bdata[q])
	return true
}