コード例 #1
0
ファイル: Model_VI.py プロジェクト: favba/python_tools
            modb = np.sqrt(
                (((2 / 3) * t11[i, j] - (1 / 3) * (t22[i, j] + t33[i, j]))**2 +
                 2 * t12[i, j]**2 + 2 * t13[i, j]**2 +
                 ((2 / 3) * t22[i, j] - (1 / 3) * (t11[i, j] + t33[i, j]))**2 +
                 2 * t23[i, j]**2 + ((2 / 3) * t33[i, j] - (1 / 3) *
                                     (t11[i, j] + t22[i, j]))**2))

            index1[i, j] = 1 - hpi * np.arccos(modm / modb)
            # index1[i,j] = modm/modb

    return


Fil = sys.argv[1]
N = sys.argv[2]
nx, ny, nz, xDomainSize, yDomainSize, zDomainSize = readglobal(
    path="../../../")

dtype = 'Float64'
if os.path.getsize(Fil + 'P11_N' + N) == nx * ny * nz * 4: dtype = 'Float32'

d11 = np.memmap(Fil + 'P11_N' + N,
                shape=(nx, ny, nz),
                order='F',
                dtype=dtype,
                mode='r')
d12 = np.memmap(Fil + 'P12_N' + N,
                shape=(nx, ny, nz),
                order='F',
                dtype=dtype,
                mode='r')
d13 = np.memmap(Fil + 'P13_N' + N,
コード例 #2
0
ファイル: vectordirec.py プロジェクト: favba/python_tools
#!/usr/bin/env python
import numpy as np
import sys
import os
from globalfile import readglobal
import subprocess

if os.path.exists('./global'): path='./'
elif os.path.exists('../global'): path='../'
elif os.path.exists('../../global'): path='../../'
elif os.path.exists('../../../global'): path='../../../'
elif os.path.exists('../../../../global'): path='../../../../'
elif os.path.exists('../../../../../global'): path='../../../../../'

nx,ny,nz,xDomainSize,yDomainSize,zDomainSize = readglobal(path=path)

files = sys.argv[1:4]
output = sys.argv[4:]

subprocess.call(["truncate","-s",str(nx*ny*nz)]+output)
subprocess.call(["vectordir",str(nx*ny*nz)]+files+output)
コード例 #3
0
ファイル: makemodT.py プロジェクト: favba/python_tools
  modt[:,:] = np.sqrt(((2/3)*t11-(1/3)*(t22+t33))**2 + 2*t12**2 + 2*t13**2 + ((2/3)*t22-(1/3)*(t11+t33))**2 + 2*t23**2 + ((2/3)*t33-(1/3)*(t11+t22))**2)
#  modt[:,:] = modt[:,:]/modt[:,:].max()
  return

@jit(nogil=True)
def tcalcmod2(t11,t12,t13,t22,t23,t33,modt):
  modt[:,:] = np.sqrt(t11**2 + 2*t12**2 + 2*t13**2 + t22**2 + 2*t23**2 + t33**2)
#  modt[:,:] = modt[:,:]/modt[:,:].max()
  return


if len(sys.argv) == 3 :
  traceless = False
  Fil = sys.argv[1]
  N = sys.argv[2]
  nx,ny,nz,xDomainSize,yDomainSize,zDomainSize = readglobal("../../../")

  sizefile = os.path.getsize(Fil+'T11_N'+N)
  if sizefile == nx*ny*nz*8 :
    dtype='float64'
  elif sizefile == nx*ny*nz*4 :
    dtype='float32'

  t11 = np.memmap(Fil+'T11_N'+N,shape=(nx,ny,nz),order='F',dtype=dtype,mode='r')
  t12 = np.memmap(Fil+'T12_N'+N,shape=(nx,ny,nz),order='F',dtype=dtype,mode='r')
  t13 = np.memmap(Fil+'T13_N'+N,shape=(nx,ny,nz),order='F',dtype=dtype,mode='r')
  t22 = np.memmap(Fil+'T22_N'+N,shape=(nx,ny,nz),order='F',dtype=dtype,mode='r')
  t23 = np.memmap(Fil+'T23_N'+N,shape=(nx,ny,nz),order='F',dtype=dtype,mode='r')
  t33 = np.memmap(Fil+'T33_N'+N,shape=(nx,ny,nz),order='F',dtype=dtype,mode='r')

  modt = np.memmap(Fil+'modT_N'+N,shape=(nx,ny,nz),order='F',dtype=dtype,mode='w+')
コード例 #4
0
                   action='store_true',
                   help="Use normal distribution")
group.add_argument('-u',
                   '--uniform',
                   action='store_true',
                   help="Use uniform distribution")

if os.path.exists('./global'): path = './'
elif os.path.exists('../global'): path = '../'
elif os.path.exists('../../global'): path = '../../'
elif os.path.exists('../../../global'): path = '../../../'
elif os.path.exists('../../../../global'): path = '../../../../'
elif os.path.exists('../../../../../global'): path = '../../../../../'
else: path = input('Please specify global file directory path: ')

nx, ny, nz, xDomainSize, yDomainSize, zDomainSize = readglobal(path=path,
                                                               tell=False)

args = parser.parse_args()

boxdim = 6 * 2 * zDomainSize * np.pi / nz

for output in args.file:
    a = np.asfortranarray(np.random.normal(size=(nx, ny, nz)))
    filterbig(a,
              xDomainSize,
              yDomainSize,
              zDomainSize,
              boxdim,
              fil='G',
              dtype='float64')
    if args.uniform: