001    //$HeadURL: svn+ssh://jwilden@svn.wald.intevation.org/deegree/base/branches/2.5_testing/src/org/deegree/tools/shape/IndexFrame.java $
002    /*----------------------------------------
003     Coyright 2003 IDgis BV
004    
005     This library is free software; you can redistribute it and/or
006     modify it under the terms of the GNU Lesser General Public
007     License as published by the Free Software Foundation; either
008     version 2.1 of the License, or (at your option) any later version.
009    
010     This library is distributed in the hope that it will be useful,
011     but WITHOUT ANY WARRANTY; without even the implied warranty of
012     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
013     Lesser General Public License for more details.
014    
015     You should have received a copy of the GNU Lesser General Public
016     License along with this library; if not, write to the Free Software
017     Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
018     ----------------------------------------*/
019    
020    package org.deegree.tools.shape;
021    
022    import java.awt.BorderLayout;
023    import java.awt.Container;
024    import java.awt.GridLayout;
025    import java.awt.event.ActionEvent;
026    import java.awt.event.ActionListener;
027    import java.io.File;
028    
029    import javax.swing.JButton;
030    import javax.swing.JCheckBox;
031    import javax.swing.JFrame;
032    import javax.swing.JLabel;
033    import javax.swing.JOptionPane;
034    import javax.swing.JPanel;
035    import javax.swing.JProgressBar;
036    import javax.swing.event.ChangeEvent;
037    import javax.swing.event.ChangeListener;
038    
039    import org.deegree.datatypes.QualifiedName;
040    import org.deegree.framework.log.ILogger;
041    import org.deegree.framework.log.LoggerFactory;
042    import org.deegree.io.dbaseapi.DBaseIndex;
043    import org.deegree.io.rtree.HyperBoundingBox;
044    import org.deegree.io.rtree.HyperPoint;
045    import org.deegree.io.rtree.RTree;
046    import org.deegree.io.shpapi.ShapeFile;
047    import org.deegree.model.feature.Feature;
048    import org.deegree.model.spatialschema.Envelope;
049    import org.deegree.model.spatialschema.Geometry;
050    import org.deegree.model.spatialschema.GeometryFactory;
051    import org.deegree.model.spatialschema.Point;
052    
053    public class IndexFrame extends JFrame implements ActionListener, ChangeListener {
054    
055        private static final ILogger LOG = LoggerFactory.getLogger( IndexFrame.class );
056    
057        protected JCheckBox geometryCheckBox;
058    
059        protected JCheckBox[] checkboxes;
060    
061        protected ShapeFile shapeFile;
062    
063        protected String[] properties;
064    
065        protected JButton cancel;
066    
067        protected Indexing indexing = null;
068    
069        protected String fileName;
070    
071        protected boolean hasGeometry;
072    
073        protected boolean[] hasIndex;
074    
075        protected JCheckBox[] uniqueBoxes;
076    
077        class Indexing extends Thread {
078            boolean stop = false;
079    
080            IndexFrame frame;
081    
082            Indexing( IndexFrame frame ) {
083                this.frame = frame;
084            }
085    
086            public void run() {
087                Container container = getContentPane();
088                container.removeAll();
089                container.setLayout( new GridLayout( 3, 1 ) );
090    
091                int features = shapeFile.getRecordNum();
092                container.add( new JLabel( "Indexing..." ) );
093                JProgressBar progressBar = new JProgressBar( 1, features );
094                progressBar.setStringPainted( true );
095                container.add( progressBar );
096                cancel = new JButton( "Cancel" );
097                cancel.addActionListener( frame );
098                container.add( cancel );
099                pack();
100    
101                boolean geometry = false;
102                DBaseIndex[] index = new DBaseIndex[properties.length];
103                RTree rtree = null;
104    
105                try {
106                    String[] dataTypes = shapeFile.getDataTypes();
107                    int[] lengths = shapeFile.getDataLengths();
108    
109                    if ( geometryCheckBox.isSelected() && !hasGeometry ) {
110                        geometry = true;
111                        rtree = new RTree( 2, 11, fileName + ".rti" );
112                    }
113    
114                    boolean indexes = false;
115                    for ( int i = 0; i < index.length; i++ ) {
116                        if ( checkboxes[i].isSelected() && !hasIndex[i] ) {
117                            index[i] = DBaseIndex.createIndex(
118                                                               fileName + "$" + properties[i],
119                                                               properties[i],
120                                                               lengths[i],
121                                                               uniqueBoxes[i].isSelected(),
122                                                               ( dataTypes[i].equalsIgnoreCase( "N" )
123                                                                 || dataTypes[i].equalsIgnoreCase( "I" ) || dataTypes[i].equalsIgnoreCase( "F" ) ) );
124                            indexes = true;
125                        } else
126                            index[i] = null;
127                    }
128    
129                    if ( geometry || indexes ) {
130                        for ( int i = 1; i < features + 1; i++ ) {
131                            Feature feature = shapeFile.getFeatureByRecNo( i );
132    
133                            if ( geometry ) {
134                                Geometry[] geometries = feature.getGeometryPropertyValues();
135                                if ( geometries.length == 0 ) {
136                                    LOG.logInfo( "no geometries at recno" + i );
137                                    continue;
138                                }
139                                Envelope envelope = null;
140                                // TODO: deal with more than one geometry; handle geometry=null (allowed
141                                // in shapefile)
142                                envelope = ( feature.getDefaultGeometryPropertyValue() ).getEnvelope();
143                                if ( envelope == null ) { // assume a Point-geometry
144                                    Point pnt = (Point) geometries[0];
145                                    envelope = GeometryFactory.createEnvelope( pnt.getX(), pnt.getY(), pnt.getX(),
146                                                                               pnt.getY(), null );
147                                }
148                                HyperBoundingBox box = new HyperBoundingBox(
149                                                                             new HyperPoint( envelope.getMin().getAsArray() ),
150                                                                             new HyperPoint( envelope.getMax().getAsArray() ) );
151                                rtree.insert( new Integer( i ), box );
152                            }
153    
154                            for ( int j = 0; j < index.length; j++ ) {
155                                if ( index[j] != null ) {
156                                    QualifiedName qn = new QualifiedName( properties[j] );
157                                    index[j].addKey( (Comparable) feature.getDefaultProperty( qn ), i );
158                                }
159                            }
160    
161                            progressBar.setValue( i );
162    
163                            synchronized ( this ) {
164                                if ( stop ) {
165                                    shapeFile.close();
166                                    if ( geometry ) {
167                                        rtree.close();
168                                        new File( fileName + ".rti" ).delete();
169                                    }
170                                    for ( int j = 0; j < index.length; j++ ) {
171                                        if ( index[j] != null ) {
172                                            index[j].close();
173                                            new File( fileName + "$" + properties[j] + ".ndx" ).delete();
174                                        }
175                                    }
176                                    System.exit( 3 );
177                                }
178                            }
179                        }
180                    }
181    
182                    try {
183                        if ( geometry ) {
184                            rtree.close();
185                        }
186                        shapeFile.close();
187    
188                        for ( int i = 0; i < index.length; i++ )
189                            if ( index[i] != null )
190                                index[i].close();
191                    } catch ( Exception e ) {
192                        e.printStackTrace();
193                        JOptionPane.showMessageDialog( frame, e );
194                        System.exit( 1 );
195                    }
196    
197                    if ( !geometryCheckBox.isSelected() && hasGeometry ) {
198                        new File( fileName + ".rti" ).delete();
199                    }
200    
201                    for ( int i = 0; i < index.length; i++ ) {
202                        if ( !checkboxes[i].isSelected() && hasIndex[i] ) {
203                            new File( fileName + "$" + properties[i] + ".ndx" ).delete();
204                        }
205                    }
206    
207                    System.exit( 0 );
208                } catch ( Exception ex ) {
209                    ex.printStackTrace();
210                    JOptionPane.showMessageDialog( frame, ex );
211                    System.exit( 1 );
212                }
213            }
214    
215            public void stopIndexing() {
216                synchronized ( this ) {
217                    stop = true;
218                }
219            }
220        }
221    
222        public IndexFrame( File file ) throws Exception {
223            super( "Deegree" );
224            fileName = file.getPath();
225            fileName = fileName.substring( 0, fileName.length() - 4 );
226            shapeFile = new ShapeFile( fileName );
227            properties = shapeFile.getProperties();
228    
229            Container container = getContentPane();
230            container.setLayout( new BorderLayout() );
231            int size = properties.length;
232            JPanel panel = new JPanel( new GridLayout( 10 + 2, 2 ) );
233    
234            panel.add( new JLabel( "Attributes" ) );
235            panel.add( new JLabel( "Unique" ) );
236    
237            hasGeometry = shapeFile.hasRTreeIndex();
238            hasIndex = new boolean[size];
239            for ( int i = 0; i < size; i++ )
240                hasIndex[i] = shapeFile.hasDBaseIndex( properties[i] );
241    
242            checkboxes = new JCheckBox[size];
243            uniqueBoxes = new JCheckBox[size];
244    
245            geometryCheckBox = new JCheckBox( "Geometry" );
246            geometryCheckBox.setSelected( hasGeometry );
247    
248            panel.add( geometryCheckBox );
249            panel.add( new JCheckBox( "" ) ).setEnabled( false );
250    
251            for ( int i = 0; i < checkboxes.length; i++ ) {
252                checkboxes[i] = new JCheckBox( properties[i] );
253                checkboxes[i].setSelected( hasIndex[i] );
254                if ( i < 10 ) {
255                    panel.add( checkboxes[i] );
256    
257                    uniqueBoxes[i] = (JCheckBox) panel.add( new JCheckBox( "" ) );
258                    if ( hasIndex[i] ) {
259                        uniqueBoxes[i].setSelected( shapeFile.isUnique( properties[i] ) );
260                    } else {
261                        checkboxes[i].addChangeListener( this );
262                    }
263                    uniqueBoxes[i].setEnabled( false );
264                }
265            }
266    
267            JButton start = new JButton( "Start indexing" );
268            start.addActionListener( this );
269            container.add( "South", start );
270            container.add( "Center", panel );
271    
272            setBounds( 100, 100, 100, 100 );
273            pack();
274    
275            setResizable( true );
276        }
277    
278        /**
279         * @param e
280         */
281        public void actionPerformed( ActionEvent e ) {
282            if ( e.getSource() == cancel ) {
283                if ( indexing != null ) {
284                    if ( JOptionPane.showConfirmDialog( this, "Cancel indexing?", "Question", JOptionPane.YES_NO_OPTION ) == JOptionPane.YES_OPTION )
285                        indexing.stopIndexing();
286                }
287            } else {
288                if ( hasGeometry != geometryCheckBox.isSelected() ) {
289                    indexing = new Indexing( this );
290                    indexing.start();
291                    return;
292                }
293    
294                for ( int i = 0; i < hasIndex.length; i++ ) {
295                    if ( hasIndex[i] != checkboxes[i].isSelected() ) {
296                        indexing = new Indexing( this );
297                        indexing.start();
298                        return;
299                    }
300                }
301    
302                JOptionPane.showMessageDialog( this, "Nothing changed" );
303            }
304        }
305    
306        /**
307         * @param e
308         */
309        public void stateChanged( ChangeEvent e ) {
310            JCheckBox checkbox = (JCheckBox) e.getSource();
311    
312            for ( int i = 0; i < checkboxes.length; i++ ) {
313                if ( checkbox == checkboxes[i] ) {
314                    uniqueBoxes[i].setEnabled( checkbox.isSelected() );
315                    if ( !checkbox.isSelected() ) {
316                        uniqueBoxes[i].setSelected( false );
317                    }
318                    break;
319                }
320            }
321        }
322    }