# Path: DiscEvolution/star.py
# class PhotoStar(SimpleStar):
#     def __init__(self, LX=1e30, Phi=0, **kwargs):
#         super().__init__(**kwargs)
#         self._L_X = LX
#         self._Phi = Phi
# 
#     @property
#     def L_X(self):
#         """X-ray Luminosity"""
#         return self._L_X
# 
#     @property
#     def Phi(self):
#         """EUV Photon Luminosity"""
#         return self._Phi


import numpy as np
import argparse
import json
import matplotlib.pyplot as plt
from DiscEvolution.constants import *
from DiscEvolution.star import PhotoStar
from scipy.signal import argrelmin
    from control_scripts import run_model
Designed for plotting to test things out
"""""""""
#################################################################################
        
class DummyDisc(object):
    def __init__(self, R, star, MD=10, RC=100):
        self._M = MD * Mjup
        self.Rc = RC
        self.R_edge = R
        self.R = 0.5*(self.R_edge[1:]+self.R_edge[:-1])
        self._Sigma = self._M / (2 * np.pi * self.Rc * self.R * AU**2) * np.exp(-self.R/self.Rc)
        self.star = star

    def Rout(self, thresh=None):
        return max(self.R_edge)

    @property
    def Sigma(self):
        return self._Sigma

    @property
    def Sigma_G(self):
        return self._Sigma

def main():
    Sigma_dot_plot()
    Test_Removal()

def Test_Removal():
    """Removes gas fom a power law disc in regular timesteps without viscous evolution etc"""

# Path: DiscEvolution/star.py
# class PhotoStar(SimpleStar):
#     def __init__(self, LX=1e30, Phi=0, **kwargs):
#         super().__init__(**kwargs)
#         self._L_X = LX
#         self._Phi = Phi
# 
#     @property
#     def L_X(self):
#         """X-ray Luminosity"""
#         return self._L_X
# 
#     @property
#     def Phi(self):
#         """EUV Photon Luminosity"""
#         return self._Phi


# Path: DiscEvolution/internal_photo.py
import numpy as np
import argparse
import json
import matplotlib.pyplot as plt
from DiscEvolution.constants import *
from DiscEvolution.star import PhotoStar
from scipy.signal import argrelmin
    from control_scripts import run_model

Designed for plotting to test things out
"""""""""
#################################################################################
        
class DummyDisc(object):
    def __init__(self, R, star, MD=10, RC=100):
        self._M = MD * Mjup
        self.Rc = RC
        self.R_edge = R
        self.R = 0.5*(self.R_edge[1:]+self.R_edge[:-1])
        self._Sigma = self._M / (2 * np.pi * self.Rc * self.R * AU**2) * np.exp(-self.R/self.Rc)
        self.star = star

    def Rout(self, thresh=None):
        return max(self.R_edge)

    @property
    def Sigma(self):
        return self._Sigma

    @property
    def Sigma_G(self):
        return self._Sigma

def main():
    Sigma_dot_plot()
    Test_Removal()

def Test_Removal():
    """Removes gas fom a power law disc in regular timesteps without viscous evolution etc"""

    star1 = PhotoStar(LX=1e30, M=1.0, R=2.5, T_eff=4000)