init with yearinfo

1 files changed, 288 insertions, 0 deletions

diff --git a/src/lib.rs b/src/lib.rs
new file mode 100644
index 0000000..809fae5
--- /dev/null
+++ b/src/lib.rs
@@ -0,0 +1,288 @@
+extern crate chrono;
+
+mod masks;
+
+use chrono::prelude::*;
+use chrono::{DateTime, Duration};
+use masks::Masks;
+
+struct RRule {}
+
+#[derive(Debug)]
+struct YearInfo {
+    yearlen: usize,
+    nextyearlen: usize,
+    yearordinal: i64,
+    yearweekday: usize,
+    mmask: Vec<u32>,
+    mrange: Vec<u32>,
+    mdaymask: Vec<u32>,
+    nmdaymask: Vec<i32>,
+    wdaymask: Vec<u32>,
+    wnomask: Option<Vec<u32>>,
+}
+
+#[derive(Debug)]
+enum Frequenzy {
+    YEARLY,
+    MONTHLY,
+    WEEKLY,
+    DAILY,
+    HOURLY,
+    MINUTELY,
+    SECONDLY,
+}
+
+#[derive(Debug)]
+struct ParsedOptions {
+    freq: Frequenzy,
+    interval: usize,
+    count: Option<u32>,
+    until: Option<DateTime<Utc>>,
+    tzid: Option<String>,
+    dtstart: DateTime<Utc>,
+    wkst: usize,
+    bysetpos: Vec<u32>,
+    bymonth: Vec<u32>,
+    bymonthday: Vec<u32>,
+    bynmonthday: Vec<u32>,
+    byyearday: Vec<u32>,
+    byweekno: Vec<isize>,
+    byweekday: Vec<u32>,
+    byhour: Vec<u32>,
+    byminute: Vec<u32>,
+    bysecond: Vec<u32>,
+}
+
+fn is_leap_year(year: i32) -> bool {
+    year % 4 == 0 && (year % 100 != 0 || year % 400 == 0)
+}
+
+fn get_year_len(year: i32) -> usize {
+    if is_leap_year(year) {
+        return 366;
+    }
+    365
+}
+
+fn to_ordinal(date: &DateTime<Utc>) -> i64 {
+    date.timestamp() / 60 / 60 / 24
+}
+
+fn get_weekday_val(wk: &Weekday) -> usize {
+    match wk {
+        Weekday::Mon => 0,
+        Weekday::Tue => 1,
+        Weekday::Wed => 2,
+        Weekday::Thu => 3,
+        Weekday::Fri => 4,
+        Weekday::Sat => 5,
+        Weekday::Sun => 6,
+    }
+}
+
+struct BaseMasks {
+    mmask: Vec<u32>,
+    mdaymask: Vec<u32>,
+    nmdaymask: Vec<i32>,
+    wdaymask: Vec<u32>,
+    mrange: Vec<u32>,
+}
+
+fn base_year_masks(year: i32) -> BaseMasks {
+    let masks = Masks::new();
+    let firstyday = Utc.ymd(year, 1, 1).and_hms_milli(0, 0, 0, 0);
+    let yearlen = get_year_len(year);
+    let wday = get_weekday_val(&firstyday.weekday()) as usize;
+
+    if yearlen == 365 {
+        return BaseMasks {
+            mmask: masks.M365,
+            mdaymask: masks.MDAY365,
+            nmdaymask: masks.NMDAY365,
+            mrange: masks.M365RANGE,
+            wdaymask: Vec::from(&masks.WDAY[wday..]),
+        };
+    }
+
+    BaseMasks {
+        mmask: masks.M366,
+        mdaymask: masks.MDAY366,
+        nmdaymask: masks.NMDAY366,
+        mrange: masks.M366RANGE,
+        wdaymask: Vec::from(&masks.WDAY[wday..]),
+    }
+}
+
+pub fn pymod(a: isize, b: isize) -> isize {
+    let r = a % b;
+    // If r and b differ in sign, add b to wrap the result to the correct sign.
+    if (r > 0 && b < 0) || (r < 0 && b > 0) {
+        return r + b;
+    }
+    r
+}
+
+fn rebuild_year(year: i32, options: &ParsedOptions) -> YearInfo {
+    let firstyday = Utc.ymd(year, 1, 1).and_hms_milli(0, 0, 0, 0);
+
+    let yearlen = get_year_len(year);
+    let nextyearlen = get_year_len(year + 1);
+    let yearordinal = to_ordinal(&firstyday);
+    let yearweekday = get_weekday_val(&firstyday.weekday());
+
+    let base_masks = base_year_masks(year);
+
+    let mut result = YearInfo {
+        yearlen,
+        nextyearlen,
+        yearordinal,
+        yearweekday,
+        wnomask: None,
+        mmask: base_masks.mmask,
+        mdaymask: base_masks.mdaymask,
+        nmdaymask: base_masks.nmdaymask,
+        mrange: base_masks.mrange,
+        wdaymask: base_masks.wdaymask,
+    };
+
+    if options.byweekno.is_empty() {
+        return result;
+    }
+
+    let mut wnomask = vec![0; yearlen + 7];
+    let wyearlen;
+    let mut no1wkst = pymod((7 - yearweekday + options.wkst) as isize, 7);
+    let firstwkst = no1wkst;
+    if no1wkst >= 4 {
+        no1wkst = 0;
+        // Number of days in the year, plus the days we got
+        // from last year.
+        wyearlen = result.yearlen as isize + pymod(yearweekday as isize - options.wkst as isize, 7);
+    } else {
+        // Number of days in the year, minus the days we
+        // left in last year.
+        wyearlen = yearlen as isize - no1wkst;
+    }
+
+    let div = (wyearlen as f32 / 7.).round() as isize;
+    let year_mod = pymod(div, 7);
+    //const numweeks = Math.floor(div + mod / 4)
+    let numweeks = div + (year_mod / 4);
+
+    for j in 0..options.byweekno.len() {
+        let mut n = options.byweekno[j];
+        if n < 0 {
+            n += (numweeks + 1) as isize;
+        }
+        if !(n > 0 && n <= numweeks) {
+            continue;
+        }
+        let mut i;
+        if n > 1 {
+            i = no1wkst + (n - 1) * 7;
+            if no1wkst != firstwkst {
+                i -= 7 - firstwkst;
+            }
+        } else {
+            i = no1wkst;
+        }
+
+        for _ in 0..7 {
+            wnomask[i as usize] = 1;
+            i += 1;
+            if result.wdaymask[i as usize] as usize == options.wkst {
+                break;
+            }
+        }
+    }
+
+    if options.byweekno.iter().any(|&wkno| wkno == 1) {
+        // Check week number 1 of next year as well
+        // orig-TODO : Check -numweeks for next year.
+        let mut i = no1wkst + numweeks * 7;
+        if no1wkst != firstwkst {
+            i -= 7 - firstwkst;
+        }
+        if i < yearlen as isize {
+            // If week starts in next year, we
+            // don't care about it.
+            for _ in 0..7 {
+                wnomask[i as usize] = 1;
+                i += 1;
+                if result.wdaymask[i as usize] as usize == options.wkst {
+                    break;
+                }
+            }
+        }
+    }
+
+    if no1wkst > 0 {
+        // Check last week number of last year as
+        // well. If no1wkst is 0, either the year
+        // started on week start, or week number 1
+        // got days from last year, so there are no
+        // days from last year's last week number in
+        // this year.
+        let lnumweeks;
+        if options.byweekno.iter().any(|&weekno| weekno == -1) {
+            let lyearweekday = get_weekday_val(&Utc.ymd(year - 1, 1, 1).weekday());
+
+            let lno1wkst = pymod((7 - lyearweekday + options.wkst) as isize, 7);
+
+            let lyearlen = get_year_len(year - 1);
+            let weekst;
+            if lno1wkst >= 4 {
+                //lno1wkst = 0;
+                weekst = lyearlen as isize + pymod((lyearweekday - options.wkst) as isize, 7);
+            } else {
+                weekst = yearlen as isize - no1wkst;
+            }
+
+            lnumweeks = 52 + (pymod(weekst, 7) / 4) as isize;
+        } else {
+            lnumweeks = -1 as isize;
+        }
+
+        if options.byweekno.iter().any(|&weekno| weekno == lnumweeks) {
+            for i in 0..no1wkst {
+                wnomask[i as usize] = 1;
+            }
+        }
+    }
+
+    result.wnomask = Some(wnomask);
+
+    result
+}
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    #[test]
+    fn it_works() {
+        let options = ParsedOptions {
+            freq: Frequenzy::YEARLY,
+            interval: 1,
+            count: Some(10),
+            until: None,
+            tzid: None,
+            dtstart: Utc.ymd(2020, 1, 1).and_hms(0, 0, 0),
+            wkst: 0,
+            bysetpos: vec![],
+            bymonth: vec![],
+            bymonthday: vec![],
+            bynmonthday: vec![],
+            byyearday: vec![],
+            byweekno: vec![1],
+            byweekday: vec![],
+            byhour: vec![],
+            byminute: vec![],
+            bysecond: vec![],
+        };
+        let res = rebuild_year(2020, &options);
+        println!("Res: {:?}", res);
+        assert_eq!(2 + 2, 4);
+    }
+}