run.py

"""
  fracture: Simulation of a scalar fracture model based on Freitas (2007).
  Copyright (C) 2017 Italo Silva

  This program is free software: you can redistribute it and/or modify
  it under the terms of the GNU General Public License as published by
  the Free Software Foundation, either version 3 of the License, or
  (at your option) any later version.

  This program is distributed in the hope that it will be useful,
  but WITHOUT ANY WARRANTY; without even the implied warranty of
  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  GNU General Public License for more details.

  You should have received a copy of the GNU General Public License
  along with this program.  If not, see <http://www.gnu.org/licenses/>.
"""

import argparse
import matplotlib.pyplot as plt
import numpy as np
import os
import sys
from subprocess import run, DEVNULL

opt = argparse.ArgumentParser(
    description='Execute fracture simulations.',
    formatter_class=argparse.ArgumentDefaultsHelpFormatter
)

opt.add_argument(
    '-g', '--geometry',
    nargs='+',
    choices=['t', 'h', 's'],
    default=['t']
)

opt.add_argument(
    '-l', '--length',
    nargs='+',
    type=int,
    default=[14]
)

opt.add_argument(
    '-d', '--disorder',
    nargs='+',
    type=float,
    default=[0, 0.2, 0.4, 0.6, 1.0]
)

opt.add_argument(
    '--show',
    action='store_true'
)

opt.add_argument(
    '--graph',
    action='store_true'
)

datafolder = 'data'
pdffolder = 'pdf'
graphfolder = 'graphs'
results = 'output.csv'
pdf = 'output.pdf'
pdf2 = 'result.pdf'
graph = 'graph-begin.dot'
graphend = 'graph-end.dot'

os.makedirs(datafolder, exist_ok=True)
os.makedirs(pdffolder, exist_ok=True)
os.makedirs(graphfolder, exist_ok=True)

exename = 'fracture'
if sys.platform == 'win32':
    exename += '.exe'

exe = os.path.join('out', exename)

plt.style.use('ggplot')

args = opt.parse_args();

for G in args.geometry:
    for L in args.length:
        print('')
        plt.figure()
        plt.suptitle(f'L = {L}, G = {G}')

        for D in args.disorder:
            run(f'{exe} {L} {D} {G}')

            if args.graph:
                begin_path = os.path.join(pdffolder, str(D) + pdf)
                run(f'dot {graph} -Tpdf -o{begin_path}')
                os.remove(graph)
                run(f'start {begin_path}', shell=True)

                end_path = os.path.join(pdffolder, str(D) + pdf2)
                run(f'dot {graphend} -Tpdf -o{end_path}')
                os.remove(graphend)
                run(f'start {end_path}', shell=True)

            result_path = os.path.join(datafolder, f'{L}.{G}.{D}.csv')
            os.replace(results, result_path)

            out = np.genfromtxt(result_path, delimiter=',',
                                skip_header=1, names=['V', 'I'])
            plt.plot(out['V'], out['I'], label=f'D={int(D*100)}%')

        plt.legend(loc='upper left')
        if args.show:
            plt.show()
        fig_path = os.path.join(graphfolder, f'{G}.{L}.png')
        plt.savefig(fig_path, bbox_inches='tight')