#!/usr/bin/env python3
# -*- coding: utf-8 -*-
import os
import shutil
import numpy as np
import pandas as pd
import radmc3dPy
import radmc3dPy.analyze as rmca
from envos import tools
import envos.nconst as nc
from envos import gpath
from envos.log import set_logger
logger = set_logger(__name__)
"""
Classes
"""
class RadmcController:
def __init__(
self,
config=None,
run_dir: str = None,
radmc_dir: str = None,
storage_dir: str = None,
#
):
if config is not None:
import os
import numpy as np
from dataclasses import dataclass, field
from scipy import interpolate, integrate
from envos import log
from envos.gpath import run_dir
import envos.nconst as nc
logger = log.set_logger(__name__)
def calc_physical_values_along_streamlines(model, start_points, **kwargs):
calc_streamlines_from_model(model, ["rhogas", "Tgas"], ["g cm^-3", "K"],
start_points, **kwargs)
def calc_streamlines_from_model(model, name_list, unit_list, start_points,
**kwargs):
values = [(n, getattr(model, n), u) for n, u in zip(name_list, unit_list)]
calc_streamlines(
model.rc_ax,
model.tc_ax,
model.vr[:, :, 0],
model.vt[:, :, 0],
start_points,
values=values,
**kwargs,
)
def calc_streamlines(