module LogControl::Functions

# File lib/vizkit/widgets/log_control/log_control.rb, line 326
def auto_replay
  @log_replay.reset_time_sync
  display_info              #display info --> otherwise info is maybe not up to date
  timeline.setSliderIndex(@log_replay.sample_index)
  @timer.start
end

# File lib/vizkit/widgets/log_control/log_control.rb, line 382
def bback_clicked
  return if !@log_replay.replay?
  if playing?
            update_target_speed @log_replay.speed*0.5
  else
    if actionNone.isChecked
        @log_replay.step_back
    else
        @port ||= @log_replay.current_port
        @log_replay.step_back
        timeline.setSliderIndex(@log_replay.sample_index)
        display_info
        if @port != @log_replay.current_port
            Orocos::Async.event_loop.once do 
                bback_clicked
            end
        else
            @port = nil
        end
    end
  end
  timeline.setSliderIndex(@log_replay.sample_index)
  display_info
end

# File lib/vizkit/widgets/log_control/log_control.rb, line 357
def bnext_clicked
  return if !@log_replay.replay?
  if playing?
    update_target_speed @log_replay.speed*2
  else
    if actionNone.isChecked
        @log_replay.step(false)
    else
        @port ||= @log_replay.current_port
        @log_replay.step(false)
        timeline.setSliderIndex(@log_replay.sample_index)
        display_info
        if @port != @log_replay.current_port
            Orocos::Async.event_loop.once do 
                bnext_clicked
            end
        else
            @port = nil
        end
    end
  end
  timeline.setSliderIndex(@log_replay.sample_index)
  display_info
end

# File lib/vizkit/widgets/log_control/log_control.rb, line 436
def bplay_clicked
  return if !@log_replay.replay?
  if playing?
    bplay.icon = @play_icon
    @timer.cancel
  else
    bplay.icon = @pause_icon
    if(timeline.getSliderIndex < timeline.getStartMarkerIndex || timeline.getSliderIndex >= timeline.getEndMarkerIndex)
        seek_to(timeline.getStartMarkerIndex)
    end
    bstop_clicked if @log_replay.eof?
    self.speed = @user_speed
    auto_replay
  end
end

# File lib/vizkit/widgets/log_control/log_control.rb, line 407
def bstop_clicked
   return if !@log_replay.replay?
   bplay_clicked if playing?
   if timeline.getStartMarkerIndex == 0
     @log_replay.rewind
   else
     seek_to(timeline.getStartMarkerIndex)
   end
   @log_replay.reset_time_sync
   timeline.setSliderIndex(@log_replay.sample_index)
   display_info
end

# File lib/vizkit/widgets/log_control/log_control.rb, line 38
def config(replay,options=Hash.new)
  raise "Cannot control #{replay.class}" if !replay.instance_of?(Orocos::Log::Replay)
  raise "LogControl: config was called more than once!" if @log_replay

  #workaround because qt objects created via an ui File
  #cannot be overloaded
  setObjectName("LogControl")
  @event_filter = CloseAllFilter.new(self)
  installEventFilter(@event_filter)
  
  @log_replay = replay
  @replay_on = false 
  @user_speed = @log_replay.speed
  
  dir = File.dirname(__FILE__)
  @pause_icon =  Qt::Icon.new(File.join(dir,'pause.png'))
  @play_icon = Qt::Icon.new(File.join(dir,'play.png'))
  setAttribute(Qt::WA_QuitOnClose, true);

  connect(timeline, SIGNAL('indexSliderMoved(int)'), index, SLOT('setValue(int)'))
  connect(timeline, SIGNAL('endMarkerMoved(int)'), index, SLOT('setValue(int)'))
  connect(timeline, SIGNAL('startMarkerMoved(int)'), index, SLOT('setValue(int)'))
  timeline.connect(SIGNAL('indexSliderReleased(int)'),self,:slider_released)
  bnext.connect(SIGNAL('clicked()'),self,:bnext_clicked)
  bback.connect(SIGNAL('clicked()'),self,:bback_clicked)
  bstop.connect(SIGNAL('clicked()'),self,:bstop_clicked)
  bplay.connect(SIGNAL('clicked()'),self,:bplay_clicked)
  dtarget_speed.connect(SIGNAL('valueChanged(double)'),self,:update_target_speed)

  timeline.connect(SIGNAL("endMarkerReleased(int)")) do |value| 
    index.setValue(@log_replay.sample_index)
  end
  timeline.connect(SIGNAL("startMarkerReleased(int)")) do |value| 
    index.setValue(@log_replay.sample_index)
  end
  timeline.connect SIGNAL('indexSliderClicked()') do 
      bplay_clicked if playing?
  end

  index.connect SIGNAL('editingFinished()') do
    if @log_replay.sample_index != index.value && !playing?
      #prevents the bottom play from starting replay
      #because of enter
      @replay_on = true
      seek_to(index.value)
    end
  end

  @log_replay.align unless @log_replay.aligned?
  return if !@log_replay.replay?
  @log_replay.process_qt_events = true
  timeline.setSteps(@log_replay.size-1)
  timeline.setStepSize 1
  timeline.setSliderIndex(@log_replay.sample_index)
  index.setMaximum(@log_replay.size-1)

  #add replayed streams to tree view 
  Vizkit.setup_tree_view treeView
  model = Vizkit::VizkitItemModel.new
  treeView.setModel model
  model.setHorizontalHeaderLabels ["Replayed Tasks","Information"]


  @global_meta_data = Vizkit::GlobalMetaItem.new @log_replay
  @global_meta_data2 = Vizkit::GlobalMetaItem.new @log_replay,:item_type => :value
  model.appendRow [@global_meta_data, @global_meta_data2]
  @log_replay.tasks.each do |task|
      next unless task.used?
      task = task.to_async
      @item1 = Vizkit::LogTaskItem.new(task)
      @item2 = Vizkit::LogTaskItem.new(task,:item_type => :value) 
      model.appendRow [@item1, @item2]
  end
  treeView.resizeColumnToContents(0)

  actionNone.connect(SIGNAL("triggered(bool)")) do |checked|
    if checked 
        actionCurrent_Port.setChecked(false)
    else 
        actionNone.setChecked(true)
    end
  end

  actionCurrent_Port.connect(SIGNAL("triggered(bool)")) do |checked|
    if checked 
        actionNone.setChecked(false)
    else 
        actionCurrent_Port.setChecked(true)
    end
  end

  actionExport.connect(SIGNAL("triggered()")) do 
    setDisabled(true)
    file = Qt::FileDialog::getSaveFileName(nil,"Export Log File",File.expand_path("."),"Log Files (*.log)")
    if file
      bstop_clicked
      progress = Qt::ProgressDialog.new
      progress.setLabelText "exporting streams"
      progress.show       
      @log_replay.export_to_file(file,timeline.getStartMarkerIndex, timeline.getEndMarkerIndex) do |index,max|
        progress.setMaximum(max)
        progress.setValue(index)
        Vizkit.process_events
        progress.wasCanceled
      end
      progress.close if progress.isVisible
    end
    setEnabled(true)
  end

  actionMovie.connect(SIGNAL("triggered()")) do
    export_to_images
  end

  actionViewer.connect(SIGNAL("triggered(bool)")) do |checked|
    widget = Vizkit.default_loader.LogMarkerViewer
    widget.config2(@log_replay.log_markers)
    widget.show
  end

  @last_info = Time.now
  @timer = Orocos::Async.event_loop.every 0.001,false do
      begin
          # make sure we only process steps for around 10ms
          # and dont block here
          update_time = Time.now
          while @log_replay.sync_step? && Time.now - update_time < 0.01
              sample = @log_replay.step
              if @log_replay.sample_index >= timeline.getEndMarkerIndex || !sample
                  bplay_clicked
                  break
              end
          end
      rescue Exception => e
          bplay_clicked
          Qt::MessageBox::warning(nil,"Corrupted logfile", e.to_s)
          Vizkit.debug do
              Vizkit.debug e.message
              e.backtrace.each do |bt|
                  Vizkit.debug "  #{bt}"
              end
              nil
          end
      end
      #we do not display the info every step to save cpu time
      if Time.now - @last_info > 0.1
          @last_info = Time.now
          display_info      
          timeline.setSliderIndex(@log_replay.sample_index)
      end
  end
  @timer.doc = "Log::Replay"
  display_info
end

# File lib/vizkit/widgets/log_control/log_control.rb, line 305
def display_info
  if @log_replay.time
    timestamp.text = @log_replay.time.strftime("%a %D %H:%M:%S." + "%06d" % @log_replay.time.usec)
    dcurrent_speed.text = ( '%.1f' % @log_replay.actual_speed )
    index.setValue(@log_replay.sample_index)
    last_port.text = @log_replay.current_port.full_name if @log_replay.current_port
  else
    timestamp.text = "0"
  end
end

# File lib/vizkit/widgets/log_control/log_control.rb, line 196
def enable_grabbing(object)
  # respond_to? is broken on qtruby ... do it the hard way
  begin object.enableGrabbing
  rescue NoMethodError
  end
  begin
    object.grab # see comment on qtruby and respond_to above
    Grabber.new(object.objectName, lambda { object.grab })
  rescue NoMethodError
    # Generic way to grab widgets, but we do it only on those that don't
    # have a parent
    if object.kind_of?(Qt::Widget) && !object.parent
      image = Qt::Image.new(object.size, Qt::Image::Format_RGB888)
      Grabber.new(object.objectName, lambda { object.render(image); image })
    end
  end
end

# File lib/vizkit/widgets/log_control/log_control.rb, line 218
def export_to_images(destination_path = nil, sampling_period = 0.04)
  specs = Vizkit.default_loader.all_plugin_specs
  setup = Array.new
  seen_widgets = Array.new

  specs.each do |spec|
    widgets = spec.created_plugins.map do |w|
      next if seen_widgets.include?(w)

      if grabber = enable_grabbing(w)
        seen_widgets << w
        grabber
      end
    end.compact
    widgets.each_with_index do |w, i|
      setup << [w, File.join(setup.size.to_s, "%06i.png")]
    end
  end

  # Create the subdirectories
  setup.each do |_, p|
    FileUtils.mkdir_p File.dirname(p)
  end

  destination_path ||= Qt::FileDialog.getExistingDirectory(self)
  if destination_path.empty?
      return
  end

  frame_count = Integer(@log_replay.duration / sampling_period)
  progress = Qt::ProgressDialog.new(self)
  bar = Qt::ProgressBar.new(progress)
  progress.bar = bar

  progress.setLabelText "Exporting #{frame_count} frames for #{setup.size} widgets in #{destination_path}\n" +
      setup.map { |w,p| " #{w.name}: #{p}" }.join("\n")
  progress.minimum = 0
  progress.maximum = frame_count
  bar.format = "%v/%m"
  progress.show       

  grab_period = 0.1
  grab_index  = 0
  next_grab_time = nil
  while true
    sample = @log_replay.step
    if @log_replay.sample_index >= timeline.getEndMarkerIndex || !sample
      bplay_clicked
      break
    end

    current_time = @log_replay.current_time
    next_grab_time ||= current_time
    if next_grab_time <= current_time
      Vizkit.step
      frames = setup.map do |grabber, path|
        [grabber.call, path]
      end
      while next_grab_time <= current_time
        frames.each do |image, path|
          image.save(path % [grab_index])
        end
        grab_index += 1
        next_grab_time += grab_period
      end
    end
    progress.value = grab_index
    break if progress.wasCanceled
  end

  setup.each do |w, _|
    begin w.disableGrabbing
    rescue NoMethodError
    end
  end
  progress.close
end

# File lib/vizkit/widgets/log_control/log_control.rb, line 296
def playing?
    @timer.running?
end

# File lib/vizkit/widgets/log_control/log_control.rb, line 420
def refresh
    @log_replay.refresh
end

# File lib/vizkit/widgets/log_control/log_control.rb, line 424
def seek_to(index)
  bplay_clicked if playing?
  if index.is_a? Time
    @log_replay.seek(index)
    timeline.setSliderIndex(@log_replay.sample_index)
    display_info
  else
    timeline.setSliderIndex index
    slider_released(index)
  end
end

# File lib/vizkit/widgets/log_control/log_control.rb, line 333
  def slider_released(index)
#    return if !@log_replay.replay?
      @log_replay.reset_time_sync
      @log_replay.seek(timeline.getSliderIndex)
      display_info
  rescue Exception => e
      Qt::MessageBox::warning(nil,"Corrupted logfile",e.to_s)
      Vizkit.debug do
          Vizkit.debug e.message
          e.backtrace.each do |bt|
              Vizkit.debug "  #{bt}"
          end
          nil
      end
  end

# File lib/vizkit/widgets/log_control/log_control.rb, line 322
def speed
  @user_speed
end

# File lib/vizkit/widgets/log_control/log_control.rb, line 316
def speed=(double)
  @user_speed = double
  @log_replay.speed = double
  @log_replay.reset_time_sync
end

# File lib/vizkit/widgets/log_control/log_control.rb, line 300
def timeline_marker(start_time,end_time)
    timeline.setStartMarkerIndex(@log_replay.sample_index_for_time(start_time))
    timeline.setEndMarkerIndex(@log_replay.sample_index_for_time(end_time))
end

# File lib/vizkit/widgets/log_control/log_control.rb, line 349
def update_target_speed value
    if value >= 0.001 && value != @log_replay.speed
        @log_replay.speed = value.to_f
        @log_replay.reset_time_sync
        dtarget_speed.value = @log_replay.speed if value != dtarget_speed.value
    end
end