X-Git-Url: http://source.jalview.org/gitweb/?a=blobdiff_plain;f=src%2Fjalview%2Fgui%2FFinder.java;h=7483a0ad543bd74a091e7bee267c995aaec2ea47;hb=3f578998521236c208e9d3c7790d4d5ac4da0a50;hp=c5a2cf4e0c090630bf0e255f5ecd119432b4e065;hpb=3e313eab680734bb0bd49d3366c5ae6ad8330ef4;p=jalview.git

diff --git a/src/jalview/gui/Finder.java b/src/jalview/gui/Finder.java
index c5a2cf4..7483a0a 100755
--- a/src/jalview/gui/Finder.java
+++ b/src/jalview/gui/Finder.java
@@ -6,7 +6,7 @@ import java.awt.event.*;
 import javax.swing.*;
 import javax.swing.event.*;
 import java.util.*;
-
+import java.awt.*;
 
 public class Finder extends GFinder
 {
@@ -14,6 +14,7 @@ public class Finder extends GFinder
   AlignmentPanel ap;
   JInternalFrame frame;
   String searchString;
+  SuperGroup searchGroup;
 
   int seqIndex = 0;
   int resIndex = 0;
@@ -32,11 +33,14 @@ public class Finder extends GFinder
         SwingUtilities.invokeLater(new Runnable()
         {
           public void run()
-          {
-            textfield.requestFocus();
-          }
+          {   textfield.requestFocus();      }
        });
-   }
+     }
+
+     public void internalFrameClosing(InternalFrameEvent evt)
+     {
+       cancel_actionPerformed(null);
+     }
 });
 
   }
@@ -61,34 +65,126 @@ public class Finder extends GFinder
   public void cancel_actionPerformed(ActionEvent e)
   {
     try{
-      frame.setClosed(true);
-    }catch(Exception ex){}
+      // if allResults is null, this effectively switches displaySearch flag in seqCanvas
+      ap.highlightSearchResults( null );
+     // frame.setClosed(true);
+    }catch(Exception ex){ }
+  }
+
+
+  public void createNewGroup_actionPerformed(ActionEvent e)
+  {
+    for(int i=0; i<searchGroup.getSize(); i++)
+      av.alignment.addGroup( searchGroup.getGroupeAt(i));
 
+    ap.av.alignment.addSuperGroup(searchGroup);
+    ap.highlightSearchResults( null );
   }
 
+
   void doSearch(boolean findAll)
   {
+    createNewGroup.setEnabled(false);
     searchString = textfield.getText().toUpperCase();
 
     ArrayList searchResults = new ArrayList();
     int [] allResults = null;
 
     Sequence seq;
-    String item;
+    String item=null;
     boolean found = false;
-    while( !found && seqIndex<av.getAlignment().getHeight())
+
+    ////// is the searchString a residue number?
+    try{
+      int res = Integer.parseInt(searchString);
+      found = true;
+
+      if(av.getSelectionGroup().getSize()>0)
+        seq = (Sequence)(av.getSelectionGroup().getSequenceAt(0));
+      else
+        seq = (Sequence)av.getAlignment().getSequenceAt(0);
+
+      searchResults.add( Integer.toString( av.getAlignment().findIndex(seq) ) );
+      searchResults.add( Integer.toString( seq.findIndex(res)-1 ) );
+      searchResults.add( Integer.toString( seq.findIndex(res)-1 ) );
+
+    }catch(NumberFormatException ex){}
+    ///////////////////////////////////////////////
+
+
+    jalview.schemes.UserColourScheme ucs = new jalview.schemes.UserColourScheme();
+    Color [] newColors = new Color[24];
+    for(int i=0; i<24; i++)
+      newColors[i] = new Color(60,160,115);
+
+    ucs.setColourScheme(newColors);
+
+    searchGroup = new SuperGroup(searchString, ucs , true, true, false);
+
+    int end = av.alignment.getHeight();
+
+    SequenceGroup selection = av.getSelectionGroup();
+    if(selection!=null)
+     if(selection.getSize()<1 || (selection.getEndRes()-selection.getStartRes()<2))
+      selection = null;
+
+    while( !found && seqIndex<end)
     {
-      seq = (Sequence)av.getAlignment().getSequenceAt(seqIndex);
-      item = seq.getSequence();
+      seq = (Sequence)av.alignment.getSequenceAt(seqIndex);
+
 
-      for(int r = resIndex; r<item.length(); r++)
+      if(selection!=null && !selection.sequences.contains(seq))
       {
-       if(item.indexOf(searchString,r)>-1)
+        seqIndex++;
+        resIndex=0;
+        continue;
+      }
+
+      item = seq.getSequence().toUpperCase();
+
+      if(selection!=null && selection.getEndRes()<av.alignment.getWidth())
+          item = item.substring(0, selection.getEndRes()+1);
+
+      ///Shall we ignore gaps????
+      StringBuffer noGaps = new StringBuffer();
+      int insertCount=0;
+      ArrayList spaces = new ArrayList();
+
+      for (int j=0; j < item.length(); j++)
+      {
+
+        if(!jalview.util.Comparison.isGap(item.charAt(j)))
+         {
+           noGaps.append(item.charAt(j));
+           spaces.add(new Integer(insertCount));
+         }
+        else
+           insertCount++;
+      }
+
+
+      com.stevesoft.pat.Regex regex = new  com.stevesoft.pat.Regex(searchString);
+
+      for(int r = resIndex; r<noGaps.length(); r++)
+      {
+
+       if( regex.searchFrom( noGaps.toString(), r ) )
        {
-         resIndex = item.indexOf(searchString,r);
-         searchResults.add( Integer.toString(seqIndex) );
-         searchResults.add( Integer.toString( resIndex) );
-         searchResults.add( Integer.toString( (resIndex+searchString.length()-1) ) );
+         resIndex = regex.matchedFrom();
+         if(selection!=null && ( resIndex+ Integer.parseInt(spaces.get(resIndex).toString()) )<selection.getStartRes())
+           continue;
+
+         searchResults.add( Integer.toString( seqIndex) );
+         searchResults.add( Integer.toString( resIndex+ Integer.parseInt(spaces.get(resIndex).toString()) ) );
+         searchResults.add( Integer.toString( regex.matchedTo()-1 + Integer.parseInt(spaces.get(regex.matchedTo()-1).toString()) ) );
+
+         SequenceGroup sg = new SequenceGroup(searchString, ucs, true, true, false,
+                                             resIndex+ Integer.parseInt(spaces.get(resIndex).toString()) ,
+                                      regex.matchedTo()-1 + Integer.parseInt(spaces.get(regex.matchedTo()-1).toString())
+            );
+         sg.addSequence((Sequence)av.getAlignment().getSequenceAt(seqIndex));
+         searchGroup.addGroup( sg );
+
          if(!findAll)
          {
            // thats enough, break and display the result
@@ -96,6 +192,8 @@ public class Finder extends GFinder
            resIndex++;
            break;
          }
+
+         r=resIndex;
        }
       }
       if(!found)
@@ -107,22 +205,28 @@ public class Finder extends GFinder
 
     if(searchResults.size()>0)
     {
-
       allResults = new int[searchResults.size()];
       for(int i=0; i<searchResults.size(); i++)
         allResults[i] = Integer.parseInt(searchResults.get(i).toString());
 
-
-      ap.highlightSearchResults( allResults );
+      createNewGroup.setEnabled(true);
     }
     else
     {
-      JOptionPane.showInternalMessageDialog(this, "Finished searching", null,
-                                            JOptionPane.INFORMATION_MESSAGE);
+      JOptionPane.showInternalMessageDialog(this, "Finished searching", null, JOptionPane.INFORMATION_MESSAGE);
       resIndex=0;
       seqIndex=0;
     }
 
+    // if allResults is null, this effectively switches displaySearch flag in seqCanvas
+    ap.highlightSearchResults( allResults );
+
+    if (findAll)
+    {
+      String message =  (searchResults.size()/3) + " matches found.";
+      JOptionPane.showInternalMessageDialog(this, message, null,
+                                            JOptionPane.INFORMATION_MESSAGE);
+    }
 
   }