Merge pull request #1690 from luiztauffer/fix-blackrock-overflow

Blackrock rawio fix overflow

Author: zm711

neo/rawio/blackrockrawio.py

@@ -1058,8 +1058,8 @@ def __read_nsx_data_variant_a(self, nsx_nb):
         filename = ".".join([self._filenames["nsx"], f"ns{nsx_nb}"])
 
         # get shape of data
-        shape = (self.__nsx_params["2.1"](nsx_nb)["nb_data_points"], self.__nsx_basic_header[nsx_nb]["channel_count"])
-        offset = self.__nsx_params["2.1"](nsx_nb)["bytes_in_headers"]
+        shape = (int(self.__nsx_params["2.1"](nsx_nb)["nb_data_points"]), int(self.__nsx_basic_header[nsx_nb]["channel_count"]))
+        offset = int(self.__nsx_params["2.1"](nsx_nb)["bytes_in_headers"])
 
         # read nsx data
         # store as dict for compatibility with higher file specs
@@ -1078,10 +1078,10 @@ def __read_nsx_data_variant_b(self, nsx_nb):
         for data_bl in self.__nsx_data_header[nsx_nb].keys():
             # get shape and offset of data
             shape = (
-                self.__nsx_data_header[nsx_nb][data_bl]["nb_data_points"],
-                self.__nsx_basic_header[nsx_nb]["channel_count"],
+                int(self.__nsx_data_header[nsx_nb][data_bl]["nb_data_points"]),
+                int(self.__nsx_basic_header[nsx_nb]["channel_count"]),
             )
-            offset = self.__nsx_data_header[nsx_nb][data_bl]["offset_to_data_block"]
+            offset = int(self.__nsx_data_header[nsx_nb][data_bl]["offset_to_data_block"])
 
             # read data
             data[data_bl] = np.memmap(filename, dtype="int16", shape=shape, offset=offset, mode="r")
@@ -1828,7 +1828,7 @@ def __get_file_size(self, filename):
         """
         filebuf = open(filename, "rb")
         filebuf.seek(0, os.SEEK_END)
-        file_size = filebuf.tell()
+        file_size = int(filebuf.tell())
         filebuf.close()
 
         return file_size
@@ -2007,8 +2007,8 @@ def __get_nsx_param_variant_a(self, nsx_nb):
 
         nsx_parameters = {
             "nb_data_points": int(
-                (self.__get_file_size(filename) - bytes_in_headers)
-                / (2 * self.__nsx_basic_header[nsx_nb]["channel_count"])
+                (int(self.__get_file_size(filename)) - int(bytes_in_headers))
+                / int(2 * self.__nsx_basic_header[nsx_nb]["channel_count"])
                 - 1
             ),
             "labels": labels,