Пример #1
0
    def read(self, mesh, **kwargs):

        self.fd = fd = open(self.filename, "r")
        mode = "header"

        nod = conns = desc = None
        while 1:
            if mode == "header":
                line = skip_read_line(fd)

                n_tags = self._read_commented_int()
                for ii in xrange(n_tags):
                    skip_read_line(fd)
                n_types = self._read_commented_int()
                for ii in xrange(n_types):
                    skip_read_line(fd)

                skip_read_line(fd)
                assert_(skip_read_line(fd).split()[1] == "Mesh")
                skip_read_line(fd)
                dim = self._read_commented_int()
                assert_((dim == 2) or (dim == 3))
                n_nod = self._read_commented_int()
                i0 = self._read_commented_int()
                mode = "points"

            elif mode == "points":
                nod = read_array(fd, n_nod, dim, nm.float64)
                mode = "cells"

            elif mode == "cells":

                n_types = self._read_commented_int()
                conns = []
                descs = []
                mat_ids = []
                for it in xrange(n_types):
                    t_name = skip_read_line(fd).split()[1]
                    n_ep = self._read_commented_int()
                    n_el = self._read_commented_int()
                    if dim == 2:
                        aux = read_array(fd, n_el, n_ep, nm.int32)
                        if t_name == "tri":
                            conns.append(aux)
                            descs.append("2_3")
                            is_conn = True
                        else:
                            is_conn = False
                    else:
                        raise NotImplementedError

                    # Skip parameters.
                    n_pv = self._read_commented_int()
                    n_par = self._read_commented_int()
                    for ii in xrange(n_par):
                        skip_read_line(fd)

                    n_domain = self._read_commented_int()
                    assert_(n_domain == n_el)
                    if is_conn:
                        mat_id = read_array(fd, n_domain, 1, nm.int32)
                        mat_ids.append(mat_id)
                    else:
                        for ii in xrange(n_domain):
                            skip_read_line(fd)

                    # Skip up/down pairs.
                    n_ud = self._read_commented_int()
                    for ii in xrange(n_ud):
                        skip_read_line(fd)
                break

        nod = nm.concatenate((nod, nm.zeros((n_nod, 1), dtype=nm.int32)), 1)
        nod = nm.ascontiguousarray(nod)

        dim = nod.shape[1] - 1

        conns2 = []
        for ii, conn in enumerate(conns):
            conns2.append(nm.c_[conn, mat_ids[ii]])

        conns_in, mat_ids = sort_by_mat_id(conns2)
        conns, mat_ids, descs = split_by_mat_id(conns_in, mat_ids, descs)
        mesh._set_data(nod, conns, mat_ids, descs)

        #        mesh.write( 'pokus.mesh', io = 'auto' )

        self.fd = None
        return mesh
Пример #2
0
 def _read_commented_int(self):
     return int(skip_read_line(self.fd).split("#")[0])