Using python's numpy/scipy in SOPs

[Jason]

Hi all,

Ever since I read about the functionality in scipy, I've wanted to see how it could enhance volumes in Houdini. So I suddenly realized our Systems people here at work installed it sometime or other and it was available within HOM! Woo!

Anyhow, please see how easy this is... and useful.

Here is the code of the SOP too:

import numpy
import scipy
import scipy.ndimage


# This code is called when instances of this SOP cook.
geo = hou.pwd().geometry()

# Add code to modify the contents of geo.
prim_number = hou.pwd().parm("prim_number").eval()
volume = geo.prims()[prim_number]
np = numpy.array(volume.allVoxels()).reshape(volume.resolution())
# Now we have our numpy 3d array... have fun!

if hou.pwd().parm("do_gaussian").eval():
  sigma= hou.pwd().parm("gaussian_sigma").eval()
  np = scipy.ndimage.filters.gaussian_filter( np, sigma  )

if hou.pwd().parm("do_uniform").eval():
  uniform_size = hou.pwd().parm("uniform_size").eval()
  np = scipy.ndimage.filters.uniform_filter( np, uniform_size  )

if hou.pwd().parm("do_sobel").eval():
  np = scipy.ndimage.filters.sobel( np  )

if hou.pwd().parm("do_prewitt").eval():
  np = scipy.ndimage.filters.prewitt( np  )

if hou.pwd().parm("do_abs").eval():
  np = abs(np)

# set the values of the volume back in Houdini
volume.setAllVoxels( np.flatten() )

I hope someone finds this useful in some way. In fact I was surprised at the speed of this thing: a 100^3 volume gets sent to scipy and filtered and returned to Houdini in about 1 second!

Take care,

Jason

scipy_volumefilter_v1.otl

[zarti]

i thought ( or even read , esp about numpy ) that this wasn't possible , but hey .. glad to see it is 'working' .

thank you Jason ! looks fun and fast too .. =)

.cheers

Edited May 25, 2011 by zarti

[graham]

There's an example of using numpy on the HOM doc page about writing a COP operator.

http://www.sidefx.com/docs/houdini11.0/hom/pythoncop

[Jason]

There's an example of using numpy on the HOM doc page about writing a COP operator.

http://www.sidefx.com/docs/houdini11.0/hom/pythoncop

Ah, that would have helped me figure out the ...AsString() method, had I known about this beforehand!

Now wouldn't it be fun to write a Python VOPnet context in which we could process volumes, rasters, etc? Or does that exist somewhere already, Mr.Thompson?

[Jason]

There are also some nice scipy.spatial.KDTree and Delauney routines available; for those who can't wait until SESI create HOM functions for UT_KDTree and such.

[graham]

Attached is a HOM_PointTree class I created that wraps around the KDTree implementation. It's designed to be kind of similar to GEO_PointTree in that it takes a hou.Geometry input and builds a tree you can then query using a hou.Vector3 object to get the nearest N points, or points within some distance. It returns tuples of hou.Points as the result.

treeutils.zip

[Jason]

Great stuff!

[rdg]

Sweet!

Are there R-Trees as well?

[zarti]

i thought ( or even read , esp about numpy ) that this wasn't possible , ..

.. which results to be pretty easy in Ubuntu .

.cheers

[dulo]

Created x64 builds for windows of numpy, scipy, vtk and opencv (CUDA).

If you are interested in testing take a look at

http://www.sidefx.com/index.php?option=com_forum&Itemid=172&page=viewtopic&p=109229#109229

greetings dulo

[sibarrick]

When I try and import scipy I just get an error, has this been removed from Houdini since H12.5?

[ikarus]

numpy/scipy are external python libraries not included in houdini

[graham]

I don't think it was ever shipped with Houdini. Has your system python configuration changed since you last tried (assuming you are on linux) ?