示例#1
0
文件: sed.py 项目: buchbend/astrolyze
 def set_limits(self, axes=plt.gca(), nu_or_lambda='lambda'):
     # sets the limits of the plot Page
     plotSize = 0.9 # how much in percentace should the plotpage be larger
                    # than the plotted values?
     if nu_or_lambda=='nu':
         xLimNu = [min(self.flux_array[0]) -
                   min(self.flux_array[0]) * plotSize,
                   max(self.flux_array[0]) + max(self.flux_array[0])]
     if nu_or_lambda == 'lambda':
         newSelf_Flux_Array = []
         for i in self.flux_array[0]:
             newSelf_Flux_Array += [astro_functions.frequency_to_wavelength(i)]
         self.flux_array[0] = newSelf_Flux_Array
         xLimNu = [min(self.flux_array[0]) -
                   min(self.flux_array[0]) * 0.5 * plotSize,
                   max(self.flux_array[0]) +
                   max(self.flux_array[0]) * 2 *plotSize]
     ylim = [min(self.flux_array[1]) - min(self.flux_array[1]) * plotSize,
             max(self.flux_array[1]) +
             max(self.flux_array[1]) * plotSize / 2]
     # makes the plot page squared; TBD not really square yet
     axes.set_xlim(xLimNu[0],xLimNu[1])
     axes.set_ylim(ylim[0],ylim[1])
     return xLimNu,ylim
示例#2
0
文件: sed.py 项目: buchbend/astrolyze
    def create_figure(self, save=True, plotLegend=False,
                      color=['black'], marker=['x'], title=None, x_label=None,
                      y_label=None, nu_or_lambda='nu', fontdict=None,
                      textStringLoc=[1,1], lineWidth=0.5,
                      kappa='easy', x_range='normal', prefix="./", **kwargs):
        r""" Creates a quick preview of the loaded SED. TODO: extend
        documentation.
        """
        fig1 = plt.figure()
        fig1ax1 = fig1.add_subplot(111)
        textString = ''
        for i in range(len(self.p2[0])):
            textString += ('T=' + str('%1.1f' % self.p2[0][i]) +
                           ' K\nM=' + str("%1.2e" % self.p2[1][i]) + ' Msun\n')

        if len(self.p2[0])==2:
            textString+= 'N1/N2 = '+str('%i'%(self.p2[1][0]/self.p2[1][1]))+'\n'
        textString += ('beta = ' + str("%1.2f" % self.p2[2][0]) +
                       '\nchi$^2$ =' + str("%1.2f" % self.fit_chisq) + '\n')
        # sets the limits of the plot Page
        plotSize = 0.9 # how much in percentace should the plotpage be larger
                       # than the plotted values?
        if nu_or_lambda=='nu':
            xLimNu = [min(self.flux_array[0]) -
                      min(self.flux_array[0]) * plotSize,
                      max(self.flux_array[0]) + max(self.flux_array[0])]

        if nu_or_lambda == 'lambda':
            newSelf_Flux_Array = []
            for i in self.flux_array[0]:
                print i
                newSelf_Flux_Array += [astro_functions.frequency_to_wavelength(i)]

            self.flux_array[0] =newSelf_Flux_Array
            xLimNu = [min(self.flux_array[0]) -
                      min(self.flux_array[0]) * plotSize * 2,
                      max(self.flux_array[0]) +
                      max(self.flux_array[0]) * plotSize]
        ylim = [min(self.flux_array[1]) - min(self.flux_array[1]) * plotSize,
                max(self.flux_array[1]) +
                max(self.flux_array[1]) * plotSize / 2]
        # makes the plot page squared; TBD not really square yet
        fig1ax1.set_xlim(xLimNu[0],xLimNu[1])
        fig1ax1.set_ylim(ylim[0],ylim[1])

        # PLots the model given in self.p2
        self.plot_sed(axes=plt.gca(), nu_or_lambda=nu_or_lambda, color='black',
                      linewidth=0.5, x_range=x_range)

        markersize =7
        #Plotting the data points
        parted = [1]
        if len(parted)==1:
            fig1ax1.errorbar(self.flux_array[0], self.flux_array[1],
                         yerr=self.flux_array[2], fmt='o', marker='p',
                         mfc='None', mew=0.5, mec='#00ffff', ms=markersize,
                         color='black', lw=lineWidth)
        else:
            for i in range(len(parted)):
                if i == 0:
                    fig1ax1.errorbar(self.flux_array[0][0:parted[i]],
                                 self.flux_array[1][0:parted[i]],
                                 yerr=self.flux_array[2][0:parted[i]],
                                 fmt=marker[i],
                                 marker=marker[i], mfc='None', label=label[i],
                                 mew=0.5, mec=color[i], ms=markersize,
                                 color=color[i], lw=lineWidth)
                else:
                    fig1ax1.errorbar(self.flux_array[0][parted[i-1]:parted[i]],
                                self.flux_array[1][parted[i-1]:parted[i]],
                                yerr=self.flux_array[2][parted[i-1]:parted[i]],
                                fmt=marker[i], marker=marker[i],
                                mfc='None', label=label[i], mew=0.5,
                                mec=color[i], ms=markersize, color=color[i],
                                lw=lineWidth)

        # setting up legend,title, xlabel.
        if plotLegend == 'yes':
            fontdict={'size':'11'}
            plt.legend(loc='upper right', numpoints=1, fancybox=False,
                      prop=fontdict, markerscale=1)
        fontdict={'size':'17'}
        plt.text(90,0.2,s=textString, fontdict=fontdict, alpha=0.4)
        if title:
            fig1ax1.title(title)
        if x_label:
            fig1ax1.xlabel(x_label)
        if y_label:
            fig1ax1.ylabel(y_label)
        fig1ax1.axis([xLimNu[0],xLimNu[1],ylim[0],ylim[1]])
        if save:
            fig1.savefig(prefix + self.source_name + '_SED.eps', dpi=None,
                         facecolor='w', edgecolor='w',orientation='portrait',
                         papertype='a5', format='eps',bbox_inches='tight')