a4fb96be43d0601a708308316953198e073e52ea
[mir.git] / source / mircoders / media / ImageProcessor.java
1 /*
2  * Copyright (C) 2001, 2002 The Mir-coders group
3  *
4  * This file is part of Mir.
5  *
6  * Mir is free software; you can redistribute it and/or modify
7  * it under the terms of the GNU General Public License as published by
8  * the Free Software Foundation; either version 2 of the License, or
9  * (at your option) any later version.
10  *
11  * Mir is distributed in the hope that it will be useful,
12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14  * GNU General Public License for more details.
15  *
16  * You should have received a copy of the GNU General Public License
17  * along with Mir; if not, write to the Free Software
18  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
19  *
20  * In addition, as a special exception, The Mir-coders gives permission to link
21  * the code of this program with  any library licensed under the Apache Software License,
22  * The Sun (tm) Java Advanced Imaging library (JAI), The Sun JIMI library
23  * (or with modified versions of the above that use the same license as the above),
24  * and distribute linked combinations including the two.  You must obey the
25  * GNU General Public License in all respects for all of the code used other than
26  * the above mentioned libraries.  If you modify this file, you may extend this
27  * exception to your version of the file, but you are not obligated to do so.
28  * If you do not wish to do so, delete this exception statement from your version.
29  */
30
31 package mircoders.media;
32
33 import java.awt.RenderingHints;
34 import java.awt.image.ColorModel;
35 import java.awt.image.DataBuffer;
36 import java.awt.image.PixelInterleavedSampleModel;
37 import java.io.BufferedOutputStream;
38 import java.io.ByteArrayOutputStream;
39 import java.io.File;
40 import java.io.FileOutputStream;
41 import java.io.IOException;
42 import java.io.OutputStream;
43
44 import javax.media.jai.ImageLayout;
45 import javax.media.jai.InterpolationBilinear;
46 import javax.media.jai.JAI;
47 import javax.media.jai.ParameterBlockJAI;
48 import javax.media.jai.PlanarImage;
49
50 import mir.log.LoggerWrapper;
51
52 import com.sun.media.jai.codec.ByteArraySeekableStream;
53 import com.sun.media.jai.codec.FileSeekableStream;
54 import com.sun.media.jai.codec.SeekableStream;
55
56 /**
57  *
58  * <p>Title: Image processor</p>
59  * <p>Description: Temporary image processor class. (Made for the immediate needs of CMI brasil.
60  *                 Will become obsolete when mh's media handler rewrite is finished. </p>
61  * @author Zapata
62  * @version 1.0
63  */
64
65 public class ImageProcessor {
66   static final LoggerWrapper logger = new LoggerWrapper("media");
67
68   private PlanarImage image;
69   private PlanarImage scaledImage;
70
71   private byte[] iconData;
72   private byte[] imageData;
73   private int iconWidth;
74   private int iconHeight;
75
76   public ImageProcessor(SeekableStream anImageStream) throws IOException {
77     PlanarImage tempImage = JAI.create("stream", anImageStream);
78     ParameterBlockJAI params = new ParameterBlockJAI("format");
79     int bands[];
80     int nrComponents;
81
82
83     params.addSource(tempImage);
84     params.setParameter("dataType", DataBuffer.TYPE_BYTE);
85
86     ImageLayout layout = new ImageLayout();
87     nrComponents = tempImage.getColorModel().getNumColorComponents();
88
89     bands = new int[nrComponents];
90     for (int i=0; i<nrComponents; i++)
91       bands[i]=i;
92
93     layout.setColorModel(ColorModel.getRGBdefault());
94     layout.setSampleModel(
95         new PixelInterleavedSampleModel(DataBuffer.TYPE_BYTE,
96         tempImage.getWidth(),
97         tempImage.getHeight(),
98         nrComponents,
99         nrComponents*tempImage.getWidth(),
100         bands));
101
102     RenderingHints hints = new RenderingHints(JAI.KEY_IMAGE_LAYOUT, layout);
103
104     image = JAI.create("format", params, hints);
105
106     scaledImage = image;
107   }
108
109   public ImageProcessor(File aFile) throws IOException {
110     this(new FileSeekableStream(aFile));
111   }
112
113   public ImageProcessor(byte[] anImageData) throws IOException {
114     this(new ByteArraySeekableStream(anImageData));
115   }
116
117   public void descaleImage(int aMaxSize) throws java.io.IOException {
118     descaleImage(aMaxSize, 0);
119   }
120
121   public void descaleImage(int aMaxSize, float aMinDescale) throws java.io.IOException {
122     descaleImage(aMaxSize, aMaxSize, aMinDescale, 0);
123   }
124
125   public void descaleImage(int aMaxSize, int aMinResize) throws java.io.IOException {
126     descaleImage(aMaxSize, aMaxSize, 0, aMinResize);
127   }
128
129   public void descaleImage(int aMaxSize, float aMinDescale, int aMinResize) throws java.io.IOException {
130     descaleImage(aMaxSize, aMaxSize, aMinDescale, aMinResize);
131   }
132
133   /**
134    *
135    * Resizes an image to fit inside <code>aMaxWidth</code> and <code>aMaxHeight</code>, provided
136    *    this requires at least <code>aMinResize</code> pixels will be removed from either the width or
137    *    the height
138    *
139    * @param aMaxWidth
140    * @param aMaxHeight
141    * @param aMinDescale
142    * @param aMinResize
143    * @throws java.io.IOException
144    */
145   public void descaleImage(int aMaxWidth, int aMaxHeight, float aMinDescale, int aMinResize) throws java.io.IOException {
146     float scale;
147     scaledImage = image;
148
149     if ((aMaxWidth>0 && image.getWidth()>aMaxWidth+aMinResize-1) || (aMaxHeight>0 && image.getHeight()>aMaxHeight+aMinResize-1))
150     {
151       logger.info("Scaling image");
152
153       scale=1;
154
155       if (aMaxWidth>0 && image.getWidth()>aMaxWidth) {
156         scale = Math.min(scale, (float) aMaxWidth / (float) image.getWidth());
157       }
158       if (aMaxHeight>0 && image.getHeight()>aMaxHeight) {
159         scale = Math.min(scale, (float) aMaxHeight / (float) image.getHeight());
160       }
161
162       if (1-scale>aMinDescale) {
163         scaleImage(scale);
164       }
165     }
166   }
167
168   public void scaleImage(float aScalingFactor) throws java.io.IOException {
169     ParameterBlockJAI params = new ParameterBlockJAI("scale");
170     params.addSource(image);
171
172     params.setParameter("xScale", aScalingFactor);
173     params.setParameter("yScale", aScalingFactor);
174     params.setParameter("xTrans", 0.0F);
175     params.setParameter("yTrans", 0.0F);
176     params.setParameter("interpolation", new InterpolationBilinear());
177     scaledImage = JAI.create("scale", params);
178   }
179
180   public int getWidth() {
181     return image.getWidth();
182   }
183
184   public int getHeight() {
185     return image.getHeight();
186   }
187
188   public int getScaledWidth() {
189     return scaledImage.getWidth();
190   }
191
192   public int getScaledHeight() {
193     return scaledImage.getHeight();
194   }
195
196   public void writeScaledData(OutputStream aStream, String anImageType) {
197     JAI.create("encode", scaledImage, aStream, anImageType, null);
198   }
199
200   public byte[] getScaledData(String anImageType) {
201     ByteArrayOutputStream outputStream = new ByteArrayOutputStream();
202     writeScaledData(outputStream, anImageType);
203     return outputStream.toByteArray();
204   }
205
206   public void writeScaledData(File aFile, String anImageType) throws IOException {
207     writeScaledData(new BufferedOutputStream(new FileOutputStream(aFile),8192), anImageType);
208   }
209 }
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229