~njansson/dolfin/hpc

# Modified by Kent-Andre Mardal (Star-CD function)

    elif suffix == "vrt" or suffix == "cel": 

        return "StarCD"

    elif iformat == "StarCD": 

        # Convert from Star-CD tetrahedral grid format to xml format

        starcd2xml(ifilename, ofilename)

def starcd2xml(ifilename, ofilename):

    "Convert from Star-CD tetrahedral grid format to DOLFIN XML."

 

    print starcd2xml.__doc__

 

    if not path.isfile(ifilename[:-3] + "vrt") or not path.isfile(ifilename[:-3] + "cel"):

        print "StarCD format requires one .vrt file and one .cel file"

        sys.exit(2)

      

 

    # open output file 

    ofile = open(ofilename, "w")

 

    # Open file, the vertices are in a .vrt file

    ifile = open(ifilename[:-3] + "vrt", "r")

 

    write_header_mesh(ofile, "tetrahedron", 3)

 

 

    # Read & write vertices

 

    # first, read all lines (need to sweep to times through the file)

    lines = ifile.readlines()

 

    # second, find the number of vertices

    num_vertices = -1 

    for line in lines: 

        nodenr = int(line[0:15])

        if nodenr > num_vertices: num_vertices = nodenr 

    

    

    # third, run over all vertices

    write_header_vertices(ofile, num_vertices)

    for line in lines: 

        nodenr = int(line[0:15])

        vertex0 = float(line[15:31])

        vertex1 = float(line[31:47])

        vertex2 = float(line[47:63])

        write_vertex(ofile, int(nodenr)-1, float(vertex0), float(vertex1), float(vertex2))  

    write_footer_vertices(ofile)

 

 

 

 

    # Open file, the cells are in a .cel file

    ifile = open(ifilename[:-3] + "cel", "r")

 

    # Read & write cells 

 

    # first, read all lines (need to sweep to times through the file)

    lines = ifile.readlines()

 

    # second, find the number of cells 

    num_cells = -1 

    counter = 0 

    for line in lines: 

        l = [int(a) for a in line.split()]

        cellnr, node0, node1, node2, node3, node4, node5, node6, node7, tmp1, tmp2  = l 

        if node4 > 0:

                if node2 == node3 and node4 == node5 and node5 == node6 and node6 == node7: # these nodes should be equal  

                        counter += 1

                else:            

                        print "The file does contain cells that are not tetraheders. The cell number is ", cellnr, " the line read was ", line                      

        else:                              

            print "The file does contain cells that are not tetraheders node4==0. The cell number is ", cellnr, " the line read was ", line 

            #sys.exit(2)

 

    num_cells = counter

 

    # third, run over all cells 

    write_header_cells(ofile, num_cells )

    counter = 0 

    for line in lines: 

        l = [int(a) for a in line.split()]

        cellnr, node0, node1, node2, node3, node4, node5, node6, node7, tmp1, tmp2  = l 

        if (node4 > 0): 

                if node2 == node3 and node4 == node5 and node5 == node6 and node6 == node7: # these nodes should be equal  

        

                        write_cell_tetrahedron(ofile, counter, node0-1, node1-1, node2-1, node4-1)

                        counter += 1 

 

       

    write_footer_cells(ofile)

    write_footer_mesh(ofile)

 

 

    # Close files

    ifile.close()

    ofile.close()