Merge pull request #267 from storm-devs/feature/maya-plugin

Add maya plugin source

tools/maya-plugin/Animation/Animation.vcxproj.filters

@@ -0,0 +1,36 @@
+<?xml version="1.0" encoding="utf-8"?>
+<Project ToolsVersion="4.0" xmlns="http://schemas.microsoft.com/developer/msbuild/2003">
+  <ItemGroup>
+    <Filter Include="Исходные файлы">
+      <UniqueIdentifier>{4FC737F1-C7A5-4376-A066-2A32D752A2FF}</UniqueIdentifier>
+      <Extensions>cpp;c;cc;cxx;c++;cppm;ixx;def;odl;idl;hpj;bat;asm;asmx</Extensions>
+    </Filter>
+    <Filter Include="Файлы заголовков">
+      <UniqueIdentifier>{93995380-89BD-4b04-88EB-625FBE52EBFB}</UniqueIdentifier>
+      <Extensions>h;hh;hpp;hxx;h++;hm;inl;inc;ipp;xsd</Extensions>
+    </Filter>
+    <Filter Include="Файлы ресурсов">
+      <UniqueIdentifier>{67DA6AB6-F800-4c08-8B7A-83BB121AAD01}</UniqueIdentifier>
+      <Extensions>rc;ico;cur;bmp;dlg;rc2;rct;bin;rgs;gif;jpg;jpeg;jpe;resx;tiff;tif;png;wav;mfcribbon-ms</Extensions>
+    </Filter>
+  </ItemGroup>
+  <ItemGroup>
+    <ClInclude Include="anim_file.h">
+      <Filter>Файлы заголовков</Filter>
+    </ClInclude>
+    <ClInclude Include="AnimationExport.h">
+      <Filter>Файлы заголовков</Filter>
+    </ClInclude>
+    <ClInclude Include="..\Common\cvector.h">
+      <Filter>Файлы заголовков</Filter>
+    </ClInclude>
+    <ClInclude Include="..\Common\matrix.h">
+      <Filter>Файлы заголовков</Filter>
+    </ClInclude>
+  </ItemGroup>
+  <ItemGroup>
+    <ClCompile Include="AnimationExport.cpp">
+      <Filter>Исходные файлы</Filter>
+    </ClCompile>
+  </ItemGroup>
+</Project>

tools/maya-plugin/Animation/AnimationExport.cpp

@@ -0,0 +1,283 @@
+/******************************************************************************
+File:  AnimationExport.cpp
+
+Author:   Nick Chirkov
+Copyright (C) 2001 Akella corp.
+
+Comments:
+Maya export
+******************************************************************************/
+#include <d3dx9.h>
+#include "AnimationExport.h"
+
+#include <cstdarg>
+#include <maya/MGlobal.h>
+#include <maya/MFnPlugin.h>
+#include <maya/MTime.h>
+#include <maya/MAnimControl.h>
+#include <maya/MDagPath.h>
+#include <maya/MSelectionList.h>
+#include <maya/MItSelectionList.h>
+#include <maya/MFnDagNode.h>
+#include <maya/MMatrix.h>
+#include <maya/MFnMesh.h>
+#include <maya/MItDag.h>
+#include <maya/MFnTransform.h>
+#include <maya/MDagPathArray.h>
+
+#include "anim_file.h"
+#include <vector>
+#include "cvector.h"
+#include "matrix.h"
+#include <string>
+
+#define CM2M_SCALE 0.01f
+
+FILE *LogFile=0;
+void Log(const char *fmt,...)
+{
+	va_list args;
+	va_start(args,fmt);
+	vfprintf(LogFile, fmt, args);
+	va_end(args);
+}
+
+bool GetDoubleValue(MObject &obj, const char *name, double &d)
+{
+	MFnDependencyNode NodeFnDn;
+	NodeFnDn.setObject(obj);
+	MStatus status;
+#if MAYA_API_VERSION <= 500
+	MPlug rpuv = NodeFnDn.findPlug(name, &status);
+#else
+	MPlug rpuv = NodeFnDn.findPlug(name, false, &status);
+#endif
+	if(!status)	return false;
+	MObject val;
+	rpuv.getValue(d);
+	return true;
+}
+
+FILE *anifile;
+MStatus status;
+
+AniExport::AniExport()
+{
+}
+
+AniExport::~AniExport()
+{
+}
+
+MStatus AniExport::writer(const MFileObject &file,  const MString &optionsString, FileAccessMode mode)
+{
+#if MAYA_API_VERSION <= 500
+  MString  fname = file.fullName();
+#else
+  MString  fname = file.expandedFullName();
+#endif
+	ExportAll(fname.asChar());
+  return (MS::kSuccess);
+}
+bool AniExport::haveWriteMethod () const
+{
+  return (true);
+}
+MString AniExport::defaultExtension () const
+{
+  return MString("an");
+}
+
+MPxFileTranslator::MFileKind   AniExport::identifyFile( const MFileObject &file, const char *buffer,short size) const
+{
+  //const char *name = file.resolvedName().asChar();
+  //int nameLength = strlen(name);
+  return (kIsMyFileType);
+}
+
+void *AniExport::creator()
+{
+	return new AniExport();
+}
+
+//****************************************************************
+__declspec(dllexport) MStatus initializePlugin (MObject obj)
+{
+	using namespace std::string_literals;
+	MStatus status;
+	MFnPlugin plugin (obj, "Nick Chirkov (C) 2001", "1.0");
+	auto pixmap_name = ""s;
+	auto options_script_name = "SD2AniExportMel"s;
+	status = plugin.registerFileTranslator("SD2AniExport", &pixmap_name[0], 	AniExport::creator,&options_script_name[0]);
+	return status;
+}
+
+__declspec(dllexport) MStatus uninitializePlugin (MObject obj)
+{
+	MFnPlugin plugin (obj);
+	plugin.deregisterFileTranslator ("SD2AniExport");
+	return (MS::kSuccess);
+}
+
+//------------------------------------------------------------------------
+//find root
+//------------------------------------------------------------------------
+void AniExport::ExportAll(const char *fname)
+{
+	MGlobal::executeCommand("window -tlc 300 300 -w 300 -h 200 -title \"ani Export\" StatusWin",true);
+	MGlobal::executeCommand("columnLayout -adjustableColumn true",true);
+	MGlobal::executeCommand("text -label \"Default\" MyLable",true);
+	MGlobal::executeCommand("showWindow StatusWin",true);
+	MGlobal::executeCommand("text -edit -label \"Search root\" MyLable",true);
+	MGlobal::executeCommand("showWindow StatusWin",true);
+
+	char newname[256];
+	sprintf(newname, "%s.ant", fname);
+	LogFile = fopen(newname,"w+");
+	char aniname[256];
+	strcpy(aniname, newname);
+	aniname[strlen(aniname)-4] = 0;
+	anifile = fopen(aniname,"w+b");
+
+	//get selected locator
+	MSelectionList slist;
+	MGlobal::getActiveSelectionList(slist);
+	MItSelectionList iter(slist);
+	if(iter.isDone())	return;
+	MDagPath locator;
+	iter.getDagPath(locator);
+	if(!locator.hasFn(MFn::kLocator))	return;
+
+	//------------------------------------------------------------------------------
+	//EXPORT FRAMES
+	//------------------------------------------------------------------------------
+  MTime startFrame( MAnimControl::minTime().value(), MTime::uiUnit() );
+  float startTime = (float)startFrame.value();
+
+  MTime endFrame( MAnimControl::maxTime().value(), MTime::uiUnit() );
+  float endTime = (float)endFrame.value();
+
+	ANIFILE::HEADER ah;
+	fwrite(&ah, sizeof ah, 1, anifile);
+	ah.nJoints = 0;
+	ah.nFrames = 0;
+	ah.framesPerSec = 15.0f;
+
+	//--------------------------------------------------------
+	//all joints
+	//--------------------------------------------------------
+	//find all joints
+	MDagPathArray mJointsList;
+	MItDag dagIterator( MItDag::kBreadthFirst, MFn::kJoint, &status);
+	if(!status)	return;
+	for ( ; !dagIterator.isDone(); dagIterator.next() ) 
+	{
+		MDagPath dagPath;
+		status = dagIterator.getPath(dagPath);  // Get DagPath
+		if(!status) continue;
+
+		status = mJointsList.append(dagPath);
+		ah.nJoints++;
+	}
+
+	//-------------------------------------------------------------
+	//parent index to handle hierarchy
+	//-------------------------------------------------------------
+	long *parIdx = new long[ah.nJoints];
+	long j;
+	for(j=0; j<ah.nJoints; j++)
+	{
+		MDagPath jp = mJointsList[j];
+		//go to the parent
+		jp.pop();
+
+		parIdx[j] = -1;
+		for(long i=0; i<ah.nJoints; i++)
+		{
+			MDagPath gp = mJointsList[i];
+			if(gp==jp)
+			{
+				parIdx[j] = i;
+				break;
+			}
+		}
+	}
+	fwrite(parIdx, sizeof(*parIdx), ah.nJoints, anifile);
+
+	MTime Time(startTime, MTime::uiUnit());
+	MAnimControl::setCurrentTime(Time);
+
+	//-------------------------------------------------------------
+	//export frames
+	//-------------------------------------------------------------
+	std::vector<CVECTOR> begPos;
+	std::vector<CVECTOR> rootPos;
+	std::vector<D3DXQUATERNION> jointAngle;
+
+  for(float time = startTime; time<endTime; time += 1.0f)
+  {
+		Log("------------------------------------------------------------------\n");
+		Log("%f\n", time);
+		//set current time
+		MTime Time(time, MTime::uiUnit());
+		MAnimControl::setCurrentTime(Time);
+
+		//--------------------------------------------------------
+		//frames
+		//--------------------------------------------------------
+		for(j=0; j<ah.nJoints; j++)
+		{
+			MDagPath jp = mJointsList[j];
+			double dblv;
+
+			//----------------------------------
+			CVECTOR jPos;
+			auto path = jp.node(&status);
+			GetDoubleValue(path, "translateX", dblv);	jPos.x = float(CM2M_SCALE*dblv);
+			GetDoubleValue(path, "translateY", dblv);	jPos.y = float(CM2M_SCALE*dblv);
+			GetDoubleValue(path, "translateZ", dblv);	jPos.z = float(CM2M_SCALE*dblv);
+
+			CVECTOR rotAng;
+			GetDoubleValue(path, "rotateX", dblv);	rotAng.x = float(dblv);
+			GetDoubleValue(path, "rotateY", dblv);	rotAng.y = float(dblv);
+			GetDoubleValue(path, "rotateZ", dblv);	rotAng.z = float(dblv);
+			CMatrix rotate;
+			rotate.BuildMatrixXYZ(rotAng.x, rotAng.y, rotAng.z, 0.0f, 0.0f, 0.0f);
+
+			CVECTOR orAng;
+			GetDoubleValue(path, "jointOrientX", dblv);	orAng.x = float(dblv);
+			GetDoubleValue(path, "jointOrientY", dblv);	orAng.y = float(dblv);
+			GetDoubleValue(path, "jointOrientZ", dblv);	orAng.z = float(dblv);
+			CMatrix orient;
+			orient.BuildMatrixXYZ(orAng.x, orAng.y, orAng.z, jPos.x, jPos.y, jPos.z);
+
+			CMatrix mtxGlob = rotate*orient;
+			D3DXQUATERNION rp;
+			D3DXQuaternionRotationMatrix(&rp, mtxGlob);
+			jointAngle.push_back(rp);
+
+			if(j==0)	rootPos.push_back(mtxGlob.Pos());
+
+			//joints' begin positions
+			if(ah.nFrames==0)	begPos.push_back(mtxGlob.Pos());
+
+		}
+		ah.nFrames++;
+	}
+
+	fwrite(&begPos[0], sizeof CVECTOR, ah.nJoints, anifile);
+	fwrite(&rootPos[0], sizeof CVECTOR, ah.nFrames, anifile);
+
+	for(j=0; j<ah.nJoints; j++)
+		for(long f=0; f<ah.nFrames; f++)
+			fwrite(&jointAngle[f*ah.nJoints + j], sizeof jointAngle[0], 1, anifile);
+
+	Log("%d\n", ah.nFrames);
+	fseek(anifile, 0, SEEK_SET);
+	fwrite(&ah, sizeof ah, 1, anifile);
+	//------------------------------------------------------------------------------
+	fclose(LogFile);
+	fclose(anifile);
+
+  MGlobal::executeCommand("deleteUI -window StatusWin;");
+}

tools/maya-plugin/Animation/AnimationExport.h

@@ -0,0 +1,33 @@
+/******************************************************************************
+File:  animationExport.h
+
+Author:   Nick Chirkov
+Copyright (C) Akella corp.
+
+Comments:
+Maya interface for animation export
+******************************************************************************/
+#ifndef _AniExport_H_
+#define _AniExport_H_
+
+#pragma warning ( disable : 4786)
+
+#include <maya/MPxFileTranslator.h>
+
+class AniExport : public MPxFileTranslator
+{
+public:
+  AniExport();
+  ~AniExport();
+
+  MStatus writer(const MFileObject &file,	const MString &optionsString,	FileAccessMode mode); 
+  bool haveWriteMethod() const;
+  MString defaultExtension() const;
+  MFileKind	identifyFile (const MFileObject &, const char *buffer, short size) const;
+  static void *creator();
+protected:
+	void ExportAll(const char *fname);
+
+};
+
+#endif

tools/maya-plugin/Animation/anim_file.h

@@ -0,0 +1,20 @@
+#pragma once
+
+//ANIMATION_HEADER
+//long parentIndex[nJoints]
+//CVECTOR jointPos[nJoints]
+
+//Frames:
+//CVECTOR rootPos[nFrames]
+//CVECTOR jointAngle[nJoints][nFrames]
+
+namespace ANIFILE
+{
+	struct HEADER
+	{
+		long nFrames;
+		long nJoints;
+		float framesPerSec;
+	};
+
+};

tools/maya-plugin/Common/chunk.h

@@ -0,0 +1,16 @@
+#ifndef _CHUNK_H_
+#define _CHUNK_H_
+
+#pragma pack(push,1)
+
+#define CHUNK_RDF_GEOMETRY	0
+
+struct CHUNK
+{
+	unsigned long id;
+	unsigned long size;
+};
+
+#pragma pack(pop)
+
+#endif