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

// DESCRIPTION:
// 
// Produces the MEL command "referenceQuery".
//
// This example provides useful information about referenced files
// in the main scene. For each referenced file, the output format is as follows:
//
//  Referenced File: filename1
//  Connections Made
//      sourceAttribute -> destinationAttribute
//      ...
//  Connections Broken
//      sourceAttribute -> destinationAttribute
//      ...
//  Attributes Changed Since File Referenced
//      attribute1
//      attribute2
//      ...
//
// To use the plug-in:
//  (1) Open a scene file that contains file references.
//  (2) Execute the MEL command "referenceQuery".
//      The reference information is written to standard output. 
//

#include <maya/MString.h>
#include <maya/MStatus.h>
#include <maya/MObject.h>
#include <maya/MFileIO.h>
#include <maya/MSelectionList.h>
#include <maya/MGlobal.h>
#include <maya/MPxCommand.h>
#include <maya/MFnPlugin.h>

#include <maya/MIOStream.h>

class referenceQuery : public MPxCommand { 
public:
                        referenceQuery();
    virtual             ~referenceQuery();

    MStatus             doIt( const MArgList & );
    bool                isUndoable() const;

    static      void   *creator();
};


referenceQuery::referenceQuery()
{
}

referenceQuery::~referenceQuery()
{
}

void *referenceQuery::creator()
{
    return new referenceQuery;
}

bool referenceQuery::isUndoable() const {
    return false;
};

MStatus referenceQuery::doIt( const MArgList &  )
{
    MStringArray referenceFiles;
    MFileIO::getReferences( referenceFiles );

    // Print some useful information about the files referenced
    // in the main scene.
    //
    // Output format is as follows:
    // 
    // Referenced File: filename1
    //      Connections Made
    //          source -> destination
    //          ...
    // 
    //      Connections Broken
    //          source ->destination
    //          ...
    //
    //      Attributes Changed Since File Referenced
    //          attribute1
    //          attribute2
    //          ...
    //
    for( unsigned i = 0; i < referenceFiles.length(); i++ ) {
        MStringArray    connectionsMade;
        MFileIO::getReferenceConnectionsMade( referenceFiles[i],
                                              connectionsMade );

        cout << "Referenced File: " << referenceFiles[i].asChar() << ":\n";
        cout << "   Connections Made:\n";
        unsigned j;
        for( j = 0; j < connectionsMade.length(); j+=2 ) {
            cout << "   ";
            cout << connectionsMade[j].asChar() << " -> ";
            if( j + 1 < connectionsMade.length() ) {
                cout << connectionsMade[j+1].asChar();
            }
            cout << "\n" ;
        }
        cout << "\n";

        MStringArray    connectionsBroken;
        MFileIO::getReferenceConnectionsBroken( referenceFiles[i],
                                              connectionsBroken );
        cout << "   Connections Broken: \n";
        for( j = 0; j < connectionsBroken.length(); j+=2 ) {
            cout << "   ";
            cout << connectionsBroken[j].asChar() << " -> ";
            if( j + 1 < connectionsBroken.length() ) {
                cout << connectionsBroken[j+1].asChar();
            }
            cout << "\n" ;
        }
        cout << "\n";

        MStringArray    referencedNodes;
        
        cout << "   Attrs Changed Since File Open:\n";
        MFileIO::getReferenceNodes( referenceFiles[i], referencedNodes );
        for( j = 0; j < referencedNodes.length(); j++ ) {
            // For each node, call a MEL command to get its
            // attributes.  Say we're only interested in scalars.
            //
            MString cmd( "listAttr -s -cfo " );
            cmd = cmd + referencedNodes[j];
            
            MSelectionList referencedPlugs;
            MStringArray   referencedAttributes;    

            MGlobal::executeCommand( cmd, referencedAttributes );
            for( unsigned k = 0; k < referencedAttributes.length(); k++ ) {
                MString plugName( referencedNodes[j] );
                plugName = plugName + "." + referencedAttributes[k];
                cout << "       " << plugName.asChar() << "\n";
            }
        }
        cout << "\n";
    }

    // End of output 
    //
    cout << "=====================================\n";

    return MS::kSuccess;
}

MStatus initializePlugin( MObject obj )
{
    MStatus   status;
    MFnPlugin plugin( obj, PLUGIN_COMPANY, "4.0", "Any");

    // NOTE: referenceQuery is already a Maya cmd
    status = plugin.registerCommand( "refQuery", 
                                     referenceQuery::creator );
    if (!status) {
        status.perror("registerCommand");
        return status;
    }

    return status;
}

MStatus uninitializePlugin( MObject obj )
{
    MStatus   status;
    MFnPlugin plugin( obj );

    status = plugin.deregisterCommand( "refQuery" );
    if (!status) {
        status.perror("deregisterCommand");
    }
    return status;
}