iterinfo first port

2 files changed, 71 insertions, 1 deletions

diff --git a/src/iterinfo.rs b/src/iterinfo.rs
index bf7f1d7..98edd45 100644
--- a/src/iterinfo.rs
+++ b/src/iterinfo.rs
@@ -1,5 +1,7 @@
 use crate::monthinfo::*;
 use crate::yearinfo::*;
+use chrono::prelude::*;
+use chrono::*;
 
 struct IterInfo<'a> {
     pub yearinfo: Option<YearInfo>,
@@ -115,4 +117,72 @@ impl<'a> IterInfo<'a> {
             None => None,
         }
     }
+
+    pub fn ydayset(&self) -> (Vec<usize>, usize, usize) {
+        let yearlen = self.yearlen().unwrap();
+        let mut v = Vec::with_capacity(yearlen);
+        for i in 0..yearlen {
+            v.push(i);
+        }
+        (v, 0, yearlen)
+    }
+
+    pub fn mdayset(&self, month: usize) -> (Vec<usize>, usize, usize) {
+        let mrange = self.mrange().unwrap();
+        let start = mrange[month - 1];
+        let end = mrange[month];
+        let mut set = vec![0; self.yearlen().unwrap()];
+        for i in start..end {
+            set[i] = 1;
+        }
+        (set, start, end)
+    }
+
+    pub fn wdayset(&self, year: isize, month: usize, day: usize) -> (Vec<usize>, usize, usize) {
+        let mut set = vec![0; self.yearlen().unwrap()];
+
+        // should it be month - 1 ??????
+        let mut i = (to_ordinal(
+            &Utc.ymd(year as i32, month as u32 - 1, day as u32)
+                .and_hms(0, 0, 0),
+        ) - self.yearordinal().unwrap()) as usize;
+
+        let start = i;
+        for _ in 0..7 {
+            set[i] = i;
+            i += 1;
+            if self.wdaymask().unwrap()[i] == self.options.wkst {
+                break;
+            }
+        }
+        (set, start, i)
+    }
+
+    pub fn ddayset(&self, year: isize, month: usize, day: usize) -> (Vec<usize>, usize, usize) {
+        let mut set = vec![0; self.yearlen().unwrap()];
+
+        // should it be month - 1 ??????
+        let mut i = (to_ordinal(
+            &Utc.ymd(year as i32, month as u32 - 1, day as u32)
+                .and_hms(0, 0, 0),
+        ) - self.yearordinal().unwrap()) as usize;
+
+        (set, i, i + 1)
+    }
+
+    pub fn getdayset(
+        &self,
+        freq: Frequenzy,
+        year: isize,
+        month: usize,
+        day: usize,
+    ) -> (Vec<usize>, usize, usize) {
+        match freq {
+            Frequenzy::YEARLY => self.ydayset(),
+            Frequenzy::MONTHLY => self.mdayset(month),
+            Frequenzy::WEEKLY => self.wdayset(year, month, day),
+            Frequenzy::DAILY => self.ddayset(year, month, day),
+            _ => self.ydayset(),
+        }
+    }
 }
diff --git a/src/yearinfo.rs b/src/yearinfo.rs
index 37767c1..0628485 100644
--- a/src/yearinfo.rs
+++ b/src/yearinfo.rs
@@ -60,7 +60,7 @@ fn get_year_len(year: i32) -> usize {
     365
 }
 
-fn to_ordinal(date: &DateTime<Utc>) -> isize {
+pub fn to_ordinal(date: &DateTime<Utc>) -> isize {
     (date.timestamp() / 60 / 60 / 24) as isize
 }