def animate(self, alpha, beta, L, n, div=32, particle='de+'):
     s = settings.Settings()
     field = SineField()
     field.alpha = alpha
     field.beta = beta
     field.L = L
     field.n = n
     
     outfile = self.filename(particle, alpha, beta, L, n)
     s.outfile = outfile
     with open(s.outpath()) as f:
         t, x, y, z = extractData(f, [0, 1, 2, 3])
         length = len(z) / div
         for i in xrange(div):
             start = i * length
             if i > 0:
                 start = (i-1) * length
             end = (i+1) * length
             plt.plot(z[start:end], y[start:end], 'b-', linewidth=1, label=self.label(particle))
             xmin, xmax = [ -1.15, 1.15 ]
             ymin, ymax = [ -1.0, 1.0 ]
             self.plotField(field, xmin, xmax, ymin, ymax)
             plt.xlabel('$z$ (m)')
             plt.ylabel('$y$ (m)')
             plt.tight_layout()
             s.outext = '_{index:03}.png'.format(index=i)
             plt.text(0.25, 0.8, r'$B_z = B_{z0} \left( 1 + \alpha r + \beta z\ \sin (\frac{n \pi z}{L}) \right)$',
                      fontsize=15)
             plt.text(0.25, 0.7, '$\\alpha = {alp},\\ \\beta = {beta}$'.format(alp=alpha, beta=beta),
                      fontsize=15)
             plt.savefig(s.outpath())
             plt.clf()
    def plotSuperimposed(self, alpha, beta, L, n):
        s = settings.Settings()
        field = SineField()
        field.alpha = alpha
        field.beta = beta
        field.L = L
        field.n = n
        
        for particle in self.particles:
            outfile = self.filename(particle, alpha, beta, L, n)
            s.outfile = outfile
            with open(s.outpath()) as f:
                t, x, y, z = extractData(f, [0, 1, 2, 3])
                #start = len(z) / 4 + 250
                #end = len(z) / 4 + 180
                start = 0
                end = len(z)
                plt.plot(z[start:end], y[start:end], self.style[particle], 
                         linewidth=1, label=self.label(particle))
        xmin, xmax = [ -0.4, 0.4 ]
        ymin, ymax = [ 0.1, 0.3 ]
#        xmin, xmax = plt.xlim()
#        ymin, ymax = plt.ylim()
        self.plotField(field, xmin, xmax, ymin, ymax)
                    
        plt.xlabel('$z$ (m)')
        plt.ylabel('$y$ (m)')
#        plt.text(0.25, 0.8, r'$B_z = B_{z0} \left( 1 + \alpha r + \beta z\ \sin (\frac{n \pi z}{L}) \right)$',
#                 fontsize=15)
#        plt.text(0.25, 0.7, '$\\alpha = {alp},\\ \\beta = {beta}$'.format(alp=alpha, beta=beta),
#                 fontsize=15)
        plt.tight_layout()
        s.outext = '.pdf'
        plt.savefig(s.outpath())
        plt.show()
Exemple #3
0
    def animate(self, alpha, beta, L, n, div=32, particle='de+'):
        s = settings.Settings()
        field = SineField()
        field.alpha = alpha
        field.beta = beta
        field.L = L
        field.n = n

        outfile = self.filename(particle, alpha, beta, L, n)
        s.outfile = outfile
        with open(s.outpath()) as f:
            t, x, y, z = extractData(f, [0, 1, 2, 3])
            length = len(z) / div
            for i in xrange(div):
                start = i * length
                if i > 0:
                    start = (i - 1) * length
                end = (i + 1) * length
                plt.plot(z[start:end],
                         y[start:end],
                         'b-',
                         linewidth=1,
                         label=self.label(particle))
                xmin, xmax = [-1.15, 1.15]
                ymin, ymax = [-1.0, 1.0]
                self.plotField(field, xmin, xmax, ymin, ymax)
                plt.xlabel('$z$ (m)')
                plt.ylabel('$y$ (m)')
                plt.tight_layout()
                s.outext = '_{index:03}.png'.format(index=i)
                plt.text(
                    0.25,
                    0.8,
                    r'$B_z = B_{z0} \left( 1 + \alpha r + \beta z\ \sin (\frac{n \pi z}{L}) \right)$',
                    fontsize=15)
                plt.text(0.25,
                         0.7,
                         '$\\alpha = {alp},\\ \\beta = {beta}$'.format(
                             alp=alpha, beta=beta),
                         fontsize=15)
                plt.savefig(s.outpath())
                plt.clf()
Exemple #4
0
    def plotSuperimposed(self, alpha, beta, L, n):
        s = settings.Settings()
        field = SineField()
        field.alpha = alpha
        field.beta = beta
        field.L = L
        field.n = n

        for particle in self.particles:
            outfile = self.filename(particle, alpha, beta, L, n)
            s.outfile = outfile
            with open(s.outpath()) as f:
                t, x, y, z = extractData(f, [0, 1, 2, 3])
                #start = len(z) / 4 + 250
                #end = len(z) / 4 + 180
                start = 0
                end = len(z)
                plt.plot(z[start:end],
                         y[start:end],
                         self.style[particle],
                         linewidth=1,
                         label=self.label(particle))
        xmin, xmax = [-0.4, 0.4]
        ymin, ymax = [0.1, 0.3]
        #        xmin, xmax = plt.xlim()
        #        ymin, ymax = plt.ylim()
        self.plotField(field, xmin, xmax, ymin, ymax)

        plt.xlabel('$z$ (m)')
        plt.ylabel('$y$ (m)')
        #        plt.text(0.25, 0.8, r'$B_z = B_{z0} \left( 1 + \alpha r + \beta z\ \sin (\frac{n \pi z}{L}) \right)$',
        #                 fontsize=15)
        #        plt.text(0.25, 0.7, '$\\alpha = {alp},\\ \\beta = {beta}$'.format(alp=alpha, beta=beta),
        #                 fontsize=15)
        plt.tight_layout()
        s.outext = '.pdf'
        plt.savefig(s.outpath())
        plt.show()