/*
 * ArcGIS Engine Developer Sample
 * Application Name: AttributeQuery.java
 */
package com.esri.arcgis.samples.beans.mapcontrol;

import java.awt.BorderLayout;
import java.awt.event.ActionEvent;
import java.awt.event.ActionListener;
import java.io.File;
import java.io.IOException;

import javax.swing.BorderFactory;
import javax.swing.Box;
import javax.swing.BoxLayout;
import javax.swing.JButton;
import javax.swing.JFileChooser;
import javax.swing.JFrame;
import javax.swing.JLabel;
import javax.swing.JOptionPane;
import javax.swing.JPanel;
import javax.swing.JTextArea;
import javax.swing.JTextField;
import javax.swing.UIManager;
import javax.swing.filechooser.FileFilter;

import com.esri.arcgis.beans.map.IMapControlEvents2Adapter;
import com.esri.arcgis.beans.map.IMapControlEvents2OnMouseDownEvent;
import com.esri.arcgis.beans.map.MapBean;
import com.esri.arcgis.carto.FeatureLayer;
import com.esri.arcgis.carto.IActiveView;
import com.esri.arcgis.carto.ILayer;
import com.esri.arcgis.carto.esriSelectionResultEnum;
import com.esri.arcgis.carto.esriViewDrawPhase;
import com.esri.arcgis.geodatabase.QueryFilter;
import com.esri.arcgis.system.AoInitialize;
import com.esri.arcgis.system.EngineInitializer;
import com.esri.arcgis.system.esriLicenseProductCode;



/**
* This sample illustrates how to programatically select features on
* the Map Control using a QueryFilter.
*
*/

public class AttributeQuery extends JFrame implements ActionListener{

    JPanel topPanel = null;
    JPanel mainPanel = null;
    JPanel rightPanel = null;
    JPanel bottomPanel = null;
    JButton addLayerButton = null;
    JLabel queryTextLabel = null;
    JTextField queryTextField = null;
    JButton selectButton = null;
    MapBean mapBean = null;
    JTextArea help = null;
    String helpString ="1.Add a layer using addLayer button."+"\n"+"2. Enter the where clause in the text"+ "\n"+"field provided"+"\n"+"For ex : If you have added States layer,"+"\n"+"then your where clause to select California\n"+ "state will be STATE_NAME='California'.\n" +
  "Use the left hand mouse button to zoom in."+"\n"+"Use the Right mouse button to pan."+"\n";

    public AttributeQuery() {
     super("Attribute Query");
     buildFrame();
     setSize(600, 500);
     setVisible(true);
     initControl();
    }

    /**This method builds 'this' frame as per the following diagram:
    *
    *   /----------------------------------------------------------------------------------
    *   |              BorderLayout.NORTH                                                 |
    *   |                                                                                 |
    *   |  JButton : AddLayerButton                                                       |
    *   |                                                                                 |
    *   |---------------------------------------------------------------------------------|
    *   |                                              |                                  |
    *   |                                              |      BorderLayer:EAST            |
    *   |                    MapBean                |      JTextArea                      |
    *   |                   BorderLayout.CENTER        |                                  |
    *   |                                              |                                  |
    *   |                                              |                                  |
    *   |                                              |                                  |
    *   |                                              |                                  |
    *   |----------------------------------------------|----------------------------------|
    *   |                                                                                 |
    *   |  BorderLayout : SOUTH                                                           |
    *   |  JLabel                    JTextField                     JButton               |
    *   |                                                                                 |
    *   |                                                                                 |
    *   \---------------------------------------------------------------------------------/
    */


   public void buildFrame() {
      topPanel = new JPanel();
      mainPanel = new JPanel();
      rightPanel = new JPanel();
      bottomPanel = new JPanel();

      topPanel.setLayout(new BoxLayout(topPanel, BoxLayout.X_AXIS));
      mainPanel.setLayout(new BorderLayout(10, 10));
      bottomPanel.setLayout(new BoxLayout(bottomPanel, BoxLayout.X_AXIS));

      addLayerButton = new JButton("Add Data Layer");
      addLayerButton.addActionListener(this);
      topPanel.add(addLayerButton);

      queryTextLabel = new JLabel("Where Clause");
      queryTextField = new JTextField();
      selectButton = new JButton("Select Features");
      selectButton.addActionListener(this);
      bottomPanel.add(queryTextLabel);
      bottomPanel.add(Box.createHorizontalStrut(5));
      bottomPanel.add(queryTextField);
      bottomPanel.add(Box.createHorizontalStrut(5));
      bottomPanel.add(selectButton);

      help = new JTextArea();
      help.setText(helpString);
      rightPanel.setLayout(new BoxLayout(rightPanel, BoxLayout.Y_AXIS));
      rightPanel.add(help);
      rightPanel.add(Box.createVerticalGlue());
      rightPanel.setBorder(BorderFactory.createEmptyBorder(5, 5, 5, 5));

      //Create map control add it to the center of the main panel.
      mapBean = new MapBean();
      mainPanel.add(topPanel, BorderLayout.NORTH);
      mainPanel.add(mapBean, BorderLayout.CENTER);
      mainPanel.add(rightPanel, BorderLayout.EAST);
      mainPanel.add(bottomPanel, BorderLayout.SOUTH);
      mainPanel.setBorder(BorderFactory.createEmptyBorder(10, 10, 10, 10));

      getContentPane().add(mainPanel, BorderLayout.CENTER);
   }

   /** Initilizes control
    *  creates and intializes member variables
    * */


   public void initControl() {
    try {
       mapBean.addIMapControlEvents2Listener(new MapControlListener());
    }
    catch (IOException ex) {
       System.out.println("Exception in initControl :" + ex);
       ex.printStackTrace();
    }


   }

   /**@see java.awt.event.ActionListener#actionPerformed(ActionEvent event)
      * @param event
      */

     public void actionPerformed(ActionEvent event) {
        if(event.getSource() == addLayerButton) {
           try {
               if (!loadFile()) return;

            }
            catch (IOException ex) {
               System.out.println(
                   "Exception in addLayerButton#actionPerformed" + ex);
               ex.printStackTrace();
            }
        }

        if(event.getSource() == selectButton) {
            processQuery();
        }


     }

     /** To perform attribute query based on the user input
      *
      */

     void processQuery() {
      try {
         // Make sure that a layer has been added
         int layerCount = mapBean.getLayerCount();
         if (layerCount < 0) {
            JOptionPane.showMessageDialog(this,
                                          "Please add a feature layer first");
         }

         ILayer layer = null;
         FeatureLayer featureLayer = null;

         try {
            // Just use 1st layer, but make sure it's a featurelayer
            layer = mapBean.getLayer(0);
            featureLayer = new FeatureLayer(layer);
            String queryString = queryTextField.getText();
            if (queryString != null || queryString.length() > 0) {
               // Do a query
               QueryFilter queryFilter = new QueryFilter();
               queryFilter.setWhereClause(queryString);
               featureLayer.selectFeatures(queryFilter,
                                           esriSelectionResultEnum.esriSelectionResultNew, false);
               IActiveView activeView = mapBean.getActiveView();
               activeView.partialRefresh(esriViewDrawPhase.esriViewGeoSelection,
                                         layer, null);
            }

         }   catch (IOException ex) {
               JOptionPane.showMessageDialog(this,
                                       "Please add a feature layer first");
         }

      }
      catch (IOException ex1) {
         ex1.printStackTrace();
      }

     }




   /**
      * Method loadFile loads the specified mxd file
      *
      */
    public boolean loadFile() throws IOException {
         //Open a JFileChooser for selecting PMF documents
       JFileChooser chooser = new JFileChooser();
       chooser.setFileFilter(new FileFilter() {
          public boolean accept(File f) {
             return (f.isDirectory() ||
                     f.getAbsolutePath().toUpperCase().endsWith(".LYR"));
          }

          public String getDescription() {
             return "Layer Documents(*.lyr)";
          }
       });

       boolean loaded = false;
       int returnVal = 0;
       returnVal = chooser.showOpenDialog(null);
       if (returnVal == JFileChooser.APPROVE_OPTION) {
            String fileChosen =
                chooser.getCurrentDirectory()
                    + File.separator
                    + chooser.getSelectedFile().getName();
            System.out.println("File picked: [" + fileChosen + "]");
            //check if the selected layer document can be loaded into MapBean
            try {
               mapBean.addLayerFromFile(fileChosen, 0);
            } catch(Exception ex) {

            }
            loaded = true;
            System.out.println("Layer Added");
       }
       return loaded;
    }

    /**
   * Description: Class which extends map control event class IMapControlEvents2Adapter
   * @see com.esri.arcgis.beans.map.IMapControlEvents2Adapter
   * */

  class MapControlListener
      extends IMapControlEvents2Adapter {

     /**
      * @see com.esri.arcgis.beans.map.IMapControlEvents2Adapter#onMouseDown(IMapControlEvents2OnMouseDownEvent theEvent)
      * @param theEvent
      */

     public void onMouseDown(IMapControlEvents2OnMouseDownEvent theEvent) {
        try {
           //If left mouse button then zoom in
           if (theEvent.getButton() == 1) {
              mapBean.setExtent(mapBean.trackRectangle());
           }
           else if(theEvent.getButton() == 2) {
              mapBean.pan();
           }

        }
        catch (Exception ex) {
           System.out.println("Exception in MapControlListener#onMouseDown : " +
                              ex);
           ex.printStackTrace();
        }

     }

  } //End of MapControlListener class


  /**
    * Main program to start the program execution.
    *
    * @param s
    */


   public static void main(String s[]) {
      try {
         //Set the system look and feel
         UIManager.setLookAndFeel(UIManager.getSystemLookAndFeelClassName());
         EngineInitializer.initializeVisualBeans();

         AoInitialize aoInit = new AoInitialize();
         aoInit.initialize(esriLicenseProductCode.esriLicenseProductCodeEngine);

         AttributeQuery attributeQuery = new AttributeQuery();
         attributeQuery.setDefaultCloseOperation(AttributeQuery.EXIT_ON_CLOSE);
      }
      catch (Exception ex) {
         ex.printStackTrace();
      }
   }



}