use crate::shared::shared;
use crate::time::calendar::{Calendar, CalendarImpl, is_weekend_sat_sun, western_easter_monday};
use crate::time::date::{Date, Month};
use crate::time::weekday::Weekday;
#[derive(Clone, Copy, Debug, PartialEq, Eq)]
pub enum Market {
Settlement,
Tsx,
}
pub struct Canada;
impl Canada {
pub fn new(market: Market) -> Calendar {
let imp: crate::shared::Shared<dyn CalendarImpl> = match market {
Market::Settlement => shared(SettlementImpl),
Market::Tsx => shared(TsxImpl),
};
Calendar::from_impl(imp)
}
}
struct SettlementImpl;
struct TsxImpl;
impl CalendarImpl for SettlementImpl {
fn name(&self) -> String {
"Canada".to_string()
}
fn is_weekend(&self, w: Weekday) -> bool {
is_weekend_sat_sun(w)
}
fn is_business_day(&self, date: Date) -> bool {
let w = date.weekday();
let d = date.day_of_month();
let dd = date.day_of_year();
let m = date.month();
let y = date.year();
let em = western_easter_monday(y);
!(is_weekend_sat_sun(w)
|| ((d == 1 || ((d == 2 || d == 3) && w == Weekday::Monday)) && m == Month::January)
|| ((d >= 15 && d <= 21) && w == Weekday::Monday && m == Month::February && y >= 2008)
|| (dd == em - 3)
|| (d > 17 && d <= 24 && w == Weekday::Monday && m == Month::May)
|| ((d == 1 || ((d == 2 || d == 3) && w == Weekday::Monday)) && m == Month::July)
|| (d <= 7 && w == Weekday::Monday && m == Month::August)
|| (d <= 7 && w == Weekday::Monday && m == Month::September)
|| (((d == 30 && m == Month::September)
|| (d <= 2 && m == Month::October && w == Weekday::Monday))
&& y >= 2021)
|| (d > 7 && d <= 14 && w == Weekday::Monday && m == Month::October)
|| ((d == 11 || ((d == 12 || d == 13) && w == Weekday::Monday)) && m == Month::November)
|| ((d == 25 || (d == 27 && (w == Weekday::Monday || w == Weekday::Tuesday)))
&& m == Month::December)
|| ((d == 26 || (d == 28 && (w == Weekday::Monday || w == Weekday::Tuesday)))
&& m == Month::December))
}
}
impl CalendarImpl for TsxImpl {
fn name(&self) -> String {
"TSX".to_string()
}
fn is_weekend(&self, w: Weekday) -> bool {
is_weekend_sat_sun(w)
}
fn is_business_day(&self, date: Date) -> bool {
let w = date.weekday();
let d = date.day_of_month();
let dd = date.day_of_year();
let m = date.month();
let y = date.year();
let em = western_easter_monday(y);
!(is_weekend_sat_sun(w)
|| ((d == 1 || ((d == 2 || d == 3) && w == Weekday::Monday)) && m == Month::January)
|| ((d >= 15 && d <= 21) && w == Weekday::Monday && m == Month::February && y >= 2008)
|| (dd == em - 3)
|| (d > 17 && d <= 24 && w == Weekday::Monday && m == Month::May)
|| ((d == 1 || ((d == 2 || d == 3) && w == Weekday::Monday)) && m == Month::July)
|| (d <= 7 && w == Weekday::Monday && m == Month::August)
|| (d <= 7 && w == Weekday::Monday && m == Month::September)
|| (d > 7 && d <= 14 && w == Weekday::Monday && m == Month::October)
|| ((d == 25 || (d == 27 && (w == Weekday::Monday || w == Weekday::Tuesday)))
&& m == Month::December)
|| ((d == 26 || (d == 28 && (w == Weekday::Monday || w == Weekday::Tuesday)))
&& m == Month::December))
}
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn names_match_quantlib() {
assert_eq!(Canada::new(Market::Settlement).name(), "Canada");
assert_eq!(Canada::new(Market::Tsx).name(), "TSX");
}
#[test]
fn weekend_rule() {
let c = Canada::new(Market::Settlement);
assert!(c.is_weekend(Weekday::Saturday));
assert!(c.is_weekend(Weekday::Sunday));
}
}