summaryrefslogtreecommitdiff
path: root/src/lib.rs
diff options
context:
space:
mode:
Diffstat (limited to 'src/lib.rs')
-rw-r--r--src/lib.rs286
1 files changed, 1 insertions, 285 deletions
diff --git a/src/lib.rs b/src/lib.rs
index 809fae5..ede868c 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -1,288 +1,4 @@
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);
- }
-}
+mod yearinfo;
generated by cgit v1.2.3 (git 2.25.1) at 2024-11-01 10:35:11 +0000