[client] Add option to run mrd2gif after session is complete

Author: kspaceKelvin

client.py

@@ -27,7 +27,8 @@
     'ignore_json_config': False,
     'send_waveforms':     False,
     'verbose':            False,
-    'logfile':            ''
+    'logfile':            '',
+    'mrd2gif':            False
 }
 
 def connection_receive_loop(sock, outfile, outgroup, verbose, logfile, recvAcqs, recvImages, recvWaveforms):
@@ -128,7 +129,7 @@ def main(args):
 
         if ('data' in group):
             hasRaw = True
-        
+
         if len([key for key in group.keys() if (key.startswith('image_') or key.startswith('images_'))]) > 0:
             hasImage = True
 
@@ -324,6 +325,20 @@ def main(args):
     logging.info("Results written to %s", args.outfile)
     logging.info("Session complete")
 
+    if args.mrd2gif:
+        try:
+            import mrd2gif
+            from types import SimpleNamespace
+
+            mrd2gifargs = SimpleNamespace(**mrd2gif.defaults)
+            mrd2gifargs.filename = args.outfile
+            mrd2gifargs.in_group = args.out_group
+
+            logging.info('Calling mrd2gif...')
+            mrd2gif.main(mrd2gifargs)
+        except:
+            logging.error('Failed to import or call mrd2gif')
+
     return
 
 if __name__ == '__main__':
@@ -342,6 +357,7 @@ def main(args):
     parser.add_argument('-v', '--verbose',            action='store_true', help='Verbose mode')
     parser.add_argument('-l', '--logfile',            type=str,            help='Path to log file')
     parser.add_argument(      '--ignore-json-config', action='store_true', help='Ignore config specified in JSON')
+    parser.add_argument(      '--mrd2gif',            action='store_true', help='Run mrd2gif on output file')
 
     parser.set_defaults(**defaults)