Ejemplo n.º 1
0
 def save_plots(self, pdf_dir, png_dir):
     #create spectrum list
     spectrums = []
     for i in range(self._refls.shape[0]):
         dm = np.vstack((self._waves, self._refls[i, :])).transpose()
         spectrums.append(Spectrum(data = dm))
     
     #plot spectrums and save
     fig1_title = self._name
     fig1 = plot_spectrums(spectrums,
                           spectrums_color = 'cyan',
                           title = fig1_title)
     fig1.savefig(os.path.join(pdf_dir, fig1_title + ".pdf"))
     fig1.savefig(os.path.join(png_dir, fig1_title + ".png"))
     
     #plot spectrums + mean and save
     fig2_title = self._name + "_with_mean"
     fig2 = plot_spectrums(spectrums,
                           spectrums_color = 'cyan',
                           title = fig2_title, 
                           stat_spectrum = self.mean,
                           stat_spectrum_color = 'black')
     fig2.savefig(os.path.join(pdf_dir, fig2_title + ".pdf"))
     fig2.savefig(os.path.join(png_dir, fig2_title + ".png"))
     
     #plot spectrums + median and save
     fig3_title = self._name + "_with_median"
     fig3 = plot_spectrums(spectrums,
                           spectrums_color = 'cyan',
                           title = fig3_title, 
                           stat_spectrum = self.median,
                           stat_spectrum_color = 'black')
     fig3.savefig(os.path.join(pdf_dir, fig3_title + ".pdf"))
     fig3.savefig(os.path.join(png_dir, fig3_title + ".png"))    
Ejemplo n.º 2
0
    def save_plots(self, pdf_dir, png_dir):
        #create spectrum list
        spectrums = []
        for i in range(self._refls.shape[0]):
            dm = np.vstack((self._waves, self._refls[i, :])).transpose()
            spectrums.append(Spectrum(data=dm))

        #plot spectrums and save
        fig1_title = self._name
        fig1 = plot_spectrums(spectrums,
                              spectrums_color='cyan',
                              title=fig1_title)
        fig1.savefig(os.path.join(pdf_dir, fig1_title + ".pdf"))
        fig1.savefig(os.path.join(png_dir, fig1_title + ".png"))

        #plot spectrums + mean and save
        fig2_title = self._name + "_with_mean"
        fig2 = plot_spectrums(spectrums,
                              spectrums_color='cyan',
                              title=fig2_title,
                              stat_spectrum=self.mean,
                              stat_spectrum_color='black')
        fig2.savefig(os.path.join(pdf_dir, fig2_title + ".pdf"))
        fig2.savefig(os.path.join(png_dir, fig2_title + ".png"))

        #plot spectrums + median and save
        fig3_title = self._name + "_with_median"
        fig3 = plot_spectrums(spectrums,
                              spectrums_color='cyan',
                              title=fig3_title,
                              stat_spectrum=self.median,
                              stat_spectrum_color='black')
        fig3.savefig(os.path.join(pdf_dir, fig3_title + ".pdf"))
        fig3.savefig(os.path.join(png_dir, fig3_title + ".png"))
Ejemplo n.º 3
0
# -*- coding: utf-8 -*-
"""
Created on Mon Aug  3 13:18:15 2015

@author: pravindran
"""

from __future__ import print_function
import os, os.path
from SdalReader import SdalReader
from sdal_plot_spectrums import plot_spectrums
from helpers import get_extension

if __name__ == "__main__":
    print("asd_trek.py")

    data_dir = "/home/pravindran/mycode/sdal_data/tests_data/asd_mixed_test/"
    files = [os.path.join(data_dir, f) for f in os.listdir(data_dir)]
    print("\n".join(files))

    specs = [SdalReader().read_spectrum(f) for f in files if get_extension(f).lower() == "asd"]
    fig = plot_spectrums(specs, spectrums_color="cyan", title="Spectra from ASD Trek + FieldSpec")
    fig.savefig(os.path.join(data_dir, "plots.pdf"))
Ejemplo n.º 4
0
# -*- coding: utf-8 -*-
"""
Created on Mon Aug  3 13:18:15 2015

@author: pravindran
"""

from __future__ import print_function
import os, os.path
from SdalReader import SdalReader
from sdal_plot_spectrums import plot_spectrums
from helpers import get_extension

if __name__ == "__main__":
    print("asd_trek.py")

    data_dir = '/home/pravindran/mycode/sdal_data/tests_data/asd_mixed_test/'
    files = [os.path.join(data_dir, f) for f in os.listdir(data_dir)]
    print("\n".join(files))

    specs = [
        SdalReader().read_spectrum(f) for f in files
        if get_extension(f).lower() == 'asd'
    ]
    fig = plot_spectrums(specs,
                         spectrums_color='cyan',
                         title="Spectra from ASD Trek + FieldSpec")
    fig.savefig(os.path.join(data_dir, "plots.pdf"))