package jalview.ws.jabaws;

import static org.junit.Assert.*;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.Vector;

import jalview.api.AlignCalcManagerI;
import jalview.datamodel.AlignmentAnnotation;
import jalview.datamodel.AlignmentI;
import jalview.datamodel.Annotation;
import jalview.io.AnnotationFile;
import jalview.io.FormatAdapter;
import jalview.io.StockholmFileTest;
import jalview.ws.jws2.AADisorderClient;
import jalview.ws.jws2.Jws2Discoverer;
import jalview.ws.jws2.RNAalifoldClient;
import jalview.ws.jws2.jabaws2.Jws2Instance;

import org.junit.AfterClass;
import org.junit.BeforeClass;
import org.junit.Test;

public class RNAStructExportImport 
{
  public static String testseqs = "examples/unfolded_RF00031.aln";

  public static Jws2Discoverer disc;

  public static Jws2Instance rnaalifoldws;

  jalview.ws.jws2.RNAalifoldClient alifoldClient;

  public static jalview.gui.AlignFrame af = null;

  @BeforeClass
  public static void setUpBeforeClass() throws Exception
  {
  	
  	
    jalview.bin.Cache.initLogger();
    disc = JalviewJabawsTestUtils.getJabawsDiscoverer();
    
    for (Jws2Instance svc : disc.getServices())
    {
    	
      if (svc.getServiceTypeURI().toLowerCase().contains("rnaalifoldws"))
      {
        rnaalifoldws = svc;
      }
    }
    
    System.out.println("State of rnaalifoldws: " + rnaalifoldws);
    
    if (rnaalifoldws == null) System.exit(0);
    
    jalview.io.FileLoader fl = new jalview.io.FileLoader(false);
    
    af = fl.LoadFileWaitTillLoaded(testseqs, jalview.io.FormatAdapter.FILE);
    
    assertNotNull("Couldn't load test data ('" + testseqs + "')", af);
    
  }

  @AfterClass
  public static void tearDownAfterClass() throws Exception
  {
    if (af != null)
    {
      af.setVisible(false);
      af.dispose();
    }
  }

  @Test
  public void testRNAStructExport()
  {
  	
  	
    alifoldClient = new RNAalifoldClient(rnaalifoldws, af, null, null);
    
    af.getViewport().getCalcManager().startWorker(alifoldClient);
    
    
    do
    {
      try
      {
        Thread.sleep(50);
      } catch (InterruptedException x)
      {
      }
      ;
    } while (af.getViewport().getCalcManager().isWorking());
    
    
    AlignmentI orig_alig = af.getViewport().getAlignment();
    
    testAnnotationFileIO("Testing RNAalifold Annotation IO", orig_alig);

  }

  public static void testAnnotationFileIO(String testname, AlignmentI al)
  {
    try
    {
    	// what format would be appropriate for RNAalifold annotations?
      String aligfileout = new FormatAdapter().formatSequences("PFAM",
              al.getSequencesArray());
      
      
      String anfileout = new AnnotationFile().printAnnotations(
              al.getAlignmentAnnotation(), al.getGroups(),
              al.getProperties());
      assertTrue(
              "Test "
                      + testname
                      + "\nAlignment annotation file was not regenerated. Null string",
              anfileout != null);
      assertTrue(
              "Test "
                      + testname
                      + "\nAlignment annotation file was not regenerated. Empty string",
              anfileout.length() > "JALVIEW_ANNOTATION".length());

      System.out.println("Output annotation file:\n" + anfileout
              + "\n<<EOF\n");

      // again what format would be appropriate?
      AlignmentI al_new = new FormatAdapter().readFile(aligfileout,
              FormatAdapter.PASTE, "PFAM");
      assertTrue(
              "Test "
                      + testname
                      + "\nregenerated annotation file did not annotate alignment.",
              new AnnotationFile().readAnnotationFile(al_new, anfileout,
                      FormatAdapter.PASTE));

      // test for consistency in io
      StockholmFileTest.testAlignmentEquivalence(al, al_new);
      return;
    } catch (Exception e)
    {
      e.printStackTrace();
    }
    fail("Test "
            + testname
            + "\nCouldn't complete Annotation file roundtrip input/output/input test.");
  }

}