OTOY Forums

Code: Select all

/*
Script : octaneInstancer.mel
Date : 05/12/2012
Version : 0.5
About : write a .csv file with matrices coming from selected transforms
and/or particleInstancers
Usage : mel command 'octaneInstancer'
Author : francois aka tchoa
with improvement since v0.5 of Clinton
http://www.scaphandrier.net/?p=303#comment-388 
Contact : [email protected] 
Copyrights(C)TreeX-2012
*/


global proc octaneInstancer()
{
if (!`window -ex octaneInstancer`){
window -s 1 -rtf 1 -tlb 1 -t "Octane Instancer" octaneInstancer;
columnLayout -adj 1;
floatSliderGrp -l "Scale" -cw3 30 50 30 -cal 3 "right" -ad3 3 -f 1 -pre 4 -min 0.01 -max 1 -fmn 0.0001 -fmx 10000 -v 1 octaneInstanceScale;
checkBox -l "Hierarchy" -v 1 octaneInstanceHi;
checkBox -l "Meshes Only" -v 0 octaneInstanceMo;
button -l "Export" -c "octaneInstancerDialog";
}
showWindow octaneInstancer;
}

global proc octaneInstancerDialog()
{
string $version = `about -v`;
string $buffer[];
tokenize $version " " $buffer;
int $year = $buffer[0];
if ($year < 2011){
string $path = `fileDialog -m 1 -dm "*.csv" -t "Export"`;
if ($path != "")
octaneInstancerFile($path);
}
else{
string $paths[] = `fileDialog2 -fm 0 -ff "octaneInstancerFile (*.csv)" -sff "Octane" -cap "Export"`;
if (`size $paths`)
octaneInstancerFile($paths[0]);
}
}

proc string getInstancesAttribute(string $attributes[], string $search)
{
string $attribute = "";
int $count = size($attributes); 
for ($i = 0; $i < $count; $i += 2){
if ($attributes[$i] == $search){
$attribute = $attributes[$i+1];
break;
}
}
return $attribute;
}

proc int getInstancerId(string $connection)
{
string $buffer[];
tokenize $connection "." $buffer;
tokenize $buffer[1] "[" $buffer;
tokenize $buffer[1] "]" $buffer;
return $buffer[0];
}

proc string getInstanceMatrix(string $transform)
{
float $scaleFactor = `floatSliderGrp -q -v octaneInstanceScale`;
string $data;
float $matrix[16] = `xform -q -ws -m $transform`;
for ($mid = 12; $mid < 15; $mid++)
$matrix[$mid] *= $scaleFactor;
$data += $matrix[0] + " " + $matrix[4] + " " + $matrix[8] + " " + $matrix[12] + " ";
$data += $matrix[1] + " " + $matrix[5] + " " + $matrix[9] + " " + $matrix[13] + " ";
$data += $matrix[2] + " " + $matrix[6] + " " + $matrix[10] + " " + $matrix[14] + "\n";
return $data;
}

proc getInstances(string $path, string $instancer)
{
float $scaleFactor = `floatSliderGrp -q -v octaneInstanceScale`;
string $particles = plugNode(`connectionInfo -sfd ($instancer + ".inputPoints")`);
string $instancerId = getInstancerId(`connectionInfo -sfd ($instancer + ".inputPoints")`);
string $attributes[] = `getAttr ($particles + ".idt["+ $instancerId +"].iam")`;
string $objects[];
int $num = `getAttr ($instancer + ".hierarchyCount")`;
for ($i = 0; $i < $num; $i++)
$objects[$i] = plugNode(`connectionInfo -sfd ($instancer + ".inputHierarchy[" + $i + "]")`);
string $ids = getInstancesAttribute($attributes, "objectIndex");
string $position = getInstancesAttribute($attributes, "position");
string $rotation = getInstancesAttribute($attributes, "rotation");
string $scale = getInstancesAttribute($attributes, "scale");
int $count = `getAttr ($instancer + ".instanceCount")`;
string $temp = `createNode "transform"`;
int $id = 0;
// Use integers for testing the existance of an attributes value (faster than comparing string
// values on every loop).
int $doPos = 0;
int $doScale = 0;
int $doRot = 0;
int $doIds = 0;
if ($position != "") $doPos = 1;
if ($rotation != "") $doRot = 1;
if ($scale != "") $doScale = 1;
if ($ids != "") $doIds = 1;

global string $gMainProgressBar;
progressBar -e -bp -ii 1 -max ($num * $count) -st "Parsing…" $gMainProgressBar;
for ($object in $objects){
string $_path = plugNode($path) + "_" + $object + ".csv";

// Build the particle data once, not every loop. This will have issues with massive data sets, though
// has been tested on 100,000 particles.
float $rotArray[];
float $posArray[];
float $scaleArray[];
if ($doRot)
$rotArray = `getParticleAttr -at $rotation -array true ($particles+".pt[0:"+$count+"]")`;
if ($doPos)
$posArray = `getParticleAttr -at $position -array true ($particles+".pt[0:"+$count+"]")`;
if ($doScale)
$scaleArray = `getParticleAttr -at $scale -array true ($particles+".pt[0:"+$count+"]")`;

$fileId = `fopen $_path "w"`;
for ($i = 0; $i < $count; $i++){
int $current = 0;
if ($doIds){
$tmpId = `particle -or $i -at $ids -q $particles`;
$current = $tmpId[0];
}
if ($id == $current){
if ($doPos)
setAttr ($temp + ".translate") $posArray[$i*3] $posArray[($i*3)+1] $posArray[($i*3)+2];
else
setAttr ($temp + ".translate") 0 0 0;

if ($doRot)
setAttr ($temp + ".rotate") $rotArray[$i*3] $rotArray[($i*3)+1] $rotArray[($i*3)+2];
else
setAttr ($temp + ".rotate") 0 0 0;

if ($doScale)
setAttr ($temp + ".scale") $scaleArray[($i*3)] $scaleArray[($i*3)+1] $scaleArray[($i*3)+2];
else
setAttr ($temp + ".scale") 1 1 1;

// Do the matrix construction in local scope.
float $matrix[16] = `xform -q -ws -m $temp`;
for ($mid = 12; $mid < 15; $mid++)
$matrix[$mid] *= $scaleFactor;
string $data = $matrix[0] + " " + $matrix[4] + " " + $matrix[8] + " " + $matrix[12] + " ";
$data += $matrix[1] + " " + $matrix[5] + " " + $matrix[9] + " " + $matrix[13] + " ";
$data += $matrix[2] + " " + $matrix[6] + " " + $matrix[10] + " " + $matrix[14] + "\n";

fprint $fileId $data;
}
if (`progressBar -q -ic $gMainProgressBar`)
break;
// Limit the progressBar updates (speeds up execution by 15%).
if ($i % 1000 == 0)
progressBar -e -s 1000 $gMainProgressBar;
}
fclose $fileId;

$id++;
if (`progressBar -q -ic $gMainProgressBar`)
break;
}
progressBar -e -ep $gMainProgressBar;
delete $temp;
}

proc string getInstance(string $path, string $transform)
{
if (`nodeType $transform` == "instancer"){
getInstances($path, $transform);
return "";
}
else{
if (`checkBox -q -v octaneInstanceMo`){
string $childs[] = `listRelatives -c -type "mesh" $transform`;
if (size($childs))
return getInstanceMatrix($transform);
}
else
return getInstanceMatrix($transform);
}
return "";
}

global proc octaneInstancerFile(string $path)
{
int $hi = `checkBox -q -v octaneInstanceHi`;
string $data;
string $sel[] = `ls -sl -tr`;
$fileId = `fopen $path "w"`;
for ($obj in $sel){
fprint $fileId (getInstance($path, $obj));
$items = `listRelatives -c -ad -type "transform" $obj`;
if ($hi){
for ($item in $items)
fprint $fileId (getInstance($path, $item));
}
} 
fclose $fileId;
}

Code: Select all

import octaneInstancer
octaneInstancer.ui()

Code: Select all

internalVar -usd

Code: Select all

rehash