def wrap_blender_call(only_verbose=False):
sys.argv = [sys.argv[0]]
args = read_args(dict(subject='empty_subject'))
su.call_script(__file__,
args,
blend_fname='empty_subject.blend',
only_verbose=only_verbose)
def run(subject='',
atlas='dkt',
run_in_background=False,
debug=None,
raise_exp=True,
run_blender=True):
args = None
if sys.argv[0] == __file__:
args = read_args(raise_exp=raise_exp)
if args is None:
sys.argv = [__file__, '-s', subject, '-a', atlas]
args = read_args()
if subject != '':
args.subject = subject
if atlas != '':
args.atlas = atlas
if debug is not None:
args.debug = debug
if run_blender:
su.call_script(__file__, args, run_in_background=run_in_background)
mmvt_agent = su.get_mmvt_object(args.subject)
if mmvt_agent is not None:
print('We got the mmvt object ({})!'.format(
list(mmvt_agent._proxy_agent.connections.keys())[0]))
return MMVT(mmvt_agent)
def wrap_blender_call(subject='',
modality='',
load_data=None,
overwrite_sensors=None,
run_in_background=True,
debug=None):
args = read_args()
if args is None:
sys.argv = [__file__, '-s', subject]
args = read_args()
if subject != '':
args.subject = subject
if modality != '':
args.modality = modality
if load_data is not None:
args.load_data = load_data
if overwrite_sensors is not None:
args.overwrite_sensors = overwrite_sensors
if debug is not None:
args.debug = debug
if args.modality not in ['meg', 'eeg']:
print('args.modality should be eeg or meg!')
su.call_script(__file__,
args,
run_in_background=run_in_background,
err_pipe=sys.stdin)
def wrap_blender_call():
args = read_args()
# todo: check args
args = pre_blender_call(args)
if args.call_mmvt_calls:
su.call_script(__file__, args, run_in_background=False)
post_blender_call(args)
def wrap_blender_call():
# Empty argv, otherwise cannot be called from setup (which already has args)
sys.argv = [sys.argv[1]]
args = read_args()
args.subject = 'empty_subject'
su.call_script(__file__, args, blend_fname='empty_subject.blend'
) #, call_args='-python_cmd {}'.format(sys.executable))
def wrap_blender_call():
args = read_args()
if isinstance(args.pos_file, str):
args.pos_file = [args.pos_file]
for subject, pos_file in zip(args.subjects, args.pos_file):
args.subjects = []
args.subject = subject
args.pos_file = pos_file
su.call_script(__file__, args)
def wrap_blender_call():
args = read_args()
if isinstance(args.pos_file, str):
args.pos_file = [args.pos_file]
for subject, pos_file in zip(args.subjects, args.pos_file):
args.subjects = []
args.subject = subject
args.pos_file = pos_file
su.call_script(__file__, args)
def run(subject='', atlas='', run_in_background=False):
args = read_args(dict(
subject=subject,
atlas=atlas,
))
if subject != '':
args.subject = subject
if atlas != '':
args.atlas = atlas
su.call_script(__file__, args, run_in_background=run_in_background)
def wrap_blender_call(args=None):
if args is None:
args = read_args()
args = pre_blender_call(args)
if args.save_views:
if args.img_name_prefix == '' and args.use_fol_for_img_prefix:
args.img_name_prefix = su.namebase(su.get_parent_fol(args.nii))
args = save_views.pre_blender_call(args)
su.call_script(__file__, args, run_in_background=False, err_pipe=sys.stdin)
if args.save_views:
save_views.post_blender_call(args)
def wrap_blender_call():
args = read_args()
ret = create_new_subject_file(args)
if ret:
args.log_fname = op.join(su.get_logs_fol(args.subject), 'create_new_subject_script.log')
if op.isfile(args.log_fname):
os.remove(args.log_fname)
if args.import_in_blender:
su.call_script(__file__, args, run_in_background=False, err_pipe=sys.stdin)
su.waits_for_file(args.log_fname)
return args
else:
return None
def run(subject='', atlas='', run_in_background=False, debug=None, raise_exp=True):
args = None
if sys.argv[0] == __file__:
args = read_args(raise_exp=raise_exp)
if args is None:
sys.argv = [__file__ , '-s', subject, '-a', atlas]
args = read_args()
if subject != '':
args.subject = subject
if atlas != '':
args.atlas = atlas
if debug is not None:
args.debug = debug
su.call_script(__file__, args, run_in_background=run_in_background)
mmvt = su.get_mmvt_object(args.subject)
if mmvt is not None:
print('We got the mmvt object!')
return mmvt
def render_image(subject,
atlas,
image_name,
output_path,
quality=60,
inflated=False,
inflated_ratio=1,
background_color='black',
lighting=1.0,
smooth_figure=False,
hide_lh=False,
hide_rh=False,
hide_subs=False,
show_elecs=False,
bipolar=False,
show_only_lead=False,
curr_elec='',
show_connections=False,
filter_nodes=True,
interactive=True,
blender_fol='',
subjects=''):
image_name = [image_name] if isinstance(image_name, str) else image_name
camera = [
op.join(
su.get_mmvt_dir(), subject, 'camera',
'camera_{}{}.pkl'.format(camera_name, '_inf' if inflated else ''))
for camera_name in image_name
]
image_name = [
'{}_{}_{}'.format(
name, 'inflated_{}'.format(inflated_ratio) if inflated else 'pial',
background_color) for name in image_name
]
camera = ','.join(camera)
image_name = ','.join(image_name)
args = su.Bag(locals())
su.call_script(__file__, args)
def wrap_blender_call():
args = read_args()
create_new_subject_file(args)
su.call_script(__file__, args)
def wrap_blender_call():
args = read_args()
should_render_figures = check_if_all_figures_were_rendered(args)
if should_render_figures:
su.call_script(__file__, args)
post_script(args)
def wrap_blender_call(only_verbose=False):
# Empty argv, otherwise cannot be called from setup (which already has args)
sys.argv = [sys.argv[0]]
args = read_args()
args.subject = 'empty_subject'
su.call_script(__file__, args, blend_fname='empty_subject.blend', only_verbose=only_verbose)#, call_args='-python_cmd {}'.format(sys.executable))
def wrap_blender_call():
args = read_args()
su.call_script(__file__, args)
def wrap_blender_call(args=None):
if args is None:
args = read_args()
su.call_script(__file__, args, run_in_background=True, err_pipe=sys.stdin)
def wrap_blender_call():
args = read_args()
create_new_subject_file(args)
su.call_script(__file__, args)
def wrap_blender_call():
args = read_args()
su.call_script(__file__, args)
def wrap_blender_call():
args = read_args()
create_new_subject_file(args)
if args.import_in_blender:
su.call_script(__file__, args)
def wrap_blender_call():
args = read_args()
should_render_figures = check_if_all_figures_were_rendered(args)
if should_render_figures:
su.call_script(__file__, args)
post_script(args)