Fixed camera stream null image error

Author: Kajetan Jonak

GUI_v2.Model/CameraStreamClient.cs

@@ -144,7 +144,7 @@ private void ReceiveFrame()
                     return;
                 var l = stream.Read(buffer, 0, 4);
                 int lenght = BitConverter.ToInt32(buffer, 0);
-                _frame = new byte[lenght + chunkSize - lenght % chunkSize];
+                _frame = new byte[lenght + chunkSize];// - lenght % chunkSize];
                 int receivedBytes = 0;
                 int chunk;
                 while (receivedBytes < lenght)

GUI_v2.Model/JetsonClient/Receiver.cs

@@ -58,7 +58,7 @@ void Receive()
                         if (rx_state == HEADER)
                         {
                             ReadAllFromStream(stream, header_buffer, 4);
-                            msg_length = BitConverter.ToInt32(header_buffer, 0);
+                            msg_length = BitConverter.ToInt32(header_buffer, 0);                        
                             rx_state = DATA;
                         }else if (rx_state == DATA)
                         {

GUI_v2.ViewModel/Common/CameraStreamViewModel.cs

@@ -24,8 +24,9 @@ public void SwapImage(Byte[] frame)
             //   CurrentImage = ImageHelpers.BytesToBitmapImage(frame);
             Application.Current.Dispatcher.Invoke(new Action(() =>
             {
-              
-                CurrentImage = ImageHelpers.BytesToBitmapImage(frame);
+              var temp = ImageHelpers.BytesToBitmapImage(frame);
+                if (temp != null)
+                    CurrentImage = temp; 
 
 
             }), DispatcherPriority.Render);

GUI_v2.ViewModel/Tools/ImageHelpers.cs

@@ -30,10 +30,18 @@ public static BitmapImage BytesToBitmapImage(Byte[] data)
         {
             Stream ms = new MemoryStream(data);
             BitmapImage img = new BitmapImage();
-            img.BeginInit();
-            img.StreamSource = ms;
-            img.EndInit();
-            return img;
+            try
+            {
+                img.BeginInit();
+                img.StreamSource = ms;
+                img.EndInit();
+                return img;
+            }
+            catch
+            {
+                return null;
+            }
+            
         }