InputMesh.h

#ifndef __InputMesh_h__
#define __InputMesh_h__

#include "Input.h"

#include <HAPI/HAPI.h>

#include <maya/MFnMesh.h>

class InputMesh : public Input
{
public:
    InputMesh();
    virtual ~InputMesh();

    virtual AssetInputType assetInputType() const;

    virtual void setInputGeo(MDataBlock &dataBlock, const MPlug &plug);

    virtual void setInputComponents(MDataBlock &dataBlock,
                                    const MPlug &geoPlug,
                                    const MPlug &compPlug,
                                    const MPlug &primGroupPlug,
                                    const MPlug &pointGroupPlug);

protected:
    bool processPoints(const MFnMesh &meshFn);
    bool processNormals(const MObject &meshObj,
                        const MFnMesh &meshFn,
                        std::vector<int> vertexCount);
    bool processUVs(const MFnMesh &meshFn,
                    std::vector<int> vertexCount,
                    std::vector<int> vertexList);
    bool processColorSets(const MFnMesh &meshFn,
                          std::vector<int> vertexCount,
                          std::vector<int> vertexList);
    bool processSets(const MPlug &plug, const MFnMesh &meshFn);
    bool processShadingGroups(const MFnMesh &meshFn,
                              const MStringArray &sgNames,
                              const MObjectArray &sgCompObjs);
};

#endif