#!/usr/bin/env python3 import os import sys from optparse import OptionParser import numpy as np import damask scriptName = os.path.splitext(os.path.basename(__file__))[0] scriptID = ' '.join([scriptName,damask.version]) minimal_surfaces = ['primitive','gyroid','diamond'] surface = { 'primitive': lambda x,y,z: np.cos(x)+np.cos(y)+np.cos(z), 'gyroid': lambda x,y,z: np.sin(x)*np.cos(y)+np.sin(y)*np.cos(z)+np.cos(x)*np.sin(z), 'diamond': lambda x,y,z: np.cos(x-y)*np.cos(z)+np.sin(x+y)*np.sin(z), } # -------------------------------------------------------------------- # MAIN # -------------------------------------------------------------------- parser = OptionParser(option_class=damask.extendableOption, usage='%prog options [geomfile]', description = """ Generate a bicontinuous structure of given type. """, version = scriptID) parser.add_option('-t','--type', dest = 'type', choices = minimal_surfaces, metavar = 'string', help = 'type of minimal surface [primitive] {%s}' %(','.join(minimal_surfaces))) parser.add_option('-f','--threshold', dest = 'threshold', type = 'float', metavar = 'float', help = 'threshold value defining minimal surface [%default]') parser.add_option('-g', '--grid', dest = 'grid', type = 'int', nargs = 3, metavar = 'int int int', help = 'a,b,c grid of hexahedral box [%default]') parser.add_option('-s', '--size', dest = 'size', type = 'float', nargs = 3, metavar = 'float float float', help = 'x,y,z size of hexahedral box [%default]') parser.add_option('-p', '--periods', dest = 'periods', type = 'int', metavar = 'int', help = 'number of repetitions of unit cell [%default]') parser.add_option('--homogenization', dest = 'homogenization', type = 'int', metavar = 'int', help = 'homogenization index to be used [%default]') parser.add_option('--m', dest = 'microstructure', type = 'int', nargs = 2, metavar = 'int int', help = 'two microstructure indices to be used [%default]') parser.set_defaults(type = minimal_surfaces[0], threshold = 0.0, periods = 1, grid = (16,16,16), size = (1.0,1.0,1.0), homogenization = 1, microstructure = (1,2), ) (options,filenames) = parser.parse_args() if filenames == []: filenames = [None] for name in filenames: damask.util.report(scriptName,name) X = options.periods*2.0*np.pi*(np.arange(options.grid[0])+0.5)/options.grid[0] Y = options.periods*2.0*np.pi*(np.arange(options.grid[1])+0.5)/options.grid[1] Z = options.periods*2.0*np.pi*(np.arange(options.grid[2])+0.5)/options.grid[2] microstructure = np.empty(options.grid,dtype='int') for x in range(options.grid[0]): for y in range(options.grid[1]): for z in range(options.grid[2]): microstructure[x,y,z]=options.microstructure[options.threshold < surface[options.type](X[x],Y[y],Z[z])] geom=damask.Geom(microstructure,options.size,options.homogenization, comments=[scriptID + ' ' + ' '.join(sys.argv[1:])]) damask.util.croak(geom) if name is None: sys.stdout.write(str(geom.show())) else: geom.to_file(name)