spacepaste

Paste details

reply | raw

  1.  
  2. from cffi import FFI
  3. import numpy as np
  5. s = '''
  6. typedef struct
  7. {
  8. int a;
  9. int b;
  10. float c;
  11. double d;
  12. } mystruct;
  13. '''
  14. ffi = FFI()
  15. ffi.cdef(s)
  17. m_arr = ffi.new("mystruct[2]")
  19. #create array and fill with dummy data
  20. for k in range(2):
  22. m = m_arr[k]
  24. m.a = k
  25. m.b = k + 1
  26. m.c = k + 2.0
  27. m.d = k + 3.0
  29. print(m_arr)
  31. # dtype for structured array in Numpy
  32. dt = np.dtype({'names': ('a', 'b', 'c', 'd'),
  33. 'formats': ('i4', 'i4', 'f4', 'f8'),
  34. 'offset': (ffi.offsetof('mystruct', 'a'),
  35. ffi.offsetof('mystruct', 'b'),
  36. ffi.offsetof('mystruct', 'c'),
  37. ffi.offsetof('mystruct', 'd'),
  38. )
  39. }, align=True)
  41. # member size, 20 bytes
  42. print('size, manually', 4 + 4 + 4 + 8)
  44. # total size of struct, 24 bytes
  45. print('sizeof', ffi.sizeof(m_arr[0]))
  48. #reason is member padding in structs
  50. buf = ffi.buffer(m_arr)
  51. print(buf)
  53. x = np.frombuffer(buf, dtype=dt)
  54. print(x)
  56.  
Powered by spacepaste. Paste removal.