!pip install -I "phoebe>=2.0,<2.1"

"""
Explanation: General Concepts
HOW TO RUN THIS FILE: if you're running this in a Jupyter notebook or Google Colab session, you can click on a cell and then shift+Enter to run the cell and automatically select the next cell.  Alt+Enter will run a cell and create a new cell below it.  Ctrl+Enter will run a cell but keep it selected.  To restart from scratch, restart the kernel/runtime.
This tutorial introduces all the general concepts of dealing with Parameters, ParameterSets, and the Bundle.  This tutorial aims to be quite complete - covering almost everything you can do with Parameters, so on first read you may just want to try to get familiar, and then return here as a reference for any details later.
All of these tutorials assume basic comfort with Python in general - particularly with the concepts of lists, dictionaries, and objects as well as basic comfort with using the numpy and matplotlib packages.
Setup
Let's first make sure we have the latest version of PHOEBE 2.0 installed. (You can comment out this line if you don't use pip for your installation or don't want to update to the latest release).
End of explanation
"""


import phoebe
from phoebe import u # units
import numpy as np
import matplotlib.pyplot as plt

"""
Explanation: Let's get started with some basic imports
End of explanation
"""


logger = phoebe.logger(clevel='WARNING', flevel='DEBUG', filename='tutorial.log')

"""
Explanation: If running in IPython notebooks, you may see a "ShimWarning" depending on the version of Jupyter you are using - this is safe to ignore.
PHOEBE 2 uses constants defined in the IAU 2015 Resolution which conflict with the constants defined in astropy.  As a result, you'll see the warnings as phoebe.u and phoebe.c "hijacks" the values in astropy.units and astropy.constants.
Whenever providing units, please make sure to use phoebe.u instead of astropy.units, otherwise the conversions may be inconsistent.
Logger
Before starting any script, it is a good habit to initialize a logger and define which levels of information you want printed to the command line (clevel) and dumped to a file (flevel).
The levels from most to least information are:

DEBUG
INFO
WARNING
ERROR
CRITICAL
End of explanation
"""


param = phoebe.parameters.StringParameter(qualifier='myparameter', 
                                          description='mydescription',
                                          value='myvalue')

"""
Explanation: All of these arguments are optional and will default to clevel='WARNING' if not provided.  There is therefore no need to provide a filename if you don't provide a value for flevel.
So with this logger, anything with "WARNING, ERROR, or CRITICAL levels will be printed to the screen.  All messages of any level will be written to a file named 'tutorial.log' in the current directory.
Note: the logger messages are not included in the outputs shown below.
Parameters
Parameters hold a single value, but need to be aware about their own types, limits, and connection with other Parameters (more on this later when we discuss ParameterSets).
Note that generally you won't ever have to "create" or "define" your own Parameters, those will be created for you by helper functions, but we have to start somewhere... so let's create our first Parameter.
We'll start with creating a StringParameter since it is the most generic, and then discuss and specific differences for each type of Parameter.
End of explanation
"""


print type(param)

"""
Explanation: If you ever need to know the type of a Parameter, you can always use python's built-in type functionality:
End of explanation
"""


print param

"""
Explanation: If we print the parameter object we can see a summary of information
End of explanation
"""


print param.meta

"""
Explanation: You can see here that we've defined three a few things about parameter: the qualifier, description, and value (others do exist, they just don't show up in the summary).
These "things" can be split into two groups: tags and attributes (although in a pythonic sense, both can be accessed as attributes).  Don't worry too much about this distinction - it isn't really important except for the fact that tags are shared across all Parameters whereas attributes are dependent on the type of the Parameter.
The tags of a Parameter define the Parameter and how it connects to other Parameters (again, more on this when we get to ParameterSets).  For now, just now that you can access a list of all the tags as follows:
End of explanation
"""


print param['qualifier'], param.qualifier

"""
Explanation: and that each of these is available through both a dictionary key and an object attribute.  For example:
End of explanation
"""


param.attributes

"""
Explanation: The 'qualifier' attribute is essentially an abbreviated name for the Parameter.
These tags will be shared across all Parameters, regardless of their type.
Attributes, on the other hand, can be dependent on the type of the Parameter and tell the Parameter its rules and how to interpret its value.  You can access a list of available attributes as follows:
End of explanation
"""


print param['description'], param.description

"""
Explanation: and again, each of these are available through both a dictionary key and as an object attribute.  For example, all parameters have a 'description' attribute which gives additional information about what the Parameter means:
End of explanation
"""


print param.get_value(), param['value'], param.value

"""
Explanation: For the special case of the 'value' attribute, there is also a get method (will become handy later when we want to be able to request the value in a specific unit).
End of explanation
"""


param.set_value('newvalue')
print param.get_value()

"""
Explanation: The value attribute is also the only attribute that you'll likely want to change, so it also has a set method:
End of explanation
"""


param = phoebe.parameters.ChoiceParameter(qualifier='mychoiceparameter',
                                          description='mydescription',
                                          choices=['choice1', 'choice2'],
                                          value='choice1')

print param

print param.attributes

print param['choices'], param.choices

print param.get_value()

#param.set_value('not_a_choice') # would raise a ValueError
param.set_value('choice2')
print param.get_value()

"""
Explanation: The 'visible_if' attribute only comes into play when the Parameter is a member of a ParameterSet, so we'll discuss it at the end of this tutorial when we get to ParameterSets.
The 'copy_for' attribute is only used when the Parameter is in a particular type of ParameterSet called a Bundle (explained at the very end of this tutorial).  We'll see the 'copy_for' capability in action later in the Datasets Tutorial, but for now, just know that you can view this property only and cannot change it... and most of the time it will just be an empty string.
StringParameters
We'll just mention StringParameters again for completeness, but we've already seen about all they can do - the value must cast to a valid string but no limits or checks are performed at all on the value.
ChoiceParameters
ChoiceParameters are essentially StringParameters with one very important exception: the value must match one of the prescribed choices.
Therefore, they have a 'choice' attribute and an error will be raised if trying to set the value to any string not in that list.
End of explanation
"""


param = phoebe.parameters.FloatParameter(qualifier='myfloatparameter',
                                         description='mydescription',
                                         default_unit=u.m,
                                         limits=[None,20],
                                         value=5)

print param

"""
Explanation: FloatParameters
FloatParameters are probably the most common Parameter used in PHOEBE and hold both a float and a unit, with the ability to retrieve the value in any other convertible unit.
End of explanation
"""


print param.attributes

"""
Explanation: You'll notice here a few new mentions in the summary... "Constrained by", "Constrains", and "Related to" are all referring to constraints which will be discussed in a future tutorial.
End of explanation
"""


print param['limits'], param.limits

#param.set_value(30) # would raise a ValueError
param.set_value(2)
print param.get_value()

"""
Explanation: FloatParameters have an attribute which hold the "limits" - whenever a value is set it will be checked to make sure it falls within the limits.  If either the lower or upper limit is None, then there is no limit check for that extreme.
End of explanation
"""


print param['default_unit'], param.default_unit

"""
Explanation: FloatParameters have an attribute which holds the "default_unit" - this is the unit in which the value is stored and the unit that will be provided if not otherwise overriden.
End of explanation
"""


print param.get_value()

"""
Explanation: Calling get_value will then return a float in these units
End of explanation
"""


print param.get_value(unit=u.km), param.get_value(unit='km')

"""
Explanation: But we can also request the value in a different unit, by passing an astropy Unit object or its string representation.
End of explanation
"""


print param.get_quantity(), param.get_quantity(unit=u.km)

"""
Explanation: FloatParameters also have their own method to access an astropy Quantity object that includes both the value and the unit
End of explanation
"""


param.set_value(10)
print param.get_quantity()

param.set_value(0.001*u.km)
print param.get_quantity()

param.set_value(10, unit='cm')
print param.get_quantity()

"""
Explanation: The set_value method also accepts a unit - this doesn't change the default_unit internally, but instead converts the provided value before storing.
End of explanation
"""


param.set_default_unit(u.km)
print param.get_quantity()

"""
Explanation: If for some reason you want to change the default_unit, you can do so as well:
End of explanation
"""


print param.limits

"""
Explanation: But note that the limits are still stored as a quantity object in the originally defined default_units
End of explanation
"""


param = phoebe.parameters.IntParameter(qualifier='myintparameter',
                                       description='mydescription',
                                       limits=[0,None],
                                       value=1)

print param

print param.attributes

"""
Explanation: IntParameters
IntParameters are essentially the same as FloatParameters except they always cast to an Integer and they have no units.
End of explanation
"""


print param['limits'], param.limits

"""
Explanation: Like FloatParameters above, IntParameters still have limits
End of explanation
"""


param.set_value(1.9)
print param.get_value()

"""
Explanation: Note that if you try to set the value to a float it will not raise an error, but will cast that value to an integer (following python rules of truncation, not rounding)
End of explanation
"""


param = phoebe.parameters.BoolParameter(qualifier='myboolparameter',
                                        description='mydescription',
                                        value=True)

print param

print param.attributes

"""
Explanation: Bool Parameters
Boolean Parameters are even simpler - they accept True or False.
End of explanation
"""


param.set_value(0)
print param.get_value()

param.set_value(None)
print param.get_value()

"""
Explanation: Note that, like IntParameters, BoolParameters will attempt to cast anything you give it into True or False.
End of explanation
"""


param.set_value('')
print param.get_value()

param.set_value('some_string')
print param.get_value()

"""
Explanation: As with Python, an empty string will cast to False and a non-empty string will cast to True
End of explanation
"""


param.set_value('False')
print param.get_value()

param.set_value('false')
print param.get_value()

"""
Explanation: The only exception to this is that (unlike Python), 'true' or 'True' will cast to True and 'false' or 'False' will cast to False.
End of explanation
"""


param = phoebe.parameters.FloatArrayParameter(qualifier='myfloatarrayparameters',
                                              description='mydescription',
                                              default_unit=u.m,
                                              value=np.array([0,1,2,3]))

print param

print param.attributes

print param.get_value(unit=u.km)

"""
Explanation: FloatArrayParameters
FloatArrayParameters are essentially the same as FloatParameters (in that they have the same unit treatment, although obviously no limits) but hold numpy arrays rather than a single value.
By convention in Phoebe, these will (almost) always have a pluralized qualifier.
End of explanation
"""


param1 = phoebe.parameters.FloatParameter(qualifier='param1',
                                          description='param1 description',
                                          default_unit=u.m,
                                          limits=[None,20],
                                          value=5,
                                          context='context1',
                                          kind='kind1')

param2 = phoebe.parameters.FloatParameter(qualifier='param2',
                                          description='param2 description',
                                          default_unit=u.deg,
                                          limits=[0,2*np.pi],
                                          value=0,
                                          context='context2',
                                          kind='kind2')

param3 = phoebe.parameters.FloatParameter(qualifier='param3',
                                          description='param3 description',
                                          default_unit=u.kg,
                                          limits=[0,2*np.pi],
                                          value=0,
                                          context='context1',
                                          kind='kind2')

ps = phoebe.parameters.ParameterSet([param1, param2, param3])

print ps.to_list()

"""
Explanation: FloatArrayParameters also allow for built-in interpolation... but this requires them to be a member of a Bundle, so we'll discuss this in just a bit.
ParametersSets
ParameterSets are a collection of Parameters that can be filtered by their tags to return another ParameterSet.
For illustration, let's create 3 random FloatParameters and combine them to make a ParameterSet.
End of explanation
"""


print ps

"""
Explanation: If we print a ParameterSet, we'll see a listing of all the Parameters and their values.
End of explanation
"""


print ps.get('param1@kind1')

"""
Explanation: Twigs
The string notation used for the Parameters is called a 'twig' - its simply a combination of all the tags joined with the '@' symbol and gives a very convenient way to access any Parameter.  
The order of the tags doesn't matter, and you only need to provide enough tags to produce a unique match.  Since there is only one parameter with kind='kind1', we do not need to provide the extraneous context='context1' in the twig to get a match.
End of explanation
"""


print ps.get('param1@kind1').description

"""
Explanation: Note that this returned the ParameterObject itself, so you can now use any of the Parameter methods or attributes we saw earlier.  For example:
End of explanation
"""


ps.set_value('param1@kind1', 10)
print ps.get_value('param1@kind1')

"""
Explanation: But we can also use set and get_value methods from the ParameterSet itself:
End of explanation
"""


print ps.meta.keys()

"""
Explanation: Tags
Each Parameter has a number of tags, and the ParameterSet has the same tags - where the value of any given tag is None if not shared by all Parameters in that ParameterSet.
So let's just print the names of the tags again and then describe what each one means.
End of explanation
"""


print ps.context

"""
Explanation: Most of these "metatags" act as labels - for example, you can give a component tag to each of the components for easier referencing.
But a few of these tags are fixed and not editable:

qualifier: literally the name of the parameter.
kind: tells what kind a parameter is (ie whether a component is a star or an orbit).
context: tells what context this parameter belongs to
twig: a shortcut to the parameter in a single string.
uniquetwig: the minimal twig needed to reach this parameter.
uniqueid: an internal representation used to reach this parameter

These contexts are (you'll notice that most are represented in the tags):

setting
history
system
component
feature
dataset
constraint
compute
model
fitting [not yet supported]
feedback [not yet supported]
plugin [not yet supported]

One way to distinguish between context and kind is with the following question and answer:
"What kind of [context] is this?  It's a [kind] tagged [context]=[tag-with-same-name-as-context]."
In different cases, this will then become:

"What kind of component is this?  It's a star tagged component=starA." (context='component', kind='star', component='starA')
"What kind of feature is this?  It's a spot tagged feature=spot01." (context='feature', kind='spot', feature='spot01')
"What kind of dataset is this?  It's a LC (light curve) tagged dataset=lc01." (context='dataset', kind='LC', dataset='lc01')
"What kind of compute (options) are these?  They're phoebe (compute options) tagged compute=preview." (context='compute', kind='phoebe', compute='preview')

As we saw before, these tags can be accessed at the Parameter level as either a dictionary key or as an object attribute.  For ParameterSets, the tags are only accessible through object attributes.
End of explanation
"""


print ps.contexts

"""
Explanation: This returns None since not all objects in this ParameterSet share a single context.  But you can see all the options for a given tag by providing the plural version of that tag name:
End of explanation
"""


print ps.filter(context='context1')

"""
Explanation: Filtering
Any of the tags can also be used to filter the ParameterSet:
End of explanation
"""


print ps.filter(context='context1', kind='kind1')

"""
Explanation: Here we were returned a ParameterSet of all Parameters that matched the filter criteria.  Since we're returned another ParameterSet, we can chain additional filter calls together.
End of explanation
"""


print ps.filter(context='context1', kind='kind1')

"""
Explanation: Now we see that we have drilled down to a single Parameter.  Note that a ParameterSet is still returned - filter will always return a ParameterSet.
We could have accomplished the exact same thing with a single call to filter:
End of explanation
"""


print ps.filter(context='context1', kind='kind1').get()

print ps.get(context='context1', kind='kind1')

"""
Explanation: If you want to access the actual Parameter, you must use get instead of (or in addition to) filter.  All of the following lines do the exact same thing:
End of explanation
"""


print ps['context1@kind1']

print ps['context1']['kind1']

"""
Explanation: Or we can use those twigs.  Remember that twigs are just a combination of these tags separated by the @ symbol.  You can use these for dictionary access in a ParameterSet - without needing to provide the name of the tag, and without having to worry about order.  And whenever this returns a ParameterSet, these are also chainable, so the following two lines will do the same thing:
End of explanation
"""


print ps['context1']

print ps['context1@kind1']

"""
Explanation: You may notice that the final result was a Parameter, not a ParameterSet.  Twig dictionary access tries to be smart - if exactly 1 Parameter is found, it will return that Parameter instead of a ParameterSet.  Notice the difference between the two following lines:
End of explanation
"""


print ps['context1@kind1']['description']

"""
Explanation: Of course, once you get the Parameter you can then use dictionary keys to access any attributes of that Parameter.
End of explanation
"""


print ps['description@context1@kind1']

"""
Explanation: So we decided we might as well allow access to those attributes directly from the twig as well
End of explanation
"""


b = phoebe.Bundle()
print b

"""
Explanation: The Bundle
The Bundle is nothing more than a glorified ParameterSet with some extra methods to compute models, add new components and datasets, etc.
You can initialize an empty Bundle as follows:
End of explanation
"""


print b.filter(context='system')

"""
Explanation: and filter just as you would for a ParameterSet
End of explanation
"""


param1 = phoebe.parameters.ChoiceParameter(qualifier='what_is_this',
                                           choices=['matter', 'aether'],
                                           value='matter',
                                           context='context1')
param2 = phoebe.parameters.FloatParameter(qualifier='mass',
                                          default_unit=u.kg,
                                          value=5,
                                          visible_if='what_is_this:matter',
                                          context='context1')

b = phoebe.Bundle([param1, param2])

print b.filter()

"""
Explanation: Visible If
As promised earlier, the 'visible_if' attribute of a Parameter controls whether its visible to a ParameterSet... but it only does anything if the Parameter belongs to a Bundle.
Let's make a new ParameterSet in which the visibility of one parameter is dependent on the value of another.
End of explanation
"""


b.set_value('what_is_this', 'aether')
print b.filter()

"""
Explanation: It doesn't make much sense to need to define a mass if this thing isn't baryonic.  So if we change the value of 'what_is_this' to 'aether' then the 'mass' Parameter will temporarily hide itself.
End of explanation
"""


xparam = phoebe.parameters.FloatArrayParameter(qualifier='xs',
                                               default_unit=u.d,
                                               value=np.linspace(0,1,10),
                                               context='context1')

yparam = phoebe.parameters.FloatArrayParameter(qualifier='ys',
                                               default_unit=u.m,
                                               value=np.linspace(0,1,10)**2,
                                               context='context1')

b = phoebe.Bundle([xparam, yparam])


b.filter('ys').get().twig

b['ys'].get_value()

"""
Explanation: FloatArrayParameters: interpolation
As mentioned earlier, when a part of a Bundle, FloatArrayParameters can handle simple linear interpolation with respect to another FloatArrayParameter in the same Bundle.
End of explanation
"""


b['ys'].interp_value(xs=0)

b['ys'].interp_value(xs=0.2)

"""
Explanation: Now we can interpolate the 'ys' param for any given value of 'xs'
End of explanation
"""


