display.tcl

#
# display related macro. \file display.tcl  \ingroup premodeler
#

package require gfm
package require image
package require datareader
package require display

namespace eval premodeler {
    ##
    # displays tracked harris points on every image of the processed sequence provide you clicked enough :)
    ##
    proc displayTrackedPointsOnSequence { engine } {
  set result [::premodeler::Engine_getProcessResult $engine]
  set dataReader [premodeler::Engine_getDataReader $engine]
  set sensor [premodeler::Engine_getSensorName $engine]
  set monoReader [datareader::DataReader_getMonoReader $dataReader $sensor]
  set i 0
  ::premodeler::displayTrackedPointsOnImage $result $monoReader $i
    }

    ##
    # displays tracked harris points on an image of the sequence matching result
    ##
    proc displayTrackedPointsOnImage { processResult reader counter  {enableClick "true"} } {
  if {$counter < [premodeler::sizeSequenceMR $processResult]} {
      set vmr [premodeler::getVMR $processResult $counter]
      set index [premodeler::ViewMR_vIndex_get $vmr]
      set mr [premodeler::ViewMR_vMatchingResult_get $vmr]
      set dr [gfm::MatchingResult_pointsMatch $mr]
      set img [datareader::MonoReader_loadImage $reader $index]
      display::show $img
      gfm::displayTrackedHarrisPoints $img $dr
  }

  if $enableClick {
      bind .$img <ButtonPress-1> "::premodeler::displayTrackedPointsOnImage $result $reader [expr $counter + 1]"
  }
    }

    ##
    # displays tracked harris points on an image given its index.
    # note that first image of the process can't be used since it's the reference
    ##
    proc displayTrackedPointsOnView { engine imgIndex } {
  set result [::premodeler::Engine_getProcessResult $engine]
  set begin [premodeler::Engine_getBegin $engine]
  set end [premodeler::Engine_getEnd $engine]
  set step [premodeler::Engine_getStep $engine]
  if {$imgIndex <= $end} {
      if {$imgIndex >= [expr $begin + $step]} {
    if {[expr [expr $imgIndex - $begin ] % $step] == 0} {
        set vmr [premodeler::getVMRofImg $result $imgIndex]
        set mr [premodeler::ViewMR_vMatchingResult_get $vmr]
        set dr [gfm::MatchingResult_pointsMatch $mr]
        set dataReader [premodeler::Engine_getDataReader $engine]
        set sensor [premodeler::Engine_getSensorName $engine]
        set reader [datareader::DataReader_getMonoReader $dataReader $sensor]
        set img [datareader::MonoReader_loadImage $reader $imgIndex]
        display::show $img
        gfm::displayTrackedHarrisPoints $img $dr
    } else {
        puts "don't fit within step!"
    }
      } else {
    puts "prior to begin!"
      }
  } else {
      puts "exceeds end!"
  }
    }
    
    ##
    # displays matched points between request image and best fitting one.
    ##
    proc displayRecognition { recognizer engine {color "green"} } {
  set img_file [premodeler::ObjectRecognizer_getRequestImageFileName $recognizer]
  set ref_img [image::Image_loadImage $img_file]
  set match_img_id [premodeler::ObjectRecognizer_getBestView $recognizer]
  set dataReader [premodeler::Engine_getDataReader $engine]
  set sensor [premodeler::Engine_getSensorName $engine]
  set reader [datareader::DataReader_getMonoReader $dataReader $sensor]
  set match_img [datareader::MonoReader_loadImage $reader $match_img_id]
  set matches [premodeler::ObjectRecognizer_getBestMatchesVector $recognizer]
  set result_img [premodeler::mergeImages $ref_img $match_img]
  display::show $result_img
  for {set i 0} {$i < [gfm::sizeMatchesVec $matches]} {incr i} {
      set hp1 [gfm::getMatchFirst [ gfm::getMatchesVec $matches $i]]
      set hp2 [gfm::getMatchSecond [ gfm::getMatchesVec $matches $i]]
      set u1 [expr round([ gfm::InterestPoint_u $hp1 ])]
      set v1 [expr round([ gfm::InterestPoint_v $hp1 ])]
      display::drawSquare $result_img $u1 $v1 5 1 $color {jafar hpm_match}
      display::drawText $result_img $u1 [expr $v1 - 10] "$i" $color "c" {jafar hpm_match}
      set u2 [expr round([ gfm::InterestPoint_u $hp2 ])]
      set offset [::image::CvImage_width $ref_img]
      set u2 [expr $u2 + $offset]
      set v2 [expr round([ gfm::InterestPoint_v $hp2 ])]
      display::drawSquare $result_img $u2 $v2 5 1 $color {jafar hpm_match}
      display::drawText $result_img $u2 [expr $v2 - 10] "$i" $color "c" {jafar hpm_match}
      display::drawLine $result_img $u1 $v1 $u2 $v2 1 "blue"
  }
    }

}
package provide premodeler 0.9