runscript specified for a_80

import sys

import os

import numpy as np

import mechnet as mn

from multiprocessing import Pool

from functools import partial

sys.path.append("/home/lpyka/hierarchical_interface/")

from lib.push_data import push_data_to, copy_runfile_to 

from lib.my_plot_edges import plot_edges_mechnet_network 

os.environ["OMP_NUM_THREADS"] = "1"

os.environ["OPENBLAS_NUM_THREADS"] = "1"

os.environ["MKL_NUM_THREADS"] = "1"

def get_parametercollection(seed,crack_length, offset):

    par = mn.ParameterCollection()

    nx = 128 # needs to == el**h

    ny = nx

    nz = 9 + offset # need to be >=  h +2 + o_set (+2 bc bound cond)    

    n = nx*ny*nz

    par.set_N(n)

    par.set_Nx(nx)

    par.set_Ny(ny)

    par.set_Nz(nz)

    par.set_NxNy()

    #set precracks x_start=0, x_end=crack_length, y_start = 0, y_end=ny

    par.set_listofhorizontalprecracks([[0,crack_length, 0, ny-1, offset]])

    #new parameters necessary for the structure

    par.set_xhierarchicalelementsize(2) # log_el(nx) = h <-> nx = el**h 

    par.set_yhierarchicalelementsize(2)

    par.set_gapzoffset(offset) # == o_set

    par.set_thresholdrng(seed)

    par.set_structurerng(seed + 1_000_000)

    par.set_weibullparameter(4)

    par.set_scalingfactor(1.00)

    par.set_uniformupperdirichlet(1.0)

    return par

def run_a_simulation_fuse(seed, crack_length, offset, targetdirectory):

    constructor = mn.cubic3D.Cubic3DFullConnectionConstructor()

    #new decorators to modify network structure:

    #random positioning of missing edges fitting to the amount that would be missing in a hierarchical network with

    #element sizes as determined by set_xhierarchicalelementsize() and set_yhierarchicalelementsize()

    #in the corresponding directions; arangement dependent on seed given by set_structurerng()

    #for the random positioning

    constructor = mn.cubic3D.ShuffledHierarchicalOffsetDecorator(constructor)

    constructor = mn.cubic3D.InitialHorizontalPrecracksDecorator(constructor)

    constructor = mn.cubic3D.ScalingWeibullThresholdDecorator(constructor)

    boundaries = mn.general.EmptyBoundariesConstructor()

    boundaries = mn.cubic3D.FixedLowerBoundaryDecorator(boundaries)

    boundaries = mn.cubic3D.UniformDisplacedUpperBoundaryDecorator(boundaries)

    parametercollection = get_parametercollection(seed,crack_length=crack_length, offset=offset)

    network = constructor.start_construction(parametercollection)

    bounded_network = boundaries.start_assign_boundaries(network)

    #switch to fuse simulation to make it easier to read the resulting stiffness matrix

    dict_of_simulation_parameters = {"niter":10_000_000, "externalV":1.0, "accuracy":12}

    builder = mn.sim.FuseDirichletBuilder()

    solver = mn.sim.PARDISOBindingsSolver()

    intrescal = mn.sim.StartInterimResultsCalculator() #InterimResultsCalculator stack pre-computes data needed for output and simulation step

    intrescal = mn.sim.FuseCurrentsVoltagesCalculator(intrescal)

    outgen = mn.sim.StartOutputGenerator()

    outgen = mn.sim.IVCurveGenerator(outgen)

    applier = mn.sim.HottestFuseBreakApplier()

    checker = mn.sim.NonInitialChecker(mn.sim.ConnectedPathChecker())

    simulation = mn.sim.RunTillStopconditionSimulation(builder, solver, intrescal, outgen, applier, checker)

    dict_of_crs_edge_data = bounded_network.get_network_data() 

    simdata = simulation.start_simulation(dict_of_crs_edge_data, dict_of_simulation_parameters, maximum_runtime=4*85500)

    simdata_dict = simdata.get_simdata_dict()

    network_descriptor_dict = bounded_network.get_construction_data()

    mn.datman.save_simulation_output(network_descriptor_dict, simdata_dict, simulation.simulationname, simulation.simulationdoc,#

                                     suffix=str(seed), overwritemode = False, targetdirectory=targetdirectory)

    print("done with sim of seed", seed)

if __name__=="__main__":

    list_of_seeds =  []

    offset = 5

    for seed in range(1010, 1050):

            list_of_seeds.append(seed) 

    crack_lengths = [40]

    for crack_length in crack_lengths:

        destination = f"/FASTTEMP/p7/lpyka/hierarchical_interface/5_precracked/a_{crack_length}/"

        with Pool(20) as p:

            p.map(partial(run_a_simulation_fuse, crack_length=crack_length, offset=offset, targetdirectory = destination), list_of_seeds)

 #   for data in results:

 #       mn.datman.save_simulation_output(data[0], data[1], data[2], data[3], targetdirectory="./", suffix="", overwritemode=False)

    #print(data[1]["current_data"]["stiffnessmatrix"].toarray())

        copy_runfile_to(runfile_name=__file__, destination=destination)

        print(f"done with crack_lenght {crack_length}")

    print("\nRun successfull.")