def render_movie(subject_fname): args = read_args(su.get_python_argv()) if args.debug: su.debug() if args.rel_output_path: mmvt_dir = op.join(su.get_links_dir(), 'mmvt') if args.output_path == '': args.output_path = args.play_type args.output_path = op.join(mmvt_dir, args.subject, 'movies', args.output_path) su.make_dir(args.output_path) mmvt = su.init_mmvt_addon() mmvt.show_hide_hemi(args.hide_lh, 'lh') mmvt.show_hide_hemi(args.hide_rh, 'rh') mmvt.show_hide_sub_corticals(args.hide_subs) mmvt.set_render_quality(args.quality) mmvt.set_render_output_path(args.output_path) mmvt.set_render_smooth_figure(args.smooth_figure) mmvt.set_light_layers_depth(args.light_layers) mmvt.set_brain_transparency(args.brain_trans) mmvt.filter_nodes(args.filter_nodes) mark_electrodes(mmvt, args) camera_fname = su.load_camera(mmvt, mmvt_dir, args) if not op.isfile(op.join(args.output_path, 'data.pkl')): try: mmvt.capture_graph(args.play_type, args.output_path, args.selection_type) except: print("Graph couldn't be captured!") su.save_blend_file(subject_fname) mmvt.render_movie(args.play_type, args.play_from, args.play_to, camera_fname, args.play_dt, args.set_to_camera_mode) su.exit_blender()
def render_image(subject_fname): args = read_args(su.get_python_argv()) if args.output_path == '': mmvt_dir = op.join(su.get_links_dir(), 'mmvt') args.output_path = op.join(mmvt_dir, args.subject, 'figures') su.make_dir(args.output_path) mmvt = su.init_mmvt_addon() mmvt.set_render_quality(args.quality) mmvt.set_render_output_path(args.output_path) mmvt.set_render_smooth_figure(args.smooth_figure) if not args.hide_lh is None: mmvt.show_hide_hemi(args.hide_lh, 'lh') if not args.hide_rh is None: mmvt.show_hide_hemi(args.hide_rh, 'rh') if not args.hide_subs is None: mmvt.show_hide_sub_corticals(args.hide_subs) if not args.show_elecs is None: mmvt.show_electrodes(args.show_elecs) if not args.show_only_lead is None: print('Showing only the current lead: {}'.format(args.show_only_lead)) mmvt.set_show_only_lead(args.show_only_lead) if args.curr_elec not in ['', 'None']: print('Setting current electrode to {}'.format(args.curr_elec)) mmvt.set_current_electrode(args.curr_elec) if op.isfile(op.join(args.output_path, 'camera.pkl')): mmvt.load_camera() else: cont = input('No camera file was detected in the output folder, continue?') if not su.is_true(cont): return # su.save_blend_file(subject_fname) mmvt.render_image(args.image_name, args.output_path, args.quality, args.smooth_figure, render_background=False) su.exit_blender()
def render_electrodes_probs(subject_fname): args = read_args(su.get_python_argv()) figures_dir = su.get_figures_dir(args) if args.rel_output_path: args.output_path = op.join(figures_dir, args.output_path) su.make_dir(args.output_path) # Set the labeling file labeling_fname = '{}_{}_electrodes_cigar_r_*_l_*{}*.pkl'.format( args.subject, args.real_atlas, '_bipolar' if args.bipolar else '') labels_fol = op.join(su.get_mmvt_dir(), args.subject, 'electrodes') labeling_files = glob.glob(op.join(labels_fol, labeling_fname)) if len(labeling_files) == 0: print('No labeling files in {}!'.format(labels_fol)) return if len(labeling_files) > 1: print( 'More than one labeling files in {}, please choose one using the --labeling flag' .format(labels_fol)) return else: labeling_file = labeling_files[0] mmvt = su.init_mmvt_addon() mmvt.show_hide_hemi(args.hide_lh, 'lh') mmvt.show_hide_hemi(args.hide_rh, 'rh') mmvt.show_hide_sub_corticals(args.hide_subs) mmvt.set_render_quality(args.quality) mmvt.set_render_output_path(args.output_path) mmvt.set_render_smooth_figure(args.smooth_figure) if op.isfile(op.join(args.output_path, 'camera.pkl')): mmvt.load_camera() else: # Try to find the camera in the figures folder if op.isfile(op.join(figures_dir, 'camera.pkl')): mmvt.set_render_output_path(figures_dir) mmvt.load_camera() mmvt.set_render_output_path(args.output_path) else: cont = input( 'No camera file was detected in the output folder, continue?') if not su.is_true(cont): return mmvt.set_electrodes_labeling_file(labeling_file) mmvt.show_electrodes() mmvt.color_the_relevant_lables(True) leads = mmvt.get_leads() for lead in leads: electrodes = mmvt.get_lead_electrodes(lead) for electrode in electrodes: print(electrode) mmvt.clear_cortex() mmvt.set_current_electrode(electrode, lead) mmvt.render_image(electrode) su.save_blend_file(subject_fname) su.exit_blender()
def pre_blender_call(args): if args.output_path == '': mmvt_dir = su.get_link_dir(su.get_links_dir(), 'mmvt') args.output_path = op.join(mmvt_dir, args.subject, 'figures') su.make_dir(args.output_path) args.log_fname = op.join(args.output_path, 'mmvt_calls.log') args.images_log_fname = op.join(args.output_path, 'images_names.txt') if op.isfile(args.log_fname): os.remove(args.log_fname) return args
def render_electrodes_probs(subject_fname): args = read_args(su.get_python_argv()) figures_dir = su.get_figures_dir(args) if args.rel_output_path: args.output_path = op.join(figures_dir, args.output_path) su.make_dir(args.output_path) # Set the labeling file labeling_fname = '{}_{}_electrodes_cigar_r_*_l_*{}*.pkl'.format(args.subject, args.real_atlas, '_bipolar' if args.bipolar else '') labels_fol = op.join(su.get_mmvt_dir(), args.subject, 'electrodes') labeling_files = glob.glob(op.join(labels_fol, labeling_fname)) if len(labeling_files) == 0: print('No labeling files in {}!'.format(labels_fol)) return if len(labeling_files) > 1: print('More than one labeling files in {}, please choose one using the --labeling flag'.format(labels_fol)) return else: labeling_file = labeling_files[0] mmvt = su.init_mmvt_addon() mmvt.show_hide_hemi(args.hide_lh, 'lh') mmvt.show_hide_hemi(args.hide_rh, 'rh') mmvt.show_hide_sub_corticals(args.hide_subs) mmvt.set_render_quality(args.quality) mmvt.set_render_output_path(args.output_path) mmvt.set_render_smooth_figure(args.smooth_figure) if op.isfile(op.join(args.output_path, 'camera.pkl')): mmvt.load_camera() else: # Try to find the camera in the figures folder if op.isfile(op.join(figures_dir, 'camera.pkl')): mmvt.set_render_output_path(figures_dir) mmvt.load_camera() mmvt.set_render_output_path(args.output_path) else: cont = input('No camera file was detected in the output folder, continue?') if not su.is_true(cont): return mmvt.set_electrodes_labeling_file(labeling_file) mmvt.show_electrodes() mmvt.color_the_relevant_lables(True) leads = mmvt.get_leads() for lead in leads: electrodes = mmvt.get_lead_electrodes(lead) for electrode in electrodes: print(electrode) mmvt.clear_cortex() mmvt.set_current_electrode(electrode, lead) mmvt.render_image(electrode) su.save_blend_file(subject_fname) su.exit_blender()
def render_image(subject_fname): args = read_args(su.get_python_argv()) if args.output_path == '': mmvt_dir = op.join(su.get_links_dir(), 'mmvt') args.output_path = op.join(mmvt_dir, args.subject, 'figures') su.make_dir(args.output_path) mmvt = su.init_mmvt_addon() mmvt.set_render_quality(args.quality) mmvt.set_render_output_path(args.output_path) mmvt.set_render_smooth_figure(args.smooth_figure) if not args.hide_lh is None: mmvt.show_hide_hemi(args.hide_lh, 'lh') if not args.hide_rh is None: mmvt.show_hide_hemi(args.hide_rh, 'rh') if not args.hide_subs is None: mmvt.show_hide_sub_corticals(args.hide_subs) if not args.show_elecs is None: mmvt.show_electrodes(args.show_elecs) if not args.show_only_lead is None: print('Showing only the current lead: {}'.format(args.show_only_lead)) mmvt.set_show_only_lead(args.show_only_lead) if args.curr_elec not in ['', 'None']: print('Setting current electrode to {}'.format(args.curr_elec)) mmvt.set_current_electrode(args.curr_elec) if not args.show_connections is None: mmvt.show_hide_connections(args.show_connections) if op.isfile(op.join(args.output_path, 'camera.pkl')): mmvt.load_camera() else: cont = input( 'No camera file was detected in the output folder, continue?') if not su.is_true(cont): return # su.save_blend_file(subject_fname) mmvt.render_image(args.image_name, args.output_path, args.quality, args.smooth_figure, render_background=False) su.exit_blender()
def render_movie(subject_fname): args = read_args(su.get_python_argv()) if args.rel_output_path: mmvt_dir = op.join(su.get_links_dir(), 'mmvt') args.output_path = op.join(mmvt_dir, args.subject, 'movies', args.output_path) su.make_dir(args.output_path) mmvt = su.init_mmvt_addon() mmvt.show_hide_hemi(args.hide_lh, 'lh') mmvt.show_hide_hemi(args.hide_rh, 'rh') mmvt.show_hide_sub_corticals(args.hide_subs) mmvt.set_render_quality(args.quality) mmvt.set_render_output_path(args.output_path) mmvt.set_render_smooth_figure(args.smooth_figure) if op.isfile(op.join(args.output_path, 'camera.pkl')): mmvt.load_camera() else: cont = input('No camera file was detected in the output folder, continue?') if not su.is_true(cont): return if not op.isfile(op.join(args.output_path, 'data.pkl')): mmvt.capture_graph(args.play_type, args.output_path, args.selection_type) su.save_blend_file(subject_fname) mmvt.render_movie(args.play_type, args.play_from, args.play_to, args.play_dt) su.exit_blender()
def render_image_blender(subject_fname): args = read_args(su.get_python_argv()) if args.debug: su.debug() mmvt_dir = su.get_link_dir(su.get_links_dir(), 'mmvt') subject = su.get_subject_name(subject_fname) if args.output_path == '': args.output_path = op.join(mmvt_dir, args.subject, 'figures') su.make_dir(args.output_path) mmvt = su.init_mmvt_addon() mmvt.set_render_quality(args.quality) mmvt.set_render_output_path(args.output_path) mmvt.set_render_smooth_figure(args.smooth_figure) if not args.hide_lh is None: mmvt.show_hide_hemi(args.hide_lh, 'lh') if not args.hide_rh is None: mmvt.show_hide_hemi(args.hide_rh, 'rh') if not args.hide_subs is None: mmvt.show_hide_sub_corticals(args.hide_subs) if not args.show_elecs is None: mmvt.show_electrodes(args.show_elecs) if not args.show_only_lead is None: print('Showing only the current lead: {}'.format(args.show_only_lead)) mmvt.set_show_only_lead(args.show_only_lead) if args.curr_elec not in ['', 'None']: print('Setting current electrode to {}'.format(args.curr_elec)) mmvt.set_current_electrode(args.curr_elec) if not args.show_connections is None: mmvt.show_hide_connections(args.show_connections) mmvt.filter_nodes(args.filter_nodes) if args.inflated: mmvt.show_inflated() mmvt.set_inflated_ratio = args.inflated_ratio mmvt.set_background_color(args.background_color) mmvt.set_lighting(args.lighting) for image_name, camera in zip(args.image_name, args.camera): if camera == '': camera = op.join(mmvt_dir, subject, 'camera', 'camera.pkl') print('Camera fname: {}'.format(camera)) if op.isfile(camera): mmvt.load_camera(camera) else: camera = op.join(mmvt_dir, subject, 'camera', 'camera.pkl') print('Camera was not found, loading default camera') mmvt.load_camera(camera) # if args.interactive: # cont = input('No camera file was detected in the output folder, continue?') # if not su.is_true(cont): # continue # else: # su.stdout_print('No camera file was detected in the output folder!!!') # continue # su.save_blend_file(subject_fname) print('Rendering {}->{}'.format(image_name, args.output_path)) mmvt.render_image(image_name, args.output_path, args.quality, args.smooth_figure, render_background=False, overwrite=args.overwrite) su.stdout_print('*** finish rendering! ***') su.exit_blender()