apiSimpleShapeUI.h

//-
// ==========================================================================
// Copyright 1995,2006,2008 Autodesk, Inc. All rights reserved.
//
// Use of this software is subject to the terms of the Autodesk
// license agreement provided at the time of installation or download,
// or which otherwise accompanies this software in either electronic
// or hard copy form.
// ==========================================================================
//+

//
// apiSimpleShapeUI.h
//
// Encapsulates the UI portion of a user defined shape. All of the
// drawing and selection code goes here.
//

#include <maya/MPxSurfaceShapeUI.h> 
#include <apiSimpleShape.h> 

class apiSimpleShapeUI : public MPxSurfaceShapeUI
{
public:
        apiSimpleShapeUI();
        virtual ~apiSimpleShapeUI(); 

        //
        // Overrides
        //

        // Puts draw request on the draw queue
        //
        virtual void    getDrawRequests( const MDrawInfo & info,
                                                                         bool objectAndActiveOnly,
                                                                         MDrawRequestQueue & requests );

        // Main draw routine. Gets called by maya with draw requests.
        //
        virtual void    draw( const MDrawRequest & request,
                                                  M3dView & view ) const;

        // Main selection routine
        //
        virtual bool    select( MSelectInfo &selectInfo,
                                                        MSelectionList &selectionList,
                                                        MPointArray &worldSpaceSelectPts ) const;

        //
        // Helper routines
        //

        void    drawVertices( const MDrawRequest & request, M3dView & view ) const;

        bool    selectVertices( MSelectInfo &selectInfo,
                                MSelectionList &selectionList,
                                MPointArray &worldSpaceSelectPts ) const;

        static  void *      creator();

private:
        // Draw Tokens
        //
        enum {
                kDrawVertices, // component token
                kDrawWireframe,
                kDrawWireframeOnShaded,
                kDrawSmoothShaded,
                kDrawFlatShaded,
                kLastToken
        };
};