Enrollment.java

import com.digitalpersona.uareu.*;
import com.digitalpersona.uareu.Reader.CaptureResult;
import java.awt.Dimension;
import java.awt.event.ActionEvent;
import java.awt.event.ActionListener;
import java.io.BufferedOutputStream;
import java.io.File;
import java.io.FileNotFoundException;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.OutputStream;
import java.lang.reflect.InvocationTargetException;
import java.sql.SQLException;
import javax.imageio.ImageIO;
import javax.swing.Box;
import javax.swing.BoxLayout;
import javax.swing.JButton;
import javax.swing.JDialog;
import javax.swing.JFileChooser;
import javax.swing.JLabel;
import javax.swing.JOptionPane;
import javax.swing.JPanel;
import javax.swing.JScrollPane;
import javax.swing.JTextArea;

public class Enrollment
        extends JPanel
        implements ActionListener {

    public class EnrollmentThread
            extends Thread
            implements Engine.EnrollmentCallback {

        public static final String ACT_PROMPT = "enrollment_prompt";
        public static final String ACT_CAPTURE = "enrollment_capture";
        public static final String ACT_FEATURES = "enrollment_features";
        public static final String ACT_DONE = "enrollment_done";
        public static final String ACT_CANCELED = "enrollment_canceled";
        public static final String ACT_SAVE = "save";

        public class EnrollmentEvent extends ActionEvent {

            private static final long serialVersionUID = 102;

            public Reader.CaptureResult capture_result;
            public Reader.Status reader_status;
            public UareUException exception;
            public Fmd enrollment_fmd;

            public EnrollmentEvent(Object source, String action, Fmd fmd, Reader.CaptureResult cr, Reader.Status st, UareUException ex) {
                super(source, ActionEvent.ACTION_PERFORMED, action);
                capture_result = cr;
                reader_status = st;
                exception = ex;
                enrollment_fmd = fmd;
            }
        }

        private final Reader m_reader;
        private CaptureThread m_capture;
        private ActionListener m_listener;
        private boolean m_bCancel;

        protected EnrollmentThread(Reader reader, ActionListener listener) {
            m_reader = reader;
            m_listener = listener;
        }

        public Engine.PreEnrollmentFmd GetFmd(Fmd.Format format) {
            Engine.PreEnrollmentFmd prefmd = null;

            while (null == prefmd && !m_bCancel) {
                //start capture thread
                m_capture = new CaptureThread(m_reader, false, Fid.Format.ANSI_381_2004, Reader.ImageProcessing.IMG_PROC_DEFAULT);
                m_capture.start(null);

                //prompt for finger
                SendToListener(ACT_PROMPT, null, null, null, null);

                //wait till done
                m_capture.join(0);

                //check result
                CaptureThread.CaptureEvent evt = m_capture.getLastCaptureEvent();
                if (null != evt.capture_result) {
                    if (Reader.CaptureQuality.CANCELED == evt.capture_result.quality) {
                        //capture canceled, return null
                        break;
                    } else if (null != evt.capture_result.image && Reader.CaptureQuality.GOOD == evt.capture_result.quality) {
                        //Send image
                       
                        SendToListener(ACT_CAPTURE, null, evt.capture_result, null, null);

                        //acquire engine
                        Engine engine = UareUGlobal.GetEngine();

                        try {
							//extract features

                            Fmd fmd = engine.CreateFmd(evt.capture_result.image, Fmd.Format.DP_PRE_REG_FEATURES);

                            //return prefmd 
                            prefmd = new Engine.PreEnrollmentFmd();
                            prefmd.fmd = fmd;
                            prefmd.view_index = 0;

                            //send success
                            SendToListener(ACT_FEATURES, null, null, null, null);
                        } catch (UareUException e) {
                            //send extraction error
                            SendToListener(ACT_FEATURES, null, null, null, e);
                        }
                    } else {
                        //send quality result
                        SendToListener(ACT_CAPTURE, null, evt.capture_result, evt.reader_status, evt.exception);
                    }
                } else {
                    //send capture error
                    SendToListener(ACT_CAPTURE, null, evt.capture_result, evt.reader_status, evt.exception);
                }
            }

            return prefmd;
        }

        public void cancel() {
            m_bCancel = true;
            if (null != m_capture) {
                m_capture.cancel();
            }
        }

        private void SendToListener(String action, Fmd fmd, Reader.CaptureResult cr, Reader.Status st, UareUException ex) {
            if (null == m_listener || null == action || action.equals("")) {
                return;
            }

            final EnrollmentEvent evt = new EnrollmentEvent(this, action, fmd, cr, st, ex);

            //invoke listener on EDT thread
            try {
                javax.swing.SwingUtilities.invokeAndWait(new Runnable() {
                    public void run() {
                        m_listener.actionPerformed(evt);
                    }
                });
            } catch (InvocationTargetException e) {
                e.printStackTrace();
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
        }

        public void run() {
            //acquire engine
            Engine engine = UareUGlobal.GetEngine();

            try {
                m_bCancel = false;
                while (!m_bCancel) {
                    //run enrollment
                    Fmd fmd = engine.CreateEnrollmentFmd(Fmd.Format.DP_REG_FEATURES, this);

                    //send result
                    if (null != fmd) {

                        SendToListener(ACT_DONE, fmd, null, null, null);
                    } else {
                        SendToListener(ACT_CANCELED, null, null, null, null);
                        break;
                    }
                }
            } catch (UareUException e) {
                JOptionPane.showMessageDialog(null, "Exception during creation of data and import");
                SendToListener(ACT_DONE, null, null, null, e);
            }
        }
    }

    private static final long serialVersionUID = 6;
    private static final String ACT_BACK = "back";
    private static final String ACT_SAVE = "save";
    private static final String ACT_SAVE_DB = "save2db";

    public com.digitalpersona.uareu.Fmd enrollmentFMD;
    private EnrollmentThread m_enrollment;
    private Reader m_reader;
    private JDialog m_dlgParent;
    private JTextArea m_text;
    private JLabel infoUsername_text;
    private JTextArea username_text;
    private boolean m_bJustStarted;
    private JButton m_save;
    private JButton m_save2DB;
    private ImagePanel m_imagePanel;

    private Enrollment(Reader reader) {
        m_reader = reader;
        m_bJustStarted = true;
        m_enrollment = new EnrollmentThread(m_reader, this);

        final int vgap = 5;
        final int width = 380;

        BoxLayout layout = new BoxLayout(this, BoxLayout.Y_AXIS);
        setLayout(layout);

        m_imagePanel = new ImagePanel();
        m_imagePanel.setPreferredSize(new Dimension(100, 300));
        add(m_imagePanel);

        m_text = new JTextArea(22, 1);
        m_text.setEditable(false);
        JScrollPane paneReader = new JScrollPane(m_text);
        add(paneReader);
        Dimension dm = paneReader.getPreferredSize();
        dm.width = width;
        paneReader.setPreferredSize(dm);

        infoUsername_text = new JLabel("Enter Username:");
        add(infoUsername_text);

        username_text = new JTextArea(1, 40);
        add(username_text);

        add(Box.createVerticalStrut(vgap));

        JButton btnBack = new JButton("Back");
        btnBack.setActionCommand(ACT_BACK);
        btnBack.addActionListener(this);
        add(btnBack);

        m_save = new JButton("Save to File");
        m_save.setActionCommand(ACT_SAVE);
        m_save.addActionListener(this);
        m_save.setEnabled(false);
        add(m_save);

        m_save2DB = new JButton("Save to DB");
        m_save2DB.setActionCommand(ACT_SAVE_DB);
        m_save2DB.addActionListener(this);
        m_save2DB.setEnabled(false);
        add(m_save2DB);

        add(Box.createVerticalStrut(vgap));

        setOpaque(true);
    }

    public void actionPerformed(ActionEvent e) {
        if (e.getActionCommand().equals(ACT_BACK)) {
            //destroy dialog to cancel enrollment
            m_dlgParent.setVisible(false);

            return;
        } else if (e.getActionCommand().equals(ACT_SAVE_DB)) {
            FingerDB db = new FingerDB("localhost", "uareu", "root", "password");
            try {
                db.Open();
                if (this.username_text.getText().isEmpty() == true) {
                    //Check if user already exists
                    if (db.UserExists(this.username_text.getText()) == false) {
                        //Save user to database along with fingerprint minutiae
                        db.Insert(this.username_text.getText(), this.enrollmentFMD.getData());
                        m_dlgParent.setVisible(false);
                    } else {
                        JOptionPane.showMessageDialog(null, "Username already taken.");
                        this.username_text.requestFocusInWindow();
                    }

                } else {
                    JOptionPane.showMessageDialog(null, "Please enter a unique username");
                    this.username_text.requestFocusInWindow();
                }

            } catch (SQLException e3) {
                JOptionPane.showMessageDialog(null, e3.getMessage());
            }
            try {
                db.Close();
            } catch (SQLException e1) {
                // TODO Auto-generated catch block
                e1.printStackTrace();
            }
        } else if (e.getActionCommand().equals(ACT_SAVE)) {
            saveDataToFile(enrollmentFMD.getData());
            this.m_save.setEnabled(false);
            return;
        } else {

            EnrollmentThread.EnrollmentEvent evt = (EnrollmentThread.EnrollmentEvent) e;

            if (e.getActionCommand().equals(EnrollmentThread.ACT_PROMPT)) {
                if (m_bJustStarted) {
                    m_text.append("Enrollment started\n");
                    m_text.append("    Put your any finger on the reader\n");
                } else {
                    m_text.append("    Put the same finger on the reader\n");
                }
                m_bJustStarted = false;
            } else if (e.getActionCommand().equals(EnrollmentThread.ACT_CAPTURE)) {

                if (null != evt.capture_result) {
                    if (evt.capture_result.image != null) {
                        m_imagePanel.showImage(evt.capture_result.image);
                    }
                }

                if (null != evt.capture_result) {
                    //MessageBox.BadQuality(evt.capture_result.quality);
                } else if (null != evt.exception) {

                    //MessageBox.DpError("Capture", evt.exception);
                } else if (null != evt.reader_status) {
                    //MessageBox.BadStatus(evt.reader_status);
                }

                m_bJustStarted = false;
            } else if (e.getActionCommand().equals(EnrollmentThread.ACT_FEATURES)) {
                if (null == evt.exception) {
                    m_text.append("    fingerprint captured, features extracted\n\n");
                } else {
                    MessageBox.DpError("Feature extraction", evt.exception);
                }
                m_bJustStarted = false;
            } else if (e.getActionCommand().equals(EnrollmentThread.ACT_DONE)) {
                if (null == evt.exception) {
                    String str = String.format("    Enrollment template created, size: %d\n\n\nPlease save to file or verify.", evt.enrollment_fmd.getData().length);
                    enrollmentFMD = evt.enrollment_fmd;
                    m_enrollment.cancel();
                    this.m_save.setEnabled(true);
                    this.m_save2DB.setEnabled(true);
                    m_text.append(str);
                } else {
                    MessageBox.DpError("Enrollment template creation", evt.exception);
                }
                m_bJustStarted = true;
            } else if (e.getActionCommand().equals(EnrollmentThread.ACT_CANCELED)) {
                //canceled, destroy dialog
                m_dlgParent.setVisible(false);
            }

            //cancel enrollment if any exception or bad reader status
            if (null != evt.exception) {
                m_dlgParent.setVisible(false);
            } else if (null != evt.reader_status && Reader.ReaderStatus.READY != evt.reader_status.status && Reader.ReaderStatus.NEED_CALIBRATION != evt.reader_status.status) {
                m_dlgParent.setVisible(false);
            }
        }
    }

    private void saveDataToFile(byte[] data) {
        // TODO Auto-generated method stub
        JFileChooser fc = new JFileChooser(new File("test"));

        fc.showSaveDialog(this);
        if (fc.getSelectedFile() != null) {
            OutputStream output = null;
            try {
                output = new BufferedOutputStream(new FileOutputStream(fc.getSelectedFile()));
                output.write(data);
                output.close();
            } catch (FileNotFoundException e) {
                // TODO Auto-generated catch block
                e.printStackTrace();
            } catch (Exception e) {
                // TODO Auto-generated catch block
                JOptionPane.showMessageDialog(null, "Error saving file.");
            }
        }
    }

    private void doModal(JDialog dlgParent) {
        //open reader
        try {
            m_reader.Open(Reader.Priority.COOPERATIVE);
        } catch (UareUException e) {
            MessageBox.DpError("Reader.Open()", e);
        }

        //start enrollment thread
        m_enrollment.start();

        //bring up modal dialog
        m_dlgParent = dlgParent;
        m_dlgParent.setContentPane(this);
        m_dlgParent.pack();
        m_dlgParent.setLocationRelativeTo(null);
        m_dlgParent.setVisible(true);
        m_dlgParent.dispose();

        //stop enrollment thread
        m_enrollment.cancel();

        //close reader
        try {
            m_reader.Close();
        } catch (UareUException e) {
            MessageBox.DpError("Reader.Close()", e);
        }
    }

    public static Fmd Run(Reader reader) {
        JDialog dlg = new JDialog((JDialog) null, "Enrollment", true);
        Enrollment enrollment = new Enrollment(reader);
        enrollment.doModal(dlg);
        return enrollment.enrollmentFMD;
    }
}