Python openの例

コード例 #1

ファイル: optics.py プロジェクト: nbiancac/pyoptics

 def open(cls,fn):
   try:
     if fn.endswith('tfs.gz'):
       return cls(tfsdata.load(gzip.open(fn)))
     elif fn.endswith('tfs'):
       if os.path.exists(fn):
         return cls(tfsdata.open(fn))
       elif os.path.exists(fn+'.gz'):
         return cls(tfsdata.load(gzip.open(fn+'.gz')))
     else:
       return cls(tfsdata.open(fn))
     raise IOError
   except IOError:
     raise IOError,"%s does not exists or wrong format" % fn

コード例 #2

 def open(cls, fn):
     try:
         if fn.endswith('tfs.gz'):
             return cls(tfsdata.load(gzip.open(fn)))
         elif fn.endswith('tfs'):
             if os.path.exists(fn):
                 return cls(tfsdata.open(fn))
             elif os.path.exists(fn + '.gz'):
                 return cls(tfsdata.load(gzip.open(fn + '.gz')))
         else:
             return cls(tfsdata.open(fn))
         raise IOError
     except IOError:
         raise IOError, "%s does not exists or wrong format" % fn

コード例 #3

ファイル: madbeam.py プロジェクト: Landau1908/pyoptics

 def open(cls, filename, twissfile=None):
     self = cls(**tfsdata.open(filename))
     self.filename = filename
     if twissfile is not None:
         self.twiss = optics.open(twissfile)
     self.twissfile = twissfile
     return self

コード例 #4

ファイル: optics.py プロジェクト: nbiancac/pyoptics

 def reload(self):
   if 'filename' in self._data:
      fdate=os.stat(self.filename).st_ctime
      if fdate>self._fdate:
        self._data=tfsdata.open(self.filename)
        self._fdate=fdate
        return True
   return False

コード例 #5

 def reload(self):
     if 'filename' in self._data:
         fdate = os.stat(self.filename).st_ctime
         if fdate > self._fdate:
             self._data = tfsdata.open(self.filename)
             self._fdate = fdate
             return True
     return False

コード例 #6

ファイル: optics.py プロジェクト: nbiancac/pyoptics

 def __init__(self,dynapfn,nangles=7,nsigma=12,label='dynap'):
   self.label=label.replace('_',' ')
   t=tfsdata.open(dynapfn)
   self.tunx=t['tunx']
   self.tuny=t['tuny']
   self.tx=t['x']
   self.ty=t['y']
   self.nangles=nangles
   self.nsigma=nsigma

コード例 #7

 def __init__(self, dynapfn, nangles=7, nsigma=12, label='dynap'):
     self.label = label.replace('_', ' ')
     t = tfsdata.open(dynapfn)
     self.tunx = t['tunx']
     self.tuny = t['tuny']
     self.tx = t['x']
     self.ty = t['y']
     self.nangles = nangles
     self.nsigma = nsigma

コード例 #8

ファイル: foot.py プロジェクト: piotrskowronski/pyoptics

 def __init__(self, base, nangles=7, nsigma=12):
     self.base = base
     self.label = base.replace('_', ' ')
     dynapfn = os.path.join(base, 'dynaptune.tfs')
     t = tfsdata.open(dynapfn)
     self.tunx = t['tunx']
     self.tuny = t['tuny']
     self.tx = t['x']
     self.ty = t['y']
     self.nangles = nangles
     self.nsigma = nsigma

コード例 #9

ファイル: foot.py プロジェクト: nbiancac/pyoptics

 def __init__(self,base,nangles=7,nsigma=12):
   self.base=base
   self.label=base.replace('_',' ')
   dynapfn=os.path.join(base,'dynaptune.tfs')
   t=tfsdata.open(dynapfn)
   self.tunx=t['tunx']
   self.tuny=t['tuny']
   self.tx=t['x']
   self.ty=t['y']
   self.nangles=nangles
   self.nsigma=nsigma

コード例 #10

ファイル: squeeze_result.py プロジェクト: Landau1908/pyoptics

 def open(cls,fn):
     data=tfsdata.open(fn)
     data['filename']=fn
     return cls(**data)

コード例 #11

 def open(cls, fn):
     obj = cls(tfsdata.open(fn))
     return obj

コード例 #12

import sys; sys.path.append('../')

import sixtracklib
import numpy as np

from scipy.constants import c as c_light
pmass_eV = 938.272046e6

V_RF = 10e6
lag_deg = 180
h_RF = 35000


import tfsdata

twdict = tfsdata.open('twiss.out')

machine = sixtracklib.CBlock()

p0c_eV = 450e9
gamma0 = np.sqrt(p0c_eV**2+pmass_eV**2)/pmass_eV
beta0 = p0c_eV/np.sqrt(p0c_eV**2+pmass_eV**2)

length = twdict['param']['length'] 
f_RF = h_RF*c_light*beta0/(length)

# I want to start in a place with 0 dispersion
start_at = 'AT_IP5'
i_start = np.where(twdict['name']==start_at)[0][0]

indices = range(i_start, len(twdict['name']))+range(0, i_start)

コード例 #13

ファイル: strtable.py プロジェクト: nbiancac/pyoptics

 def open(cls,fn):
   obj=cls(tfsdata.open(fn))
   return obj