blob: 0d310b47a18e8fcd7ccca36376141963f905950b [file] [log] [blame]
Tim Peters2a799bf2002-12-16 20:18:38 +00001/* datetime.h
2 */
3
4#ifndef DATETIME_H
5#define DATETIME_H
Tim Peters9ddf40b2004-06-20 22:41:32 +00006#ifdef __cplusplus
7extern "C" {
8#endif
Tim Peters2a799bf2002-12-16 20:18:38 +00009
10/* Fields are packed into successive bytes, each viewed as unsigned and
11 * big-endian, unless otherwise noted:
12 *
13 * byte offset
14 * 0 year 2 bytes, 1-9999
15 * 2 month 1 byte, 1-12
16 * 3 day 1 byte, 1-31
17 * 4 hour 1 byte, 0-23
18 * 5 minute 1 byte, 0-59
19 * 6 second 1 byte, 0-59
20 * 7 usecond 3 bytes, 0-999999
21 * 10
22 */
23
24/* # of bytes for year, month, and day. */
25#define _PyDateTime_DATE_DATASIZE 4
26
27/* # of bytes for hour, minute, second, and usecond. */
28#define _PyDateTime_TIME_DATASIZE 6
29
30/* # of bytes for year, month, day, hour, minute, second, and usecond. */
31#define _PyDateTime_DATETIME_DATASIZE 10
32
Tim Petersa032d2e2003-01-11 00:15:54 +000033
34typedef struct
35{
36 PyObject_HEAD
37 long hashcode; /* -1 when unknown */
38 int days; /* -MAX_DELTA_DAYS <= days <= MAX_DELTA_DAYS */
39 int seconds; /* 0 <= seconds < 24*3600 is invariant */
40 int microseconds; /* 0 <= microseconds < 1000000 is invariant */
41} PyDateTime_Delta;
42
43typedef struct
44{
45 PyObject_HEAD /* a pure abstract base clase */
46} PyDateTime_TZInfo;
Tim Peters37f39822003-01-10 03:49:02 +000047
Tim Petersa032d2e2003-01-11 00:15:54 +000048
49/* The datetime and time types have hashcodes, and an optional tzinfo member,
50 * present if and only if hastzinfo is true.
51 */
52#define _PyTZINFO_HEAD \
53 PyObject_HEAD \
54 long hashcode; \
55 char hastzinfo; /* boolean flag */
56
57/* No _PyDateTime_BaseTZInfo is allocated; it's just to have something
58 * convenient to cast to, when getting at the hastzinfo member of objects
59 * starting with _PyTZINFO_HEAD.
60 */
61typedef struct
62{
63 _PyTZINFO_HEAD
64} _PyDateTime_BaseTZInfo;
65
66/* All time objects are of PyDateTime_TimeType, but that can be allocated
67 * in two ways, with or without a tzinfo member. Without is the same as
68 * tzinfo == None, but consumes less memory. _PyDateTime_BaseTime is an
69 * internal struct used to allocate the right amount of space for the
70 * "without" case.
71 */
72#define _PyDateTime_TIMEHEAD \
73 _PyTZINFO_HEAD \
74 unsigned char data[_PyDateTime_TIME_DATASIZE];
75
76typedef struct
77{
78 _PyDateTime_TIMEHEAD
79} _PyDateTime_BaseTime; /* hastzinfo false */
80
81typedef struct
82{
83 _PyDateTime_TIMEHEAD
84 PyObject *tzinfo;
85} PyDateTime_Time; /* hastzinfo true */
86
Tim Petersa9bc1682003-01-11 03:39:11 +000087
88/* All datetime objects are of PyDateTime_DateTimeType, but that can be
89 * allocated in two ways too, just like for time objects above. In addition,
90 * the plain date type is a base class for datetime, so it must also have
91 * a hastzinfo member (although it's unused there).
92 */
93typedef struct
94{
95 _PyTZINFO_HEAD
96 unsigned char data[_PyDateTime_DATE_DATASIZE];
97} PyDateTime_Date;
98
99#define _PyDateTime_DATETIMEHEAD \
100 _PyTZINFO_HEAD \
101 unsigned char data[_PyDateTime_DATETIME_DATASIZE];
Tim Petersa032d2e2003-01-11 00:15:54 +0000102
Tim Peters2a799bf2002-12-16 20:18:38 +0000103typedef struct
104{
Tim Petersa9bc1682003-01-11 03:39:11 +0000105 _PyDateTime_DATETIMEHEAD
106} _PyDateTime_BaseDateTime; /* hastzinfo false */
Tim Peters2a799bf2002-12-16 20:18:38 +0000107
108typedef struct
109{
Tim Petersa9bc1682003-01-11 03:39:11 +0000110 _PyDateTime_DATETIMEHEAD
Tim Peters2a799bf2002-12-16 20:18:38 +0000111 PyObject *tzinfo;
Tim Petersa9bc1682003-01-11 03:39:11 +0000112} PyDateTime_DateTime; /* hastzinfo true */
Tim Peters2a799bf2002-12-16 20:18:38 +0000113
Tim Peters37f39822003-01-10 03:49:02 +0000114
Tim Petersa9bc1682003-01-11 03:39:11 +0000115/* Apply for date and datetime instances. */
Tim Peters2a799bf2002-12-16 20:18:38 +0000116#define PyDateTime_GET_YEAR(o) ((((PyDateTime_Date*)o)->data[0] << 8) | \
117 ((PyDateTime_Date*)o)->data[1])
118#define PyDateTime_GET_MONTH(o) (((PyDateTime_Date*)o)->data[2])
119#define PyDateTime_GET_DAY(o) (((PyDateTime_Date*)o)->data[3])
120
121#define PyDateTime_DATE_GET_HOUR(o) (((PyDateTime_DateTime*)o)->data[4])
122#define PyDateTime_DATE_GET_MINUTE(o) (((PyDateTime_DateTime*)o)->data[5])
123#define PyDateTime_DATE_GET_SECOND(o) (((PyDateTime_DateTime*)o)->data[6])
124#define PyDateTime_DATE_GET_MICROSECOND(o) \
125 ((((PyDateTime_DateTime*)o)->data[7] << 16) | \
126 (((PyDateTime_DateTime*)o)->data[8] << 8) | \
127 ((PyDateTime_DateTime*)o)->data[9])
128
Tim Peters37f39822003-01-10 03:49:02 +0000129/* Apply for time instances. */
Tim Peters2a799bf2002-12-16 20:18:38 +0000130#define PyDateTime_TIME_GET_HOUR(o) (((PyDateTime_Time*)o)->data[0])
131#define PyDateTime_TIME_GET_MINUTE(o) (((PyDateTime_Time*)o)->data[1])
132#define PyDateTime_TIME_GET_SECOND(o) (((PyDateTime_Time*)o)->data[2])
133#define PyDateTime_TIME_GET_MICROSECOND(o) \
134 ((((PyDateTime_Time*)o)->data[3] << 16) | \
135 (((PyDateTime_Time*)o)->data[4] << 8) | \
136 ((PyDateTime_Time*)o)->data[5])
137
Tim Peters9ddf40b2004-06-20 22:41:32 +0000138
139/* Define structure for C API. */
140typedef struct {
141 /* type objects */
142 PyTypeObject *DateType;
143 PyTypeObject *DateTimeType;
144 PyTypeObject *TimeType;
145 PyTypeObject *DeltaType;
146 PyTypeObject *TZInfoType;
147
148 /* constructors */
149 PyObject *(*Date_FromDate)(int, int, int, PyTypeObject*);
150 PyObject *(*DateTime_FromDateAndTime)(int, int, int, int, int, int, int,
151 PyObject*, PyTypeObject*);
152 PyObject *(*Time_FromTime)(int, int, int, int, PyObject*, PyTypeObject*);
153 PyObject *(*Delta_FromDelta)(int, int, int, int, PyTypeObject*);
154
155 /* constructors for the DB API */
156 PyObject *(*DateTime_FromTimestamp)(PyObject*, PyObject*, PyObject*);
157 PyObject *(*Date_FromTimestamp)(PyObject*, PyObject*);
158
159} PyDateTime_CAPI;
160
161
162/* "magic" constant used to partially protect against developer mistakes. */
163#define DATETIME_API_MAGIC 0x414548d5
164
165#ifdef Py_BUILD_CORE
166
167/* Macros for type checking when building the Python core. */
Tim Peters2a799bf2002-12-16 20:18:38 +0000168#define PyDate_Check(op) PyObject_TypeCheck(op, &PyDateTime_DateType)
Christian Heimese93237d2007-12-19 02:37:44 +0000169#define PyDate_CheckExact(op) (Py_TYPE(op) == &PyDateTime_DateType)
Tim Peters2a799bf2002-12-16 20:18:38 +0000170
171#define PyDateTime_Check(op) PyObject_TypeCheck(op, &PyDateTime_DateTimeType)
Christian Heimese93237d2007-12-19 02:37:44 +0000172#define PyDateTime_CheckExact(op) (Py_TYPE(op) == &PyDateTime_DateTimeType)
Tim Peters2a799bf2002-12-16 20:18:38 +0000173
Tim Peters2a799bf2002-12-16 20:18:38 +0000174#define PyTime_Check(op) PyObject_TypeCheck(op, &PyDateTime_TimeType)
Christian Heimese93237d2007-12-19 02:37:44 +0000175#define PyTime_CheckExact(op) (Py_TYPE(op) == &PyDateTime_TimeType)
Tim Peters2a799bf2002-12-16 20:18:38 +0000176
Tim Peters2a799bf2002-12-16 20:18:38 +0000177#define PyDelta_Check(op) PyObject_TypeCheck(op, &PyDateTime_DeltaType)
Christian Heimese93237d2007-12-19 02:37:44 +0000178#define PyDelta_CheckExact(op) (Py_TYPE(op) == &PyDateTime_DeltaType)
Tim Peters2a799bf2002-12-16 20:18:38 +0000179
180#define PyTZInfo_Check(op) PyObject_TypeCheck(op, &PyDateTime_TZInfoType)
Christian Heimese93237d2007-12-19 02:37:44 +0000181#define PyTZInfo_CheckExact(op) (Py_TYPE(op) == &PyDateTime_TZInfoType)
Tim Peters2a799bf2002-12-16 20:18:38 +0000182
Tim Peters9ddf40b2004-06-20 22:41:32 +0000183#else
184
185/* Define global variable for the C API and a macro for setting it. */
186static PyDateTime_CAPI *PyDateTimeAPI;
187
188#define PyDateTime_IMPORT \
189 PyDateTimeAPI = (PyDateTime_CAPI*) PyCObject_Import("datetime", \
190 "datetime_CAPI")
191
192/* This macro would be used if PyCObject_ImportEx() was created.
193#define PyDateTime_IMPORT \
194 PyDateTimeAPI = (PyDateTime_CAPI*) PyCObject_ImportEx("datetime", \
195 "datetime_CAPI", \
196 DATETIME_API_MAGIC)
197*/
198
199/* Macros for type checking when not building the Python core. */
200#define PyDate_Check(op) PyObject_TypeCheck(op, PyDateTimeAPI->DateType)
Christian Heimese93237d2007-12-19 02:37:44 +0000201#define PyDate_CheckExact(op) (Py_TYPE(op) == PyDateTimeAPI->DateType)
Tim Peters9ddf40b2004-06-20 22:41:32 +0000202
203#define PyDateTime_Check(op) PyObject_TypeCheck(op, PyDateTimeAPI->DateTimeType)
Christian Heimese93237d2007-12-19 02:37:44 +0000204#define PyDateTime_CheckExact(op) (Py_TYPE(op) == PyDateTimeAPI->DateTimeType)
Tim Peters9ddf40b2004-06-20 22:41:32 +0000205
206#define PyTime_Check(op) PyObject_TypeCheck(op, PyDateTimeAPI->TimeType)
Christian Heimese93237d2007-12-19 02:37:44 +0000207#define PyTime_CheckExact(op) (Py_TYPE(op) == PyDateTimeAPI->TimeType)
Tim Peters9ddf40b2004-06-20 22:41:32 +0000208
209#define PyDelta_Check(op) PyObject_TypeCheck(op, PyDateTimeAPI->DeltaType)
Christian Heimese93237d2007-12-19 02:37:44 +0000210#define PyDelta_CheckExact(op) (Py_TYPE(op) == PyDateTimeAPI->DeltaType)
Tim Peters9ddf40b2004-06-20 22:41:32 +0000211
212#define PyTZInfo_Check(op) PyObject_TypeCheck(op, PyDateTimeAPI->TZInfoType)
Christian Heimese93237d2007-12-19 02:37:44 +0000213#define PyTZInfo_CheckExact(op) (Py_TYPE(op) == PyDateTimeAPI->TZInfoType)
Tim Peters9ddf40b2004-06-20 22:41:32 +0000214
215/* Macros for accessing constructors in a simplified fashion. */
216#define PyDate_FromDate(year, month, day) \
217 PyDateTimeAPI->Date_FromDate(year, month, day, PyDateTimeAPI->DateType)
218
219#define PyDateTime_FromDateAndTime(year, month, day, hour, min, sec, usec) \
220 PyDateTimeAPI->DateTime_FromDateAndTime(year, month, day, hour, \
221 min, sec, usec, Py_None, PyDateTimeAPI->DateTimeType)
222
223#define PyTime_FromTime(hour, minute, second, usecond) \
224 PyDateTimeAPI->Time_FromTime(hour, minute, second, usecond, \
225 Py_None, PyDateTimeAPI->TimeType)
226
227#define PyDelta_FromDSU(days, seconds, useconds) \
Brett Cannon1ed57052004-06-28 00:48:30 +0000228 PyDateTimeAPI->Delta_FromDelta(days, seconds, useconds, 1, \
Tim Peters9ddf40b2004-06-20 22:41:32 +0000229 PyDateTimeAPI->DeltaType)
230
231/* Macros supporting the DB API. */
232#define PyDateTime_FromTimestamp(args) \
233 PyDateTimeAPI->DateTime_FromTimestamp( \
234 (PyObject*) (PyDateTimeAPI->DateTimeType), args, NULL)
235
236#define PyDate_FromTimestamp(args) \
237 PyDateTimeAPI->Date_FromTimestamp( \
238 (PyObject*) (PyDateTimeAPI->DateType), args)
239
240#endif /* Py_BUILD_CORE */
241
242#ifdef __cplusplus
243}
244#endif
Tim Peters2a799bf2002-12-16 20:18:38 +0000245#endif