Test32.java

/*
VisAD system for interactive analysis and visualization of numerical
data.  Copyright (C) 1996 - 2017 Bill Hibbard, Curtis Rueden, Tom
Rink, Dave Glowacki, Steve Emmerson, Tom Whittaker, Don Murray, and
Tommy Jasmin.

This library is free software; you can redistribute it and/or
modify it under the terms of the GNU Library General Public
License as published by the Free Software Foundation; either
version 2 of the License, or (at your option) any later version.

This library is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
Library General Public License for more details.

You should have received a copy of the GNU Library General Public
License along with this library; if not, write to the Free
Software Foundation, Inc., 59 Temple Place - Suite 330, Boston,
MA 02111-1307, USA
*/

import java.rmi.RemoteException;

import visad.*;

import visad.data.fits.FitsForm;

import visad.java3d.DisplayImplJ3D;

public class Test32
  extends TestSkeleton
{
  private String fileName;

  public Test32() { }

  public Test32(String[] args)
    throws RemoteException, VisADException
  {
    super(args);
  }

  public void initializeArgs() { fileName = null; }

  public int checkKeyword(String testName, int argc, String[] args)
  {
    if (fileName == null) {
      fileName = args[argc];
    } else {
      System.err.println(testName + ": Ignoring extra filename \"" +
                         args[argc] + "\"");
    }

    return 1;
  }

  public String keywordUsage()
  {
    return super.keywordUsage() + " file";
  }

  private DataReferenceImpl loadFile()
    throws RemoteException, VisADException
  {
    if (fileName == null) {
      return null;
    }

    FitsForm fits = new FitsForm();
    Data data;
    try {
      data = fits.open(fileName);
    } catch (VisADException ve) {
      System.err.println("Couldn't load \"" + fileName + "\"");
      ve.printStackTrace();
      System.exit(1);
      return null;
    }

    if (!(data instanceof FieldImpl)) {
      System.err.println("File \"" + fileName + "\" resolves to " +
                         data.getClass().getName() + ", not " +
                         FieldImpl.class.getName());
      System.exit(1);
      return null;
    }

    //System.out.println("data type = " + data.getType());

    DataReferenceImpl ref = new DataReferenceImpl("fits");
    ref.setData(data);
    return ref;
  }

  DataReference[] getClientDataReferences()
    throws RemoteException, VisADException
  {
    DataReference ref = loadFile();
    if (ref == null) {
      return null;
    }

    return new DataReference[] { ref };
  }

  DisplayImpl[] setupServerDisplays()
    throws RemoteException, VisADException
  {
    DisplayImpl[] dpys = new DisplayImpl[1];
    dpys[0] = new DisplayImplJ3D("display", DisplayImplJ3D.APPLETFRAME);
    return dpys;
  }

  void setupServerData(LocalDisplay[] dpys)
    throws RemoteException, VisADException
  {
    DataReference ref = loadFile();
    if (ref == null) {
      System.err.println("Must specify FITS file name");
      return;
    }

    FieldImpl fits_data = (FieldImpl )ref.getData();

    // compute ScalarMaps from type components
    FunctionType ftype = (FunctionType) fits_data.getType();
    RealTupleType dtype = ftype.getDomain();
    MathType rntype = ftype.getRange();
    int n = dtype.getDimension();
    dpys[0].addMap(new ScalarMap((RealType) dtype.getComponent(0),
                                  Display.XAxis));
    if (n > 1) {
      dpys[0].addMap(new ScalarMap((RealType) dtype.getComponent(1),
                                    Display.YAxis));
    }
    if (n > 2) {
      dpys[0].addMap(new ScalarMap((RealType) dtype.getComponent(2),
                                    Display.ZAxis));
    }
    if (rntype instanceof RealType) {
      dpys[0].addMap(new ScalarMap((RealType) rntype, Display.Green));
    }
    else if (rntype instanceof RealTupleType) {
      int m = ((RealTupleType) rntype).getDimension();
      RealType rr = (RealType) ((RealTupleType) rntype).getComponent(0);
      dpys[0].addMap(new ScalarMap(rr, Display.Green));
      if (n <= 2) {
        if (m > 1) {
          rr = (RealType) ((RealTupleType) rntype).getComponent(1);
        }
        dpys[0].addMap(new ScalarMap(rr, Display.ZAxis));
      }
    }
    dpys[0].addMap(new ConstantMap(0.5, Display.Red));
    dpys[0].addMap(new ConstantMap(0.0, Display.Blue));

    dpys[0].addReference(ref, null);
  }

  public String toString() { return " file_name: FITS adapter"; }

  public static void main(String[] args)
    throws RemoteException, VisADException
  {
    new Test32(args);
  }
}