| 1 | #import genera operating system related functions
|
|---|
| 2 | import os
|
|---|
| 3 | #import atnf_sd module
|
|---|
| 4 | from atnf_sd import *
|
|---|
| 5 | # create a reader
|
|---|
| 6 | r = sdreader()
|
|---|
| 7 | # open an RPFITS file
|
|---|
| 8 | r.open('/u/mmarquar/zorro/singledish/data/2001-09-01_0332_P363.rpf')
|
|---|
| 9 | # create a vector with numbers [0..109]
|
|---|
| 10 | integrations = range(110)
|
|---|
| 11 | r.read(integrations)
|
|---|
| 12 | # get the data out of the reader
|
|---|
| 13 | scans = r.getdata()
|
|---|
| 14 | # close the reader
|
|---|
| 15 | r = None
|
|---|
| 16 | # Test sdwriter.
|
|---|
| 17 | print 'Begin sdwriter tests...'
|
|---|
| 18 | # Create an MS2 writer.
|
|---|
| 19 | w = sdwriter('MS2')
|
|---|
| 20 | # Change to SDFITS output format (the default).
|
|---|
| 21 | w.setformat()
|
|---|
| 22 | # Write out the spectra.
|
|---|
| 23 | w.write(scans, 'test_SDWriter.sdfits')
|
|---|
| 24 | # clean up
|
|---|
| 25 | print "removing test_SDWriter.sdfits ..."
|
|---|
| 26 | os.remove('test_SDWriter.sdfits')
|
|---|
| 27 | # print a short summary of the data
|
|---|
| 28 | scans.summary()
|
|---|
| 29 | # get the scan with the number '1'
|
|---|
| 30 | scan = scans.getscan(0)
|
|---|
| 31 | # get the scan with the number '1'
|
|---|
| 32 | ref = scans.getscan(1)
|
|---|
| 33 | # close the data table
|
|---|
| 34 | scans = None
|
|---|
| 35 | # open the math server
|
|---|
| 36 | # average the "on" scan
|
|---|
| 37 | scanav = average(scan)
|
|---|
| 38 | # get rid of the original scan
|
|---|
| 39 | scan = None
|
|---|
| 40 | # print a summary of the scan
|
|---|
| 41 | scanav.summary()
|
|---|
| 42 | # average the "off" scan
|
|---|
| 43 | refav = average(ref)
|
|---|
| 44 | # get rid of the original scan
|
|---|
| 45 | ref = None
|
|---|
| 46 | # print a summary of the scan
|
|---|
| 47 | refav.summary()
|
|---|
| 48 | # form the quotione spectrum
|
|---|
| 49 | quot = quotient(scanav,refav)
|
|---|
| 50 | # set the cursor to polarisation 0
|
|---|
| 51 | quot.setpol(0)
|
|---|
| 52 | # get the spectrum for polarisation 0
|
|---|
| 53 | v0 = quot.getspectrum()
|
|---|
| 54 | #print the first ten channel
|
|---|
| 55 | print v0[0:10]
|
|---|
| 56 | # set the cursor to polarisation 1
|
|---|
| 57 | quot.setpol(1)
|
|---|
| 58 | # get the spectrum for polarisation 1
|
|---|
| 59 | v1 = quot.getspectrum()
|
|---|
| 60 | #print the first ten channel
|
|---|
| 61 | print v1[0:10]
|
|---|
| 62 | # write it to disk for further use
|
|---|
| 63 | quot.makepersistent('/tmp/myfirstquotient.tbl')
|
|---|
| 64 | # cleanup
|
|---|
| 65 | print "removing /tmp/myfirstquotient.tbl ..."
|
|---|
| 66 | os.system('rm -rf /tmp/myfirstquotient.tbl')
|
|---|
| 67 | print "Test successful."
|
|---|
