#!/usr/local/bin/python

from eigenlib import *
import re

# usage:  learn-faces.py eigs-dir learn-dir face...

fs = FaceSpace(sys.argv[1])

learndir = sys.argv[2]

name_finder = re.compile('([\w]+)\.')
for person in sys.argv[3:]:
    name = name_finder.match(person).group(1)
    mesg(name + ': loading...')
    im_face = fs.image_to_vector(person)
    mesg(' done.  ')
    fs_face = fs.project_to_face_space(im_face)
    mesg('saving...')
    save_name = learndir + '/' + name + '.face'
    fs.write_fs_face(fs_face, save_name)
    print ' done.'