#!/usr/bin/env python2
import sys, os
from SummaryJob import SummaryJob, dateDir
from datetime import date, timedelta

def relativeWeekDateDir( datedir, offset = -1 ):

    """
    datedir is a string of the format YYYY/MMDD-MMDD representing
    a Sunday and following Saturday.  This function will return a
    string corresponding to the previous week (offset = -1),
    or the nth prior week offset = -n
    """
    y,m,d = datedir[0:4],datedir[5:7],datedir[7:9]
    d0 = date(int(y),int(m),int(d))
    da = d0+timedelta(      7*offset    )
    db = d0+timedelta( -1 + 7*(offset+1))
    return '%s-%s' % (da.strftime('%Y/%m%d'), db.strftime('%m%d') )

class RefDirSummaryJob(SummaryJob):
    nweeks=52 # Search back this many weeks for a reference dir.
    _refdir_name = '<refdir>'
    def mkCommandPre(self,command):
        foundrefdir = False
        dd = self.getDateDirName()
        refdir = None
        for week in range(1,self.nweeks):
            ddold = relativeWeekDateDir( dd, -week )
            refdirbase = os.path.join(self.c.cp.get(self.jobname,'outroot'),
                                  self.subdet, ddold )
            if not os.path.exists(refdirbase):
                continue

            # Try to find in the new foo-vnn subdirectory
            for v in range(self.maxversion,0,-1):
                refdir = os.path.join(refdirbase, '%s-v%02i' %(self.jobname,v))
                if os.path.exists(refdir):
                    foundrefdir = True
                    break
            if foundrefdir:
                break

            refdir = None

        if refdir == None:
            refdir = os.path.join( os.getenv("CALIBOFFLINEROOT"), 'share/examples')
            print 'Warning: using %s as reference directory'
        command = command.replace(self._refdir_name, refdir )
        return command

c = 'python -m TPCdEdxStability -o <outdir> -r <refdir> <indir>/dq-tpc-bc_*_*.root'
j = RefDirSummaryJob('dq-tpc-bc', 'tpc', c, sys.argv[1:] )
script = j.mkJob()
if not script is None:
    j.submitJob(script)