net.sourceforge.squirrel_sql.plugins.graph.TableFramesModel Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of graph Show documentation
Graph Plugin
The newest version!
package net.sourceforge.squirrel_sql.plugins.graph;

import javax.swing.*;
import java.awt.*;
import java.awt.event.ActionEvent;
import java.awt.event.ActionListener;
import java.awt.event.ComponentAdapter;
import java.awt.event.ComponentEvent;
import java.util.ArrayList;
import java.util.Vector;

public class TableFramesModel
{
   private Vector _openTableFrameCtrls = new Vector();

   private TableFrameControllerListener _tableFrameControllerListener;
   private ConstraintViewsModelListener _constraintViewsModelListener;

   private ArrayList _listeners = new ArrayList();
   private ColumnInfoModelListener _columnInfoModelListener;

   private TableFrameController _lastRoot;
   private Timer _calcQueryRootTimer;

   public TableFramesModel()
   {
      _tableFrameControllerListener = new TableFrameControllerListener()
      {
         public void closed(TableFrameController tfc)
         {
            onTableFrameControllerClosed(tfc);
         }
      };

      _constraintViewsModelListener = new ConstraintViewsModelListener()
      {
         @Override
         public void constraintsChanged()
         {
            fireListeners(TableFramesModelChangeType.CONSTRAINT);
         }
      };

      _columnInfoModelListener = new ColumnInfoModelListener()
      {
         @Override
         public void columnInfosChanged(TableFramesModelChangeType changeType)
         {
            fireListeners(changeType);
         }
      };

      _calcQueryRootTimer = new Timer(300, new ActionListener()
      {
         @Override
         public void actionPerformed(ActionEvent e)
         {
            calculateQueryRoot();
         }
      });

      _calcQueryRootTimer.setRepeats(false);
      _calcQueryRootTimer.stop();

   }

   public Vector getTblCtrls()
   {
      return _openTableFrameCtrls;
   }

   void refreshTableNames()
   {
      for (TableFrameController openTableFrameCtrl : _openTableFrameCtrls)
      {
         openTableFrameCtrl.refreshTableName();
      }
   }

   void allTablesDbOrder()
   {
      for (TableFrameController openTableFrameCtrl : _openTableFrameCtrls)
      {
         openTableFrameCtrl.dbOrder();
      }
   }

   void allTablesByNameOrder()
   {
      for (TableFrameController openTableFrameCtrl : _openTableFrameCtrls)
      {
         openTableFrameCtrl.nameOrder();
      }
   }

   void allTablesPkConstOrder()
   {
      for (TableFrameController openTableFrameCtrl : _openTableFrameCtrls)
      {
         openTableFrameCtrl.pkConstraintOrder();
      }
   }

   void allTablesFilteredSelectedOrder()
   {
      for (TableFrameController openTableFrameCtrl : _openTableFrameCtrls)
      {
         openTableFrameCtrl.filteredSelectedOrder();
      }
   }

   private void onTableFrameControllerClosed(TableFrameController tfc)
   {
      _openTableFrameCtrls.remove(tfc);

      for (int i = 0; i < _openTableFrameCtrls.size(); i++)
      {
         TableFrameController tableFrameController = _openTableFrameCtrls.get(i);
         tableFrameController.disconnectTableFrame(tfc);
      }

      fireListeners(TableFramesModelChangeType.TABLE);
      recalculateAllConnections();
   }

   private void fireListeners(TableFramesModelChangeType changeType)
   {
      TableFramesModelListener[] listeners = _listeners.toArray(new TableFramesModelListener[_listeners.size()]);

      for (TableFramesModelListener listener : listeners)
      {
         listener.modelChanged(changeType);
      }
   }


   void refreshAllTables()
   {
      for (int i = 0; i < _openTableFrameCtrls.size(); i++)
      {
         TableFrameController tableFrameController = _openTableFrameCtrls.get(i);
         tableFrameController.refresh();
      }

      fireListeners(TableFramesModelChangeType.TABLE);
      recalculateAllConnections();
   }


   public boolean containsTable(TableFrameController tfc)
   {
      return _openTableFrameCtrls.contains(tfc);
   }

   public boolean containsTable(String tableName)
   {
      for (TableFrameController openTableFrameCtrl : _openTableFrameCtrls)
      {
         if(openTableFrameCtrl.getTableInfo().getSimpleName().equalsIgnoreCase(tableName))
         {
            return true;
         }
      }
      return false;
   }

   public void addTable(TableFrameController tfc, boolean readingXMLBean)
   {
      tfc.addTableFrameControllerListener(_tableFrameControllerListener);
      tfc.getConstraintViewsModel().addListener(_constraintViewsModelListener);
      tfc.getColumnInfoModel().addColumnInfoModelListener(_columnInfoModelListener);

      tfc.getFrame().addComponentListener(new ComponentAdapter()
      {
         @Override
         public void componentResized(ComponentEvent e)
         {
            onTableFrameMoved();
         }

         @Override
         public void componentMoved(ComponentEvent e)
         {
            onTableFrameMoved();
         }
      });


      _openTableFrameCtrls.add(tfc);

      initsAfterFrameAdded(tfc, false == readingXMLBean);
      if (false == readingXMLBean)
      {
         fireListeners(TableFramesModelChangeType.TABLE);
      }
      recalculateAllConnections();
   }

   private void initsAfterFrameAdded(TableFrameController tfc, boolean resetBounds)
   {

      for (int i = 0; i < _openTableFrameCtrls.size(); i++)
      {
         TableFrameController buf = _openTableFrameCtrls.get(i);
         if (false == buf.equals(tfc))
         {
            buf.tableFrameOpen(tfc);
         }
      }

      Vector others = new  Vector(_openTableFrameCtrls);
      others.remove(tfc);
      TableFrameController[] othersArr = others.toArray(new TableFrameController[others.size()]);
      tfc.initAfterAddedToDesktop(othersArr, resetBounds);
   }


   private void onTableFrameMoved()
   {
      if(0 == _openTableFrameCtrls.size())
      {
         return;
      }
      _calcQueryRootTimer.restart();
   }

   private void calculateQueryRoot()
   {
      TableFrameController queryRoot = _openTableFrameCtrls.get(0);

      for (int i = 1; i < _openTableFrameCtrls.size(); i++)
      {
         Point loc = _openTableFrameCtrls.get(i).getFrame().getLocation();
         Point rootLoc = queryRoot.getFrame().getLocation();

         if(loc.x * loc.x + loc.y * loc.y <  rootLoc.x * rootLoc.x + rootLoc.y * rootLoc.y )
         {
            queryRoot = _openTableFrameCtrls.get(i);
         }

         if(_lastRoot != queryRoot)
         {
            _lastRoot = queryRoot;
            fireListeners(TableFramesModelChangeType.ROOT_TABLE);
         }
      }
   }

   public void addTableFramesModelListener(TableFramesModelListener tableFramesModelListener)
   {
      _listeners.remove(tableFramesModelListener);
      _listeners.add(tableFramesModelListener);
   }

   public void removeTableFramesModelListener(TableFramesModelListener tableFramesModelListener)
   {
      _listeners.remove(tableFramesModelListener);
   }

   public void hideNoJoins(boolean b)
   {
      for (TableFrameController openTableFrameCtrl : _openTableFrameCtrls)
      {
         openTableFrameCtrl.getConstraintViewsModel().hideNoJoins(b);
      }


   }

   public boolean containsUniddenNoJoins()
   {
      for (TableFrameController openTableFrameCtrl : _openTableFrameCtrls)
      {
         if(openTableFrameCtrl.getConstraintViewsModel().containsUniddenNoJoins())
         {
            return true;
         }
      }

      return false;
   }

   public void recalculateAllConnections()
   {
      for (TableFrameController tfc : _openTableFrameCtrls)
      {
         tfc.recalculateConnections();
      }
   }

   /**
    * Needed to draw constraint lines right, when Graph was loaded from XML and colums are reordered.
    */
   public void replaceColumnClonesInConstraintsByRefrences()
   {
      for (TableFrameController openTableFrameCtrl : _openTableFrameCtrls)
      {
         openTableFrameCtrl.getConstraintViewsModel().replaceColumnClonesInConstraintsByRefrences(this);
      }
   }

   public ColumnInfo findColumn(String tableName, String columnName)
   {
      for (TableFrameController openTableFrameCtrl : _openTableFrameCtrls)
      {
         if(openTableFrameCtrl.getTableInfo().getSimpleName().equalsIgnoreCase(tableName))
         {
            for (ColumnInfo columnInfo : openTableFrameCtrl.getColumnInfos())
            {
               if(columnInfo.getColumnName().equalsIgnoreCase(columnName))
               {
                  return columnInfo;
               }
            }
         }
      }

      throw new IllegalArgumentException("Column " + tableName + "." + columnName + " does not exist.");
   }
}