use std::hash::{Hash, Hasher};
use nautilus_core::{
Params, UnixNanos,
correctness::{FAILED, check_equal_u8},
};
use rust_decimal::Decimal;
use serde::{Deserialize, Serialize};
use ustr::Ustr;
use super::{Instrument, any::InstrumentAny};
use crate::{
enums::{AssetClass, InstrumentClass, OptionKind},
identifiers::{InstrumentId, Symbol},
types::{
currency::Currency,
money::Money,
price::{Price, check_positive_price},
quantity::{Quantity, check_positive_quantity},
},
};
#[repr(C)]
#[derive(Clone, Debug, Serialize, Deserialize)]
#[cfg_attr(
feature = "python",
pyo3::pyclass(module = "nautilus_trader.core.nautilus_pyo3.model", from_py_object)
)]
#[cfg_attr(
feature = "python",
pyo3_stub_gen::derive::gen_stub_pyclass(module = "nautilus_trader.model")
)]
pub struct BinaryOption {
pub id: InstrumentId,
pub raw_symbol: Symbol,
pub asset_class: AssetClass,
pub currency: Currency,
pub activation_ns: UnixNanos,
pub expiration_ns: UnixNanos,
pub price_precision: u8,
pub size_precision: u8,
pub price_increment: Price,
pub size_increment: Quantity,
pub margin_init: Decimal,
pub margin_maint: Decimal,
pub maker_fee: Decimal,
pub taker_fee: Decimal,
pub outcome: Option<Ustr>,
pub description: Option<Ustr>,
pub max_quantity: Option<Quantity>,
pub min_quantity: Option<Quantity>,
pub max_notional: Option<Money>,
pub min_notional: Option<Money>,
pub max_price: Option<Price>,
pub min_price: Option<Price>,
pub info: Option<Params>,
pub ts_event: UnixNanos,
pub ts_init: UnixNanos,
}
impl BinaryOption {
#[allow(clippy::too_many_arguments)]
pub fn new_checked(
instrument_id: InstrumentId,
raw_symbol: Symbol,
asset_class: AssetClass,
currency: Currency,
activation_ns: UnixNanos,
expiration_ns: UnixNanos,
price_precision: u8,
size_precision: u8,
price_increment: Price,
size_increment: Quantity,
outcome: Option<Ustr>,
description: Option<Ustr>,
max_quantity: Option<Quantity>,
min_quantity: Option<Quantity>,
max_notional: Option<Money>,
min_notional: Option<Money>,
max_price: Option<Price>,
min_price: Option<Price>,
margin_init: Option<Decimal>,
margin_maint: Option<Decimal>,
maker_fee: Option<Decimal>,
taker_fee: Option<Decimal>,
info: Option<Params>,
ts_event: UnixNanos,
ts_init: UnixNanos,
) -> anyhow::Result<Self> {
check_equal_u8(
price_precision,
price_increment.precision,
stringify!(price_precision),
stringify!(price_increment.precision),
)?;
check_equal_u8(
size_precision,
size_increment.precision,
stringify!(size_precision),
stringify!(size_increment.precision),
)?;
check_positive_price(price_increment, stringify!(price_increment))?;
check_positive_quantity(size_increment, stringify!(size_increment))?;
Ok(Self {
id: instrument_id,
raw_symbol,
asset_class,
currency,
activation_ns,
expiration_ns,
price_precision,
size_precision,
price_increment,
size_increment,
margin_init: margin_init.unwrap_or_default(),
margin_maint: margin_maint.unwrap_or_default(),
maker_fee: maker_fee.unwrap_or_default(),
taker_fee: taker_fee.unwrap_or_default(),
outcome,
description,
max_quantity,
min_quantity,
max_notional,
min_notional,
max_price,
min_price,
info,
ts_event,
ts_init,
})
}
#[allow(clippy::too_many_arguments)]
pub fn new(
instrument_id: InstrumentId,
raw_symbol: Symbol,
asset_class: AssetClass,
currency: Currency,
activation_ns: UnixNanos,
expiration_ns: UnixNanos,
price_precision: u8,
size_precision: u8,
price_increment: Price,
size_increment: Quantity,
outcome: Option<Ustr>,
description: Option<Ustr>,
max_quantity: Option<Quantity>,
min_quantity: Option<Quantity>,
max_notional: Option<Money>,
min_notional: Option<Money>,
max_price: Option<Price>,
min_price: Option<Price>,
margin_init: Option<Decimal>,
margin_maint: Option<Decimal>,
maker_fee: Option<Decimal>,
taker_fee: Option<Decimal>,
info: Option<Params>,
ts_event: UnixNanos,
ts_init: UnixNanos,
) -> Self {
Self::new_checked(
instrument_id,
raw_symbol,
asset_class,
currency,
activation_ns,
expiration_ns,
price_precision,
size_precision,
price_increment,
size_increment,
outcome,
description,
max_quantity,
min_quantity,
max_notional,
min_notional,
max_price,
min_price,
margin_init,
margin_maint,
maker_fee,
taker_fee,
info,
ts_event,
ts_init,
)
.expect(FAILED)
}
}
impl PartialEq<Self> for BinaryOption {
fn eq(&self, other: &Self) -> bool {
self.id == other.id
}
}
impl Eq for BinaryOption {}
impl Hash for BinaryOption {
fn hash<H: Hasher>(&self, state: &mut H) {
self.id.hash(state);
}
}
impl Instrument for BinaryOption {
fn into_any(self) -> InstrumentAny {
InstrumentAny::BinaryOption(self)
}
fn id(&self) -> InstrumentId {
self.id
}
fn raw_symbol(&self) -> Symbol {
self.raw_symbol
}
fn asset_class(&self) -> AssetClass {
self.asset_class
}
fn instrument_class(&self) -> InstrumentClass {
InstrumentClass::BinaryOption
}
fn underlying(&self) -> Option<Ustr> {
None
}
fn base_currency(&self) -> Option<Currency> {
None
}
fn quote_currency(&self) -> Currency {
self.currency
}
fn settlement_currency(&self) -> Currency {
self.currency
}
fn isin(&self) -> Option<Ustr> {
None
}
fn exchange(&self) -> Option<Ustr> {
None
}
fn option_kind(&self) -> Option<OptionKind> {
None
}
fn is_inverse(&self) -> bool {
false
}
fn price_precision(&self) -> u8 {
self.price_precision
}
fn size_precision(&self) -> u8 {
self.size_precision
}
fn price_increment(&self) -> Price {
self.price_increment
}
fn size_increment(&self) -> Quantity {
self.size_increment
}
fn multiplier(&self) -> Quantity {
Quantity::from(1)
}
fn lot_size(&self) -> Option<Quantity> {
Some(Quantity::from(1))
}
fn max_quantity(&self) -> Option<Quantity> {
self.max_quantity
}
fn min_quantity(&self) -> Option<Quantity> {
self.min_quantity
}
fn max_price(&self) -> Option<Price> {
self.max_price
}
fn min_price(&self) -> Option<Price> {
self.min_price
}
fn ts_event(&self) -> UnixNanos {
self.ts_event
}
fn ts_init(&self) -> UnixNanos {
self.ts_init
}
fn margin_init(&self) -> Decimal {
self.margin_init
}
fn margin_maint(&self) -> Decimal {
self.margin_maint
}
fn maker_fee(&self) -> Decimal {
self.maker_fee
}
fn taker_fee(&self) -> Decimal {
self.taker_fee
}
fn strike_price(&self) -> Option<Price> {
None
}
fn activation_ns(&self) -> Option<UnixNanos> {
Some(self.activation_ns)
}
fn expiration_ns(&self) -> Option<UnixNanos> {
Some(self.expiration_ns)
}
fn max_notional(&self) -> Option<Money> {
self.max_notional
}
fn min_notional(&self) -> Option<Money> {
self.min_notional
}
}
#[cfg(test)]
mod tests {
use rstest::rstest;
use crate::{
enums::{AssetClass, InstrumentClass},
identifiers::{InstrumentId, Symbol},
instruments::{BinaryOption, Instrument, stubs::*},
types::{Currency, Price, Quantity},
};
#[rstest]
fn test_trait_accessors(binary_option: BinaryOption) {
assert_eq!(binary_option.asset_class(), AssetClass::Alternative);
assert_eq!(
binary_option.instrument_class(),
InstrumentClass::BinaryOption
);
assert_eq!(binary_option.quote_currency(), Currency::USDC());
assert!(!binary_option.is_inverse());
assert_eq!(binary_option.price_precision(), 3);
assert_eq!(binary_option.size_precision(), 2);
assert!(binary_option.activation_ns().is_some());
assert!(binary_option.expiration_ns().is_some());
}
#[rstest]
fn test_new_checked_price_precision_mismatch() {
let result = BinaryOption::new_checked(
InstrumentId::from("TEST.POLYMARKET"),
Symbol::from("TEST"),
AssetClass::Alternative,
Currency::USDC(),
0.into(),
0.into(),
4, 2,
Price::from("0.001"),
Quantity::from("0.01"),
None,
None,
None,
None,
None,
None,
None,
None,
None,
None,
None,
None,
None,
0.into(),
0.into(),
);
assert!(result.is_err());
}
#[rstest]
fn test_serialization_roundtrip(binary_option: BinaryOption) {
let json = serde_json::to_string(&binary_option).unwrap();
let deserialized: BinaryOption = serde_json::from_str(&json).unwrap();
assert_eq!(binary_option, deserialized);
}
}