Skip to content

Added getArray() python API method to get all particle attribute data in a single python tuple #31

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Closed
mattebb wants to merge 6 commits into from
Closed

Added getArray() python API method to get all particle attribute data in a single python tuple #31

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
6 commits
Select commit Hold shift + click to select a range
b911132
gitignore
mattebb Oct 10, 2012
fbce9f6
Added python API method to get all particle attribute data in a singl…
mattebb Oct 10, 2012
79f100e
Initial framework for retrieving particle data as a numpy array
mattebb Oct 14, 2012
6c8669b
gitignore
mattebb Oct 14, 2012
628e1fd
Initial code to fill data to numpy array
mattebb Oct 14, 2012
51b2a2a
Cleaned up numpy python bindings
mattebb Oct 15, 2012
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
23 changes: 0 additions & 23 deletions .gitignore
View file
Open in desktop

This file was deleted.

106 changes: 0 additions & 106 deletions CMakeLists.txt
View file
Open in desktop

This file was deleted.

67 changes: 0 additions & 67 deletions src/py/CMakeLists.txt
View file
Open in desktop

This file was deleted.

59 changes: 58 additions & 1 deletion src/py/partio.i
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -37,8 +37,8 @@ OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGES.
%module partio
%include "std_string.i"


%{
#include <numpy/arrayobject.h>
#include <Partio.h>
namespace Partio{
typedef uint64_t ParticleIndex;
Expand All @@ -57,7 +57,10 @@ struct fixedFloatArray
int i[16];
};
};
%}

%init %{
import_array();
%}

// Particle Types
Expand Down Expand Up @@ -235,6 +238,60 @@ public:
return list;
}

%feature("autodoc");
%feature("docstring","Get");
PyObject* getNDArray(const ParticleAttribute& attr)
{
unsigned int numparticles = $self->numParticles();

// 1 dimensional for now
npy_intp dims[1] = { numparticles*attr.count };
PyObject *array = PyArray_SimpleNew(1, dims, NPY_FLOAT);

if (!array) {
PyErr_SetString(PyExc_TypeError,"Unable to create array");
return NULL;
}

npy_intp size;
unsigned int i=0;
float *dptr;
size = PyArray_SIZE(array);
dptr = (float *)PyArray_DATA(array);

for (int j=0;j<size;j+=3) {
const float* p=$self->data<float>(attr,i);
for(int k=0;k<attr.count;k++) {
dptr[0] = p[k];
dptr++;
}
i++;
}

return PyArray_Return((PyArrayObject *)array);
}

%feature("autodoc");
%feature("docstring","Gets a single flattened tuple, containing attribute data for all particles");
PyObject* getArray(const ParticleAttribute& attr)
{
unsigned int numparticles = $self->numParticles();
PyObject* tuple=PyTuple_New(numparticles * attr.count);

if(attr.type==Partio::INT){
for(unsigned int i=0;i<numparticles;i++) {
const int* p=$self->data<int>(attr,i);
for(int k=0;k<attr.count;k++) PyTuple_SetItem(tuple, i*attr.count+k, PyInt_FromLong(p[k]));
}
}else{
for(unsigned int i=0;i<numparticles;i++) {
const float* p=$self->data<float>(attr,i);
for(int k=0;k<attr.count;k++) PyTuple_SetItem(tuple, i*attr.count+k, PyFloat_FromDouble(p[k]));
}
}
return tuple;
}

%feature("autodoc");
%feature("docstring","Gets attribute data for particleIndex'th particle");
PyObject* get(const ParticleAttribute& attr,const ParticleIndex particleIndex)
Expand Down