コード例 #1
0
ファイル: fields.py プロジェクト: fangohr/nmag-src
    def __init__(self,rawfield,name):
        self._name = name
        self._rawfield = rawfield

        #learn about degrees of freedom:
        tmp_dof_maxind_by_name = {}

        self.dof_names = []
        self.dofs = []
        
        dof_maxnames = ocaml.data_doftypes(self._rawfield)
        for dofname,maxind in dof_maxnames:
            tmp_dof_maxind_by_name[dofname] = maxind

        #populate degrees of freedom
        vecdofs = {}

        def cb(i, dof_name_stem, site, pos, value):
            site = tuple(site)
            dofname, ind = dof_name_stem

            #have we seen this dof before?
            if not vecdofs.has_key(dofname):
                vecdofs[dofname]={}

            #have we seen this site before?
            if not vecdofs[dofname].has_key(site):
                maxind = tmp_dof_maxind_by_name[dofname]
                data = make_dof_list_structure(maxind,None)
                dof_id = copy.copy(data)
                vecdofs[dofname][site] = [data,pos,site,dof_id]

            if ind==[]: #scalar
                vecdofs[dofname][site][0] = value
                vecdofs[dofname][site][3] = i
            else:
                vecdofs[dofname][site][0][ind[0]] = value
                vecdofs[dofname][site][3][ind[0]] = i

        nfem.field_entry_wise( self._rawfield, cb )

        #now sort into neat vectors
        for dofname,maxind in dof_maxnames:
            tmp = vecdofs[dofname].values()
            data = map( lambda a : a[0], tmp)
            pos = map( lambda a : a[1], tmp)
            site = map( lambda a : a[2], tmp)
            dof_id = map( lambda a : a[3], tmp)

            tmpdof = VectorDOF(dofname,data,pos,site,dof_id,maxind)
            self.__setattr__(dofname,tmpdof)
            self.dof_names.append(dofname)
            self.dofs.append(tmpdof)
コード例 #2
0
    def __init__(self, rawfield, name):
        self._name = name
        self._rawfield = rawfield

        #learn about degrees of freedom:
        tmp_dof_maxind_by_name = {}

        self.dof_names = []
        self.dofs = []

        dof_maxnames = ocaml.data_doftypes(self._rawfield)
        for dofname, maxind in dof_maxnames:
            tmp_dof_maxind_by_name[dofname] = maxind

        #populate degrees of freedom
        vecdofs = {}

        def cb(i, dof_name_stem, site, pos, value):
            site = tuple(site)
            dofname, ind = dof_name_stem

            #have we seen this dof before?
            if not vecdofs.has_key(dofname):
                vecdofs[dofname] = {}

            #have we seen this site before?
            if not vecdofs[dofname].has_key(site):
                maxind = tmp_dof_maxind_by_name[dofname]
                data = make_dof_list_structure(maxind, None)
                dof_id = copy.copy(data)
                vecdofs[dofname][site] = [data, pos, site, dof_id]

            if ind == []:  #scalar
                vecdofs[dofname][site][0] = value
                vecdofs[dofname][site][3] = i
            else:
                vecdofs[dofname][site][0][ind[0]] = value
                vecdofs[dofname][site][3][ind[0]] = i

        nfem.field_entry_wise(self._rawfield, cb)

        #now sort into neat vectors
        for dofname, maxind in dof_maxnames:
            tmp = vecdofs[dofname].values()
            data = map(lambda a: a[0], tmp)
            pos = map(lambda a: a[1], tmp)
            site = map(lambda a: a[2], tmp)
            dof_id = map(lambda a: a[3], tmp)

            tmpdof = VectorDOF(dofname, data, pos, site, dof_id, maxind)
            self.__setattr__(dofname, tmpdof)
            self.dof_names.append(dofname)
            self.dofs.append(tmpdof)
コード例 #3
0
def data_doftypes(x):
    """Given a raw ocaml field, this returns a list of pairs. Each pair contains first
    the name of the subfield and second the shape information for this subfield."""
    result = ocaml.data_doftypes(x)
    #log.debug("doftypes of XXX is %s" % str(result))
    return result
コード例 #4
0
ファイル: main.py プロジェクト: fangohr/nmag-src
def data_doftypes(x):
    """Given a raw ocaml field, this returns a list of pairs. Each pair contains first
    the name of the subfield and second the shape information for this subfield."""
    result = ocaml.data_doftypes(x)
    # log.debug("doftypes of XXX is %s" % str(result))
    return result