#!/usr/bin/python

#
#
# NARGal - a simple gallery program that replaces igal
#
# (c) Copyright 2003 by Nick Rusnov <projects@nspam.industrialmeats.com>
#    This program is free software; you can redistribute it and/or modify
#    it under the terms of the GNU General Public License as published by
#    the Free Software Foundation; either version 2 of the License, or
#    (at your option) any later version.
#
#    This program is distributed in the hope that it will be useful,
#    but WITHOUT ANY WARRANTY; without even the implied warranty of
#    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
#    GNU General Public License for more details.
#
#    You should have received a copy of the GNU General Public License
#    along with this program; if not, write to the Free Software
#    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
# 
#


import os
import EXIF
import md5
import string
import Image
import ImageFile
ImageFile.MAXBLOCK = 300*1024
import sys
from htmltmpl import TemplateManager, TemplateProcessor
import getopt

PROGNAME='nargal'
PROGVER='0.0.1'

def usage():
    print """
nargal - gallery program
usage:

  -s, --slide-template <template>   specify the slide template
  -i, --index-template <template>   specify the index tempalte
  -t, --thumb-dim <value>           specify thumb major dimension (def. 130)
  -l, --slide-dim <value>           specify slide major dimension (def. 600)
  --slide-qual <value>              specify slide quality (def. 88)
  --thumb-qual <value>              specify thumb quality (def. 57)
  -q, --quiet                       send no output to stderr (default is to send status)
  --title <title>                   specify the index and parent title
  -c, --columns <value>             specify the number of columns (def. 5)
  -d, --linkdirs                    generate links for subdirectories
  -x, --exclude <dirs>              don't generate links for comma-separated list
"""

imagetypes = ('jpg', 'tiff', 'jpeg', 'tif', 'png')
exiftypes = ('jpg', 'jpeg', 'tiff', 'tif')

# defaults
THUMBDIM=130
SLIDEDIM=600
SLIDEQUAL=88
THUMBQUAL=57
DEBUG=True
SLIDETEMPLATE='' 
INDEXTEMPLATE='' 
TITLE=''
TARGET='.'
COLS=5
LINKDIRS=False
LINKEXCLUDE=()

try:
    opts, args = getopt.getopt(sys.argv[1:], 's:i:t:l:c:x:qd', ['slide-template=', 'index-template=', 'thumb-dim=', 'slide-dim=', 'slide-qual=', 'thumb-qual=', 'columns=', 'title=', 'exclude=', 'quiet', 'linkdirs'])
except getopt.GetoptError:
    usage()
    sys.exit(2)

for i in opts:
    if i[0] in ('q', 'quiet'):
        DEBUG=False
    elif i[0] in ('-s', '--slide-template'):
        SLIDETEMPLATE=i[1].strip()
    elif i[0] in ('-i', '--index-template'):
        INDEXTEMPLATE=i[1].strip()
    elif i[0] in ('-t', '--thumb-dim'):
        try:
            THUMBDIM=int(i[1])
        except:
            print "%s needs a integer argument." % i[0]
            usage()
            sys.exit(3)
    elif i[0] in ('-l', '--slide-dim'):
        try:
            SLIDEDIM=int(i[1])
        except:
            print "%s needs a integer argument." % i[0]
            usage()
            sys.exit(4)
    elif i[0] in ('-c', '--columns'):
        try:
            COLS=int(i[1])
        except:
            print "%s needs a integer argument." % i[0]
            usage()
            sys.exit(5)
    elif i[0] == '--slide-qual':
        try:
            SLIDEQUAL=int(i[1])
        except:
            print "%s needs a integer argument." % i[0]
            usage()
            sys.exit(6)
    elif i[0] == '--thumb-qual':
        try:
            THUMBQUAL=int(i[1])
        except:
            print "%s needs a integer argument." % i[0]
            usage()
            sys.exit(7)
    elif i[0] == '--title':
        TITLE=i[1]
    elif i[0] in ('-d', '--linkdirs'):
        LINKDIRS=True
    elif i[0] in ('-x', '--exclude'):
        LINKEXCLUDE=string.split(i[1], ',')

if SLIDETEMPLATE=='':
    print "need to spcify a slide template"
    usage()
    sys.exit(8)

if INDEXTEMPLATE=='':
    print "need to specify an index template"
    usage()
    sys.exit(9)

#print 'debug', DEBUG, 'slidetemplate', SLIDETEMPLATE, 'indextemplate', INDEXTEMPLATE, 'thumbdim', THUMBDIM, 'slidedim', SLIDEDIM, 'slidequal', SLIDEQUAL, 'thumbqual', THUMBQUAL, 'title', TITLE
    
slidetemplate = TemplateManager().prepare(SLIDETEMPLATE)
indextemplate = TemplateManager().prepare(INDEXTEMPLATE)

def extmatch(filename, extentions):
    tok = string.split(filename, '.')
    ext = tok[len(tok)-1].lower()
    return ext in extentions

def sysmatch(filename):
    return (filename[:6] == '.SLIDE') or (filename[:6] == '.THUMB') 


def loadmeta(filename):
    try:
        f = open(filename, 'r')
    except:
        return None


    odict = dict()

    for i in f.readlines():
        line = string.strip(i)
        if line[0] == '#' or len(line) <= 1:
            continue

        tok = string.split(line, ':',1)
        if len(tok) > 1:
            odict[tok[0]] = tok[1]

    f.close()
    
    if len(odict) > 0:
        return odict

    return None

def savemeta(filename, metadata):
    out = file(filename, 'w')
    for i in metadata.keys():
        out.write('%s:%s\n' % (i, metadata[i]))
    out.close()

def generatestubmeta(fn):
    meta = dict()
    meta['md5'] = md5sum(fn)
    meta['caption'] = fn
    meta['keywords'] = fn

    return meta

def md5sum(filename):
    try:
        f = open(filename, 'r')
    except:
        return None

    sum = md5.md5(f.read())
    f.close()
    return sum.hexdigest()

def metafn(fn):
    return 'META_'+fn+'.txt'

def slidefn(fn):
    return '.SLIDE_'+fn+'.jpg'

def thumbfn(fn):
    return '.THUMB_'+fn+'.jpg'

def pagefn(fn):
    return fn+'.html'

def getdims(fn):
    sdim = tuple(Image.open(slidefn(fn)).size)
    tdim = tuple(Image.open(thumbfn(fn)).size)
    odim = tuple(Image.open(fn).size)

    return (sdim[0], sdim[1], tdim[0], tdim[1], odim[0], odim[1])

def generateslide(f, metadict, dimtuple, p, n):
    tproc = TemplateProcessor()

    tproc.set('nargal_name', PROGNAME)
    tproc.set('nargal_ver', PROGVER)

    tproc.set('title',TITLE)
    tproc.set('caption', metadict['caption'])
    tproc.set('prev_slide', pagefn(p))
    tproc.set('next_slide', pagefn(n))
    tproc.set('index', 'index.html')

    tproc.set('orig_image', f)
    tproc.set('slide_image', slidefn(f))

    tproc.set('slidex', str(dimtuple[0]))
    tproc.set('slidey', str(dimtuple[1]))
    tproc.set('origx', str(dimtuple[4]))
    tproc.set('origy', str(dimtuple[5]))

    meta = list()
    meta.append({'label':'original md5', 'value':metadict['md5']})
    meta.append({'label':'original dimensions', 'value':'%i x %i' % (dimtuple[4], dimtuple[5])})

    if extmatch(f, exiftypes):
        exif = EXIF.process_file(file(f, 'r'))
        if exif.has_key('Image Model') and exif.has_key('Image Make'):
            meta.append({'label':'Camera', 'value':exif['Image Make'].printable + ' ' + exif['Image Model'].printable})
        if exif.has_key('EXIF FNumber'):
            den = float(exif['EXIF FNumber'].values[0].den)
            num = float(exif['EXIF FNumber'].values[0].num)
            fnumber = str(num / den)
            meta.append({'label':'f-stop', 'value':'f/'+fnumber})
        if exif.has_key('EXIF ExposureTime'):
            meta.append({'label':'exposure time', 'value':exif['EXIF ExposureTime'].printable+' sec.'})
        if exif.has_key('EXIF ISOSpeedRatings'):
            meta.append({'label':'iso', 'value':exif['EXIF ISOSpeedRatings'].printable})
        if exif.has_key('EXIF FocalLength'):
            meta.append({'label':'focal length', 'value':exif['EXIF FocalLength'].printable+' mm'})
        if exif.has_key('EXIF Flash'):
            meta.append({'label':'flash', 'value':exif['EXIF Flash'].printable})

        if exif.has_key('MakerNote Color'):
            meta.append({'label':'color mode', 'value':exif['MakerNote Color'].printable})

        if exif.has_key('Image DateTime'):
            meta.append({'label':'date & time', 'value':exif['Image DateTime'].printable})

    tproc.set('Meta', meta)

    html = file(pagefn(f), 'w')
    html.write(tproc.process(slidetemplate))
    html.write('\n')
    html.close()

def generateindex(files, dimtuples, metadict):
    tproc = TemplateProcessor()

    tproc.set('nargal_name', PROGNAME) 
    tproc.set('nargal_ver', PROGVER)

    tproc.set('title',TITLE)

    rows = []
    col = 0
    cols = []
    for i in files:

        cols.append({'file':i, 'thumb_image':thumbfn(i), 'slide':pagefn(i), 'thumbx':str(dimtuples[i][2]), 'thumby':str(dimtuples[i][3]), 'caption':metadict[i]['caption']})
        
        if col == COLS-1:
            rows.append({'Cols':cols})
            col = 0
            cols = []
        else:
            col += 1
    
    if cols != []:
        rows.append({'Cols':cols})


    tproc.set('Rows', rows)


    somelinks = False
    if LINKDIRS:
        links = list()
        for i in os.listdir(TARGET):
            try:
                os.listdir(TARGET+'/'+i)
                if i in LINKEXCLUDE:
                    continue
                links.append({'subdir':i+'/'})
                somelinks = True
            except OSError:
                None
        tproc.set('Subdirs', links)

    tproc.set('subdirp',int(LINKDIRS and somelinks))
        
    html = file('index.html', 'w')
    html.write(tproc.process(indextemplate))
    html.close()

queue = list()
regen = False

meta = dict()

DEBUG and sys.stderr.write('Scanning files')
# iterate for every image
for fn in os.listdir(TARGET):
    if (not extmatch(fn, imagetypes)) or sysmatch(fn):
        continue

    DEBUG and sys.stderr.write('.')

    metafile = metafn(fn)
    
    # does the meta file exist?        
    if os.access(metafile, os.F_OK):
        # yes - load md5 sum
        meta[fn] = loadmeta(metafile)
        # compare with image

        sum = md5sum(fn)
	try:
	    csum = meta[fn]['md5']
	except KeyError:
            csum = ''
        if csum != sum:
            # different?
            # yes - queue slide, thumb, page generation
            queue.append(fn)
            meta[fn]['md5'] = sum
            savemeta(metafile, meta[fn])
            
        elif (not os.access(slidefn(fn), os.F_OK)) or (not os.access(thumbfn(fn), os.F_OK)):
            queue.append(fn)
        
    else:        
        # no
        # generate stub meta
        meta[fn] = generatestubmeta(fn)
        savemeta(metafile, meta[fn])
        
        # queue page, thumb, slide generation
        queue.append(fn)
        # set index flag
        regen = True

if not os.access('index.html', os.F_OK):
    regen = True

DEBUG and sys.stderr.write(' DONE!\n')

DEBUG and sys.stderr.write('Making thumbnails and slides... ')

dims = dict()

for fn in queue:
    img = Image.open(fn)
    dimx, dimy = img.size
    if dimx > dimy:
        slidex = SLIDEDIM
        thumbx = THUMBDIM
        slidey = int(round(float(dimy) * (float(SLIDEDIM) / float(dimx))))
        thumby = int(round(float(dimy) * (float(THUMBDIM) / float(dimx))))
    else:
        slidey = SLIDEDIM
        thumby = THUMBDIM
        slidex = int(round(float(dimx) * (float(SLIDEDIM) / float(dimy))))
        thumbx = int(round(float(dimx) * (float(THUMBDIM) / float(dimy))))

    # generate slide
    sim = img.resize((slidex, slidey), Image.BICUBIC)
    try:
        sim.save(slidefn(fn), 'JPEG', quality=SLIDEQUAL, optimize=1)
    except IOError:
        print "error!"
        sim.draft("RGB", sim.size)
        sim = sim.convert("RGB")
        sim.save(slidefn(fn), 'JPEG',quality=SLIDEQUAL, optimize=1)

    # generate thumbnail
    sim = img.resize((thumbx, thumby), Image.NEAREST)
    try:
        sim.save(thumbfn(fn), 'JPEG',quality=THUMBQUAL, optimize=1)
    except:
        print "error2!"
        sim.draft("RGB", sim.size)
        sim = sim.convert("RGB")
        sim.save(thumbfn(fn), 'JPEG',quality=THUMBQUAL, optimize=1)

    dims[fn] = (slidex, slidey, thumbx, thumby, dimx, dimy)
    
    DEBUG and sys.stderr.write('%i%% ' % int(round((float(queue.index(fn)+1) / float(len(queue))) * 100.0)))


DEBUG and sys.stderr.write(' DONE!\n')

filelist = list(meta.keys())
filelist.sort()

idx = 0
genindex = False

DEBUG and sys.stderr.write('Creating HTML slides... ')

# is index flag set?
if regen:
    DEBUG and sys.stderr.write('[complete] ')
    # yes - iterate through all meta
    # generate page for all meta
    for fn in filelist:
        if idx == 0:
            prev = filelist[len(filelist)-1]
        else:
            prev = filelist[idx-1]

        if idx == len(filelist)-1:
            next = filelist[0]
        else:
            next = filelist[idx+1]

        if not dims.has_key(fn):
            dims[fn] = getdims(fn)
            
        idx += 1

        generateslide(fn, meta[fn], dims[fn], prev, next)
        DEBUG and sys.stderr.write('%i%% ' % int(round((float(idx) / float(len(filelist))) * 100)))
        
    genindex = True
    DEBUG and sys.stderr.write('\n')
    
elif len(queue) > 0:
    DEBUG and sys.stderr.write('[queue] ')
    # no - iterate through page queue
    for fn in filelist:
        if not dims.has_key(fn):
            dims[fn] = getdims(fn)
            DEBUG and sys.stderr.write('.')

    DEBUG and sys.stderr.write(' ')
    for fn in queue:
        # generate pages
        idx = filelist.index(fn)
        if idx == 0:
            prev = filelist[len(filelist)-1]
        else:
            prev = filelist[idx-1]

        if idx == len(filelist)-1:
            next = filelist[0]
        else:
            next = filelist[idx+1]

        generateslide(fn, meta[fn], dims[fn], prev, next)
        DEBUG and sys.stderr.write('%i%% ' % int(round((float(idx) / float(len(queue))) * 100)))

    genindex = True
    DEBUG and sys.stderr.write('\n')

if genindex:
    generateindex(filelist, dims, meta)

DEBUG and sys.stderr.write(' DONE!\n')

DEBUG and sys.stderr.write('\n')