narrowPolyViewerCmd.cpp

//-
// ==========================================================================
// 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.
// ==========================================================================
//+

//      File Name: narrowPolyViewerCmd.cpp
//
//      Description:
//              A command used for testing multiple camera drawing into a single view.
//              See narrowPolyViewerCmd.h for a description.
//


#include "narrowPolyViewerCmd.h"

#include <maya/MItDag.h>
#include <maya/MGlobal.h>
#include <maya/MSyntax.h>
#include <maya/MArgParser.h>
#include <maya/MArgList.h>
#include <maya/MSelectionList.h>
#include <maya/MFnCamera.h>

#include <maya/MIOStream.h>

narrowPolyViewerCmd::narrowPolyViewerCmd()
:       MPxModelEditorCommand()
//
//      Description:
//              Class constructor.
//
{
}

narrowPolyViewerCmd::~narrowPolyViewerCmd()
//
//      Description:
//              Class destructor.
//
{
}

void* narrowPolyViewerCmd::creator()
//
//      Description:
//              Create the command.
//
{
    return new narrowPolyViewerCmd();
}

MStatus narrowPolyViewerCmd::appendSyntax()
//
//      Description:
//              Add syntax to the command. All of the parent syntax is added
//              before this call is made.
//
{
        MStatus ReturnStatus;

        MSyntax theSyntax = syntax(&ReturnStatus);
        if (MS::kSuccess != ReturnStatus) {
                MGlobal::displayError("Could not get the parent's syntax");
                return ReturnStatus;
        }

        theSyntax.addFlag(kInitFlag,
                                          kInitFlagLong);

        theSyntax.addFlag(kResultsFlag,
                                          kResultsFlagLong);

        theSyntax.addFlag(kClearFlag,
                                          kClearFlagLong);

        theSyntax.addFlag(kToleranceFlag,
                              kToleranceFlagLong,
                                          MSyntax::kDouble);

        return ReturnStatus;
}

MStatus narrowPolyViewerCmd::doEditFlags()
//
//      Description:
//              Handle edits for flags added by this class.
//              If the flag is unknown, return MS::kSuccess and the parent class
//              will attempt to process the flag. Returning MS::kUnknownParameter
//              will cause the parent class to process the flag.
//
{
        MStatus ReturnStatus = MS::kSuccess;

        MPx3dModelView *user3dModelView = modelView();
        if (NULL == user3dModelView) {
                MGlobal::displayError("NULL == user3dModelView!");
                return MS::kFailure;
        }

        if (user3dModelView->viewType() != "narrowPolyViewer") {
                MGlobal::displayError("This view is not a narrowPolyViewer");
                return MS::kFailure;
        }

        //      This is now safe to do, since the above test passed.
        //
        narrowPolyViewer *mpView = (narrowPolyViewer *)user3dModelView;

        MArgParser argData = parser();
        if (argData.isFlagSet(kInitFlag)) {
                return initTests(*mpView);
        } else if (argData.isFlagSet(kResultsFlag)) {
                return testResults(*mpView);
        } else if (argData.isFlagSet(kClearFlag)) {
                return clearResults(*mpView);
        } else if (argData.isFlagSet(kToleranceFlag)) {
                double tol;
                argData.getFlagArgument(kToleranceFlag, 0, tol);
                mpView->tol = tol;
                mpView->refresh(true, true);
        } else {
                return MS::kUnknownParameter;
        }

        return ReturnStatus;
}

MStatus narrowPolyViewerCmd::doQueryFlags()
//
//      Description:
//              Don't do anything for the query.
//
{
        return ParentClass::doQueryFlags();
}

MStatus narrowPolyViewerCmd::clearResults(narrowPolyViewer &view)
//
//      Description:
//
{       
        view.removeAllCameras();
        fCameraList.clear();
        return MS::kSuccess;
}

MStatus narrowPolyViewerCmd::initTests(narrowPolyViewer &view)
//
//      Description:
//
{
        MGlobal::displayInfo("narrowPolyViewerCmd::initTests");

        clearResults(view);

        //      Add every camera into the scene. Don't change the main camera,
        //      it is OK that it gets reused.
        //
        MDagPath cameraPath;
        MStatus status = MS::kSuccess;
        MItDag dagIterator(MItDag::kDepthFirst, MFn::kCamera);
        for (; !dagIterator.isDone(); dagIterator.next()) {
                if (!dagIterator.getPath(cameraPath)) {
                        continue;
                }

                MFnCamera camera(cameraPath, &status);
                if (MS::kSuccess != status) {
                        continue;
                }

                MGlobal::displayInfo(camera.fullPathName());
                fCameraList.append(cameraPath);
        }

        if (MS::kSuccess !=
                view.setCameraList(fCameraList)) {
                MGlobal::displayError("Could not set the list of cameras");
                return MS::kFailure;
        } else {
                view.refresh();
        }

        return MS::kSuccess;
}

MStatus narrowPolyViewerCmd::testResults(narrowPolyViewer &view)
{
        cout << "fCameraList.length() = " << fCameraList.length() << endl;
        cout << "fCameraList = " << fCameraList << endl;

        cout << "view.fTestCameraList.length() = " << view.fTestCameraList.length() << endl;
        cout << "view.fTestCameraList = " << view.fTestCameraList << endl;

        return MS::kSuccess;
}