diff options
Diffstat (limited to 'zerocurve.pyx')
| -rw-r--r-- | zerocurve.pyx | 24 |
1 files changed, 14 insertions, 10 deletions
diff --git a/zerocurve.pyx b/zerocurve.pyx index e56a38e..96e7036 100644 --- a/zerocurve.pyx +++ b/zerocurve.pyx @@ -3,7 +3,7 @@ from libc.stdlib cimport malloc, free from pyisda.zerocurve cimport JpmcdsBuildIRZeroCurve from pyisda.yearfrac cimport dcc from pyisda.date cimport (JpmcdsStringToDateInterval, JpmcdsDateIntervalToFreq, - JpmcdsDate, JpmcdsDateFwdThenAdjust, NONE) + JpmcdsDate, JpmcdsDateFwdThenAdjust, BadDay) cdef int SUCCESS = 0 @@ -11,12 +11,14 @@ cdef class ZeroCurve: cdef TCurve* _thisptr cdef TDate* _dates - def __cinit__(self, date, str types, + def __init__(self, date, str types, list periods, double[:] rates, str mm_dcc, str fixed_swap_period, str float_swap_period, - str fixed_swap_dcc, str float_swap_dcc, char bad_day_conv, - char* holidayFile): + str fixed_swap_dcc, str float_swap_dcc, BadDay bad_day_conv): + """ Initialize a zero coupon curve + instruments need to be sorted by tenor + """ cdef: double fixed_freq double float_freq @@ -29,15 +31,15 @@ cdef class ZeroCurve: else: raise ValueError - cdef TDate* _dates = <TDate*>malloc(len(periods) * sizeof(TDate)) + self._dates = <TDate*>malloc(len(periods) * sizeof(TDate)) cdef TDateInterval tmp for i, p in enumerate(periods): period_bytes = p.encode('utf-8') - if JpmcdsStringToDateInterval(p, routine, &tmp) != SUCCESS: + if JpmcdsStringToDateInterval(period_bytes, routine, &tmp) != SUCCESS: raise ValueError if JpmcdsDateFwdThenAdjust(value_date, &tmp, NONE, - "None", &_dates[i]) != SUCCESS: + "None", &self._dates[i]) != SUCCESS: raise ValueError('Invalid interval') fixed_bytes = fixed_swap_period.encode('utf-8') @@ -52,11 +54,12 @@ cdef class ZeroCurve: raise ValueError if JpmcdsDateIntervalToFreq(&ivl, &float_freq) != SUCCESS: raise ValueError + self._thisptr = JpmcdsBuildIRZeroCurve( - value_date, types_bytes, _dates, + value_date, types_bytes, self._dates, &rates[0], len(periods), dcc(mm_dcc), <long> fixed_freq, <long> float_freq, - dcc(fixed_swap_dcc), dcc(fixed_swap_dcc), <long>bad_day_conv, "None" + dcc(fixed_swap_dcc), dcc(float_swap_dcc), bad_day_conv, b"None" ) def __dealloc__(self): @@ -66,10 +69,11 @@ cdef class ZeroCurve: free(self._dates) def discount_factor(self, date): + if self._thisptr is NULL: + raise ValueError('curve is empty') cdef TDate discount_date if isinstance(date, datetime.date): discount_date = JpmcdsDate(date.year, date.month, date.day) else: raise ValueError - return JpmcdsZeroPrice(self._thisptr, discount_date) |