iffInfoCmd/iffInfoCmd.cpp

//-
// ==========================================================================
// Copyright 2020 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 "iffInfo".
//
// This command takes the name of an IFF file as an argument. 
// The file is opened and read, and general information about the file is returned as a result.
// For example: "iffInfo sphere.iff". 
// 

#include <maya/MPxCommand.h>
#include <maya/MStatus.h>
#include <maya/MArgList.h>
#include <maya/MFnPlugin.h>
#include <maya/MObject.h>
#include <maya/MGlobal.h>
#include <maya/MString.h>
#include <maya/MPoint.h>
#include "iffreader.h"
#ifndef __APPLE__
#include <maya/MIOStream.h>
#endif
#define IFFCHECKERR(stat, call) \
if (!stat) { \
    MString string = reader.errorString(); \
    string += " in method "; \
    string += #call; \
    displayError (string); \
    return MS::kFailure; \
}

class iffInfo : public MPxCommand
{
public:
                iffInfo();
        ~iffInfo() override;

    MStatus     doIt ( const MArgList& args ) override;
    MStatus     redoIt () override;
    MStatus     undoIt () override;
    bool        isUndoable() const override;

    static      void* creator();

private:
    MString     result;
};

iffInfo::iffInfo()
{
}

iffInfo::~iffInfo() {}

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

bool iffInfo::isUndoable() const
{
    return true;
}

MStatus iffInfo::undoIt()
{
    return MS::kSuccess;
}

MString itoa (int n)
{
    char buffer [256];
    sprintf (buffer, "%d", n);
    return MString (buffer);
}

MStatus iffInfo::doIt( const MArgList& args )
{
    MString componentName;
    if (args.length () != 1 ) {
        displayError ("Syntax: iffInfo file");
        return MS::kFailure;
    }

    MString fileName;

    args.get (0, fileName);

    IFFimageReader reader;
    MStatus stat;

    stat = reader.open (fileName);
    IFFCHECKERR (stat, open);

    int bytesPerChannel = reader.getBytesPerChannel ();
    int w,h;

    stat = reader.getSize (w, h);
    IFFCHECKERR (stat, getSize);

    result = "\nResolution: ";
    result += itoa (w);
    result += "x";
    result += itoa (h);
    result += "\n";
    if (reader.isRGB () || reader.isGrayscale ()) {
        if (reader.isRGB ()) {
            result += "RGB";
            if (reader.hasAlpha ())
                result += "A";
        }
        else
            result += "Grayscale";
        result += " data with ";
        if (bytesPerChannel == 2)
            result += "16";
        else
            result += "8";
        result += " bits per channel\n";
    }
    if (reader.hasDepthMap ()) {
        result += "Image has a depth map\n";
    } else {
        result += "Image does not have a depth map\n";
    }

    stat = reader.close ();
    IFFCHECKERR (stat, close);

    return redoIt();
}

MStatus iffInfo::redoIt()
{
    clearResult();
    appendToResult (result);

    return MS::kSuccess;
}

MStatus initializePlugin( MObject obj )
{
    MStatus status;
    MFnPlugin plugin( obj, PLUGIN_COMPANY, "3.0", "Any");
    status = plugin.registerCommand( "iffInfo", iffInfo::creator );

    if (!status)
        status.perror("registerCommand");

    return status;
}

MStatus uninitializePlugin( MObject obj )
{
    MStatus status;
    MFnPlugin plugin( obj );
    status = plugin.deregisterCommand( "iffInfo" );

    if (!status)
        status.perror("deregisterCommand");

    return status;
}