Statistics
| Revision:

root / branches / gvSIG_19_ext3D_osgVP_2_2_0 / libraries / lib3DMap / src / org / gvsig / gvsig3d / cacheservices / FLayerCacheService.java @ 31732

History | View | Annotate | Download (11.1 KB)

1
package org.gvsig.gvsig3d.cacheservices;
2

    
3
import java.awt.Color;
4
import java.awt.Dimension;
5
import java.awt.Graphics2D;
6
import java.awt.Image;
7
import java.awt.geom.AffineTransform;
8
import java.awt.geom.Rectangle2D;
9
import java.awt.image.BufferedImage;
10
import java.io.File;
11
import java.io.IOException;
12
import java.net.URL;
13
import java.util.HashMap;
14
import java.util.Hashtable;
15
import java.util.Vector;
16

    
17
import javax.imageio.ImageIO;
18

    
19
import org.cresques.cts.IProjection;
20
import org.gvsig.cacheservice.CacheService;
21
import org.gvsig.cacheservice.CacheServiceException;
22
import org.gvsig.cacheservice.RasterCacheService;
23
import org.gvsig.cacheservice.TileNum;
24
import org.gvsig.fmap.raster.layers.FLyrRasterSE;
25
import org.gvsig.osgvp.terrain.Terrain;
26
import org.gvsig.osgvp.util.Util;
27
import org.gvsig.raster.buffer.BufferFactory;
28
import org.gvsig.raster.buffer.RasterBuffer;
29
import org.gvsig.raster.buffer.WriterBufferServer;
30
import org.gvsig.raster.dataset.GeoRasterWriter;
31
import org.gvsig.raster.dataset.InvalidSetViewException;
32
import org.gvsig.raster.dataset.NotSupportedExtensionException;
33
import org.gvsig.raster.dataset.Params;
34
import org.gvsig.raster.dataset.io.RasterDriverException;
35
import org.gvsig.remoteClient.utils.Utilities;
36
import org.gvsig.remoteClient.wcs.WCSStatus;
37
import org.gvsig.remoteClient.wms.ICancellable;
38
import org.gvsig.remoteClient.wms.WMSStatus;
39

    
40
import com.hardcode.gdbms.driver.exceptions.ReadDriverException;
41
import com.iver.ai2.gvsig3d.map3d.layers.Layer3DProps;
42
import com.iver.cit.gvsig.exceptions.expansionfile.ExpansionFileReadException;
43
import com.iver.cit.gvsig.fmap.ViewPort;
44
import com.iver.cit.gvsig.fmap.drivers.wcs.FMapWCSDriver;
45
import com.iver.cit.gvsig.fmap.drivers.wcs.FMapWCSDriverFactory;
46
import com.iver.cit.gvsig.fmap.drivers.wms.FMapWMSDriver;
47
import com.iver.cit.gvsig.fmap.drivers.wms.FMapWMSDriverFactory;
48
import com.iver.cit.gvsig.fmap.layers.FLayer;
49
import com.iver.cit.gvsig.fmap.layers.FLyrDefault;
50
import com.iver.cit.gvsig.fmap.layers.FLyrVect;
51
import com.iver.cit.gvsig.fmap.layers.FLyrWCS;
52
import com.iver.cit.gvsig.fmap.layers.FLyrWMS;
53
import com.iver.utiles.FileUtils;
54
import com.iver.utiles.swing.threads.Cancellable;
55

    
56
public class FLayerCacheService extends RasterCacheService {
57
        private boolean         _loadFailed = true;
58
        private Rectangle2D _lyrExtentRect;
59
        private FLayer                 _layer;
60
        private IProjection _viewProj;
61
        private int                 _dataType;
62

    
63
        // WMS terrain
64
        FMapWMSDriver                 _wmsDriver = null;
65
        FMapWCSDriver                 _wcsDriver = null;
66

    
67
        WMSStatus                         _wmsStatus = null;
68
        WCSStatus                         _wcsStatus = null;
69

    
70
        public FLayerCacheService(Terrain terrain, String name, FLayer layer,
71
                        IProjection viewProj) {
72
                super(terrain.getTerrainName(), name);
73

    
74
                _layer = layer;
75
                _viewProj = viewProj;
76
                _loadFailed = true;
77

    
78
                int cacheType = CacheService.GLOBAL;
79
                if (terrain.getCoordinateSystemType() == Terrain.CoordinateSystemType.GEOCENTRIC)
80
                        cacheType += SPHERIC;
81
                else
82
                        cacheType += PLANE;
83
                setCacheType(cacheType);
84

    
85
                try {
86
                        _lyrExtentRect = _layer.getFullExtent();
87
                } catch (ExpansionFileReadException e) {
88
                        e.printStackTrace();
89
                } catch (ReadDriverException e) {
90
                        e.printStackTrace();
91
                }
92

    
93
                Layer3DProps props3D = Layer3DProps.getLayer3DProps(layer);
94
                _dataType = props3D.getType();
95
                if (_dataType == Layer3DProps.layer3DImage)
96
                        setFileExtension(".png");
97
                else
98
                        setFileExtension(".tif");
99

    
100
                _loadFailed = false;
101

    
102
        }
103

    
104
        protected class MyCancel implements Cancellable, ICancellable {
105

    
106
                public boolean isCanceled() {
107
                        return false;
108
                }
109

    
110
                public void setCanceled(boolean canceled) {
111

    
112
                }
113

    
114
                /* if you don�t put and ID the wms donwload will be fail */
115
                public Object getID() {
116
                        return this;
117
                }
118

    
119
        }
120

    
121
        private Image getTileFromLayer(String fName, Rectangle2D tileExtent) {
122
                BufferedImage image;
123
                if (_dataType == Layer3DProps.layer3DImage) {
124
                        int size = getTileSize();
125
                        image = new BufferedImage(size, size, BufferedImage.TYPE_INT_ARGB);
126

    
127
                        ViewPort vp = new ViewPort(_viewProj);
128
                        vp.setExtent(tileExtent); // the real extent of the tile
129
                        // NotificationManager.addInfo("Extent solicitado--> "
130
                        // + tileExtent.getMinX() + " " + tileExtent.getMinY() + " "
131
                        // + tileExtent.getMaxX() + " " + tileExtent.getMaxY());
132
                        vp.setAdjustable(false);
133
                        vp.setImageSize(new Dimension(size, size));
134
                        // vp.setDist1pixel(1 / 150000);
135
                        vp.setDist1pixel(0.000000001);
136
                        
137

    
138
                        Graphics2D graphics = image.createGraphics();
139
                        Color backgColor = new Color(255, 255, 255, 0); // transparent
140
                        // background
141
                        graphics.setColor(backgColor);
142
                        graphics.fillRect(0, 0, size, size);
143

    
144
                        // CancelDraw canceldraw = new MapControl.CancelDraw();
145
                        // canceldraw.setCanceled(false);
146
                        // For now we'll use a trick to make sure the layer is drawn
147
                        // even if it has scale dependency
148
                        // In the future, a scale corresponding to the tile should be passed
149
                        double scale = 0.5 * (_layer.getMaxScale() + _layer.getMinScale());
150
                        int trans = 255;
151

    
152
                        // The layers of remote services don�t have transparency.
153
                        if (_layer.getClass().equals(FLyrVect.class)) {
154
                                // Generate layers 3D properties
155
                                Layer3DProps props3D = Layer3DProps.getLayer3DProps(_layer);
156
                                // fix opaque the layer to save in disk
157
                                ((FLyrDefault) _layer).setTransparency((int) (props3D.getOpacity()*255));
158
                        }
159

    
160
                        try {
161
                                _layer.draw(image, graphics, vp, new MyCancel(), scale);
162
                        } catch (Exception e) {
163
                                return null;
164
                        }
165
                        // Restoring the real value of transparency
166
                        if (_layer.getClass().equals(FLyrVect.class)) {
167
                                // fix the real transparency to the layer
168
                                ((FLyrDefault) _layer).setTransparency(trans);
169
                        }
170

    
171
                        try {
172
                                File file = new File(fName);
173
                                String format = getFileExtension().substring(1);
174
                                saveCachedFile(image, format, file);
175
                        } catch (IOException e1) {
176
                                e1.printStackTrace();
177
                                return null;
178
                        }
179
                } else { // Elevation data
180
                        if (_layer instanceof FLyrWMS) { // WMS Elevation
181
                                try {
182
                                        FLyrWMS wmsLayer = (FLyrWMS) _layer;
183
                                        if (_wmsDriver == null) {
184

    
185
                                                URL host = wmsLayer.getHost();
186
                                                _wmsDriver = FMapWMSDriverFactory
187
                                                                .getFMapDriverForURL(host);
188
                                        }
189
                                        if (_wmsStatus == null) {
190
                                                _wmsStatus = new WMSStatus();
191
                                                _wmsStatus.setFormat(wmsLayer.getFormat());
192
                                                _wmsStatus.setHeight(32);
193
                                                _wmsStatus.setWidth(32);
194
                                                _wmsStatus.setLayerNames(Utilities.createVector(
195
                                                                wmsLayer.getLayerQuery(), ","));
196
                                                _wmsStatus.setSrs(wmsLayer.getSRS());
197

    
198
                                                HashMap props = wmsLayer.getProperties();
199
                                                Vector styles;
200
                                                Vector dimensions;
201
                                                styles = (Vector) (props.get("styles"));
202
                                                dimensions = (Vector) (props.get("dimensions"));
203
                                                _wmsStatus.setStyles(styles);
204
                                                _wmsStatus.setDimensions(dimensions);
205
                                                _wmsStatus.setTransparency(false);
206
                                                String getMapStr = wmsLayer.getOnlineResource("GetMap");
207
                                                _wmsStatus.setOnlineResource(getMapStr);
208
                                        }
209
                                        _wmsStatus.setExtent(tileExtent);
210
                                        File f = _wmsDriver.getMap(_wmsStatus, new MyCancel());
211
                                        if (f == null)
212
                                                return null;
213
                                        FileUtils.copy(f, new File(fName));
214
                                        image = ImageIO.read(new File(fName));
215
                                        return image;
216
                                } catch (Exception e) {
217
                                        return null;
218
                                }
219
                        } else if (_layer instanceof FLyrWCS) { // WCS Elevation
220
                                try {
221
                                        FLyrWCS wcsLayer = (FLyrWCS) _layer;
222
                                        if (_wcsDriver == null) {
223

    
224
                                                URL host = wcsLayer.getHost();
225
                                                _wcsDriver = FMapWCSDriverFactory
226
                                                                .getFMapDriverForURL(host);
227
                                        }
228
                                        if (_wcsStatus == null) {
229
                                                _wcsStatus = new WCSStatus();
230

    
231
                                                Hashtable props = wcsLayer.getProperties();
232

    
233
                                                _wcsStatus.setCoveraName((String) props.get("name"));
234
                                                _wcsStatus.setFormat((String) props.get("format"));
235
                                                _wcsStatus.setHeight(32);
236
                                                _wcsStatus.setWidth(32);
237
                                                _wcsStatus.setSrs((String) props.get("crs"));
238
                                                _wcsStatus.setParameters((String) props
239
                                                                .get("parameter"));
240
                                                String time = (String) props.get("time");
241
                                                if (time.length() > 0)
242
                                                        _wcsStatus.setTime(time);
243
                                        }
244
                                        _wcsStatus.setExtent(tileExtent);
245

    
246
                                        File f = _wcsDriver.getCoverage(_wcsStatus, new MyCancel());
247
                                        if (f == null)
248
                                                return null;
249

    
250
                                        FileUtils.copy(f, new File(fName));
251
                                } catch (Exception e) {
252
                                        return null;
253
                                }
254
                        }
255
                        // RASTER_GRID
256
                        if (_layer instanceof FLyrRasterSE) {
257

    
258
                                FLyrRasterSE rasterLayer = (FLyrRasterSE) _layer;
259
                                BufferFactory bufferFactory = new BufferFactory(rasterLayer
260
                                                .getDataSource());
261

    
262
                                int bands[] = { 0 };
263
                                bufferFactory.setDrawableBands(bands);
264
                                bufferFactory.setAdjustToExtent(true);
265

    
266
                                try {
267
                                        bufferFactory.setAreaOfInterest(tileExtent.getMinX(),
268
                                                        tileExtent.getMinY(), tileExtent.getMaxX(),
269
                                                        tileExtent.getMaxY(), 32, 32);
270
                                } catch (ArrayIndexOutOfBoundsException e1) {
271
                                        return null;
272
                                } catch (InvalidSetViewException e1) {
273
                                        return null;
274
                                } catch (InterruptedException e1) {
275
                                        return null;
276
                                } catch (RasterDriverException e) {
277
                                        return null;
278
                                }
279

    
280
                                RasterBuffer raster = (RasterBuffer) bufferFactory
281
                                                .getRasterBuf();
282
                                raster.setNoDataValue(0.0);
283
                                try {
284
                                        WriterBufferServer bufferServer = new WriterBufferServer(
285
                                                        raster);
286
                                        Params p = GeoRasterWriter.getWriter(fName).getParams();
287
                                        AffineTransform at = new AffineTransform(tileExtent
288
                                                        .getWidth() / 32, 0, 0,
289
                                                        -(tileExtent.getHeight() / 32), tileExtent
290
                                                                        .getMinX(), tileExtent.getMaxY());
291
                                        GeoRasterWriter grw = GeoRasterWriter.getWriter(
292
                                                        bufferServer, fName, 1, at, raster.getWidth(),
293
                                                        raster.getHeight(), raster.getDataType(), p, null);
294
                                        try {
295
                                                grw.dataWrite();
296
                                        } catch (InterruptedException e) {
297
                                                e.printStackTrace();
298
                                        }
299
                                        grw.writeClose();
300
                                        
301
                                } catch (NotSupportedExtensionException e) {
302
                                        //e.printStackTrace();
303
                                        return null;
304
                                } catch (RasterDriverException e) {
305
                                        //e.printStackTrace();
306
                                        return null;
307
                                } catch (IOException e) {
308
                                        //e.printStackTrace();
309
                                        return null;
310
                                }
311

    
312
                        }
313
                }
314
                try {
315
                        image = ImageIO.read(new File(fName));
316
                } catch (IOException e) {
317
                        return null;
318
                }
319
                return image;
320
        }
321

    
322
        public String getTileAsFName(TileNum t, Rectangle2D extent)
323
                        throws CacheServiceException {
324
                if (_loadFailed) {
325
                        throw new CacheServiceException(new Exception());
326
                }
327
                
328
                String fName = getTileFileName(t);
329
                if (!new File(fName).exists()) {
330
                        Rectangle2D tileExtent = extent;
331
                        if(getTileFromLayer(fName, tileExtent) == null) 
332
                                fName = null;
333
                }
334
                return fName;
335
        }
336

    
337
        public Image getTileAsImage(TileNum tileNum, Rectangle2D extent)
338
                        throws CacheServiceException {
339
                if ((tileNum.getLevel() == 5)) {// && (tileNum.getX())==29 &&
340
                        // (tileNum.getY()==9)){
341
                        System.err.println("pillado");
342
                }
343

    
344
                if (_loadFailed) {
345
                        throw new CacheServiceException(new Exception());
346
                }
347
                String fName = getTileFileName(tileNum);
348
                Rectangle2D tileExtent = extent;
349
                return getTileFromLayer(fName, tileExtent);
350
        }
351

    
352
        public boolean intersectsLayer(Rectangle2D extent) {
353
                return _lyrExtentRect.intersects(extent);
354
        }
355

    
356
}