#     0x03?? that means 0000_0011 and >>2 means shift twice on the right for 

import argparse

#Added on January 16, 2012 by Benoit NCEAS

def list_raster_per_month(listmaps):

    """Determine starting and ending numeric day-of-year (doy)

    asscociated with the specified month and year.

    Arguments:

    maplist -- list of raster grass-digit integer year

    month -- integer month (1-12)

    year_range -- list of years

    Returns tuple of start and end doy for that month/year.

    """

    list_maps_month=list()

    nb_month=12

    for i in range(1,nb_month+1):

        #i=i+1

        #filename = listfiles_wd2[i] #list the files of interest

        #monthlist[:]=[] #empty the list

        monthlist=list()

        #monthlist2[:]=[] #empty the list

        j=0  #counter

        for mapname in listmaps:

            #tmp_str=LST[0]

            date_map=mapname.split('_')[1][1:8]

            #date = filename[filename.rfind('_MOD')-8:filename.rfind('_MOD')]

            d=datetime.datetime.strptime(date_map,'%Y%j') #This produces a date object from a string extracted from the file name.

            month=d.strftime('%m') #Extract the month of the current map

            if int(month)==i:

                #monthlist.append(listmaps[j])

                monthlist.append(mapname)

                #monthlist2.append(listmaps[j])

            j=j+1

        list_maps_month.append(monthlist)   #This is a list of a list containing a list for each month

        #list_files2.append(monthlist2)

    return(list_maps_month)    

    ########## START OF SCRIPT ##############

    #### Modified by Benoit on May 13, 2013 

    ### INPUT Parameters

    #Inputs from R?? there are 9 parameters

   #tiles = ['h11v08','h11v07','h12v07','h12v08','h10v07','h10v08'] #These tiles correspond to Venezuela.

    tiles= ['h12v08']    

    #start_year = 2001

    #end_year = 2010

    #start_month=1

    #end_month=12

    #hdfdir =  '/home/layers/commons/modis/MOD11A1_tiles' #destination file where hdf files are stored locally after download.

    #night=1    # if 1 then produce night climatology

    #out_suffix="_03192013"

    #download=1  # if 1 then download files

    parser = argparse.ArgumentParser()

    parser.add_argument("tiles", type=str, help="list of Modis tiles")

    parser.add_argument("start_year", type=int, help="start year")

    parser.add_argument("end_year", type=int, help="end year")

    parser.add_argument("start_month", type=int, help="start month")

    parser.add_argument("end_month", type=int, help="end month")

    parser.add_argument("hdfdir", type=str, help="destination/source directory for hdf file")

    parser.add_argument("night", type=int, help="night")

    parser.add_argument("download", type=int, help="out_suffix")

    parser.add_argument("out_suffix", type=str, help="out_suffix")

    myargs = parser.parse_args()

    tiles = myargs.tiles #These tiles correspond to Venezuela.

    start_year = myargs.start_year

    end_year = myargs.start_year 

    end_month = myargs.end_month

    start_month= myargs.start_month

    hdfdir =  myargs.hdfdir

    night= myargs.night    # if 1 then produce night climatology

    out_suffix= myargs.out_suffix #"_03192013"

    download=myargs.download# if 1 then download files

    tiles =tiles.split(",") #Create a list from string

    #need to add on the fly creation of folder for each tile!!

        # clean up GRASS DATABASE 

        gs.run_command('g.remove', rast=','.join(LST))

        gs.os.environ['GRASS_OVERWRITE'] = '0'

        #clean up

        gs.run_command('g.gisenv', set='LOCATION_NAME=%s' % orig_location)

        gs.run_command('g.gisenv', set='MAPSET=%s' % orig_mapset)

        shutil.rmtree(os.path.join(gs.gisenv()['GISDBASE'], tmp_location))