org.uva.itast.blended.omr.scanners.SolidSquareMarkScanner.java Source code

Introduction

Here is the source code for org.uva.itast.blended.omr.scanners.SolidSquareMarkScanner.java

Source

/************************************************
 *  Note: Original work copyright to respective authors
*
* This file is part of Blended (c) 2009-2010 University of Valladolid..
*
* Blended 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.
*
* Blended 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.
*
*
* Module developed at the University of Valladolid http://www.eduvalab.uva.es
*
* http://www.itnt.uva.es , http://www.eduvalab.uva.es
*
* Designed and directed by Juan Pablo de Castro with 
* the effort of many other students of telecommunication 
* engineering.
* This module is provides as-is without any 
* guarantee. Use it as your own risk.
*
* @author Juan Pablo de Castro
* @author Jesus Rodilana
* @author Mara Jess Verd 
* @author Luisa Regueras 
* @author Elena Verd
* 
* @license http://www.gnu.org/copyleft/gpl.html GNU Public License
* @package blended
 ***********************************************************************/

package org.uva.itast.blended.omr.scanners;

import java.awt.BasicStroke;
import java.awt.Color;
import java.awt.Graphics2D;
import java.awt.Polygon;
import java.awt.geom.AffineTransform;
import java.awt.geom.Dimension2D;
import java.awt.geom.Rectangle2D;
import java.awt.image.BufferedImage;

import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.uva.itast.blended.omr.OMRProcessor;
import org.uva.itast.blended.omr.pages.PageImage;

public class SolidSquareMarkScanner extends TemplateMarkScanner {
    Log logger = LogFactory.getLog(SolidSquareMarkScanner.class);

    /**
     * 
     * @param grayimage
     * @param markWidth          in milimeters
     * @param markHeight         in milimeters
     * @param approxXscale
     * @param approxYscale
     */
    public SolidSquareMarkScanner(OMRProcessor omr, PageImage pageimage, double markWidth, double markHeight,
            boolean medianfilter) {
        super(omr, pageimage, markWidth, markHeight, medianfilter);
    }

    /* (non-Javadoc)
     * @see org.uva.itast.blended.omr.scanners.TemplateMarkScanner#isMark(java.awt.geom.Rectangle2D, boolean)
     */
    @Override
    protected MarkDetectionStatus isMark(Rectangle2D markArea, boolean dump) throws MarkScannerException {
        if (logger.isDebugEnabled())
            dump = true;
        return super.isMark(markArea, dump);
    }

    @Override
    protected double fillTemplate(BufferedImage templateimg, int width, int height) {
        double centerX = templateimg.getWidth() / 2;
        double centerY = templateimg.getHeight() / 2;
        int minX = (int) (centerX - width / 2);
        int maxX = (int) (centerX + width / 2);
        int minY = (int) (centerY - height / 2);
        int maxY = (int) (centerY + height / 2);

        int whites = 0;
        for (int i = 0; i < templateimg.getWidth(); i++) {
            for (int j = 0; j < templateimg.getHeight(); j++) {
                if (i > minX && i < maxX && j > minY && j < maxY) {
                    templateimg.setRGB(i, j, Color.BLACK.getRGB());
                } else {
                    templateimg.setRGB(i, j, Color.WHITE.getRGB());
                    whites++;
                }
            }
        }
        // compute autoSimilarity
        return ((double) whites) / (templateimg.getHeight() * templateimg.getWidth());
    }

    /**
     * @param pageImage
     */
    public void putEmphasisMarkOnImage(PageImage pageImage, Color color) {

        Graphics2D g = pageImage.getReportingGraphics();
        // int centerColor=imagen.getRGB(maxsimX, maxsimY);
        // g.setXORMode(new Color(centerColor));
        // g.setColor(Color.RED);
        // g.fillOval(maxsimX - markWidth/2, maxsimY - markHeight/2, markWidth,
        // markHeight);
        // g.setPaintMode();
        Dimension2D markDimsPx = pageImage.sizeInPixels(new Size(markWidth, markHeight));
        int markWidth = (int) markDimsPx.getWidth();
        int markHeight = (int) markDimsPx.getHeight();
        g.setColor(color);
        AffineTransform t = g.getTransform();
        g.drawLine(maxsimX, maxsimY - markHeight / 2 - 1, maxsimX,
                maxsimY - markHeight / 2 - (int) (20 / t.getScaleY()));
        Polygon arrowHead = new Polygon();
        arrowHead.addPoint(maxsimX, (int) (maxsimY - markHeight / 2 - 1 / t.getScaleY()));
        arrowHead.addPoint((int) (maxsimX - 6 / t.getScaleX()),
                (int) (maxsimY - markHeight / 2 - 6 / t.getScaleY()));
        arrowHead.addPoint((int) (maxsimX + 6 / t.getScaleX()),
                (int) (maxsimY - markHeight / 2 - 6 / t.getScaleY()));
        g.fillPolygon(arrowHead);

        g.setStroke(new BasicStroke(2, BasicStroke.CAP_SQUARE, BasicStroke.JOIN_ROUND, 1,
                new float[] { (float) (3 / t.getScaleX()), (float) (3 / t.getScaleY()) }, 0));
        g.drawRect(maxsimX - markWidth / 2 - 1, maxsimY - markHeight / 2 - 1, markWidth + 1, markHeight + 1);

    }

    @Override
    public String getType() {
        return "SolidSquare";
    }
}