# ----------------------------------------------------------------------
#  EXAMPLE:  Rappture optimization
#
#  This script shows the core of the Rappture optimization loop.
#  It represents a simple skeleton of the much more complicated
#  Rappture GUI, but it drives enough of the optimization process
#  to drive development and testing.
#
#  Run this as:  wish simple.tcl ?-tool path/to/tool.xml?
#
# ======================================================================
#  AUTHOR:  Michael McLennan, Purdue University
#  Copyright (c) 2004-2012  HUBzero Foundation, LLC
#
#  See the file "license.terms" for information on usage and
#  redistribution of this file, and for a DISCLAIMER OF ALL WARRANTIES.
# ======================================================================
set auto_path [linsert $auto_path 0 /tmp/opt/lib]
package require BLT
package require Itcl
package require Rappture
package require RapptureGUI
set auto_path [linsert $auto_path 0 /home/ganesh/workspace/optim_post_dir_changes/src/lib]
package require -exact RapptureOptimizer 1.1

set pop 0
set popsize 100  ;# size of each population for genetic algorithm


# ----------------------------------------------------------------------
#  Create a Tool object based on the tool.xml file...
# ----------------------------------------------------------------------
Rappture::getopts argv params {
    value -tool tool.xml
}

# open the XML file containing the tool parameters
if {![file exists $params(-tool)]} {
    puts stderr "can't find tool \"$params(-tool)\""
    exit 1
}
set xmlobj [Rappture::library $params(-tool)]

set installdir [file dirname $params(-tool)]
if {"." == $installdir} {
    set installdir [pwd]
}

set tool [Rappture::Tool ::#auto $xmlobj $installdir]

# ----------------------------------------------------------------------
#  Create some plotting stuff so we can watch the progress of the
#  optimization as it runs.
# ----------------------------------------------------------------------
blt::graph .space
.space xaxis configure -title "x1" -min -2 -max 2
.space yaxis configure -title "x2" -min -2 -max 2
.space legend configure -hide yes
pack .space -side left -expand yes -fill both

blt::graph .value
.value xaxis configure -title "job number" -min 0
.value yaxis configure -title "f(x1,x2)" -logscale yes -max 1
.value legend configure -hide yes
pack .value -side left -expand yes -fill both

button .quit -text Quit -command "optim abort yes"
pack .quit
button .restart -text Restart -command "optim restart yes"
pack .restart

set colors {magenta purple blue DeepSkyBlue cyan green yellow Gold orange tomato red FireBrick black brown grey lavender wheat navy}


#for {set pop [expr [llength $colors]-1]} {$pop >= 0} {incr pop -1} {
#    blt::vector x1vec$pop
#   blt::vector x2vec$pop
#    .space element create dot -xdata x1vec$pop -ydata x2vec$pop -color black -linewidth 0 -symbol scross
#        -color [lindex $colors $pop] -linewidth 0 -label "Population #$pop"
    blt::vector jobvec
    blt::vector fvec
    .value element create line -xdata jobvec -ydata fvec -color black -symbol none

#        -color [lindex $colors $pop] -symbol none



set jobnumber 0
set pop -1
proc add_to_plot {xmlobj} {
		
    global jobnumber popsize pop
    if {$jobnumber%$popsize == 0} {
	incr pop
	blt::vector x1vec$pop
	blt::vector x2vec$pop

	blt::graph .space$pop
	.space$pop xaxis configure -title "x1" -min -2 -max 2
	.space$pop yaxis configure -title "x2" -min -2 -max 2
	.space$pop legend configure -hide yes
	pack .space$pop -side left -expand yes -fill both
	.space$pop element create dot -xdata x1vec$pop -ydata x2vec$pop -color black -linewidth 0 -symbol scross
	
    }	
    x1vec$pop append [$xmlobj get input.number(x1).current]
    x2vec$pop append [$xmlobj get input.number(x2).current]
    jobvec append $jobnumber
    fvec append [$xmlobj get output.number(f).current]
    incr jobnumber
#    optim samples $jobnumber
}

# ----------------------------------------------------------------------
#  Create an optimization context and configure the parameters used
#  for optimization...
# ----------------------------------------------------------------------
Rappture::optimizer optim -tool $tool -using pgapack

optim add number input.number(x1) -min -2 -max 2 -randdist uniform -strictmin yes -strictmax yes
optim add number input.number(x2) -min -2 -max 2 -randdist uniform -strictmin yes -strictmax yes
optim configure -operation minimize -popsize 100 -maxruns 100 -mutnrate 0.1 -crossovrate 0.8 -randnumseed 20 -stpcriteria maxiter -mutnandcrossover no -allowdup yes -numReplPerPop 40 -mutnValue 0.1 -crossovtype sbx



optim get configure 



set status [optim perform \
    -fitness output.number(f).current \
    -updatecommand add_to_plot]
optim samples 10
puts "done: $status"