detectors/aare
25
0
Fork 0
mirror of https://github.com/slsdetectorgroup/aare.git synced 2025-06-23 11:57:58 +02:00
Code Issues Actions Packages Releases Activity

add python bindings for numpy

Browse Source
This commit is contained in:
Bechir Braham
2024-03-11 16:05:54 +01:00
parent 25d282717c
commit c4c88c50d1
5 changed files with 49 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
core/include/aare/Frame.hpp
View File

@ -22,6 +22,7 @@ template <class DataType> class Frame {
Frame(ssize_t rows, ssize_t cols);
Frame(std::byte *fp, ssize_t rows, ssize_t cols);
DataType get(int row, int col);
std::vector<std::vector<DataType>> get_array();
ssize_t rows() const{
return m_rows;
}

12
core/src/Frame.cpp
View File

@ -25,5 +25,17 @@ DataType Frame<DataType>::get(int row, int col) {
return m_data[row*m_cols + col];
}
template <typename DataType>
std::vector<std::vector<DataType>> Frame<DataType>::get_array() {
std::vector<std::vector<DataType>> array;
for (int i = 0; i < m_rows; i++) {
std::vector<DataType> row;
row.assign(m_data + i*m_cols, m_data + (i+1)*m_cols);
array.push_back(row);
}
return array;
}
template class Frame<uint16_t>;