SourceRecord.h

Go to the documentation of this file.

/*
 * SourceRecord.h
 *
 *  Created on: Jan 27, 2016
 *      Author: nakazato
 */

#ifndef SINGLEDISH_FILLER_SOURCERECORD_H_
#define SINGLEDISH_FILLER_SOURCERECORD_H_

#include <casacore/casa/Arrays/Matrix.h>
#include <casacore/casa/BasicSL/String.h>
#include <casacore/measures/Measures/MDirection.h>
#include <casacore/ms/MeasurementSets/MSSource.h>
#include <casacore/ms/MeasurementSets/MSSourceColumns.h>

namespace casa { //# NAMESPACE CASA - BEGIN
namespace sdfiller { //# NAMESPACE SDFILLER - BEGIN

struct SourceRecord {
  typedef MSSource AssociatingTable;
  typedef MSSourceColumns AssociatingColumns;

  // mandatory
  Int source_id;
  Int spw_id;
  String name;
  Double time;
  Double interval;
  MDirection direction;
  Int num_lines;

  // optional
  String code;
  Int calibration_group;
  Vector<String> transition;
  Vector<Double> rest_frequency;
  Vector<Double> sysvel;
  Vector<Double> proper_motion;

  // method
  void clear() {
    source_id = -1;
    spw_id = -1;
    name = "";
    time = -1.0;
    interval = -1.0;
    direction = MDirection();
    num_lines = 0;
    code = "";
    calibration_group = -1;
    transition.resize();
    rest_frequency.resize();
    sysvel.resize();
    proper_motion.resize();
  }

  SourceRecord &operator=(SourceRecord const &other) {
    source_id = other.source_id;
    spw_id = other.spw_id;
    name = other.name;
    time = other.time;
    interval = other.interval;
    direction = other.direction;
    num_lines = other.num_lines;
    code = other.code;
    calibration_group = other.calibration_group;
    transition = other.transition;
    rest_frequency = other.rest_frequency;
    sysvel = other.sysvel;
    proper_motion = other.proper_motion;
    return *this;
  }

  void add(AssociatingTable &table, AssociatingColumns &columns) {
    if (columns.nrow() == 0) {
      // set frame info
      TableRecord &record = columns.direction().rwKeywordSet();
      Record meas_info = record.asRecord("MEASINFO");
      meas_info.define("Ref", direction.getRefString());
      record.defineRecord("MEASINFO", meas_info);
    }

    table.addRow(1, True);
  }

  Bool fill(uInt irow, AssociatingColumns &columns) {
    if (columns.nrow() <= irow) {
      return False;
    }

    columns.sourceId().put(irow, source_id);
    columns.spectralWindowId().put(irow, spw_id);
    columns.name().put(irow, name);
    columns.time().put(irow, time);
    columns.interval().put(irow, interval);
    columns.directionMeas().put(irow, direction);
    columns.numLines().put(irow, num_lines);
    columns.calibrationGroup().put(irow, calibration_group);
    if (code.size() > 0) {
      columns.code().put(irow, code);
    }
    if (transition.size() > 0) {
      columns.transition().put(irow, transition);
    }
    if (rest_frequency.size() > 0) {
      columns.restFrequency().put(irow, rest_frequency);
    }
    if (sysvel.size() > 0) {
      columns.sysvel().put(irow, sysvel);
    }
    if (proper_motion.size() > 0) {
      columns.properMotion().put(irow, proper_motion);
    }

    return True;
  }
};

} //# NAMESPACE SDFILLER - END
} //# NAMESPACE CASA - END

#endif /* SINGLEDISH_FILLER_SOURCERECORD_H_ */