blob: 20efebe0a3b479e3312cd529bc215bd8e25f0ab8 (
plain)
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
|
#ifndef NUMPY_CORE_SRC_MULTIARRAY_DATETIME_BUSDAYCAL_H_
#define NUMPY_CORE_SRC_MULTIARRAY_DATETIME_BUSDAYCAL_H_
/*
* A list of holidays, which should be sorted, not contain any
* duplicates or NaTs, and not include any days already excluded
* by the associated weekmask.
*
* The data is manually managed with PyArray_malloc/PyArray_free.
*/
typedef struct {
npy_datetime *begin, *end;
} npy_holidayslist;
/*
* This object encapsulates a weekmask and normalized holidays list,
* so that the business day API can use this data without having
* to normalize it repeatedly. All the data of this object is private
* and cannot be modified from Python. Copies are made when giving
* the weekmask and holidays data to Python code.
*/
typedef struct {
PyObject_HEAD
npy_holidayslist holidays;
int busdays_in_weekmask;
npy_bool weekmask[7];
} NpyBusDayCalendar;
extern NPY_NO_EXPORT PyTypeObject NpyBusDayCalendar_Type;
/*
* Converts a Python input into a 7-element weekmask, where 0 means
* weekend and 1 means business day.
*/
NPY_NO_EXPORT int
PyArray_WeekMaskConverter(PyObject *weekmask_in, npy_bool *weekmask);
/*
* Sorts the array of dates provided in place and removes
* NaT, duplicates and any date which is already excluded on account
* of the weekmask.
*
* Returns the number of dates left after removing weekmask-excluded
* dates.
*/
NPY_NO_EXPORT void
normalize_holidays_list(npy_holidayslist *holidays, npy_bool *weekmask);
/*
* Converts a Python input into a non-normalized list of holidays.
*
* IMPORTANT: This function can't do the normalization, because it doesn't
* know the weekmask. You must call 'normalize_holiday_list'
* on the result before using it.
*/
NPY_NO_EXPORT int
PyArray_HolidaysConverter(PyObject *dates_in, npy_holidayslist *holidays);
#endif /* NUMPY_CORE_SRC_MULTIARRAY_DATETIME_BUSDAYCAL_H_ */
|
