Statistics
| Revision:

svn-gvsig-desktop / trunk / libraries / lib3DMap / src / com / iver / ai2 / gvsig3d / cacheservices / FLayerCacheService.java @ 18219

History | View | Annotate | Download (12.9 KB)

1
package com.iver.ai2.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 org.cresques.cts.IProjection;
18
import org.cresques.px.Extent;
19
import org.gvsig.cacheservice.CacheService;
20
import org.gvsig.cacheservice.CacheServiceException;
21
import org.gvsig.cacheservice.RasterCacheService;
22
import org.gvsig.cacheservice.TileNum;
23
import org.gvsig.fmap.raster.layers.FLyrRasterSE;
24
import org.gvsig.raster.buffer.BufferFactory;
25
import org.gvsig.raster.buffer.RasterBuffer;
26
import org.gvsig.raster.buffer.cache.WriterBufferServer;
27
import org.gvsig.raster.dataset.GeoRasterWriter;
28
import org.gvsig.raster.dataset.InvalidSetViewException;
29
import org.gvsig.raster.dataset.NotSupportedExtensionException;
30
import org.gvsig.raster.dataset.Params;
31
import org.gvsig.raster.dataset.io.RasterDriverException;
32
import org.gvsig.remoteClient.utils.Utilities;
33
import org.gvsig.remoteClient.wcs.WCSStatus;
34
import org.gvsig.remoteClient.wms.ICancellable;
35
import org.gvsig.remoteClient.wms.WMSStatus;
36

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

    
53
import es.upv.ai2.osgvp.planets.Planet;
54

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

    
63
        // WMS terrain
64
        // TODO Un toggle this comment to use WMS and WCS extensions.
65
        FMapWMSDriver _wmsDriver = null;
66
        FMapWCSDriver _wcsDriver = null;
67

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

    
71
        public FLayerCacheService(Planet planet, String name, FLayer layer,
72
                        IProjection viewProj) {
73
                super(planet.getPlanetName(), name);
74

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

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

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

    
94
                _extent = new Extent(_lyrExtentRect);
95

    
96
                Layer3DProps props3D = Layer3DProps.getLayer3DProps(layer);
97
                _dataType = props3D.getType();
98
                if (_dataType == Layer3DProps.layer3DImage)
99
                        setFileExtension(".png");
100
                else
101
                        setFileExtension(".tif");
102

    
103
                _loadFailed = false;
104

    
105
        }
106

    
107
        protected class MyCancel implements Cancellable, ICancellable {
108

    
109
                public boolean isCanceled() {
110
                        return false;
111
                }
112

    
113
                public void setCanceled(boolean canceled) {
114

    
115
                }
116

    
117
                /* if you don?t put and ID the wms donwload will be fail */
118
                public Object getID() {
119
                        return this;
120
                }
121

    
122
        }
123

    
124
        private Image getTileFromLayer(String fName, Rectangle2D tileExtent) {
125
                BufferedImage image;
126

    
127
                if (_dataType == Layer3DProps.layer3DImage) {
128
                        int size = getTileSize();
129
                        image = new BufferedImage(size, size, BufferedImage.TYPE_INT_ARGB);
130

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

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

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

    
155
                        // The layers of remote services don?t have transparency.
156
                        if (_layer.getClass().equals(FLyrVect.class)) {
157
                                // Generate layers 3D properties
158
                                Layer3DProps props3D = Layer3DProps.getLayer3DProps(_layer);
159
                                // Getting layer transparency
160
                                trans = ((FLyrDefault) _layer).getTransparency();
161
                                // Casting to float
162
                                float transFloat = (float) ((float) (255) / 255.0);
163
                                // Setting in the layers 3D props
164
                                props3D.setOpacity(transFloat);
165
                                // fix opaque the layer to save in disk
166
                                ((FLyrDefault) _layer).setTransparency(255);
167
                        }
168

    
169
                        try {
170
                                // disabling the iterator
171
//                                if (_layer instanceof FLyrVect) {
172
//                                        FLyrVect layerVect = (FLyrVect) _layer;
173
//                                        boolean strategyUse = layerVect.isUseStrategy();
174
//                                        layerVect.setUseStrategy(true);
175
//                                        layerVect.draw(image, graphics, vp, new MyCancel(), scale);
176
//                                        layerVect.setUseStrategy(strategyUse);
177
//                                } else
178
                                        _layer.draw(image, graphics, vp, new MyCancel(), scale);
179
                        } catch (ReadDriverException e) {
180
                                e.printStackTrace();
181
                        }
182
                        // Restoring the real value of transparency
183
                        if (_layer.getClass().equals(FLyrVect.class)) {
184
                                // fix the real transparency to the layer
185
                                ((FLyrDefault) _layer).setTransparency(trans);
186
                        }
187

    
188
                        try {
189
                                File file = new File(fName);
190
                                String format = getFileExtension().substring(1);
191
                                saveCachedFile(image, format, file);
192
                        } catch (IOException e1) {
193
                                e1.printStackTrace();
194
                                return null;
195
                        }
196
                        return image;
197
                } else { // Elevation data
198
                        // TODO Un toggle this comment to use WCS and WMS extensions.
199
                        if (_layer instanceof FLyrWMS) { // WMS Elevation
200
                                try {
201
                                        FLyrWMS wmsLayer = (FLyrWMS) _layer;
202
                                        if (_wmsDriver == null) {
203

    
204
                                                URL host = wmsLayer.getHost();
205
                                                _wmsDriver = FMapWMSDriverFactory
206
                                                                .getFMapDriverForURL(host);
207
                                        }
208
                                        if (_wmsStatus == null) {
209
                                                _wmsStatus = new WMSStatus();
210
                                                _wmsStatus.setFormat(wmsLayer.getFormat());
211
                                                _wmsStatus.setHeight(32);
212
                                                _wmsStatus.setWidth(32);
213
                                                _wmsStatus.setLayerNames(Utilities.createVector(
214
                                                                wmsLayer.getLayerQuery(), ","));
215
                                                _wmsStatus.setSrs(wmsLayer.getSRS());
216

    
217
                                                HashMap props = wmsLayer.getProperties();
218
                                                Vector styles;
219
                                                Vector dimensions;
220
                                                styles = (Vector) (props.get("styles"));
221
                                                dimensions = (Vector) (props.get("dimensions"));
222
                                                _wmsStatus.setStyles(styles);
223
                                                _wmsStatus.setDimensions(dimensions);
224
                                                _wmsStatus.setTransparency(false);
225
                                                String getMapStr = wmsLayer.getOnlineResource("GetMap");
226
                                                _wmsStatus.setOnlineResource(getMapStr);
227
                                        }
228
                                        _wmsStatus.setExtent(tileExtent);
229
                                        File f = _wmsDriver.getMap(_wmsStatus, new MyCancel());
230
                                        if (f == null)
231
                                                return null;
232
                                        FileUtils.copy(f, new File(fName));
233

    
234
                                        String fileName = f.getPath();
235
                                        // System.out.println(fileName);
236

    
237
                                } catch (Exception e) {
238
                                        return null;
239
                                }
240
                        } else if (_layer instanceof FLyrWCS) { // WCS Elevation
241
                                try {
242
                                        FLyrWCS wcsLayer = (FLyrWCS) _layer;
243
                                        if (_wcsDriver == null) {
244

    
245
                                                URL host = wcsLayer.getHost();
246
                                                _wcsDriver = FMapWCSDriverFactory
247
                                                                .getFMapDriverForURL(host);
248
                                        }
249
                                        if (_wcsStatus == null) {
250
                                                _wcsStatus = new WCSStatus();
251

    
252
                                                Hashtable props = wcsLayer.getProperties();
253

    
254
                                                _wcsStatus.setCoveraName((String) props.get("name"));
255
                                                _wcsStatus.setFormat((String) props.get("format"));
256
                                                _wcsStatus.setHeight(32);
257
                                                _wcsStatus.setWidth(32);
258
                                                _wcsStatus.setSrs((String) props.get("crs"));
259
                                                _wcsStatus.setParameters((String) props
260
                                                                .get("parameter"));
261
                                                String time = (String) props.get("time");
262
                                                if (time.length() > 0)
263
                                                        _wcsStatus.setTime(time);
264
                                                // _wcsStatus.setOnlineResource(wcsLayer.getOnlineResource("GetCoverage"));
265
                                        }
266
                                        _wcsStatus.setExtent(tileExtent);
267

    
268
                                        File f = _wcsDriver.getCoverage(_wcsStatus, new MyCancel());
269
                                        if (f == null)
270
                                                return null;
271

    
272
                                        FileUtils.copy(f, new File(fName));
273

    
274
                                        String fileName = f.getPath();
275
                                        // System.out.println(fileName);
276

    
277
                                } catch (Exception e) {
278
                                        return null;
279
                                }
280
                        }
281
                        // RASTER_GRID
282
                        if (_layer instanceof FLyrRasterSE) {
283
                                // Grid grid = null;
284
                                FLyrRasterSE rasterLayer = (FLyrRasterSE) _layer;
285

    
286
                                // grid = rasterLayer.getSource().getGrid();
287
                                // if (grid == null)
288
                                // return null;
289

    
290
                                BufferFactory bufferFactory = new BufferFactory(rasterLayer
291
                                                .getDataSource());
292

    
293
                                int bands[] = { 0 };
294
                                bufferFactory.setDrawableBands(bands);
295
                                //bufferFactory.setNoDataValue(0.0);
296
                                //((IBuffer) bufferFactory).setNoDataValue(0.0);
297
                                bufferFactory.setAdjustToExtent(false);
298

    
299
                                try {
300
                                        bufferFactory.setAreaOfInterest(tileExtent.getMinX(),
301
                                                        tileExtent.getMinY(), tileExtent.getWidth(),
302
                                                        tileExtent.getHeight(), 32, 32);
303
                                } catch (ArrayIndexOutOfBoundsException e1) {
304
                                        // TODO Auto-generated catch block
305
                                        e1.printStackTrace();
306
                                } catch (InvalidSetViewException e1) {
307
                                        // // TODO Auto-generated catch block
308
                                        e1.printStackTrace();
309
                                } catch (InterruptedException e1) {
310
                                        // // TODO Auto-generated catch block
311
                                        e1.printStackTrace();
312
                                } catch (RasterDriverException e) {
313
                                        // TODO Auto-generated catch block
314
                                        e.printStackTrace();
315
                                }
316

    
317
                                RasterBuffer raster = (RasterBuffer) bufferFactory
318
                                                .getRasterBuf();
319
                                raster.setNoDataValue(0.0);
320
                                try {
321
                                        // raster.save(fName, new Extent(tileExtent));
322
                                        WriterBufferServer bufferServer = new WriterBufferServer(
323
                                                        raster);
324
                                        Params p = GeoRasterWriter.getWriter(fName).getParams();
325
                                        AffineTransform at = new AffineTransform(tileExtent
326
                                                        .getWidth() / 32, 0, 0,
327
                                                        -(tileExtent.getHeight() / 32), tileExtent
328
                                                                        .getMinX(), tileExtent.getMaxY());
329
                                        GeoRasterWriter grw = GeoRasterWriter.getWriter(
330
                                                        bufferServer, fName, 1, at, raster.getWidth(),
331
                                                        raster.getHeight(), raster.getDataType(), p, null);
332
                                        // grw.setWkt(d.getWktProjection());
333
                                        try {
334
                                                grw.dataWrite();
335
                                        } catch (InterruptedException e) {
336
                                                // TODO Auto-generated catch block
337
                                                e.printStackTrace();
338
                                        }
339
                                        grw.writeClose();
340
                                } catch (NotSupportedExtensionException e) {
341
                                        e.printStackTrace();
342
                                } catch (RasterDriverException e) {
343
                                        e.printStackTrace();
344
                                } catch (IOException e) {
345
                                        e.printStackTrace();
346
                                }
347

    
348
                        }
349
                        return null; // no image
350
                }
351

    
352
        }
353

    
354
        public String getTileAsFName(TileNum t, Rectangle2D extent) throws CacheServiceException {
355
                if (_loadFailed) {
356
                        throw new CacheServiceException(new Exception());
357
                }
358
                String tileId = t.numToOpTileId();
359
                String fName = getTileFileName(t);
360
                // System.out.println("CMTexture2D: requestTexture: "+tileId);
361
                if (!new File(fName).exists()) {
362
                        Rectangle2D tileExtent = extent;
363
                        getTileFromLayer(fName, tileExtent);
364
                }
365
                return fName;
366
        }
367

    
368
        public Image getTileAsImage(TileNum tileNum, Rectangle2D extent) throws CacheServiceException {
369
                if ((tileNum.getLevel() == 5)) {// && (tileNum.getX())==29 &&
370
                                                                                // (tileNum.getY()==9)){
371
                        System.err.println("pillado");
372
                }
373

    
374
                if (_loadFailed) {
375
                        throw new CacheServiceException(new Exception());
376
                }
377
                String tileId = tileNum.numToOpTileId();
378
                String fName = getTileFileName(tileNum);
379
                // System.out.println("CMTexture2D: requestTexture: "+tileId);
380
                Rectangle2D tileExtent = extent;
381
                return getTileFromLayer(fName, tileExtent);
382
        }
383

    
384
        public boolean intersectsLayer(Rectangle2D extent) {
385
                Rectangle2D tileExtent = extent;
386

    
387
                return _lyrExtentRect.intersects(extent);
388
                // return Math.max(_lyrExtentRect.getMinX(),tileExtent.getMinX()) <=
389
                // Math.min(_lyrExtentRect.getMaxX(),tileExtent.getMaxX()) &&
390
                // Math.max(Math.min(_lyrExtentRect.getMinY(),_lyrExtentRect.getMaxY()),tileExtent.getMinY())
391
                // <=
392
                // Math.min(Math.max(_lyrExtentRect.getMinY(),_lyrExtentRect.getMaxY()),tileExtent.getMaxY());
393
        }
394

    
395
}