View Javadoc

1   /**
2    * Copyright (c) 2011, The University of Southampton and the individual contributors.
3    * All rights reserved.
4    *
5    * Redistribution and use in source and binary forms, with or without modification,
6    * are permitted provided that the following conditions are met:
7    *
8    *   * 	Redistributions of source code must retain the above copyright notice,
9    * 	this list of conditions and the following disclaimer.
10   *
11   *   *	Redistributions in binary form must reproduce the above copyright notice,
12   * 	this list of conditions and the following disclaimer in the documentation
13   * 	and/or other materials provided with the distribution.
14   *
15   *   *	Neither the name of the University of Southampton nor the names of its
16   * 	contributors may be used to endorse or promote products derived from this
17   * 	software without specific prior written permission.
18   *
19   * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
20   * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
21   * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
22   * DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR
23   * ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
24   * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
25   * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON
26   * ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
27   * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
28   * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
29   */
30  package org.openimaj.image.neardups.sim;
31  
32  import java.io.IOException;
33  
34  import org.openimaj.image.ImageUtilities;
35  import org.openimaj.image.MBFImage;
36  import org.openimaj.image.processing.resize.ResizeProcessor;
37  
38  public class WatermarkSimulation extends Simulation {
39  	protected final static float minAlpha = 0.4f;
40  	protected final static float maxAlpha = 1.0f;
41  	protected final String[] watermarks = { "sotonimages.png", "logo.png" };
42  
43  	public WatermarkSimulation(int seed) {
44  		super(seed);
45  	}
46  
47  	@Override
48  	public MBFImage applySimulation(MBFImage input) {
49  		try {
50  			final String wmark = watermarks[random.nextInt(watermarks.length)];
51  
52  			final MBFImage watermark = ImageUtilities.readMBFAlpha(WatermarkSimulation.class
53  					.getResourceAsStream("/org/openimaj/image/neardups/" + wmark));
54  
55  			if (watermark.getHeight() > input.getHeight()) {
56  				final int newY = (input.getHeight() - 10);
57  				final int newX = (int) (watermark.getWidth() * ((double) newY / (double) watermark.getHeight()));
58  				watermark.processInplace(new ResizeProcessor(newX, newY));
59  			}
60  
61  			if (watermark.getWidth() > input.getWidth()) {
62  				final int newX = (input.getWidth() - 10);
63  				final int newY = (int) (watermark.getHeight() * ((double) newX / (double) watermark.getWidth()));
64  				watermark.processInplace(new ResizeProcessor(newX, newY));
65  			}
66  
67  			final float alpha = randomFloatInRange(minAlpha, maxAlpha);
68  			watermark.getBand(3).multiplyInplace(alpha);
69  			final MBFImage output = input.clone();
70  			output.drawImage(watermark, (input.getWidth() / 2) - (watermark.getWidth() / 2), (input.getHeight() / 2)
71  					- (watermark.getHeight() / 2));
72  			return output;
73  
74  		} catch (final IOException e) {
75  			throw new Error(e);
76  		}
77  	}
78  }