refactoring: bar_filler
Vladimir Bauer
6 years ago
| 19 | 19 | |
| 20 | 20 | // DefaultBarStyle is applied when bar constructed with *Progress.AddBar method. |
| 21 | 21 | // |
| 22 | // '1th rune' stands for left boundary rune | |
| 22 | // '1st rune' stands for left boundary rune | |
| 23 | 23 | // |
| 24 | // '2th rune' stands for fill rune | |
| 24 | // '2nd rune' stands for fill rune | |
| 25 | 25 | // |
| 26 | // '3th rune' stands for tip rune | |
| 26 | // '3rd rune' stands for tip rune | |
| 27 | 27 | // |
| 28 | 28 | // '4th rune' stands for empty rune |
| 29 | 29 | // |
| 49 | 49 | style = DefaultBarStyle |
| 50 | 50 | } |
| 51 | 51 | bf := &barFiller{ |
| 52 | format: make([][]byte, utf8.RuneCountInString(style)), | |
| 52 | format: make([][]byte, utf8.RuneCountInString(style)), | |
| 53 | reverse: reverse, | |
| 53 | 54 | } |
| 54 | 55 | bf.SetStyle(style) |
| 55 | bf.SetReverse(reverse) | |
| 56 | 56 | return bf |
| 57 | 57 | } |
| 58 | 58 | |
| 65 | 65 | src = append(src, []byte(string(r))) |
| 66 | 66 | } |
| 67 | 67 | copy(s.format, src) |
| 68 | if s.reverse { | |
| 69 | s.tip = s.format[rRevTip] | |
| 70 | } else { | |
| 71 | s.tip = s.format[rTip] | |
| 72 | } | |
| 68 | s.SetReverse(s.reverse) | |
| 73 | 69 | } |
| 74 | 70 | |
| 75 | 71 | func (s *barFiller) SetReverse(reverse bool) { |
| 76 | 72 | if reverse { |
| 77 | 73 | s.tip = s.format[rRevTip] |
| 78 | s.flush = func(w io.Writer, bb [][]byte) { | |
| 79 | for i := len(bb) - 1; i >= 0; i-- { | |
| 80 | w.Write(bb[i]) | |
| 81 | } | |
| 82 | } | |
| 74 | s.flush = reverseFlush | |
| 83 | 75 | } else { |
| 84 | 76 | s.tip = s.format[rTip] |
| 85 | s.flush = func(w io.Writer, bb [][]byte) { | |
| 86 | for i := 0; i < len(bb); i++ { | |
| 87 | w.Write(bb[i]) | |
| 88 | } | |
| 89 | } | |
| 77 | s.flush = normalFlush | |
| 90 | 78 | } |
| 91 | 79 | s.reverse = reverse |
| 92 | 80 | } |
| 134 | 122 | |
| 135 | 123 | s.flush(w, bb) |
| 136 | 124 | } |
| 125 | ||
| 126 | func normalFlush(w io.Writer, bb [][]byte) { | |
| 127 | for i := 0; i < len(bb); i++ { | |
| 128 | w.Write(bb[i]) | |
| 129 | } | |
| 130 | } | |
| 131 | ||
| 132 | func reverseFlush(w io.Writer, bb [][]byte) { | |
| 133 | for i := len(bb) - 1; i >= 0; i-- { | |
| 134 | w.Write(bb[i]) | |
| 135 | } | |
| 136 | } | |