Statistics
| Revision:

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

History | View | Annotate | Download (17.8 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.ArrayList;
14
import java.util.HashMap;
15
import java.util.Hashtable;
16
import java.util.Iterator;
17
import java.util.PriorityQueue;
18
import java.util.Vector;
19

    
20
import javax.imageio.ImageIO;
21

    
22
import org.apache.log4j.Logger;
23
import org.cresques.cts.IProjection;
24
import org.gvsig.cacheservice.CacheService;
25
import org.gvsig.cacheservice.CacheServiceException;
26
import org.gvsig.cacheservice.RasterCacheService;
27
import org.gvsig.cacheservice.TileNum;
28
import org.gvsig.fmap.raster.layers.FLyrRasterSE;
29
import org.gvsig.gvsig3d.map3d.MapContext3D;
30
import org.gvsig.osgvp.terrain.Terrain;
31
import org.gvsig.raster.buffer.BufferFactory;
32
import org.gvsig.raster.buffer.RasterBuffer;
33
import org.gvsig.raster.buffer.WriterBufferServer;
34
import org.gvsig.raster.dataset.GeoRasterWriter;
35
import org.gvsig.raster.dataset.InvalidSetViewException;
36
import org.gvsig.raster.dataset.NotSupportedExtensionException;
37
import org.gvsig.raster.dataset.Params;
38
import org.gvsig.raster.dataset.io.RasterDriverException;
39
import org.gvsig.remoteClient.utils.Utilities;
40
import org.gvsig.remoteClient.wcs.WCSStatus;
41
import org.gvsig.remoteClient.wms.ICancellable;
42
import org.gvsig.remoteClient.wms.WMSStatus;
43

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

    
60
public class FLayerCacheService extends RasterCacheService {
61
        private boolean _loadFailed = true;
62
        private Rectangle2D _lyrExtentRect;
63
        private FLayer _layer;
64
        private IProjection _viewProj;
65
        private int _dataType;
66
        private Object cacheMutex = new Object();
67

    
68
        // WMS terrain
69
        FMapWMSDriver _wmsDriver = null;
70
        FMapWCSDriver _wcsDriver = null;
71

    
72
        WMSStatus _wmsStatus = null;
73
        WCSStatus _wcsStatus = null;
74
        
75
        Thread _requestThead = null;
76
        private RequestTileFromLayerThread _requestTileFromLayerThread;
77
        
78
        private static Logger _logger = Logger.getLogger(FLayerCacheService.class
79
                        .getName());
80

    
81
        public FLayerCacheService(Terrain terrain, String name, FLayer layer,
82
                        IProjection viewProj) {
83
                super(terrain.getTerrainName(), name);
84

    
85
                _layer = layer;
86
                _viewProj = viewProj;
87
                _loadFailed = true;
88

    
89
                int cacheType = CacheService.GLOBAL;
90
                if (terrain.getCoordinateSystemType() == Terrain.CoordinateSystemType.GEOCENTRIC)
91
                        cacheType += SPHERIC;
92
                else
93
                        cacheType += PLANE;
94
                setCacheType(cacheType);
95

    
96
                try {
97
                        _lyrExtentRect = _layer.getFullExtent();
98
                } catch (ExpansionFileReadException e) {
99
                        e.printStackTrace();
100
                } catch (ReadDriverException e) {
101
                        e.printStackTrace();
102
                }
103

    
104
                Layer3DProps props3D = Layer3DProps.getLayer3DProps(layer);
105
                _dataType = props3D.getType();
106
                if (_dataType == Layer3DProps.layer3DImage)
107
                        setFileExtension(".png");
108
                else
109
                        setFileExtension(".tif");
110

    
111
                _loadFailed = false;
112

    
113
        }
114

    
115
        protected class MyCancel implements Cancellable, ICancellable {
116

    
117
                public boolean isCanceled() {
118
                        return false;
119
                }
120

    
121
                public void setCanceled(boolean canceled) {
122

    
123
                }
124

    
125
                /* if you don�t put and ID the wms donwload will be fail */
126
                public Object getID() {
127
                        return this;
128
                }
129

    
130
        }
131
        
132
        protected class RequestTileFromLayerThread implements Runnable {
133
                
134
                private class Request implements Comparable<Request> {
135
                        public Request(long timeStamp, String fName, Rectangle2D tileExtent) {
136
                                _timeStamp = timeStamp;
137
                                _fName = fName;
138
                                _tileExtent = tileExtent;
139
                        }
140
                        long _timeStamp;
141
                        String _fName;
142
                        Rectangle2D _tileExtent;
143
                        public int compareTo(Request o) {
144
                                if(_timeStamp > o._timeStamp) return -1;
145
                                else if(_timeStamp < o._timeStamp) return 1;
146
                                else return 0;
147
                        }
148
                }
149
                
150
                private HashMap<String, Request> _requestList = new HashMap<String, Request>();
151
                private PriorityQueue<Request> _priorityRequestList = new PriorityQueue<Request>();
152
                private volatile boolean _done = false;
153
                
154
                protected synchronized void purgeOldRequests() {
155
                        int max_elements = 100;
156
                        if(_requestList.size() > max_elements) {
157
                                //_logger.debug("max elements reached from request list thread...");
158
                                Iterator<Request> iterator = _requestList.values().iterator();
159
                                ArrayList<Request> removeRequests = new ArrayList<Request>();
160
                                int index = 0;
161
                                while(iterator.hasNext()) {
162
                                        Request request = iterator.next();
163
                                        if(index>max_elements) {
164
                                                removeRequests.add(request);
165
                                        }
166
                                        index++;
167
                                }
168
                                for(int i=0;i<removeRequests.size();i++) {
169
                                        //_logger.debug("Removing old elements from request list thread...");
170
                                        _priorityRequestList.remove(removeRequests.get(i));
171
                                        _requestList.remove(removeRequests.get(i)._fName);
172
                                }
173
                        }
174
                }
175
                
176
                public synchronized void requestTile(String fName, Rectangle2D tileExtent) {
177
                        Request request = _requestList.get(fName);
178
                        if(request == null) {
179
                                Request req = new Request(System.currentTimeMillis(),fName,tileExtent);
180
                                _requestList.put(fName,req);
181
                                _priorityRequestList.add(req);
182
                        } else {
183
                                request._timeStamp = System.currentTimeMillis();
184
                        }
185
                        purgeOldRequests();
186
                }
187
                
188
                public void run() {
189
                        while(!_done) {
190
                                Request request = null;
191
                                synchronized (this) {
192
                                        request = _priorityRequestList.poll();
193
                                }
194
                                if(request != null) {
195
                                        _requestList.remove(request._fName);
196
                                        getTileFromLayer(request._fName, request._tileExtent);
197
                                }
198
                                else {
199
                                        try {
200
                                                Thread.sleep(1);
201
                                        } catch (InterruptedException e) {
202
                                                e.printStackTrace();
203
                                        }
204
                                }
205
                        }
206
                        
207
                }
208
                
209
                public void setDone(boolean value) { _done = value; }
210
                
211
        }
212
        
213
        private void requestTileFromLayer(String fName, Rectangle2D tileExtent) {
214
                if(_requestThead == null) {
215
                        _requestTileFromLayerThread = new RequestTileFromLayerThread();
216
                        _requestThead = new Thread(_requestTileFromLayerThread);
217
                        _requestThead.start();
218
                }
219
                _requestTileFromLayerThread.requestTile(fName, tileExtent);
220
                
221
        }
222

    
223
        private void getTileFromLayer(String fName, Rectangle2D tileExtent) {
224
                BufferedImage image;
225
                
226
                // Check if the file exists, because some files are in request queue
227
                // while the request is processing.
228
                boolean fileExists = new File(fName).exists();
229
                if(!fileExists)
230
                {
231
                        if (_dataType == Layer3DProps.layer3DImage) {
232
                                // Workaround to avoid too many System.gc() calls on layer.draw. which makes
233
                                // rendering stall when rendering the tile.
234
                                if (_layer instanceof FLyrWMS) { // WMS Elevation
235
                                        try {
236
                                                FLyrWMS wmsLayer = (FLyrWMS) _layer;
237
                                                if (_wmsDriver == null) {
238
        
239
                                                        URL host = wmsLayer.getHost();
240
                                                        _wmsDriver = FMapWMSDriverFactory
241
                                                                        .getFMapDriverForURL(host);
242
                                                }
243
                                                if (_wmsStatus == null) {
244
                                                        _wmsStatus = new WMSStatus();
245
                                                        _wmsStatus.setFormat(wmsLayer.getFormat());
246
                                                        _wmsStatus.setHeight(256);
247
                                                        _wmsStatus.setWidth(256);
248
                                                        _wmsStatus.setLayerNames(Utilities.createVector(
249
                                                                        wmsLayer.getLayerQuery(), ","));
250
                                                        _wmsStatus.setSrs(wmsLayer.getSRS());
251
        
252
                                                        HashMap props = wmsLayer.getProperties();
253
                                                        Vector styles;
254
                                                        Vector dimensions;
255
                                                        styles = (Vector) (props.get("styles"));
256
                                                        dimensions = (Vector) (props.get("dimensions"));
257
                                                        _wmsStatus.setStyles(styles);
258
                                                        _wmsStatus.setDimensions(dimensions);
259
                                                        _wmsStatus.setTransparency((Boolean)props.get("wmsTransparency"));
260
                                                        String getMapStr = wmsLayer.getOnlineResource("GetMap");
261
                                                        _wmsStatus.setOnlineResource(getMapStr);
262
                                                }
263
                                                _wmsStatus.setExtent(tileExtent);
264
                                                File f = _wmsDriver.getMap(_wmsStatus, new MyCancel());
265
                                                if (f == null)
266
                                                        return;
267
                                                synchronized (cacheMutex) {
268
                                                        FileUtils.copy(f, new File(fName));
269
                                                }
270
                                        } catch (Exception e) {
271
                                                return;
272
                                        }
273
                                } else {
274
                                        int size = getTileSize();
275
                                        image = new BufferedImage(size, size,
276
                                                        BufferedImage.TYPE_INT_ARGB);
277
        
278
                                        ViewPort vp = new ViewPort(_viewProj);
279
                                        vp.setExtent(tileExtent); // the real extent of the tile
280
                                        // NotificationManager.addInfo("Extent solicitado--> "
281
                                        // + tileExtent.getMinX() + " " + tileExtent.getMinY() + " "
282
                                        // + tileExtent.getMaxX() + " " + tileExtent.getMaxY());
283
                                        vp.setAdjustable(false);
284
                                        vp.setImageSize(new Dimension(size, size));
285
                                        // vp.setDist1pixel(1 / 150000);
286
                                        vp.setDist1pixel(0.000000001);
287
        
288
                                        Graphics2D graphics = image.createGraphics();
289
                                        Color backgColor = new Color(255, 255, 255, 0); // transparent
290
                                        // background
291
                                        graphics.setColor(backgColor);
292
                                        graphics.fillRect(0, 0, size, size);
293
        
294
                                        // CancelDraw canceldraw = new MapControl.CancelDraw();
295
                                        // canceldraw.setCanceled(false);
296
                                        // For now we'll use a trick to make sure the layer is drawn
297
                                        // even if it has scale dependency
298
                                        // In the future, a scale corresponding to the tile should be
299
                                        // passed
300
                                        double scale = 0.5 * (_layer.getMaxScale() + _layer
301
                                                        .getMinScale());
302
                                        int trans = 255;
303
        
304
                                        // The layers of remote services don�t have transparency.
305
                                        if (_layer.getClass().equals(FLyrVect.class)) {
306
                                                // Generate layers 3D properties
307
                                                Layer3DProps props3D = Layer3DProps.getLayer3DProps(_layer);
308
                                                // fix opaque the layer to save in disk
309
                                                ((FLyrDefault) _layer).setTransparency((int) (props3D
310
                                                                .getOpacity() * 255));
311
                                        
312
                                        }
313
                                        
314
                                        if (_layer instanceof FLyrRasterSE) {
315
                                                // fix opaque the layer to save in disk
316
                                                ((FLyrRasterSE) _layer).setTransparency(255);
317
                                        }
318
        
319
                                        try {
320
                                                _layer.draw(image, graphics, vp, new MyCancel(), scale);
321
                                                if ((_layer.getClass().equals(FLyrVect.class))&& ((FLyrVect)_layer).isLabeled())
322
                                                         ((FLyrVect)_layer).drawLabels(image, graphics, vp, new MyCancel(), scale,72);
323
                                                                                        
324
                                        } catch (Exception e) {
325
                                                return;
326
                                        }
327
                                        // Restoring the real value of transparency
328
                                        if (_layer.getClass().equals(FLyrVect.class)) {
329
                                                // fix the real transparency to the layer
330
                                                ((FLyrDefault) _layer).setTransparency(trans);
331
                                        }
332
        
333
                                        try {
334
                                                File file = new File(fName);
335
                                                String format = getFileExtension().substring(1);
336
                                                synchronized (cacheMutex) {
337
                                                        saveCachedFile(image, format, file);
338
                                                }
339
                                        } catch (IOException e1) {
340
                                                e1.printStackTrace();
341
                                                return;
342
                                        }
343
                                }
344
                        } else { // Elevation data
345
                                if (_layer instanceof FLyrWMS) { // WMS Elevation
346
                                        try {
347
                                                FLyrWMS wmsLayer = (FLyrWMS) _layer;
348
                                                if (_wmsDriver == null) {
349
        
350
                                                        URL host = wmsLayer.getHost();
351
                                                        _wmsDriver = FMapWMSDriverFactory
352
                                                                        .getFMapDriverForURL(host);
353
                                                }
354
                                                if (_wmsStatus == null) {
355
                                                        _wmsStatus = new WMSStatus();
356
                                                        _wmsStatus.setFormat(wmsLayer.getFormat());
357
                                                        _wmsStatus.setHeight(32);
358
                                                        _wmsStatus.setWidth(32);
359
                                                        _wmsStatus.setLayerNames(Utilities.createVector(
360
                                                                        wmsLayer.getLayerQuery(), ","));
361
                                                        _wmsStatus.setSrs(wmsLayer.getSRS());
362
        
363
                                                        HashMap props = wmsLayer.getProperties();
364
                                                        Vector styles;
365
                                                        Vector dimensions;
366
                                                        styles = (Vector) (props.get("styles"));
367
                                                        dimensions = (Vector) (props.get("dimensions"));
368
                                                        _wmsStatus.setStyles(styles);
369
                                                        _wmsStatus.setDimensions(dimensions);
370
                                                        _wmsStatus.setTransparency(false);
371
                                                        String getMapStr = wmsLayer.getOnlineResource("GetMap");
372
                                                        _wmsStatus.setOnlineResource(getMapStr);
373
                                                }
374
                                                _wmsStatus.setExtent(tileExtent);
375
                                                File f = _wmsDriver.getMap(_wmsStatus, new MyCancel());
376
                                                if (f == null)
377
                                                        return;
378
                                                synchronized (cacheMutex) {
379
                                                        FileUtils.copy(f, new File(fName));
380
                                                }
381
                                        } catch (Exception e) {
382
                                                return;
383
                                        }
384
                                } else if (_layer instanceof FLyrWCS) { // WCS Elevation
385
                                        try {
386
                                                FLyrWCS wcsLayer = (FLyrWCS) _layer;
387
                                                if (_wcsDriver == null) {
388
        
389
                                                        URL host = wcsLayer.getHost();
390
                                                        _wcsDriver = FMapWCSDriverFactory
391
                                                                        .getFMapDriverForURL(host);
392
                                                }
393
                                                if (_wcsStatus == null) {
394
                                                        _wcsStatus = new WCSStatus();
395
        
396
                                                        Hashtable props = wcsLayer.getProperties();
397
        
398
                                                        _wcsStatus.setCoveraName((String) props.get("name"));
399
                                                        _wcsStatus.setFormat((String) props.get("format"));
400
                                                        _wcsStatus.setHeight(32);
401
                                                        _wcsStatus.setWidth(32);
402
                                                        _wcsStatus.setSrs((String) props.get("crs"));
403
                                                        _wcsStatus.setParameters((String) props
404
                                                                        .get("parameter"));
405
                                                        String time = (String) props.get("time");
406
                                                        if (time.length() > 0)
407
                                                                _wcsStatus.setTime(time);
408
                                                }
409
                                                _wcsStatus.setExtent(tileExtent);
410
        
411
                                                File f = _wcsDriver.getCoverage(_wcsStatus, new MyCancel());
412
                                                if (f == null)
413
                                                        return;
414
                                                synchronized (cacheMutex) {
415
                                                        FileUtils.copy(f, new File(fName));
416
                                                }
417
                                        } catch (Exception e) {
418
                                                return;
419
                                        }
420
                                } else
421
                                // RASTER_GRID
422
                                if (_layer instanceof FLyrRasterSE) {
423
                                        
424
                                        FLyrRasterSE rasterLayer = (FLyrRasterSE) _layer;
425
                                        BufferFactory bufferFactory = rasterLayer.getBufferFactory();
426
                                        
427
                                        //int bands[] = { 0 };
428
                                        bufferFactory.setAllDrawableBands();
429
                                        bufferFactory.setAdjustToExtent(false);
430
                                        bufferFactory.setNoDataToFill(0.0);
431
                                        
432
                                        // The y max and min are mixed because there are some problems 
433
                                        // computing the adjustedDataExtent.
434
                                        try {
435
                                                bufferFactory.setAreaOfInterest(tileExtent.getMinX(),
436
                                                                tileExtent.getMaxY(), tileExtent.getMaxX(),
437
                                                                tileExtent.getMinY(), 32, 32);
438
                                        } catch (RasterDriverException e) {
439
                                                // TODO Auto-generated catch block
440
                                                e.printStackTrace();
441
                                        } catch (InvalidSetViewException e) {
442
                                                // TODO Auto-generated catch block
443
                                                e.printStackTrace();
444
                                        } catch (InterruptedException e) {
445
                                                // TODO Auto-generated catch block
446
                                                e.printStackTrace();
447
                                        } catch (Exception e) {
448
                                                return;
449
                                        }
450
        
451
                                        RasterBuffer raster = (RasterBuffer) bufferFactory
452
                                                        .getRasterBuf();
453
                                        raster.setNoDataValue(0.0);
454
                                        raster.setNotValidValue(0.0);
455
                                        try {
456
                                                WriterBufferServer bufferServer = new WriterBufferServer(
457
                                                                raster);
458
                                                Params p = GeoRasterWriter.getWriter(fName).getParams();
459
                                                AffineTransform at = new AffineTransform(tileExtent
460
                                                                .getWidth() / 32, 0, 0,
461
                                                                -(tileExtent.getHeight() / 32), tileExtent
462
                                                                                .getMinX(), tileExtent.getMaxY());
463
                                                GeoRasterWriter grw = GeoRasterWriter.getWriter(
464
                                                                bufferServer, fName, 1, at, raster.getWidth(),
465
                                                                raster.getHeight(), raster.getDataType(), p, null);
466
                                                synchronized (cacheMutex) {
467
                                                        try {
468
                                                                grw.dataWrite();
469
                                                        } catch (InterruptedException e) {
470
                                                                e.printStackTrace();
471
                                                        }
472
                                                        grw.writeClose();
473
                                                }
474
        
475
                                        } catch (NotSupportedExtensionException e) {
476
                                                // e.printStackTrace();
477
                                                return;
478
                                        } catch (RasterDriverException e) {
479
                                                // e.printStackTrace();
480
                                                return;
481
                                        } catch (IOException e) {
482
                                                // e.printStackTrace();
483
                                                return;
484
                                        }
485
        
486
                                }
487
                        }
488
                }
489
                //The method cant return a image because ImageIO crashes with tiff images.
490
                /*try {
491
                        synchronized (cacheMutex) {
492
                                image = ImageIO.read(new File(fName));
493
                        }
494
                } catch (IOException e) {
495
                        return null;
496
                }
497
                return image;*/
498
        }
499

    
500
        public String getTileAsFName(TileNum t, Rectangle2D extent)
501
                        throws CacheServiceException {
502
                if (_loadFailed) {
503
                        throw new CacheServiceException(new Exception());
504
                }
505

    
506
                String fName = getTileFileName(t);
507
                boolean fileExists = false;
508
                synchronized (cacheMutex) {
509
                        fileExists = new File(fName).exists();
510
                }
511
                if (!fileExists) {
512
                        Rectangle2D tileExtent = extent;
513
                        requestTileFromLayer(fName,tileExtent);
514
                        //if (getTileFromLayer(fName, tileExtent) == null)
515
                        fName = null;
516
                }
517
                return fName;
518
        }
519

    
520
        /**
521
         * @TODO: This method crashes with tiff images use getTileAsFileName instead
522
         */
523
        public Image getTileAsImage(TileNum tileNum, Rectangle2D extent)
524
                        throws CacheServiceException {
525

    
526
                if (_loadFailed) {
527
                        throw new CacheServiceException(new Exception());
528
                }
529
                String fName = getTileFileName(tileNum);
530
                Image image = null;
531
                boolean fileExists = false;
532
                synchronized (cacheMutex) {
533
                        fileExists = new File(fName).exists();
534
                }
535
                if (!fileExists) {
536
                        Rectangle2D tileExtent = extent;
537
                        requestTileFromLayer(fName,tileExtent);
538
                        //image = getTileFromLayer(fName, tileExtent);
539
                } else {
540
                        //ImageIO cant read tiff images. Change in the next version.
541
                        /*try {
542
                                //image = ImageIO.read(new File(fName));
543
                        } catch (IOException e) {
544
                                return null;
545
                        }*/
546
                }
547
                return image;
548
        }
549

    
550
        public boolean intersectsLayer(Rectangle2D extent) {
551
                return _lyrExtentRect.intersects(extent);
552
        }
553
        
554
        public void free() throws Throwable {
555
                _logger.info("finalizing cache service and thread");
556
                _requestTileFromLayerThread.setDone(true);
557
                _requestThead.join();
558
        }
559
        
560
        /*@Override
561
        protected void finalize() throws Throwable {
562
                _logger.info("finalizing cache service and thread");
563
                _requestTileFromLayerThread.setDone(true);
564
                _requestThead.join();
565
                super.finalize();
566
        }*/
567

    
568
}