Пример #1
0
def build_coeff_tables():
    """
    Use the latest throughput values to build Table 2. Requires LSST stack installed and
    syseng_throughputs setup
    """
    from calcM5 import calcM5
    import bandpassUtils as bu
    from lsst.utils import getPackageDir
    defaultDirs = bu.setDefaultDirs(
        rootDir=getPackageDir('syseng_throughputs'))
    addLosses = True
    atmosphere = bu.readAtmosphere(defaultDirs['atmosphere'],
                                   atmosFile='atmos_10_aerosol.dat')
    hardware, system = bu.buildHardwareAndSystem(defaultDirs,
                                                 addLosses,
                                                 atmosphereOverride=atmosphere)
    m5_vals = calcM5(hardware,
                     system,
                     atmosphere,
                     title='',
                     return_t2_values=True)
    np.savez('m5_vals.npz', m5_vals=m5_vals)
    plt.ylim(0, 1)
    plt.xlim(300, 1100)
    plt.xlabel('Wavelength (nm)')
    plt.ylabel('Fractional Throughput Response')
    plt.title('System total response curves %s' %(title))
    return m5


if __name__ == '__main__':

    # Set the directories for each component.
    # Note that this sets the detector to be the 'generic detector' (minimum of each vendor).
    throughputDir = os.getenv('LSST_THROUGHPUTS_DEFAULT')
    defaultDirs = bu.setDefaultDirs()

    # Build the system and hardware throughput curves (without aerosols, with X=1.0).
    atmosphere = bu.readAtmosphere(defaultDirs['atmosphere'], atmosFile='atmos_10_aerosol.dat')
    hardware = {}
    system = {}
    for f in ['u', 'g', 'r', 'i', 'z', 'y']:
        hardware[f] = Bandpass()
        system[f] = Bandpass()
        hardware[f].readThroughputList(componentList=['detector.dat', 'filter_'+f+'.dat','lens1.dat', 'lens2.dat', 'lens3.dat', 'm1.dat', 'm2.dat', 'm3.dat'],
                                       rootDir=throughputDir)
        system[f].wavelen, system[f].sb = hardware[f].multiplyThroughputs(atmosphere.wavelen, atmosphere.sb)

    m5 = calcM5(hardware, system, atmosphere, title='')


    plt.show()
Пример #3
0
    fwhm_eff = 1.16 * np.sqrt(fwhm_system**2 + 1.04 * fwhm_atmo**2)
    # Translate to FWHMgeom.
    fwhm_geom = 0.822 * fwhm_eff + 0.052
    return fwhm_eff, fwhm_geom



if __name__ == '__main__':

    # Set the directories for each component.
    # Note that this sets the detector to be the 'generic detector' (minimum of each vendor).
    defaultDirs = bu.setDefaultDirs(rootDir = '..')
    # To use a particular vendor, uncomment one of the following lines or edit as necessary.
    # defaultDirs['detector'] = 'vendor1'
    # defaultDirs['detector'] = 'vendor2'

    # Add losses to each component?
    addLosses = True

    # Build the system and hardware throughput curves (without aerosols, with X=1.0).
    #atmosphere = bu.readAtmosphere(defaultDirs['atmosphere'], atmosFile='atmos_10.dat')
    #hardware, system = bu.buildHardwareAndSystem(defaultDirs, addLosses, atmosphereOverride=atmosphere)
    #m5 = calcM5(hardware, system, atmosphere, title='')

    atmosphere = bu.readAtmosphere(defaultDirs['atmosphere'], atmosFile='atmos_10_aerosol.dat')
    hardware, system = bu.buildHardwareAndSystem(defaultDirs, addLosses, atmosphereOverride=atmosphere)
    m5 = calcM5(hardware, system, atmosphere, title='')


    plt.show()
    if title == "Vendor combo":
        title = ""
    plt.title("System total response curves %s" % (title))
    plt.savefig("../plots/system+sky" + title + ".png", format="png", dpi=600)
    return m5


if __name__ == "__main__":

    defaultDirs = bu.setDefaultDirs(rootDir="..")
    addLosses = True

    hardware = {}
    system = {}
    m5 = {}
    atmosphere = bu.readAtmosphere(defaultDirs["atmosphere"], atmosFile="atmos_10_aerosol.dat")
    hardware["combo"], system["combo"] = bu.buildHardwareAndSystem(defaultDirs, atmosphereOverride=atmosphere)
    m5["combo"] = calcM5(hardware["combo"], system["combo"], atmosphere, title="combo")
    genericDetector = defaultDirs["detector"]
    for det in [1, 2]:
        defaultDirs["detector"] = os.path.join(genericDetector, "vendor%d" % det)
        hardware[det], system[det] = bu.buildHardwareAndSystem(defaultDirs, atmosphereOverride=atmosphere)
        m5[det] = calcM5(hardware[det], system[det], atmosphere, title="vendor%d" % (det))

    # Show what these look like (print m5 limits on throughput curves)
    plt.figure()
    for f in filterlist:
        for det in [1, 2]:
            if det == 1:
                linestyle = ":"
                spacer = " "
# total_[ugrizy].dat
# hardware_[ugrizy].dat

# Read the data needed from the syseng_throughput repo.
defaultDirs = bu.setDefaultDirs()
addLosses = True

detector = bu.buildDetector(defaultDirs['detector'], addLosses)
filters = bu.buildFilters(defaultDirs['filters'], addLosses)
lens1 = bu.buildLens(defaultDirs['lens1'], addLosses)
lens2 = bu.buildLens(defaultDirs['lens2'], addLosses)
lens3 = bu.buildLens(defaultDirs['lens3'], addLosses)
m1 = bu.buildMirror(defaultDirs['mirror1'], addLosses)
m2 = bu.buildMirror(defaultDirs['mirror2'], addLosses)
m3 = bu.buildMirror(defaultDirs['mirror3'], addLosses)
atmos_std = bu.readAtmosphere(defaultDirs['atmosphere'],
                              atmosFile='pachonModtranAtm_12_aerosol.dat')
atmos_10 = bu.readAtmosphere(defaultDirs['atmosphere'],
                             atmosFile='atmos_10_aerosol.dat')
darksky = Sed()
darksky.readSED_flambda(os.path.join(defaultDirs['atmosphere'], 'darksky.dat'))
hardware, system = bu.buildHardwareAndSystem(defaultDirs,
                                             addLosses=addLosses,
                                             atmosphereOverride=atmos_std)

# Write the data to disk.
outDir = 'baseline'
if not os.path.isdir(outDir):
    os.makedirs(outDir)

shutil.copy('../README.md', os.path.join(outDir, 'README.md'))
Пример #6
0
        bu.plotBandpasses(atmodict)

    return atmosphere_aerosol


if __name__ == '__main__':

    parser = argparse.ArgumentParser(
        description=
        'Add a standard aerosol component to an atmosphere without aerosols')
    parser.add_argument('atmosphereFile',
                        type=str,
                        default=None,
                        help='The atmosphere file to add aerosols to.')
    parser.add_argument('airmass',
                        type=float,
                        default=None,
                        help='The airmass of the atmosphere file.')
    args = parser.parse_args()
    atmosDir, atmosFile = os.path.split(args.atmosphereFile)

    defaultDirs = bu.setDefaultDirs()
    atmosphere = bu.readAtmosphere(atmosDir, atmosFile)
    atmosphere_aerosol = addAerosol(atmosphere, args.airmass)

    outfile = atmosphere.bandpassname.replace('.dat', '') + '_aerosol.dat'
    atmosphere_aerosol.writeThroughput(outfile,
                                       print_header='Added aerosol component')

    plt.show()
# total_[ugrizy].dat
# hardware_[ugrizy].dat

# Read the data needed from the syseng_throughput repo.
defaultDirs = bu.setDefaultDirs()
addLosses = True

detector = bu.buildDetector(defaultDirs['detector'], addLosses)
filters = bu.buildFilters(defaultDirs['filters'], addLosses)
lens1 = bu.buildLens(defaultDirs['lens1'], addLosses)
lens2 = bu.buildLens(defaultDirs['lens2'], addLosses)
lens3 = bu.buildLens(defaultDirs['lens3'], addLosses)
m1 = bu.buildMirror(defaultDirs['mirror1'], addLosses)
m2 = bu.buildMirror(defaultDirs['mirror2'], addLosses)
m3 = bu.buildMirror(defaultDirs['mirror3'], addLosses)
atmos_std = bu.readAtmosphere(defaultDirs['atmosphere'], atmosFile='pachonModtranAtm_12_aerosol.dat')
atmos_10 = bu.readAtmosphere(defaultDirs['atmosphere'], atmosFile='atmos_10_aerosol.dat')
darksky = Sed()
darksky.readSED_flambda(os.path.join(defaultDirs['atmosphere'], 'darksky.dat'))
hardware, system = bu.buildHardwareAndSystem(defaultDirs, addLosses=addLosses, atmosphereOverride=atmos_std)

# Write the data to disk.
outDir = 'baseline'
if not os.path.isdir(outDir):
    os.makedirs(outDir)

version = subprocess.check_output(['git', 'describe']).strip()
sha1 = subprocess.check_output(["git", "rev-parse", "HEAD"]).strip()
print "version", version, "sha1", sha1
versioninfo = '# Version %s\n'%(version)
versioninfo += '# sha1 %s\n' %(sha1)
    tau = tau0 * np.power((wavelen0/atmosphere.wavelen), alpha)
    # Generate new atmosphere bandpass with aerosols.
    atmosphere_aerosol = Bandpass()
    atmosphere_aerosol.setBandpass(wavelen = atmosphere.wavelen,
                                   sb = atmosphere.sb * np.exp(-tau*X))

    if plotAtmosphere:
        # Plot for a check:
        atmodict = {'Original atmosphere':atmosphere,
                    'With aerosols': atmosphere_aerosol}
        bu.plotBandpasses(atmodict)

    return atmosphere_aerosol

if __name__=='__main__':

    parser = argparse.ArgumentParser(description='Add a standard aerosol component to an atmosphere without aerosols')
    parser.add_argument('atmosphereFile', type=str, default=None, help='The atmosphere file to add aerosols to.')
    parser.add_argument('airmass', type=float, default=None, help='The airmass of the atmosphere file.')
    args = parser.parse_args()
    atmosDir, atmosFile = os.path.split(args.atmosphereFile)

    defaultDirs = bu.setDefaultDirs()
    atmosphere = bu.readAtmosphere(atmosDir, atmosFile)
    atmosphere_aerosol = addAerosol(atmosphere, args.airmass)

    outfile = atmosphere.bandpassname.replace('.dat', '') + '_aerosol.dat'
    atmosphere_aerosol.writeThroughput(outfile, print_header='Added aerosol component')

    plt.show()
    bu.plotBandpasses(oldfilters, newfig=False, linestyle=':', title='Compare filters')
    # Put the individual filter bandpasses straight into 'throughputs', for easier use below.
    for f in filters:
        throughputs[f] = filters[f]

    # Build and compare the mirror curves, for mirrors 1/2/3.
    for mirror in ('mirror1', 'mirror2', 'mirror3'):
        throughputs[mirror] = bu.buildMirror(defaultDirs[mirror], addLosses)
        mfile = ('../intermediateFiles/components/telescope/' + mirror[0] + mirror[-1] +
                 'Throughput.dat')
        comparison.readThroughput(mfile)
        plotDict = {'New %s' %(mirror):throughputs[mirror], 'Old %s' %(mirror):comparison}
        bu.plotBandpasses(plotDict, title='Compare %s' %(mirror))

    # Read the atmosphere file.
    throughputs['atmosphere'] = bu.readAtmosphere(defaultDirs['atmosphere'])

    # Plot all components.
    bu.plotBandpasses(throughputs, title='All components')

    # Combine components by hand. Compare to combination returned from bandpassUtils.
    core_sb = (throughputs['detector'].sb * throughputs['lens1'].sb * throughputs['lens2'].sb
            * throughputs['lens3'].sb * throughputs['mirror1'].sb
            * throughputs['mirror2'].sb * throughputs['mirror3'].sb)
    hardware = {}
    system = {}
    for f in filters:
        hardware[f] = Bandpass()
        system[f] = Bandpass()
        wavelen = filters[f].wavelen
        hw_sb = core_sb * filters[f].sb
Пример #10
0
        throughputs[f] = filters[f]

    # Build and compare the mirror curves, for mirrors 1/2/3.
    for mirror in ('mirror1', 'mirror2', 'mirror3'):
        throughputs[mirror] = bu.buildMirror(defaultDirs[mirror], addLosses)
        mfile = ('../intermediateFiles/components/telescope/' + mirror[0] +
                 mirror[-1] + 'Throughput.dat')
        comparison.readThroughput(mfile)
        plotDict = {
            'New %s' % (mirror): throughputs[mirror],
            'Old %s' % (mirror): comparison
        }
        bu.plotBandpasses(plotDict, title='Compare %s' % (mirror))

    # Read the atmosphere file.
    throughputs['atmosphere'] = bu.readAtmosphere(defaultDirs['atmosphere'])

    # Plot all components.
    bu.plotBandpasses(throughputs, title='All components')

    # Combine components by hand. Compare to combination returned from bandpassUtils.
    core_sb = (throughputs['detector'].sb * throughputs['lens1'].sb *
               throughputs['lens2'].sb * throughputs['lens3'].sb *
               throughputs['mirror1'].sb * throughputs['mirror2'].sb *
               throughputs['mirror3'].sb)
    hardware = {}
    system = {}
    for f in filters:
        hardware[f] = Bandpass()
        system[f] = Bandpass()
        wavelen = filters[f].wavelen