B a]@s4ddlZddlZddlZddlmZddlmZddlZddlZddl m Z ddZ Gddde Z ed kr0ejd ejd ddlZdd lmZejd dZejddddejdddddejdddddejdddddeZeejZejZe eZx(ejD]Zed efe eeqWdS)!N) OptionParser)fromorg)ThreadcsRt}|tdd|dtddfdd|DddS)N|z | z |csg|]}t|qS)str).0k)dr*/home/dice/stardice/scheduler/scheduler.py szdict_to_org..)listkeyssortprintjoin)r lr)r r dict_to_orgs rc@sZeZdZdddZddZddZddd Zd d Zdd dZddZ ddZ ddZ dS) SchedulerNcsd_g__|dkr,tjj}nfdd|D}fdd|D_x jD]\}}|j gq\W t j }d|j |j|jf}tj jjj|_xjD]\}}t||qWdS)z Creates a scheduler instance that will handle a schedule and in particular handle taking the images Data will be saved in [config.Output.dir]/year-month-day/ FNcsg|]}|tj|fqSr)getattr Subsystems)rname)configrr r &sz&Scheduler.__init__..cs&g|]\}}|tt||fqSr)r allsubsystems)rrcomp)rselfrr r 'sz%04d-%02d-%02d)aborting current_jobsrr r__dict__items subsystems async_runZ configurerdatetimenowyearmonthdayospathOutput output_dirsetattr)rrZ device_listrsr#Zoutput_dir_nowsysr)rrr __init__s zScheduler.__init__cCs$t||d}|j||dS)N)targetargs)rrappendstart)rfuncr0trrr r!7s  zScheduler.async_runcCs"x|jr|j}|qWdS)N)rpopr)rr4rrr r<s zScheduler.joincCs|dkr|j}tj|d}tt|xTdD]L}x>|jD]4\}}|jrPtd||r:| t ||||gq:W| q.Wtd||j |dS)z Shoots an exposure: runs every subprocess before_exposure, exposure and after_exposure and dumps the exposure in output_dir. If output_dir isn't set, uses the scheduler output_dir N)r*)Zbefore_exposureexposureZafter_exposurezAborted observation programzdumping files -> %s) r* fitsbuilderZ FitsBuilderlogginginforr r StopIterationr!rrdump)rstater*Zfbr3r subsystemrrr shootAs   zScheduler.shootcCs8d}x.|jD]$\}}|d|d7}||7}q W|S)zPrint system status  z ---------- )r r<)rstrr=rrr r<Zs zScheduler.staterc csddlm}d|_td|t|\}}||}x`t||d|dD]F\}}d|_x6|jstd|t |f| ||} | |fVq^WqNWdS)Nr) prog_fieldsFzExecuting program %szProcessing exposure %d/%d) Zstardice.obsprogrBrr8r9rastype enumerateZacceptedlenr>) rZ program_filer* startlinerBprogramr expr6filenamerrr executeds     zScheduler.executecCs d|_dS)NT)r)rrrr abortqszScheduler.abortcCsF|jj|jj}}ttd||f}|dkr<|j|||jS)z@ Mount synchronisation with the result of last analysis z3Do you want set current pointing to %f,%f (y/n)? : y)ZanalysefrafdecevalinputZmountZ set_radecZ get_radec)rrNrOarrr mount_initts zScheduler.mount_initcCsJi}x@|jD]6\}}d||<x$|jD]}|j||d||f<q$Wq W|S)NTz%s.%s)r Zexpected_params)rresnr,r rrr get_default~s  zScheduler.get_default)N)N)Nr) __name__ __module__ __qualname__r.r!rr>r<rKrLrSrVrrrr rs    r__main__z%(asctime)s %(message)s)formatlevel) configfilesz1Supervise the execution of an observation program) descriptionprograms*zprogram to be executed)nargshelpz-cz--configz scheduler.cfgzName of the configuration file)defaultrbz-oz --output-dirz./zLPath to the directory where images will be stored. (Created if not existing)z-rz --releaseFzif True, release the servers at the end of execution; set to False if you want to reuse the scheduler after the first execution.zExecuting program )!r'r"r8optparserZstardice.orgfilesrr7r threadingrrobjectrrW basicConfigDEBUGargparsestardicer]ArgumentParserparser add_argument parse_argsr0Configrr* schedulerr_rHrrKrrrr sD   p