cosmos 4.2.4 → 4.5.0

data/lib/cosmos/gui/utilities/classification_banner.rb

@@ -0,0 +1,60 @@
+# encoding: ascii-8bit
+
+# Copyright 2014 Ball Aerospace & Technologies Corp.
+# All Rights Reserved.
+#
+# This program is free software; you can modify and/or redistribute it
+# under the terms of the GNU General Public License
+# as published by the Free Software Foundation; version 3 with
+# attribution addendums as found in the LICENSE.txt
+
+module ClassificationBanner
+
+  # Add to Qt::MainWindow the ability to display a classification banner. Typically used on every QtTool and Screen
+  class Qt::MainWindow
+
+    def add_classification_banner
+      # Add a classification banner if the system configuration called for one
+      classification_banner = Cosmos::System.instance.classificiation_banner
+      unless classification_banner.nil?
+        # Get the RGB color information from the classification_banner
+        color_red   = classification_banner['color'].red
+        color_green = classification_banner['color'].green
+        color_blue  = classification_banner['color'].blue
+        color_rgb   = "#{color_red},#{color_green},#{color_blue}"
+
+        # Create a classification toolbar
+        classification_toolbar = Qt::ToolBar.new
+        # Disable right clicking on the bar (prevents it from being hidden unintentionally)
+        classification_toolbar.setContextMenuPolicy(Qt::PreventContextMenu)
+        # Freeze the bar at the top
+        classification_toolbar.setFloatable(false)
+        classification_toolbar.setMovable(false)
+        # Specify sizes and set the style (background = background color, color = text color)
+        classification_toolbar.minimumHeight = 20
+        classification_toolbar.maximumHeight = 20
+        classification_toolbar.setStyleSheet("background:rgb(#{color_rgb});color:white;text-align:center;border:none;")
+
+        # Create a frame that will hold a horizontal layout
+        label_frame = Qt::Frame.new
+        label_layout = Qt::HBoxLayout.new(label_frame)
+        label_layout.setContentsMargins(0,1,0,0) # Centers the text nicely inside the horizontal layout
+
+        # Create a label of the classification and add it the horizontal layout
+        label = Qt::Label.new("#{classification_banner['display_text']}")
+        label.setStyleSheet("margin:0px;")
+
+        # Add stretchers on either side so it is always in the middle and looks nice
+        label_layout.addStretch(1)
+        label_layout.addWidget(label)
+        label_layout.addStretch(1)
+
+        # Add the frame to the main toolbar, then add the toolbar to the MainWindow
+        classification_toolbar.addWidget(label_frame)
+        self.addToolBar(classification_toolbar)
+      end
+
+    end
+
+  end
+end

data/lib/cosmos/gui/utilities/screenshot.rb

@@ -24,7 +24,7 @@ module Cosmos
       end
 
       # Create a Pixmap to save the screenshot into
-      pixmap = Qt::Pixmap::grabWindow(window.winId())
+      pixmap = Qt::Pixmap::grabWidget(window)
       pixmap.save(filename)
       pixmap
     end

data/lib/cosmos/gui/utilities/script_module_gui.rb

@@ -10,6 +10,7 @@
 
 require 'cosmos/script/script'
 require 'cosmos/gui/choosers/combobox_chooser'
+require 'cosmos/tools/tlm_viewer/screen'
 
 $cmd_tlm_gui_window = nil
 
@@ -163,90 +164,116 @@ module Cosmos
       return true # Aborted - Don't retry
     end
 
-    def prompt_to_continue(string)
+    def prompt_to_continue(string, text_color: nil, background_color: nil, font_size: nil, font_family: nil, informative: 'Ok to Continue?', details: nil, combo: nil)
+      result = 'Success'
       loop do
-        stop = false
         _get_main_thread_gui do |window|
-          result = Qt::MessageBox::question(window,
-                                            "COSMOS",
-                                            "#{string}\n\nOK to Continue?",
-                                            Qt::MessageBox::Ok | Qt::MessageBox::Cancel,
-                                            Qt::MessageBox::Ok)
-          if result == Qt::MessageBox::Ok
-            Logger.info "User pressed 'Ok' for '#{string}'"
-          else
-            stop = true
+          box = Qt::MessageBox.new(window)
+          box.setIcon(Qt::MessageBox::Question)
+          box.setWindowTitle("Prompt")
+          box.setText(string)
+          if informative
+            box.setInformativeText(informative)
+          elsif combo
+            box.setInformativeText(' ') # Added to ensure consistent layout when adding combo
+          end
+          box.setStandardButtons(Qt::MessageBox::Ok | Qt::MessageBox::Cancel)
+          box.setDefaultButton(Qt::MessageBox::Ok)
+          box.setDetailedText(details) if details
+          text_color = 'black' unless text_color
+          background_color = 'white' unless background_color
+          font = box.font()
+          font_size = font.pointSize unless font_size
+          font_family = font.family unless font_family
+          box.setStyleSheet("QMessageBox { background-color: #{background_color}; }\
+                             QMessageBox QLabel { color: #{text_color}; font: #{font_size}pt '#{font_family}'}")
+
+          if combo
+            result = combo[0].clone
+            # Remove the items in the layout starting with the buttons so we can insert the combo
+            items = []
+            (2...box.layout.count).each do |x|
+              items << box.layout.takeAt(x)
+            end
+            chooser = ComboboxChooser.new(box, "Select:", combo)
+            chooser.setContentsMargins(11,11,11,11)
+            chooser.sel_command_callback = lambda { |value| result = value }
+            # Insert the combo right where we removed the existing items
+            box.layout.addWidget(chooser, 2, 0, 1, 2)
+            # Add back all the items we removed
+            items.each do |item|
+              widget = item && item.widget ? item.widget : nil
+              next unless item && item.widget
+              if item.widget.is_a? Qt::DialogButtonBox
+                box.layout.addItem(item, 3, 0, 1, 2)
+              elsif item.widget.is_a? Qt::Widget
+                box.layout.addItem(item, 4, 0, 1, 2)
+              end
+            end
+          end
+          box.exec()
+          if box.clickedButton.text == "Cancel"
             Logger.warn "User pressed 'Cancel' for '#{string}'"
+            result = 'Cancel'
+          else
+            if combo
+              Logger.info "User selected '#{result}' for '#{string}'"
+            else
+              Logger.info "User pressed 'Ok' for '#{string}'"
+              result = 'Ok'
+            end
           end
+          box.dispose
         end
-        if stop
+        if result == "Cancel"
           prompt_for_script_abort()
         else
           break
         end
       end
+      result
+    end
+
+    def prompt_combo_box(string, items, options)
+      options[:combo] = items
+      prompt_to_continue(string, options)
     end
 
-    def prompt_message_box(string, buttons)
+    def prompt_message_box(string, buttons, text_color: nil, background_color: nil, font_size: nil, font_family: nil, informative: nil, details: nil, vertical: false)
       loop do
         result = nil
         _get_main_thread_gui do |window|
-          msg = Qt::MessageBox.new(window)
-          msg.setText(string)
-          msg.setWindowTitle("Message Box")
+          box = Qt::MessageBox.new(window)
+          box.setIcon(Qt::MessageBox::Question)
+          box.setWindowTitle("Message Box")
+          box.setText(string)
+          box.setInformativeText(informative) if informative
           # Check if the last parameter is false which means they don't want
           # the Cancel button to be displayed
           if buttons[-1] == false
-            buttons[0..-2].each {|text| msg.addButton(text, Qt::MessageBox::AcceptRole)}
+            buttons[0..-2].each {|text| box.addButton(text, Qt::MessageBox::AcceptRole)}
           else
-            buttons.each {|text| msg.addButton(text, Qt::MessageBox::AcceptRole)}
-            msg.addButton("Cancel", Qt::MessageBox::RejectRole)
+            buttons.each {|text| box.addButton(text, Qt::MessageBox::AcceptRole)}
+            box.addButton("Cancel", Qt::MessageBox::RejectRole)
           end
-          msg.exec()
-          if msg.clickedButton.text == "Cancel"
+          box.setDetailedText(details) if details
+          text_color = 'black' unless text_color
+          background_color = 'white' unless background_color
+          font = box.font()
+          font_size = font.pointSize unless font_size
+          font_family = font.family unless font_family
+          box.setStyleSheet("QMessageBox { background-color: #{background_color}; }\
+                             QMessageBox QLabel { color: #{text_color}; font: #{font_size}pt '#{font_family}'}")
+          # Change the layout of the buttons to vertical if necessary
+          box.layout.itemAt(2).widget.setOrientation(Qt::Vertical) if vertical
+          box.exec()
+          result = box.clickedButton.text
+          if result == "Cancel"
             Logger.warn "User pressed 'Cancel' for '#{string}'"
           else
-            Logger.info "User pressed '#{msg.clickedButton.text}' for '#{string}'"
-          end
-          result = msg.clickedButton.text
-          msg.dispose
-        end
-        if result == "Cancel"
-          prompt_for_script_abort()
-        else
-          return result
-        end
-      end
-    end
-
-    def prompt_vertical_message_box(string, buttons)
-      loop do
-        result = buttons[0].clone
-        Qt.execute_in_main_thread(true, 0.05) do
-          dialog = _build_dialog(string)
-
-          # Check if the last parameter is false which means they don't want
-          # the Cancel button to be displayed
-          if buttons[-1] == false
-            cancel = false
-            display_buttons = buttons[0..-2]
-          else
-            cancel = true
-            display_buttons = buttons
+            Logger.info "User pressed '#{box.clickedButton.text}' for '#{string}'"
           end
-          button_layout = Qt::VBoxLayout.new
-          button_layout.setContentsMargins(11,11,11,11)
-          display_buttons.each do |button_text|
-            button = Qt::PushButton.new(button_text)
-            button.connect(SIGNAL('clicked()')) do
-              result.replace(button_text)
-              dialog.accept()
-            end
-            button_layout.addWidget(button)
-          end
-          dialog.layout.addLayout(button_layout)
-          dialog.layout.addWidget(_build_dialog_buttons(dialog, false, cancel))
-          result = "Cancel" unless _exec_dialog(dialog, string, result)
+          box.dispose
         end
         if result == "Cancel"
           prompt_for_script_abort()
@@ -256,33 +283,9 @@ module Cosmos
       end
     end
 
-    def prompt_combo_box(string, options)
-      loop do
-        result = options[0].clone
-        Qt.execute_in_main_thread(true, 0.05) do
-          dialog = _build_dialog(string)
-          # Check if the last parameter is false which means they don't want
-          # the Cancel button to be displayed
-          if options[-1] == false
-            cancel = false
-            display_options = options[0..-2]
-          else
-            cancel = true
-            display_options = options
-          end
-          chooser = ComboboxChooser.new(dialog, "Select:", display_options)
-          chooser.setContentsMargins(11,11,11,11)
-          chooser.sel_command_callback = lambda { |value| result.replace(value) }
-          dialog.layout.addWidget(chooser)
-          dialog.layout.addWidget(_build_dialog_buttons(dialog, true, cancel))
-          result = "Cancel" unless _exec_dialog(dialog, string, result)
-        end
-        if result == "Cancel"
-          prompt_for_script_abort()
-        else
-          return result
-        end
-      end
+    def prompt_vertical_message_box(string, buttons, options)
+      options[:vertical] = true
+      prompt_message_box(string, buttons, options)
     end
 
     def _build_dialog(message)
@@ -358,7 +361,30 @@ module Cosmos
       $cmd_tlm_gui_window
     end
 
-  end # module Script
-  $VERBOSE = old_verbose
+    def local_screen(title = "Local Screen", screen_def = nil, x_pos = nil, y_pos = nil, &block)
+      screen = nil
+      if block_given?
+        screen_def = yield
+      end
+
+      Qt.execute_in_main_thread(true) do
+        begin
+          if block_given?
+            screen = Screen.new(title, screen_def, nil, :REALTIME, x_pos, y_pos, nil, nil, false, false, block.binding )
+          else
+            screen = Screen.new(title, screen_def, nil, :REALTIME, x_pos, y_pos, nil, nil, false, false)
+          end
+        rescue Exception => err
+          puts err.formatted
+        end
+      end
 
-end # module Cosmos
+      screen
+    end
+
+    def close_local_screens
+      Qt.execute_in_main_thread(true) { Screen.close_all_screens(nil) }
+    end
+  end
+  $VERBOSE = old_verbose
+end

data/lib/cosmos/gui/widgets/dart_meta_frame.rb

@@ -16,6 +16,13 @@ module Cosmos
   class DartMetaFrame < Qt::Widget
     @@meta_filters = []
 
+    class LabelPopup < Qt::Label
+      attr_accessor :text
+      def mousePressEvent(event)
+        Qt::MessageBox.critical(self.parent, 'Error', @text)
+      end
+    end
+
     def initialize(parent)
       super(parent)
 
@@ -27,6 +34,7 @@ module Cosmos
       @groupbox = Qt::GroupBox.new("Meta Filter Selection")
       @vbox = Qt::VBoxLayout.new(@groupbox)
       @hbox1 = Qt::HBoxLayout.new
+
       @label = Qt::Label.new("Meta Filters: ")
       @hbox1.addWidget(@label)
       @meta_filters_text = Qt::LineEdit.new
@@ -43,6 +51,10 @@ module Cosmos
       @vbox.addLayout(@hbox1)
 
       @hbox2 = Qt::HBoxLayout.new
+      @error = LabelPopup.new
+      @error.setPixmap(Qt::Application.style.standardIcon(Qt::Style::SP_MessageBoxCritical).pixmap(20, 20))
+      @error.hide
+      @hbox2.addWidget(@error)
       @meta_item_name = Qt::ComboBox.new
       @meta_item_name.setMinimumWidth(200)
       @meta_item_name.setMaxVisibleItems(6)
@@ -102,7 +114,7 @@ module Cosmos
       if !@got_meta_item_names and !@update_thread
         @update_thread = Thread.new do
           begin
-            server = JsonDRbObject.new(System.connect_hosts['DART_DECOM'], System.ports['DART_DECOM'])
+            server = JsonDRbObject.new(System.connect_hosts['DART_DECOM'], System.ports['DART_DECOM'], 1.0, Cosmos::System.x_csrf_token)
             item_names = server.item_names("SYSTEM", "META")
             Qt.execute_in_main_thread do
               unless self.disposed?
@@ -113,8 +125,15 @@ module Cosmos
               end
             end
             @got_meta_item_names = true
-          rescue Exception
-            # We tried...
+            Qt.execute_in_main_thread { @error.hide }
+          rescue Exception => e
+            case e.message
+            when /execution expired/
+              @error.text = "Could not connect to the DART Decom Server. Is it running?"
+            else
+              @error.text = e.message
+            end
+            Qt.execute_in_main_thread { @error.show }
           ensure
             @update_thread = nil
             server.disconnect if defined? server

data/lib/cosmos/interfaces.rb

@@ -20,3 +20,5 @@ require 'cosmos/interfaces/protocols/template_protocol'
 require 'cosmos/interfaces/protocols/terminated_protocol'
 
 require 'cosmos/interfaces/protocols/override_protocol'
+require 'cosmos/interfaces/protocols/crc_protocol'
+require 'cosmos/interfaces/protocols/ignore_packet_protocol'

data/lib/cosmos/interfaces/dart_status_interface.rb

@@ -36,7 +36,7 @@ module Cosmos
       @status_packet.write('PACKET_ID', 1)
       @clear_errors_command = System.commands.packet('DART', 'CLEAR_ERRORS')
       @sleeper = Sleeper.new
-      @dart = JsonDRbObject.new(System.connect_hosts['DART_DECOM'], System.ports['DART_DECOM'])
+      @dart = JsonDRbObject.new(System.connect_hosts['DART_DECOM'], System.ports['DART_DECOM'], 1.0, Cosmos::System.x_csrf_token)
     end
 
     # Indicates if the interface is connected to its target(s) or not. Must be

data/lib/cosmos/interfaces/interface.rb

@@ -46,6 +46,9 @@ module Cosmos
     # @return [Array] Array of packet logger classes for this interface
     attr_accessor :packet_log_writer_pairs
 
+    # @return [Array] Array of stored packet log writers
+    attr_accessor :stored_packet_log_writer_pairs
+
     # @return [RawLoggerPair] RawLoggerPair instance or nil
     attr_accessor :raw_logger_pair
 
@@ -122,6 +125,7 @@ module Cosmos
       @reconnect_delay = 5.0
       @disable_disconnect = false
       @packet_log_writer_pairs = []
+      @stored_packet_log_writer_pairs = []
       @raw_logger_pair = RawLoggerPair.new(@name)
       @routers = []
       @cmd_routers = []
@@ -279,7 +283,11 @@ module Cosmos
 
     # Wrap all writes in a mutex and handle errors
     def _write
-      @write_mutex.synchronize { yield }
+      if @write_mutex.owned?
+        yield
+      else
+        @write_mutex.synchronize { yield }
+      end
     rescue Exception => err
       Logger.instance.error("Error writing to interface : #{@name}")
       disconnect()

data/lib/cosmos/interfaces/linc_interface.rb

@@ -256,7 +256,7 @@ module Cosmos
       # Handle handshake warnings and errors
       if status == "OK" and code != 0
         unless @ignored_error_codes[handshake_cmd.handshake.handshake.target_name].include? code
-          Logger.warn "Warning sending command (#{code}): #{source}"
+          Logger.warn "#{@name}: Warning sending command (#{code}): #{source}"
         end
       elsif status == "ERROR"
         unless @ignored_error_codes[handshake_cmd.handshake.handshake.target_name].include? code
@@ -295,7 +295,7 @@ module Cosmos
       command.received_count += 1
 
       # Put a log of the command onto the server for the user to see
-      Logger.info("External Command: " + System.commands.format(linc_handshake.identified_command, System.targets[linc_handshake.identified_command.target_name].ignored_parameters))
+      Logger.info("#{@name}: External Command: " + System.commands.format(linc_handshake.identified_command, System.targets[linc_handshake.identified_command.target_name].ignored_parameters))
 
       # Log the command to the command log(s)
       @packet_log_writer_pairs.each do |packet_log_writer_pair|
@@ -361,7 +361,7 @@ module Cosmos
       if @handshake
         timed_out = false
       else
-        Logger.warn "No handshake - must be timeout."
+        Logger.warn "#{@name}: No handshake - must be timeout."
         timed_out = true
       end