|
12 | 12 | import math |
13 | 13 | from functools import reduce |
14 | 14 | from fractions import Fraction |
| 15 | +from ._extensions._pyedflib import FILETYPE_EDFPLUS, FILETYPE_BDFPLUS, FILETYPE_BDF, FILETYPE_EDF |
| 16 | +from ._extensions._pyedflib import open_file_writeonly, set_physical_maximum, set_patient_additional, set_digital_maximum |
| 17 | +from ._extensions._pyedflib import set_birthdate, set_digital_minimum, set_technician, set_recording_additional, set_patientname |
| 18 | +from ._extensions._pyedflib import set_patientcode, set_equipment, set_admincode, set_sex, set_datarecord_duration, set_number_of_annotation_signals |
| 19 | +from ._extensions._pyedflib import set_startdatetime, set_starttime_subsecond, set_samples_per_record, set_physical_minimum, set_label, set_physical_dimension |
| 20 | +from ._extensions._pyedflib import set_transducer, set_prefilter, write_physical_samples, close_file, write_annotation_latin1_hr, write_annotation_utf8_hr |
| 21 | +from ._extensions._pyedflib import blockwrite_physical_samples, write_errors, blockwrite_digital_samples, write_digital_short_samples, write_digital_samples, blockwrite_digital_short_samples |
15 | 22 |
|
16 | 23 | import numpy as np |
17 | 24 |
|
|
47 | 54 | set_starttime_subsecond, |
48 | 55 | set_technician, |
49 | 56 | set_transducer, |
50 | | - write_annotation_latin1, |
51 | | - write_annotation_utf8, |
| 57 | + write_annotation_latin1_hr, |
| 58 | + write_annotation_utf8_hr, |
52 | 59 | write_digital_samples, |
53 | 60 | write_digital_short_samples, |
54 | 61 | write_errors, |
@@ -1013,16 +1020,16 @@ def writeAnnotation(self, onset_in_seconds: Union[int, float], duration_in_secon |
1013 | 1020 |
|
1014 | 1021 | if str_format == 'utf_8': |
1015 | 1022 | if duration_in_seconds >= 0: |
1016 | | - return write_annotation_utf8(self.handle, np.round(onset_in_seconds*10000).astype(np.int64), np.round(duration_in_seconds*10000).astype(int), du(description)) |
| 1023 | + return write_annotation_utf8_hr(self.handle, np.round(onset_in_seconds*1000000).astype(np.int64), np.round(duration_in_seconds*1000000).astype(np.int64), du(description)) |
1017 | 1024 | else: |
1018 | | - return write_annotation_utf8(self.handle, np.round(onset_in_seconds*10000).astype(np.int64), -1, du(description)) |
| 1025 | + return write_annotation_utf8_hr(self.handle, np.round(onset_in_seconds*1000000).astype(np.int64), -1, du(description)) |
1019 | 1026 | else: |
1020 | 1027 | if duration_in_seconds >= 0: |
1021 | 1028 | # FIX: description must be bytes. string will fail in u function |
1022 | | - return write_annotation_latin1(self.handle, np.round(onset_in_seconds*10000).astype(np.int64), np.round(duration_in_seconds*10000).astype(int), u(description).encode('latin1')) # type: ignore |
| 1029 | + return write_annotation_latin1_hr(self.handle, np.round(onset_in_seconds*1000000).astype(np.int64), np.round(duration_in_seconds*1000000).astype(np.int64), u(description).encode('latin1')) # type: ignore |
1023 | 1030 | else: |
1024 | 1031 | # FIX: description must be bytes. string will fail in u function |
1025 | | - return write_annotation_latin1(self.handle, np.round(onset_in_seconds*10000).astype(np.int64), -1, u(description).encode('latin1')) # type: ignore |
| 1032 | + return write_annotation_latin1_hr(self.handle, np.round(onset_in_seconds*1000000).astype(np.int64), -1, u(description).encode('latin1')) # type: ignore |
1026 | 1033 |
|
1027 | 1034 | def close(self) -> None: |
1028 | 1035 | """ |
|
0 commit comments