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
|
use crate::datetime::*;
use crate::iterinfo::*;
use crate::yearinfo::*;
use chrono::prelude::*;
use chrono::*;
pub fn from_ordinal(ordinal: isize) -> DateTime<Utc> {
let timestamp = ordinal * 24 * 60 * 60;
let naive = NaiveDateTime::from_timestamp(timestamp as i64, 0);
DateTime::from_utc(naive, Utc)
}
pub fn build_poslist(
bysetpost: &Vec<usize>,
timeset: &Vec<Time>,
start: usize,
end: usize,
ii: &IterInfo,
dayset: &Vec<Option<isize>>,
) -> Vec<DateTime<Utc>> {
let mut poslist: Vec<DateTime<Utc>> = vec![];
for j in 0..bysetpost.len() {
let daypos;
let timepos;
let pos = bysetpost[j];
if pos < 0 {
daypos = pos / timeset.len();
timepos = pymod(pos as isize, timeset.len() as isize);
} else {
daypos = (pos - 1) / timeset.len();
timepos = pymod(pos as isize - 1, timeset.len() as isize);
}
let mut tmp = vec![];
for k in start..end {
let val = dayset[k];
if val.is_some() {
tmp.push(val.unwrap());
}
}
let i;
if daypos < 0 {
let index = (tmp.len() - daypos) as usize;
i = &tmp[index];
} else {
i = &tmp[daypos as usize];
}
let date = from_ordinal(ii.yearordinal().unwrap() + i);
// const res = dateutil.combine(date, time)
let res = Utc.ymd(date.year(), date.month(), date.day()).and_hms(
timeset[timepos as usize].hour as u32,
timeset[timepos as usize].minute as u32,
timeset[timepos as usize].second as u32,
);
// XXX: can this ever be in the array?
// - compare the actual date instead?
if !poslist.iter().any(|&p| p == res) {
poslist.push(res);
}
}
//poslist.sort();
poslist.sort_by(|a, b| a.timestamp().cmp(&b.timestamp()));
poslist
}
|
