Adding Shave-and-a-Haircut 9.6

1 files changed, 422 insertions, 0 deletions

diff --git a/mayaPlug/shaveVrayExporterImpl.cpp b/mayaPlug/shaveVrayExporterImpl.cpp
new file mode 100644
index 0000000..28c7202
--- /dev/null
+++ b/mayaPlug/shaveVrayExporterImpl.cpp
@@ -0,0 +1,422 @@
+// Shave and a Haircut
+// (c) 2019 Epic Games
+// US Patent 6720962
+
+#include <sys/stat.h>
+#include <stdio.h>
+
+#ifdef _WIN32
+#define	fileno	_fileno
+#define	fstat	_fstat
+typedef struct _stat StatType;
+#else
+#include <unistd.h>
+typedef struct stat StatType;
+#endif
+
+#include <maya/MGlobal.h>
+#include <maya/MStatus.h>
+#include <maya/MItDependencyNodes.h>
+#include <maya/MFnDependencyNode.h>
+#include <maya/MFnMesh.h>
+#include <maya/MPlug.h>
+#include <maya/MFloatArray.h>
+
+#include "utils.h"
+#include "vrayexport.h"
+#include "plugman.h"
+#include "vrayplugins.h"
+#include "defparams.h"
+#include "vraymayageom.h"
+
+#include "shaveVrayExporter.h"
+
+#ifndef VRAY_OVERRIDE
+#define VRAY_OVERRIDE
+#endif
+
+#define VR VUtils
+
+class shaveVrayExporterImpl: public shaveVrayExporter {
+public:
+	shaveVrayExporterImpl();
+	void createVRayPlugin(
+			void *voidGeomInfo,
+			const MString &libPath,
+			int stackIdx,
+			bool ownshader,
+			bool isInstanced,
+			const MObject &instance,
+			bool tipfade,
+			bool squirrel,
+			const MColor &spec_tint,
+			const MColor &spec_tint2,
+			bool camVis,
+			bool lightVis,
+			bool giVis,
+			bool reflVis,
+			bool refrVis,
+			float selfShadow,
+			bool recvShadow
+			) VRAY_OVERRIDE;
+	bool readDraToParam(const char *draFilename) VRAY_OVERRIDE;
+private:
+	// V-Ray plugin parameters
+	VR::DefIntParam m_stackIndexParam;
+	VR::DefIntParam m_instancedParam;
+	VR::DefIntParam m_ownshaderParam;
+	VR::DefStringParam m_draFileParam;
+	VR::DefIntListParam m_dataParam;
+	VR::DefIntParam m_dataSizeParam;
+	VR::DefStringParam m_libPathParam;
+	VR::DefIntParam m_squirrelParam;
+	VR::DefIntParam m_tipfadeParam;
+	VR::DefColorParam m_spectintParam;
+	VR::DefColorParam m_spectint2Param;
+	//instance uvs-related params
+	VR::DefIntParam m_numFacesPerInstParam;
+	VR::DefIntParam m_numUVSetsParam;
+	VR::DefVectorListParam m_uvsParam;
+
+	//ray visibility params
+	VR::DefIntParam m_cameraVisibilityParam;
+	VR::DefIntParam m_reflVisibilityParam;
+	VR::DefIntParam m_refrVisibilityParam;
+	VR::DefIntParam m_lightVisibilityParam;
+	VR::DefIntParam m_GiVisibilityParam;
+
+	VR::DefFloatParam m_selfShadowParam;
+	VR::DefIntParam m_recvShadowParam;
+
+	VR::DefIntParam m_useGlobalHairTreeParam; // store all voxels in a global hair tree
+	VR::DefIntParam m_dynamicHairTesselation; // compute sub-segments based on distance to camera
+	VR::DefFloatParam m_hairTesselMaxEdgeLen; // max desired sub-segment length (in pixels)
+};
+
+shaveVrayExporterImpl::shaveVrayExporterImpl() :
+	m_stackIndexParam("stackId",0),
+	m_instancedParam("instanced",0),
+	m_ownshaderParam("ownshader",1),
+	m_draFileParam("draFile",""),
+	m_dataParam("draData"),
+	m_dataSizeParam("draSize",0),
+	m_libPathParam("libPath", ""),
+	m_squirrelParam("squirrel",0),
+	m_tipfadeParam("tipfade",0),
+	m_spectintParam("spectint",VR::Color(1.0f,1.0f,1.0f)),
+	m_spectint2Param("spectint2",VR::Color(0.0f,0.0f,0.0f)),
+	m_numFacesPerInstParam("numFacesPerInst",0),
+	m_numUVSetsParam("numUVSets",0),
+	m_uvsParam("uvs"),
+	m_cameraVisibilityParam("cameraVisibility",1),
+	m_reflVisibilityParam("reflVisibility",1),
+	m_refrVisibilityParam("refrVisibility",1),
+	m_lightVisibilityParam("lightVisibility",1),
+	m_GiVisibilityParam("GiVisibility",1),
+	m_selfShadowParam("selfShadow",1.0f),
+	m_recvShadowParam("recvShadow",1),
+	m_useGlobalHairTreeParam("useGlobalHairTree",1),
+	m_dynamicHairTesselation("dynamicHairTesselation",0),
+	m_hairTesselMaxEdgeLen("hairTesselMaxEdgleLen",4.0f)
+{}
+
+void shaveVrayExporterImpl::createVRayPlugin(
+		void *voidGeomInfo,
+		const MString &libPath,
+		int stackIdx,
+		bool ownshader,
+		bool isInstanced,
+		const MObject &instance,
+		bool tipfade,
+		bool squirrel,
+		const MColor &spec_tint,
+		const MColor &spec_tint2,
+		bool camVis,
+		bool lightVis,
+		bool giVis,
+		bool  reflVis,
+		bool  refrVis,
+		float selfShadow,
+		bool  recvShadow) 
+{
+	if (!voidGeomInfo) 
+		return;
+
+	MStatus stat;
+	VR::VRayGeomInfo *geomInfo=reinterpret_cast<VR::VRayGeomInfo*>(voidGeomInfo);
+
+	// Create the VRay plugin
+	bool existing=false;
+	VR::VRayPluginInterface* vrplug = geomInfo->newPlugin("hairVRay30Shader", existing);
+	if(!vrplug)
+	{
+		//  Check for the VRay 2.4 shader.
+		//
+		vrplug = geomInfo->newPlugin("hairVRayShader", existing);
+	}
+
+	if(!vrplug)
+	{
+		MGlobal::displayError("shaveVrayExporterImpl: can not create 'hairVRayShader' or 'hairVRay30Shader' V-Ray plugin.");
+		return;
+	}
+
+	m_stackIndexParam.setInt(stackIdx,0,0);
+	m_ownshaderParam.setInt(ownshader ? 1 : 0,0,0);
+	m_instancedParam.setInt(isInstanced ? 1 : 0,0,0);
+	m_tipfadeParam.setInt(tipfade ? 1 : 0,0,0);
+	m_squirrelParam.setInt(squirrel ? 1 : 0,0,0);
+	m_spectintParam.setColor(VUtils::Color(spec_tint.r, spec_tint.g, spec_tint.b), 0, 0);
+	m_spectint2Param.setColor(VUtils::Color(spec_tint2.r, spec_tint2.g, spec_tint2.b), 0, 0);
+	m_libPathParam.setString(libPath.asChar(),0,0);
+
+	/////// ray visibiltity
+	m_cameraVisibilityParam.setInt(camVis ? 1 : 0,0,0);
+	m_reflVisibilityParam.setInt(reflVis ? 1 : 0,0,0);
+	m_refrVisibilityParam.setInt(refrVis ? 1 : 0,0,0);
+	m_lightVisibilityParam.setInt(lightVis ? 1 : 0,0,0);
+	m_GiVisibilityParam.setInt(giVis ? 1 : 0,0,0);
+	m_selfShadowParam.setFloat(selfShadow,0,0);
+	//printf("set selfShadow %f\n",selfShadow);fflush(stdout);
+	m_recvShadowParam.setInt(recvShadow ? 1 : 0,0,0);
+
+	if(isInstanced && !instance.isNull())
+	{
+		MGlobal::displayInfo("shaveVrayExporterImpl: instance not NULL.");
+		MFnMesh	meshFn(instance);
+		
+		int nf = meshFn.numPolygons();
+		
+		MStringArray	uvSetNames;
+		meshFn.getUVSetNames(uvSetNames);
+		int ns = (int)uvSetNames.length();
+
+		m_numFacesPerInstParam.setInt(nf,0,0);
+		//MGlobal::displayInfo(MString("num faces ") + nf );
+		m_numUVSetsParam.setInt(ns,0,0);
+		
+		{
+			int count = 0;
+			for (unsigned int i = 0; i < uvSetNames.length(); i++)
+			{
+				//MFloatArray	u;
+				//MFloatArray	v;
+				MIntArray	numUVsPerFace;
+				MIntArray	uvIds;
+				MString		uvSet = uvSetNames[i];
+
+				stat = meshFn.getAssignedUVs(numUVsPerFace, uvIds, &uvSet);
+				count += uvIds.length();
+			}
+			m_uvsParam.setCount(count);
+#ifdef WIN32
+			MGlobal::displayInfo(MString("memory ") + (_CrtCheckMemory() ? 1 : 0));
+#endif			
+
+			int k = 0;
+			for (unsigned int i = 0; i < uvSetNames.length(); i++)
+			{
+				MFloatArray	u;
+				MFloatArray	v;
+				MIntArray	numUVsPerFace;
+				MIntArray	uvIds;
+				MString		uvSet = uvSetNames[i];
+
+				stat = meshFn.getUVs(u, v, &uvSet);
+				stat = meshFn.getAssignedUVs(numUVsPerFace, uvIds, &uvSet);
+
+				for (unsigned int j = 0; j < uvIds.length(); j++)
+				{
+					VR::Vector V(u[uvIds[j]],v[uvIds[j]],0.0f);
+					m_uvsParam[k] = V;
+					k++;
+				}
+			}
+		}
+	}
+
+	vrplug->setParameter(&m_stackIndexParam);
+	vrplug->setParameter(&m_instancedParam);
+	vrplug->setParameter(&m_ownshaderParam);
+	vrplug->setParameter(&m_squirrelParam);
+	vrplug->setParameter(&m_tipfadeParam);
+	vrplug->setParameter(&m_spectintParam);
+	vrplug->setParameter(&m_spectint2Param);
+	vrplug->setParameter(&m_draFileParam);
+	vrplug->setParameter(&m_dataParam);
+	vrplug->setParameter(&m_dataSizeParam);
+	vrplug->setParameter(&m_libPathParam);
+
+	//instance uvs-related params
+	vrplug->setParameter(&m_numFacesPerInstParam);
+	vrplug->setParameter(&m_numUVSetsParam);
+	vrplug->setParameter(&m_uvsParam);
+
+	//ray visibility params
+	vrplug->setParameter(&m_cameraVisibilityParam);
+	vrplug->setParameter(&m_reflVisibilityParam);
+	vrplug->setParameter(&m_refrVisibilityParam);
+	vrplug->setParameter(&m_lightVisibilityParam);
+	vrplug->setParameter(&m_GiVisibilityParam);
+	vrplug->setParameter(&m_selfShadowParam);
+	vrplug->setParameter(&m_recvShadowParam);
+	vrplug->setParameter(&m_useGlobalHairTreeParam);
+	vrplug->setParameter(&m_dynamicHairTesselation);
+	vrplug->setParameter(&m_hairTesselMaxEdgeLen);
+
+	if (!existing) 
+	{
+		// The plugin did not exist before - we can do additional set up here, if necessary
+	}
+}
+
+bool shaveVrayExporterImpl::readDraToParam(const char *draFilename)
+{
+#ifdef LINUX
+		printf("shaveVrayExporterImpl: readDraToParam\n");fflush(stdout);
+#endif
+	VR::DefIntListParam *li = &m_dataParam;
+	VR::DefIntParam *sa = &m_dataSizeParam;
+	if(!li)
+	{
+#if 0	//we do not want to fire this warning if dummy shape is located at hidden display layer
+		MGlobal::displayError(MString("shaveVrayExporterImpl: NULL draData parameter"));
+#ifdef LINUX
+		printf("shaveVrayExporterImpl: NULL draData parameter\n");fflush(stdout);
+#endif
+#endif
+		return false;
+	}
+	if(!sa)
+	{
+		MGlobal::displayError(MString("shaveVrayExporterImpl: NULL draDataSize parameter"));
+#ifdef LINUX
+		printf("shaveVrayExporterImpl: NULL draDataSize parameter\n");fflush(stdout);
+#endif
+		return false;
+	}
+	FILE*	draFile = fopen(draFilename, "r+b");
+	if(!draFile)
+	{
+		MGlobal::displayError(MString("shaveVrayExporterImpl: can not open DRA file\n")+draFilename);
+#ifdef LINUX
+		printf("shaveVrayExporterImpl: can not open DRA file %s\n",draFilename);fflush(stdout);
+#endif
+		return false;
+	}
+	else
+	{
+		//Get the size of the DRA file.
+		StatType    fileInfo;
+		fstat(fileno(draFile), &fileInfo);
+		if(fileInfo.st_size == 0)
+		{
+			MGlobal::displayError("shaveVrayExporterImpl: DRA file is empty.");
+#ifdef LINUX
+			printf("shaveVrayExporterImpl: DRA file is empty.");fflush(stdout);
+#endif
+			return false;
+		}
+		else
+		{
+			long size = fileInfo.st_size;
+			char* data = (char*)malloc(size);
+			int sum = 0;
+			int c;
+			int p = 0;
+			while ((c = fgetc(draFile)) != EOF)
+			{
+				data[p] = c;
+				p++;
+				sum+=c;
+			}
+#ifdef LINUX
+			printf("shaveVrayExporterImpl: checksum %i \n",sum);fflush(stdout);
+#endif
+
+			fclose(draFile);
+			if(p != size)
+			{
+				MGlobal::displayError(MString("shaveVrayExporterImpl: dra read error, bytes expected: ")+size+" bytes read: "+p);
+#ifdef LINUX
+				printf("shaveVrayExporterImpl: dra read error, bytes expected %i bytes read %i \n",size,p);fflush(stdout);
+#endif
+			}
+			
+			//pass data to vray parameter
+			
+			long n = size/sizeof(int);
+			long N = n;
+			long extra = 0;
+			if(n*sizeof(int) < size)
+			{
+				N = n+1;
+				extra = size - n*sizeof(int);
+			}
+			li->setCount(N);
+//			MGlobal::displayInfo(MString("shaveVrayExporterImpl: (debug) extra bytes: ") + extra);
+//#ifdef LINUX
+//			printf("shaveVrayExporterImpl: (debug) extra bytes:  %i \n",extra);fflush(stdout);
+//#endif
+
+			int* idata = (int*)data;
+			long chpos = 0;
+			long intsum = 0;
+			for(long oo = 0; oo < N; oo++)
+			{
+				if(oo < n)
+				{
+					int v = idata[oo];
+					li->setInt(v,oo,0.0);
+					chpos += sizeof(int);
+
+					intsum += v;
+				}
+				else
+				{
+					int A = 0;
+					char* AA = (char*)(&A);
+					for(int j = 0; j < extra; j++)
+					{
+						AA[j] = data[chpos + j];
+						//MGlobal::displayInfo(MString("shaveVrayExporterImpl: (debug) idx: ") + (chpos + j));
+						//MGlobal::displayInfo(MString("shaveVrayExporterImpl: (debug) byte: ") + data[chpos + j]);
+
+					}
+					li->setInt(A,oo,0.0);
+					chpos += extra;
+
+					intsum += A;
+				}
+			}
+			/////// checksum //
+#ifdef LINUX
+			sum = 0;
+			for(int oo = 0; oo < size; oo++) 
+			{
+				sum += (unsigned char)data[oo];
+			}
+			printf("shaveVrayExporterImpl: checksum %i\n",sum); fflush(stdout);
+			printf("shaveVrayExporterImpl: intsum %i\n",intsum); fflush(stdout);
+			/////////////
+#endif	
+
+			free(data);
+			sa->setInt(size,0,0);
+
+#ifdef LINUX
+			printf("shaveVrayExporterImpl: readDraToParam - done\n");fflush(stdout);
+#endif
+			return true;
+		}
+	}
+}
+
+extern "C" SHAVEVRAYEXPORTER_EXPORT shaveVrayExporter *newShaveVrayExporter() {
+	return new shaveVrayExporterImpl();
+}
+
+extern "C" SHAVEVRAYEXPORTER_EXPORT void deleteShaveVrayExporter(shaveVrayExporter *exporter) {
+	delete exporter;
+}
+