read.particles Class Reference

Load binary particles data. More...

Public Member Functions
def	__init__ (self, offset, file='data', components=3)
def	makefilename (self)
def	openfile (self)
def	closefile (self)
def	readheader (self)
def	read_next_particle (self)
def	read (self)
def	particle (self, index)
def	get_sorted (self, i)

Public Attributes
	offset
	filenameout
	isLoaded
	components
	data
	mynparticles
	filename
	file
	index
	sorted

Detailed Description

Load binary particles data.

Definition at line 590 of file read.py.

Constructor & Destructor Documentation

__init__()

def read.particles.__init__	(		self,
			offset,
			file = 'data',
			components = 3
	)

Definition at line 592 of file read.py.

    def __init__(self,offset,file='data',components=3):
        self.offset=offset
        self.filenameout = file
        self.isLoaded = False
        self.components = components

        self.data=[]
        self.mynparticles = 0

        self.makefilename()
        self.openfile()
        self.read()
        self.closefile()

Member Function Documentation

closefile()

def read.particles.closefile

(

self

)

Definition at line 616 of file read.py.

    def closefile(self):
        self.file.close()
        self.isLoaded = False

get_sorted()

def read.particles.get_sorted	(		self,
			i
	)

Definition at line 675 of file read.py.

    def get_sorted(self,i):
        try:
            self.sorted
        except AttributeError:
            self.index=np.array(self.nparticles)
            index=[]
            for particle in self.data:
                index.append(particle['index'])
            index=np.array(index)
            self.sorted=index.argsort()
        return self.sorted[i-1]

#=============================================================================

makefilename()

def read.particles.makefilename

(

self

)

Definition at line 607 of file read.py.

    def makefilename(self):
        self.filename = self.filenameout+'_particles'+str(self.offset).zfill(4)+'.dat'

openfile()

def read.particles.openfile

(

self

)

Definition at line 611 of file read.py.

    def openfile(self):
        self.file = open(self.filename,'rb')
        self.isLoaded = True

particle()

def read.particles.particle	(		self,
			index
	)

Definition at line 668 of file read.py.

    def particle(self,index):
        for particle in self.data:
            if particle['index'] == index :
                return particle
        return False

read()

def read.particles.read

(

self

)

Definition at line 662 of file read.py.

    def read(self):
        self.readheader()
        while (self.mynparticles < self.nparticles):
            self.read_next_particle()

read_next_particle()

def read.particles.read_next_particle

(

self

)

Definition at line 628 of file read.py.

    def read_next_particle(self):
        x       = np.empty(self.components)
        u       = np.empty(self.components)
        payload = np.empty(self.npayload)

        (index,)  = struct.unpack('i',self.file.read(4))

        (ifollow,) = struct.unpack('i',self.file.read(4))
        if (ifollow == -1):
            follow = True
        else:
            follow = False

        (q,)      = struct.unpack('d',self.file.read(8))
        (m,)      = struct.unpack('d',self.file.read(8))
        (t,)      = struct.unpack('d',self.file.read(8))
        (dt,)     = struct.unpack('d',self.file.read(8))

        for icomp in range(self.components):
            (x[icomp],) = struct.unpack('d',self.file.read(8))

        for icomp in range(self.components):
            (u[icomp],) = struct.unpack('d',self.file.read(8))

        for ipayload in range(self.npayload):
            (payload[ipayload],) = struct.unpack('d',self.file.read(8))
        
        self.data.append({'index':index, 'q':q, 
                          'follow':follow, 'm':m, 't':t, 
                          'dt':dt, 'x':x, 'u':u, 'payload':payload})
        
        self.mynparticles= self.mynparticles + 1

readheader()

def read.particles.readheader

(

self

)

Definition at line 621 of file read.py.

    def readheader(self):
        self.file.seek(0)
        (self.nparticles,) = struct.unpack('i',self.file.read(4))
        (self.itparticles,) = struct.unpack('i',self.file.read(4))
        (self.npayload,) = struct.unpack('i',self.file.read(4))

Member Data Documentation

components

read.particles.components

Definition at line 596 of file read.py.

data

read.particles.data

Definition at line 598 of file read.py.

file

read.particles.file

Definition at line 612 of file read.py.

filename

read.particles.filename

Definition at line 608 of file read.py.

filenameout

read.particles.filenameout

Definition at line 594 of file read.py.

index

read.particles.index

Definition at line 679 of file read.py.

isLoaded

read.particles.isLoaded

Definition at line 595 of file read.py.

mynparticles

read.particles.mynparticles

Definition at line 599 of file read.py.

offset

read.particles.offset

Definition at line 593 of file read.py.

sorted

read.particles.sorted

Definition at line 684 of file read.py.

---

The documentation for this class was generated from the following file:

• read.py