-
Notifications
You must be signed in to change notification settings - Fork 8
/
quarter.go
77 lines (68 loc) · 1.87 KB
/
quarter.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
package quarter
import (
"time"
"github.com/grokify/mogo/math/mathutil"
)
// YearQuarterToQuarterContinuous converts a year and quarteer to
// a continuous quarter integer. This is useful when an even
// even spacing between months is desired, such as with
// charting x-axis values.
func YearQuarterToQuarterContinuous(year, quarter uint64) uint64 {
return year*4 + quarter
}
// QuarterContinuousToYearQuarter converts a continuous quarter
// value (e.g. numerof months from year 0).
func QuarterContinuousToYearQuarter(quarterc uint64) (uint64, uint64) {
quotient, remainder := mathutil.DivideInt64(
int64(quarterc-1), int64(4))
return uint64(quotient), uint64(remainder + 1)
}
// TimeToQuarterContinuous converts a `time.Time` value
// to a continuous month.
func TimeToQuarterContinuous(t time.Time) uint64 {
t = t.UTC()
return YearQuarterToQuarterContinuous(
uint64(t.Year()), MonthToQuarter(uint64(t.Month())))
}
func MonthToQuarter(month uint64) uint64 {
if month <= 3 {
return uint64(1)
} else if month <= 6 {
return uint64(2)
} else if month <= 9 {
return uint64(3)
}
return uint64(4)
}
// QuarterContinuousToTime converts a continuous month
// value to a `time.Time` value.
func QuarterContinuousToTime(monthc uint64) time.Time {
year, quarter := QuarterContinuousToYearQuarter(monthc)
month := 1
if quarter == 2 {
month = 4
} else if quarter == 3 {
month = 7
} else if quarter == 4 {
month = 10
}
return time.Date(
int(year), time.Month(int(month)), 1,
0, 0, 0, 0, time.UTC)
}
func QuarterContinuousIsQuarterBegin(quarterc uint64) bool {
t := QuarterContinuousToTime(quarterc)
month := t.Month()
if month == 1 || month == 4 || month == 7 || month == 10 {
return true
}
return false
}
func QuarterContinuousIsYearBegin(quarterc uint64) bool {
t := QuarterContinuousToTime(quarterc)
month := t.Month()
if month == 1 {
return true
}
return false
}