# -*- coding: utf-8 -*-

from vamdctap.unitconv import invcm2eV

# In the RESTRICTABLES dictionary, keys are names from the dictionary and values are 
# names from the Django model for the transitions table.
#
# Where a field is actually in the transitions model (i.e. in the transitions table and
# not in a linked table) then the name of the field appears without punctuation. Where
# the field is in a related model (meaning that Django must join tables to get at it),
# the double-underscore operator expressed the chain of relations.
#
# E.g., radtranswavelengthexperimentalvalue refers a field directly in the transitions model.
# finalstateindex__species__atomsymbol refers to the atomsymbol field in the
# species model. The transitions model has a foreign key to the states model in a field
# called finalstateindex. The states model has a foreign key to the species model in a
# field called species. All three parts of finalstateindex__species__atomsymbol are names of 
# fields in models; names of entire models are not part of this syntax.

RESTRICTABLES = {\
'AtomSymbol':'finalstate__species__element',
'AtomNuclearCharge':'finalstate__species__nuclearcharge',
'IonCharge':'finalstate__species__ioncharge',
'RadTransWavelength':'wavelength'
}

RETURNABLES = {\
'NodeID':u'tignanillo', # Constant value
'XSAMSVersion':u'1.0',

'AtomSpeciesId':'Atom.id',
'AtomSymbol':'Atom.element',
'AtomNuclearCharge':'Atom.nuclearcharge',
'AtomIonCharge':'Atom.ioncharge',

'AtomStateId':'AtomState.id',
'AtomStateTotalAngMom':'AtomState.j',
'AtomStateStatisticalWeight':'(2*AtomState.j)+1',
'AtomStateEnergy':'AtomState.energy',
'AtomStateEnergyMethod':'AtomState.energyMethod',
'AtomStateEnergyUnit':u'1/cm',
'AtomStateDescription':'AtomState.configuration',
'AtomStateConfigurationLabel':'AtomState.configuration',

'RadTransWavelength':'RadTran.wavelength',
'RadTransWavelengthUnit':u'A', # Constant: Angstroms
'RadTransProbabilityWeightedOscillatorStrength':'RadTran.weightedoscillatorstrength',
'RadTransProbabilityA':'RadTran.probabilitya',
'RadTransProbabilityAUnit':u'1/s',
'RadTransSpeciesRef':'RadTran.initialstate.species.id',
'RadTransUpperStateRef':'RadTran.upperStateRef()',
'RadTransLowerStateRef':'RadTran.lowerStateRef()',
'RadTransID':'RadTran.id'
}