imaya_helper.h

Go to the documentation of this file.

#ifndef _IMAYA_HELPER_H_
#define _IMAYA_HELPER_H_

// includes, maya
#include <maya/MPoint.h>
#include <maya/MDagPath.h>
#include <maya/MGlobal.h>
#include <maya/MItSelectionList.h>

// types, exported (class, enum, struct, union, typedef)

namespace Util {


//@brief For now we assume only one point light source position will be returned.
inline void
getPointLightPosition( MPoint &Translation ) {
  
  // Status and intermediate status
  MStatus status( MS::kFailure );

  // Create a selection list of all the objects in the scene.
  MSelectionList selectionList;
  MString allLights("*");

  // save the current selection mask
  MStatus stat;
  MSelectionMask select_mask_prev( MGlobal::objectSelectionMask( &stat));
  CONDITION( MS::kSuccess == stat);


  MSelectionMask select_mask( MSelectionMask::kSelectLights );
  MGlobal::setObjectSelectionMask( select_mask);
  MGlobal::getSelectionListByName( allLights, selectionList );

  // DAG path of point light source and it's transform,
  // and transform object.
  MDagPath pLTransformDagPath;

  // Iterate over the selection list but filter 
  // everything except for Point Light Sources.
  MItSelectionList iterator( selectionList );
  iterator.setFilter( MFn::kPointLight );

  for( iterator.reset(); !iterator.isDone(); iterator.next() ) {

    iterator.getDagPath( pLTransformDagPath );

    // Pop off the shape node from the path,
    // leaving the transform node.
    pLTransformDagPath.pop();

    MFnTransform theTransform( pLTransformDagPath );
    Translation = MPoint( MVector(theTransform.getTranslation( MSpace::kTransform , &status )) );

    // Note: at this point, we only keep the first
    // point light source iterated through.
    break;
  }

  // reset selection mask
  MGlobal::setObjectSelectionMask( select_mask_prev);

}

}

// variables, exported (extern)

// functions, inlined (inline)
  
// functions, exported (extern)
  
#endif // #ifndef _PROTO_H_

---