|
33
|
1 /*
|
|
|
2 fx.go: make MA/RSI/BB-Band graph
|
|
|
3 */
|
|
31
|
4 package main
|
|
|
5
|
|
|
6 import (
|
|
|
7 "bytes"
|
|
|
8 "encoding/csv"
|
|
|
9 "fmt"
|
|
|
10 "io"
|
|
|
11 "io/ioutil"
|
|
|
12 "log"
|
|
|
13 "math"
|
|
|
14 "strconv"
|
|
|
15 "strings"
|
|
|
16 )
|
|
|
17
|
|
|
18 const debug = false
|
|
33
|
19 const ratefile = "./fx_result.csv"
|
|
31
|
20
|
|
|
21 func main() {
|
|
|
22 n, m := 20, 14
|
|
|
23 var x, xl, xh []float64
|
|
|
24 var date []string
|
|
|
25
|
|
|
26 // Read Exchange-data
|
|
33
|
27 content, err := ioutil.ReadFile(ratefile)
|
|
31
|
28 if err != nil {
|
|
|
29 log.Fatal(err)
|
|
|
30 }
|
|
|
31
|
|
|
32 r := csv.NewReader(bytes.NewReader(content))
|
|
33
|
33 r.Comment = '#'
|
|
31
|
34 for {
|
|
|
35 record, err := r.Read()
|
|
|
36 if err == io.EOF {
|
|
|
37 break
|
|
|
38 }
|
|
|
39 if err != nil {
|
|
|
40 log.Fatal(err)
|
|
|
41 }
|
|
|
42
|
|
|
43 f, err := strconv.ParseFloat(record[4], 64)
|
|
|
44 if err != nil {
|
|
|
45 continue
|
|
|
46 }
|
|
|
47 x = append(x, f)
|
|
|
48
|
|
|
49 f, err = strconv.ParseFloat(record[3], 64)
|
|
|
50 xl = append(xl, f)
|
|
|
51
|
|
|
52 f, err = strconv.ParseFloat(record[2], 64)
|
|
|
53 xh = append(xh, f)
|
|
|
54
|
|
33
|
55 date = append(date, record[0])
|
|
31
|
56 }
|
|
|
57
|
|
|
58 // Calc Moving-Average
|
|
|
59 xa := make([]float64, len(x))
|
|
|
60 for i := n; i < len(x); i++ {
|
|
|
61 for j := 0; j < n; j++ {
|
|
|
62 xa[i] += x[i-j]
|
|
|
63 }
|
|
|
64 xa[i] /= float64(n)
|
|
|
65 }
|
|
|
66
|
|
|
67 // Calc Variance & SD
|
|
|
68 v := make([]float64, len(x))
|
|
|
69 sd := make([]float64, len(x))
|
|
|
70 for i := n; i < len(x); i++ {
|
|
|
71 // variance
|
|
|
72 for j := 0; j < n; j++ {
|
|
|
73 v[i] += (x[i-j] - xa[i-j]) * (x[i-j] - xa[i-j])
|
|
|
74 }
|
|
|
75 v[i] /= float64(n)
|
|
|
76
|
|
|
77 // standard deviation
|
|
|
78 sd[i] = math.Sqrt(v[i])
|
|
|
79 }
|
|
|
80
|
|
|
81 // Calc Bollinger-Bands
|
|
|
82 bb1 := make([]float64, len(x))
|
|
|
83 bb2 := make([]float64, len(x))
|
|
|
84 for i := n * 2; i < len(x); i++ {
|
|
|
85 bb1[i], bb2[i] = x[i] - sd[i], x[i] + sd[i]
|
|
|
86 }
|
|
|
87
|
|
|
88 // Calc RSI
|
|
|
89 rsi := make([]float64, len(x))
|
|
|
90 for i := m; i < len(x); i++ {
|
|
|
91 var ru, rd float64
|
|
|
92 for j := 0; j < m; j++ {
|
|
|
93 z := x[i-j-1] - x[i-j]
|
|
|
94 if z > 0 {
|
|
|
95 ru += z
|
|
|
96 } else {
|
|
|
97 rd += -z
|
|
|
98 }
|
|
|
99 }
|
|
|
100 rsi[i] = rd / (rd + ru) * 100
|
|
|
101 }
|
|
|
102
|
|
|
103 // Make Graph
|
|
|
104 // [chart.svg]
|
|
|
105 w, h := 800, 600
|
|
|
106
|
|
|
107 term := 200
|
|
|
108 min, max := 200.0, 1.0
|
|
|
109 for i := len(x) - term; i < len(x); i++ {
|
|
|
110 min = math.Min(min, x[i])
|
|
|
111 max = math.Max(max, x[i])
|
|
|
112 min = math.Min(min, bb1[i])
|
|
|
113 max = math.Max(max, bb2[i])
|
|
|
114 }
|
|
|
115 sx := float64(w) / float64(term)
|
|
|
116 sy := float64(h) / (max - min)
|
|
|
117
|
|
|
118 svg := fmt.Sprintf("<svg xmlns='http://www.w3.org/2000/svg' width='%d' height='%d'>\n", w, h)
|
|
|
119 for y := int(min); y < int(max) + 1; y++ {
|
|
|
120 svg += make_line(h, 0, float64(y), w, float64(y), min, 1.0, sy, "gray", 0.2)
|
|
|
121 svg += make_label(h, 10, y, min, 1.0, sy, fmt.Sprintf("%d", y))
|
|
|
122 }
|
|
|
123 for i, j, lm := len(x) - term, 0, ""; i < len(x); i, j = i + 1, j + 1 {
|
|
|
124 if !strings.EqualFold(lm, date[i][5:7]) {
|
|
|
125 svg += make_line(h, j, 0.0, j, float64(h), 0, sx, 1.0, "gray", 0.2)
|
|
|
126 svg += make_label(h, j, 20, 0, sx, 1.0, date[i][0:7])
|
|
|
127 lm = date[i][5:7]
|
|
|
128 }
|
|
|
129 }
|
|
|
130
|
|
|
131 for i, j := len(x) - term, 1; i < len(x); i, j = i + 1, j + 1 {
|
|
|
132 svg += make_line(h, j-1, x[i-1], j, x[i], min, sx, sy, "black", 1.5)
|
|
|
133 svg += make_line(h, j-1, xa[i-1], j, xa[i], min, sx, sy, "green", 0.7)
|
|
|
134 svg += make_line(h, j-1, bb1[i-1], j, bb1[i], min, sx, sy, "red", 0.2)
|
|
|
135 svg += make_line(h, j-1, bb2[i-1], j, bb2[i], min, sx, sy, "blue", 0.2)
|
|
|
136 }
|
|
|
137 svg += fmt.Sprint("</svg>\n")
|
|
|
138 if err := ioutil.WriteFile("chart.svg", []byte(svg), 0644); err != nil {
|
|
|
139 log.Fatal(err)
|
|
|
140 }
|
|
|
141
|
|
|
142 // [rsi.svg]
|
|
|
143 w, h = w, 150
|
|
|
144 sy = float64(h) / 100
|
|
|
145 svg = fmt.Sprintf("<svg xmlns='http://www.w3.org/2000/svg' width='%d' height='%d'>\n", w, h)
|
|
|
146 svg += make_line(h, 0, 30.0, w, 30.0, 0, 1.0, sy, "gray", 0.2)
|
|
|
147 svg += make_label(h, 10, 30, 0, 1.0, sy, fmt.Sprintf("30%%"))
|
|
|
148 svg += make_line(h, 0, 50.0, w, 50.0, 0, 1.0, sy, "gray", 0.2)
|
|
|
149 svg += make_label(h, 10, 50, 0, 1.0, sy, fmt.Sprintf("50%%"))
|
|
|
150 svg += make_line(h, 0, 70.0, w, 70.0, 0, 1.0, sy, "gray", 0.2)
|
|
|
151 svg += make_label(h, 10, 70, 0, 1.0, sy, fmt.Sprintf("70%%"))
|
|
|
152
|
|
|
153 for i, j, lm := len(x) - term, 0, ""; i < len(x); i, j = i + 1, j + 1 {
|
|
|
154 if !strings.EqualFold(lm, date[i][5:7]) {
|
|
|
155 svg += make_line(h, j, 0.0, j, float64(h), 0, sx, 1.0, "gray", 0.2)
|
|
|
156 svg += make_label(h, j, 20, 0, sx, 1.0, date[i][0:7])
|
|
|
157 lm = date[i][5:7]
|
|
|
158 }
|
|
|
159 }
|
|
|
160
|
|
|
161 for i, j := len(x) - term, 1; i < len(x); i, j = i + 1, j + 1 {
|
|
|
162 svg += make_line(h, j-1, rsi[i-1], j, rsi[i], 0, sx, sy, "black", 1.5)
|
|
|
163 }
|
|
|
164 svg += fmt.Sprint("</svg>\n")
|
|
|
165 if err := ioutil.WriteFile("rsi.svg", []byte(svg), 0644); err != nil {
|
|
|
166 log.Fatal(err)
|
|
|
167 }
|
|
|
168
|
|
|
169 // Make Html
|
|
|
170 html := "<html><body>"
|
|
|
171 html += "<h2>Daily Chart</h2>"
|
|
|
172 html += "<img src='chart.svg' />"
|
|
|
173 html += "<h2>RSI</h2>"
|
|
|
174 html += "<img src='rsi.svg' />"
|
|
|
175 html += "</body><html>"
|
|
|
176 if err := ioutil.WriteFile("fx.html", []byte(html), 0644); err != nil {
|
|
|
177 log.Fatal(err)
|
|
|
178 }
|
|
|
179
|
|
|
180 // Print Data
|
|
33
|
181 if debug {
|
|
31
|
182 fmt.Println("[j] n x bb1 bb2 rsi | n x bb1 bb2")
|
|
|
183 fmt.Println("---")
|
|
|
184 for i, j := len(x) - term, 0; i < len(x); i++ {
|
|
|
185 fmt.Printf("[%04d] %04d %5.2f %5.2f %5.2f %3.1f | ", j, i, x[i], bb1[i], bb2[i], rsi[i])
|
|
|
186 fmt.Printf("%04.0f %5.2f %5.2f %5.2f ", float64(i) * sx, (x[i] - min) * sy, (bb1[i] - min) * sy, (bb2[i] - min) * sy)
|
|
|
187 fmt.Println()
|
|
|
188 j++
|
|
|
189 }
|
|
|
190 fmt.Println("---")
|
|
|
191 fmt.Printf("min=%5.2f max=%5.2f\n", min, max)
|
|
|
192 }
|
|
|
193 }
|
|
|
194
|
|
|
195 func make_line(h, x1 int, y1 float64, x2 int, y2 float64, os, sx, sy float64, c string, sw float64) string {
|
|
|
196 x1, y1 = int(float64(x1) * sx), (y1 - os) * sy
|
|
|
197 x2, y2 = int(float64(x2) * sx), (y2 - os) * sy
|
|
|
198 y1, y2 = float64(h) - y1, float64(h) - y2
|
|
|
199 return fmt.Sprintf("<line x1='%d' y1='%d' x2='%d' y2='%d' stroke='%s' stroke-width='%0.1f' />\n", x1, int(y1), x2, int(y2), c, sw)
|
|
|
200 }
|
|
|
201
|
|
|
202 func make_label(h, x, y int, os, sx, sy float64, label string) string {
|
|
|
203 x = int(float64(x) * sx)
|
|
|
204 y = int((float64(y) - os) * sy)
|
|
|
205 y = h - y
|
|
|
206 return fmt.Sprintf("<text x='%d' y='%d'>%s</text>", x, y, label)
|
|
|
207 }
|
|
|
208
|
