plot_phase_fractions.py

#!/usr/bin/env python3
#! -*- coding: utf-8 -*-

"""
Plot TTT (or CCT) diagram and transformed fraction
"""
import argparse
import matplotlib.pyplot as plt
from transformation_models import Alloy, TransformationDiagrams

if __name__ == '__main__':
    parser = argparse.ArgumentParser(description='Script for plotting phase fraction curves for a given thermal cycle',
                                     formatter_class=argparse.ArgumentDefaultsHelpFormatter)
    
    required_named = parser.add_argument_group('required arguments')
    required_named.add_argument('-Tini', '--Tini', type=float, required=True, help='Initial temperature (oC)')
    required_named.add_argument('-t', '--t', type=float, required=True, help='Total time (s)')
    required_named.add_argument('-phi', '--phi', type=float, default=0., help='Cooling rate (oC/s; if 0, isothermal)')

    parser.add_argument('-g', '--gs', type=float, default=7, help='ASTM grain size number')
    parser.add_argument('-C', '--C', type=float, default=0., help='Carbon wt.%%')
    parser.add_argument('-Si', '--Si', type=float, default=0., help='Silicon wt.%%')
    parser.add_argument('-Mn', '--Mn', type=float, default=0., help='Manganese wt.%%')
    parser.add_argument('-Ni', '--Ni', type=float, default=0., help='Nickel wt.%%')
    parser.add_argument('-Mo', '--Mo', type=float, default=0., help='Molybdenum wt.%%')
    parser.add_argument('-Cr', '--Cr', type=float, default=0., help='Chromium wt.%%')
    parser.add_argument('-V', '--V', type=float, default=0., help='Vanadium wt.%%')
    parser.add_argument('-Co', '--Co', type=float, default=0., help='Cobalt wt.%%')
    parser.add_argument('-Cu', '--Cu', type=float, default=0., help='Copper wt.%%')
    parser.add_argument('-Al', '--Al', type=float, default=0., help='Aluminium wt.%%')
    parser.add_argument('-W', '--W', type=float, default=0., help='Tungsten wt.%%')

    args = parser.parse_args()

    comp = vars(args)
    gs = comp.pop('gs')
    Tini = comp.pop('Tini')
    t = comp.pop('t')
    phi = comp.pop('phi')

    # Defines alloy (grain size gs and composition)
    alloy = Alloy(gs=gs, **comp)

    # Initializes diagrams object
    diagrams = TransformationDiagrams(alloy)

    fig, (ax1, ax2) = plt.subplots(1, 2, figsize=(12, 6))
    fig.subplots_adjust(wspace=.2)

    if phi == 0:
        # If isothermal, plot TTT
        diagrams.TTT(ax=ax1)
        xaxis = 't'
    else:
        diagrams.TTT(ax=ax1)
        t_min, t_max = ax1.get_xlim()
        ax1.clear()
        
        # Otherwise, plot CCT
        diagrams.CCT(Tini=Tini, ax=ax1, phi_min=Tini/t_max, phi_max=Tini/t_min)
        xaxis = 'T'

    t_, T_ = [0, t], [Tini, Tini - phi*t]
    diagrams.draw_thermal_cycle(ax1, t_, T_)
    diagrams.plot_phase_fraction(t_, T_, xaxis=xaxis, ax=ax2)

    fig.suptitle(ax1.get_title())
    ax1.set_title('')
    ax2.set_title('')

    plt.show()