DiskSer.java

import java.nio.file.Files;
import java.nio.file.Paths;
import java.io.IOException;
import java.util.HashMap;
import java.util.Map;
import org.apache.thrift.TException;
import org.apache.thrift.transport.TTransport;
import org.apache.thrift.transport.TSimpleFileTransport;
import org.apache.thrift.protocol.TBinaryProtocol;
import org.apache.thrift.protocol.TProtocol;
import radio_observation.ICRFPosition;
import radio_observation.Position;
import radio_observation.RadioObservation;
import radio_observation.RadioObservationSystem;
import radio_observation.radio_observationConstants;

public class DiskSer {
  public static void FakeInit(RadioObservation ro) {
      ro.telescope_count = 1;
      ro.system = RadioObservationSystem.Arecibo;
      ro.time = System.currentTimeMillis() / 1000.0;
      ICRFPosition pos = new ICRFPosition(270.3, 45.24);
      pos.setEcliptic_year((short)2000);
      ro.pos = new Position();
      ro.pos.setIcrfpos(pos);
      ro.freq_amp = new HashMap<>();
      ro.freq_amp.put(20500000L, 75.456);
      ro.freq_amp.put(50000000L, 29.321);
      ro.freq_amp.put(75000000L, 51.526);
  }
  
  public static void DumpICRFPosition(ICRFPosition pos) {
      System.out.println("Position       : " + 
                         pos.declination + " dec - " + 
                         pos.right_ascension + " ra [" + 
                         ((pos.isSetEcliptic_year())?pos.ecliptic_year:"") + "]");    
  }
  
  public static void DumpObservation(RadioObservation ro) {
      System.out.println("Telescope Count: " + ro.telescope_count);
      System.out.println("System         : " + ro.system.name());
      System.out.println("Time           : " + ro.time);
      if (ro.pos.isSetIcrfpos()) {
        DumpICRFPosition(ro.pos.getIcrfpos());
      }
      System.out.println("Frequency   Magnitude");
      for (Map.Entry<Long, Double> entry : ro.freq_amp.entrySet()) {
          System.out.println("  "+entry.getKey()+"  "+entry.getValue());
      }      
  }
  
  public static void WriteRadioObservation(TProtocol proto, 
                                           boolean writeBMP) 
          throws TException, IOException {
      System.out.println("\nWriting Observations");
      System.out.println("-------------------------");
      RadioObservation ro = new RadioObservation();
      FakeInit(ro);
      if (writeBMP){
        ro.setSky_bmp(Files.readAllBytes(Paths.get("quasar.bmp")));       
      }
      ro.write(proto);
      DumpObservation(ro);    
  }

  public static void ReadRadioObservation(TProtocol proto) throws TException {
      System.out.println("\nReading Observations");
      System.out.println("-------------------------");
      RadioObservation ro = new RadioObservation();
      ro.read(proto);
      DumpObservation(ro);    
  }

  public static void main(String[] args) {
    TTransport trans = null;
    try {
      System.out.println("Radio Observation Disk Serializer " + radio_observationConstants.Version);
      trans = new TSimpleFileTransport("data", true, true);
      trans.open();
      TProtocol proto = new TBinaryProtocol(trans);
      if (args.length > 0 && 0 == args[0].compareTo("write")) {
        WriteRadioObservation(proto, args.length > 1);        
      } else if (args.length>0 && 0==args[0].compareToIgnoreCase("read")) {
        ReadRadioObservation(proto);      
      }
      else {
         System.out.println("Usage: DiskSer (read | write [bmp])");        
      }
    } catch (TException | IOException ex) {
      System.out.println("Error: " + ex.getMessage());
    }
    if (null != trans) {
      trans.close();
    }
  }
}