360394c78d22de83d2f526dd7ae26661b5e97d9c
[mir.git] / source / mircoders / entity / EntityImages.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.entity;
32
33 import java.io.ByteArrayOutputStream;
34 import java.io.InputStream;
35 import java.sql.SQLException;
36
37 import mir.log.LoggerWrapper;
38 import mir.storage.DatabaseFailure;
39 import mir.util.StreamCopier;
40 import mir.media.image.ImageProcessor;
41 import mir.media.image.JAIImageProcessor;
42
43 public class EntityImages extends EntityUploadedMedia
44 {
45   private int maxImageSize = configuration.getInt("Producer.Image.MaxSize");
46   private int maxIconSize = configuration.getInt("Producer.Image.MaxIconSize");
47   private float minDescaleRatio = configuration.getFloat("Producer.Image.MinDescalePercentage")/100;
48   private int minDescaleReduction = configuration.getInt("Producer.Image.MinDescaleReduction");
49
50
51   public EntityImages()
52   {
53     super();
54
55     logger = new LoggerWrapper("Entity.UploadedMedia.Images");
56   }
57
58   /**
59    * Retrieves the image data
60    */
61   public byte[] getImage() throws DatabaseFailure {
62     try {
63       return database.getBinaryField("select image_data from images where id="+getId());
64     }
65     catch (SQLException e) {
66       throw new DatabaseFailure(e);
67     }
68   }
69
70   /**
71    * Processes and saves image data
72    */
73   public void setImage(InputStream anInputStream, String type) throws DatabaseFailure {
74     if (anInputStream != null) {
75       try {
76         ByteArrayOutputStream inputData = new ByteArrayOutputStream();
77         StreamCopier.copy(anInputStream, inputData);
78
79         ImageProcessor processor = new JAIImageProcessor(inputData.toByteArray());
80         processor.descaleImage(maxImageSize, minDescaleRatio, minDescaleReduction);
81
82         ByteArrayOutputStream imageData = new ByteArrayOutputStream();
83         processor.writeScaledData(imageData, type);
84         database.setBinaryField("image_data", getId(), imageData.toByteArray());
85
86         setFieldValue("img_height", new Integer(processor.getScaledHeight()).toString());
87         setFieldValue("img_width", new Integer(processor.getScaledWidth()).toString());
88
89         imageData.reset();
90         processor.descaleImage(maxIconSize, minDescaleRatio, minDescaleReduction);
91         processor.writeScaledData(imageData, type);
92         database.setBinaryField("icon_data", getId(), imageData.toByteArray());
93
94         setFieldValue("icon_height", new Integer(processor.getScaledHeight()).toString());
95         setFieldValue("icon_width", new Integer(processor.getScaledWidth()).toString());
96
97         update();
98       }
99       catch (Exception e) {
100         throw new DatabaseFailure(e);
101       }
102     }
103   }
104
105   /**
106    * Takes an OutputStream as an argument and reads in the data
107    * from the DB and writes it to the OutputStream.
108    *
109    * It will also take care of closing the OutputStream.
110    */
111   public byte[] getIcon() throws DatabaseFailure {
112     try {
113       return database.getBinaryField("select icon_data from images where id="+getId());
114     }
115     catch (SQLException e) {
116       throw new DatabaseFailure(e);
117     }
118   }
119 }