viewRenderOverrideShadows/viewRenderOverrideShadows.h

//-
// Copyright 2012 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.
//+

#include <stdio.h>

#include <maya/MString.h>
#include <maya/MColor.h>
#include <maya/MViewport2Renderer.h>
#include <maya/MRenderTargetManager.h>
#include <maya/MShaderManager.h>
#include <maya/MSelectionList.h>

class viewRenderOverrideShadows : public MHWRender::MRenderOverride
{
public:
    enum
    {
        kShadowPrePass,
        kMaya3dSceneRender,
        kPresentOp,
        kOperationCount
    };

    viewRenderOverrideShadows(const MString& name);
    virtual ~viewRenderOverrideShadows();

    virtual MHWRender::DrawAPI supportedDrawAPIs() const;
    virtual bool startOperationIterator();
    virtual MHWRender::MRenderOperation * renderOperation();
    virtual bool nextRenderOperation();
    virtual MStatus setup(const MString& destination);
    virtual MStatus cleanup();
    virtual MString uiName() const
    {
        return mUIName;
    }

protected:
    MStatus updateRenderOperations();
    MStatus updateShaders(const MHWRender::MShaderManager* shaderMgr);
    MStatus updateLightList();

    MString mUIName;
    MColor mClearColor;

    MHWRender::MRenderOperation * mRenderOperations[kOperationCount];
    MString mRenderOperationNames[kOperationCount];
    bool mRenderOperationEnabled[kOperationCount];
    int mCurrentOperation;

    MHWRender::MShaderInstance * mLightShader;
    MSelectionList mLightList;
};

// Scene render to output to targets
class sceneRender : public MHWRender::MSceneRender
{
public:
    sceneRender(const MString &name);
    virtual ~sceneRender();

    virtual const MHWRender::MShaderInstance* shaderOverride();
    virtual void preSceneRender(const MHWRender::MDrawContext & context);

    void setShader( MHWRender::MShaderInstance *shader)
    {
        mLightShader = shader;
    }
    void setLightList( MSelectionList *val )
    {
        mLightList = val;
    }

protected:
    MHWRender::MShaderInstance *mLightShader;
    MSelectionList *mLightList;
};

// Shadow prepass operation
class shadowPrepass : public MHWRender::MUserRenderOperation
{
public:
    shadowPrepass(const MString &name);
    virtual ~shadowPrepass();

    virtual MStatus execute( const MHWRender::MDrawContext & drawContext );
    virtual bool requiresLightData() const
    {
        return true;
    }
    void setLightList( MSelectionList *val )
    {
        mLightList = val;
    }

    MSelectionList *mLightList;
};