net.sourceforge.javaocr.ocrPlugins.handWriting.HandwritingOCR Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of javaocr-plugin-awt Show documentation
encapsulate AWT based image processing classes
The newest version!
/*
 * Copyright (c) 2003-2012, Ronald B. Cemer , Konstantin Pribluda, William Whitney, Andrea De Pasquale
 *
 *
 *
 *  Licensed under the Apache License, Version 2.0 (the "License");
 *  you may not use this file except in compliance with the License.
 *  You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 *  Unless required by applicable law or agreed to in writing,
 *  software distributed under the License is distributed on an "AS IS" BASIS,
 *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *  See the License for the specific language governing permissions and
 *  limitations under the License.
 */
package net.sourceforge.javaocr.ocrPlugins.handWriting;

import java.awt.image.BufferedImage;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.logging.Logger;
import net.sourceforge.javaocr.ocrPlugins.aspectRatio.AspectRatioOCR;
import net.sourceforge.javaocr.ocrPlugins.mseOCR.OCRScanner;
import net.sourceforge.javaocr.ocrPlugins.mseOCR.TrainingImage;
import net.sourceforge.javaocr.scanner.accuracy.AccuracyListenerInterface;
import net.sourceforge.javaocr.scanner.accuracy.OCRComp;
import net.sourceforge.javaocr.scanner.accuracy.OCRIdentification;

/**
 * Employs the help of many plug-ins to help recognize hand writing.
 * @author William Whitney
 */
public class HandwritingOCR implements AccuracyListenerInterface
{

    private final HashMap> trainingImages;
    private boolean isDoMSE;
    private boolean isDoAspect;
    private final ArrayList mseCharIdentList;
    private final ArrayList aspectCharIdentList;

    public HandwritingOCR(HashMap> trainingImages)
    {
        this.trainingImages = trainingImages;
        this.mseCharIdentList = new ArrayList();
        this.aspectCharIdentList = new ArrayList();
      
    }

    public String scan(BufferedImage targetBfImage)
    {
        if (isDoMSE)
        {
            //Make scan pass using MSE OCR
            doMSEScan(targetBfImage);
        }

        if (isDoAspect)
        {
            //Make aspect ratio pass
            doAspectRatioScan(targetBfImage);
        }

        //Analyze the info that came back from processCharOrSpace
        ResultAnalyzer resultAnalyzer = new ResultAnalyzer(mseCharIdentList, aspectCharIdentList);
        return resultAnalyzer.calculateResultAndReturnString();

    }

    public void processCharOrSpace(OCRIdentification charIdent)
    {
        if (charIdent.getOCRType() == OCRComp.MSE)
        {
            this.mseCharIdentList.add(charIdent);
        }
        else if (charIdent.getOCRType() == OCRComp.ASPECT_RATIO)
        {
            this.aspectCharIdentList.add(charIdent);
        }
       
    }

   

    private void doMSEScan(BufferedImage targetBfImage)
    {
        OCRScanner ocrScanner = new OCRScanner();
        ocrScanner.acceptAccuracyListener(this);
        ocrScanner.addTrainingImages(trainingImages);
        ocrScanner.scan(targetBfImage, 0, 0, 0, 0, null);
    }

    private void doAspectRatioScan(BufferedImage targetBfImage)
    {
        AspectRatioOCR ocrScanner = new AspectRatioOCR(trainingImages);
        ocrScanner.acceptAccuracyListener(this);
        ocrScanner.scan(targetBfImage);
    }

    public void setEnableMSEOCR(boolean enable)
    {
        this.isDoMSE = enable;
    }

    public void setEnableAspectOCR(boolean enable)
    {
        this.isDoAspect = enable;
    }

   
    private static final Logger LOG = Logger.getLogger(HandwritingOCR.class.getName());
}