// Assumes the shape node is a curve with 4 cv's
global proc string curveEncode(string $shapeNode)
{
string $newname = $shapeNode;
for($i = 0; $i < 4; $i++)
    {
    vector $v = `pointOnCurve -nn -pr ($i * 0.333) $shapeNode`;
    $newname += "_" + floatToStr($v.x);
    $newname += "_" + floatToStr($v.y);
    $newname += "_" + floatToStr($v.z);
    }
rename -ignoreShape $shapeNode $newname;
return $newname;
}
  
// Input: 0.02 returned as p02
//       -0.02 returned as n02
global proc string floatToStr(float $f)
{
string $out;
if($f >= 0 && $f < 0.001)
    $out = "p001";
else if($f < 0 && $f > -0.001)
    $out = "n001";
else if($f > 0.999)
    $out = "p999";
else if($f < -0.999)
    $out = "n999";
else if($f < 0)
    $out = "n" + abs(trunc($f * 1000));
else
    $out = "p" + trunc($f * 1000);
return $out;
}

// Creates an arbitary number of curves. Returns the names
// of the transform nodes of the curves.
global proc string[] curveCreate(int $num, vector $dir, vector $position)
{
string $shapeNode[];
string $transNodes[];
  
for($n = 0; $n < $num; $n++) {
    $transNodes[$n] = doCurve($dir);
    $shapeNode = `listRelatives -s`;
    $shapeNode[0] = curveEncode($shapeNode[0]);
    }
return $transNodes;
}
  
// Creates a curve and returns the name of its transform node.
proc string doCurve(vector $dir)
{
$dir = $dir + unit(sphrand(1) * 0.25);
vector $d = $dir/3;
vector $cv0 = <<rand(-0.25,0.25), 0, rand(-0.25,0.25)>>;
vector $cv1 = $d * 1;
vector $cv2 = $d * 2;
vector $cv3 = $d * 3;
float $len = length($dir.x, $dir.z);
float $droop3 = $len * rand(2, 0.950);
float $droop2 = $len * 0.25;
float $droop1 = $len * 0.15;
$cv1 = <<$cv1.x, $cv1.y - $droop1, $cv1.z>>;
$cv2 = <<$cv2.x, $cv2.y - $droop2, $cv2.z>>;
$cv3 = <<$cv3.x, $cv3.y - $droop3, $cv3.z>>;
return  `curve -d 3  -p ($cv0.x) ($cv0.y) ($cv0.z)
       -p ($cv1.x) ($cv1.y) ($cv1.z)
       -p ($cv2.x) ($cv2.y) ($cv2.z)
       -p ($cv3.x) ($cv3.y) ($cv3.z)`;
}
  
// Returns the length of a 2d vector
proc float length(float $a, float $b)
{
return sqrt($a * $a + $b * $b);
}

global proc curveAddSharedAttrRI(string $transformNodes[], 
                                 float $base, 
                                 float $tip)
{
rmanEd_CreateSettings kGeometric;
customReqAttrsRefresh("rmanSettings.rmanSettings");
rmanAddAttr "rmanSettings" rman__torattr___curveBaseWidth $base;
rmanAddAttr "rmanSettings" rman__torattr___curveTipWidth  $tip;
string $attrnm=`rmanGetAttrName customShadingGroup`;
rmanAddAttr "rmanSettings" $attrnm "";
  
select $transformNodes;
rmanEd_AttachSettings kGeometric;
}

proc decode { input } {
    if { [string equal [string index $input 0] p] } {
        set out 0.[string trimleft $input p]
        append out " "
    } else {
        set out -0.[string trimleft $input n]
        append out " "
        }
    return $out
    }
    
proc processJobFiles { src_dir } {
    # Get a preliminary list of job files.
    set files [glob -nocomplain -directory $src_dir *.job.rib]
    foreach item $files {
        # Paths ending with "_attr.job.rib" or "jobCompile.job.rib"
        # will be ignored
        if { [regexp {(?w)_attr.job.rib} $item] == 0 && 
             [regexp {(?w)jobCompile.job.rib} $item] == 0} {
            set filename [file tail $item]
            set tokens [split $filename _]
            # Skip the filename if it does not have exactly 14 tokens
            if {[llength $tokens] != 14} {
                continue
                }
            # Ignore the first token it will be the curve name, such
            # as "curveShape1", also ignore the last token is will be
            # something similiar to "0.job.rib"
            set tokens [lrange $tokens 1 12]
            set N "\"N\" \[ "
            # Because RfM Pro repeats the first and last cv's
            # we must also repeat the first and last normals
            for {set i 0} {$i < 12} {incr i 3} {
                set x [lindex $tokens $i]
                set y [lindex $tokens [expr $i + 1]]
                set z [lindex $tokens [expr $i + 2]]
                set xyz "[decode $x][decode $y][decode $z]"
                if {$i == 0 || $i == 9} {
                    append N $xyz$xyz
                } else {
                    append N $xyz
                    }
                }
            append N "\]"
            # Open the rib file and append the string that specifies
            # the list of normals
            set f [open $item a]
            puts $f $N
            close $f
            }
        }
    }
  
set src PATH_TO/maya/projects/default/renderman/untitled/rib/job
set dst PATH_TO/maya/projects/default/renderman/untitled/rib/job    
  
processJobFiles $src