display.tcl

# $Id$ 

package require jmath
package require image
package require display
package require sams

namespace eval sams {
  proc showFrame { fr } {
    ::display::show [$fr img]
    set size [::sams::Frame_size $fr]
    for { set i 0 } { $i < $size } { incr i } {
      set f [::sams::Frame_get $fr $i]
      set angle [::sams::Feature_o $f]
      ::display::drawCircle [$fr img] [::sams::Feature_x $f] [::sams::Feature_y $f]\
      [::sams::Feature_s $f] 

      set xR [expr [::sams::Feature_s $f] * cos($angle)]
      set yR [expr [::sams::Feature_s $f] * sin($angle)]
      ::display::drawLine [$fr img] [::sams::Feature_x $f] [::sams::Feature_y $f] \
      [expr [::sams::Feature_x $f]+$xR] [expr [::sams::Feature_y $f]+$yR]
    }
  }

  proc drawProjectedHypothesis { img hc } {
    set pm [$hc view2Frame]
    display::drawLine $img [jmath::vec_get $pm 0] [jmath::vec_get $pm 1] [jmath::vec_get $pm 2] [jmath::vec_get $pm 3] 2 "green"
    display::drawLine $img [jmath::vec_get $pm 0] [jmath::vec_get $pm 1] [jmath::vec_get $pm 4] [jmath::vec_get $pm 5] 2 "green"
    display::drawLine $img [jmath::vec_get $pm 6] [jmath::vec_get $pm 7] [jmath::vec_get $pm 2] [jmath::vec_get $pm 3] 2 "green"
    display::drawLine $img [jmath::vec_get $pm 6] [jmath::vec_get $pm 7] [jmath::vec_get $pm 4] [jmath::vec_get $pm 5] 2 "green"
    display::drawText $img [jmath::vec_get $pm 8] [jmath::vec_get $pm 9] [sams::Hypothesis_m_view_id_get $hc] "green"
    set mt [sams::Hypothesis_m_matches_get $hc]
    set size [::sams::MatchesVec_size $mt]
    for {set i 0} {$i < $size} {incr i} {
      set ff [::sams::Match_frameRef_get [sams::MatchesVec_get $mt $i]]
      ::display::drawCircle $img [::sams::Feature_x $ff] [::sams::Feature_y $ff] [::sams::Feature_s $ff] 
      set f [$hc part2Frame $i]
      ::display::drawCircle $img [$f get 0] [$f get 1] [$f get 3] 1 "green"
      set xR [expr [$f get 3] * cos([$f get 2])]
      set yR [expr [$f get 3] * sin([$f get 2])]
      ::display::drawLine $img [$f get 0] [$f get 1] \
      [expr [$f get 0]+$xR] [expr [$f get 1]+$yR] 1 "green"
      ::display::drawLine $img [::sams::Feature_x $ff] [::sams::Feature_y $ff] [$f get 0] [$f get 1] 1 "green"
    }
  }

  proc drawProjectedMatches { img fr mt tr } {
    set size [::sams::MatchesVec_size $mt]
    for {set i 0} {$i < $size} {incr i} {
      set m [::sams::MatchesVec_get $mt $i]
      if { [sams::Match_valid_get $m] } {
        kernel::putsDbg "draw matched feature"
        set ff [::sams::Match_frameRef_get $m]
        ::display::drawCircle $img [::sams::Feature_x $ff] [::sams::Feature_y $ff] [::sams::Feature_s $ff] 2
        kernel::putsDbg "draw matched part"
        set vpi [::sams::Match_modelRef_get $m]
        set f [$vpi transform $tr]
        ::display::drawCircle $img [$f get 0] [$f get 1] [$f get 3] 1 "green"
        kernel::putsDbg "link match"
        ::display::drawLine $img [::sams::Feature_x $ff] [::sams::Feature_y $ff] [$f get 0] [$f get 1] 1 "green"
      }
    }
  }

}

package provide sams 1.0