transition.h

Go to the documentation of this file.

/* This file is part of Cloudy and is copyright (C)1978-2013 by Gary J. Ferland and
 * others.  For conditions of distribution and use see copyright notice in license.txt */

#ifndef TRANSITION_H_
#define TRANSITION_H_

#include "count_ptr.h"
#include "emission.h"
#include "collision.h"
#include "proxy_iterator.h"
#include "quantumstate.h"

/* create a dummy emission structure.  Non-radiative transitions will point to this */
extern EmissionProxy DummyEmis;

// Proxy class provides 'object' access to struct-of-vectors
class TransitionList;
class TransitionListImpl;
class TransitionConstProxy;
class TransitionProxy
{
public:
        typedef TransitionListImpl list_type;
        typedef ProxyIterator<TransitionProxy,TransitionConstProxy> iterator;
        typedef ProxyIterator<TransitionConstProxy,TransitionConstProxy> const_iterator;  
private:
        friend class  ProxyIterator<TransitionProxy,TransitionConstProxy>;
        list_type *m_list;
        int m_index;
public:
        //TransitionProxy()
        //{
        //      Junk();
        //}

        explicit TransitionProxy(): m_list(0), m_index(-1) {}
        explicit TransitionProxy(TransitionListImpl *list, int index) : 
        m_list(list), m_index(index) {}
        void copy(const TransitionProxy& other) const;
        qList::iterator Lo() const;
        qList::iterator Hi() const;
        void setLo(int ipLo) const;
        void setHi(int ipHi) const;
        void AddLine2Stack() const;
        EmissionList::reference Emis() const;   
        int &ipEmis() const;
        bool associated() const
        {
                return m_list != NULL && m_index >= 0;
        }
        bool hasEmis() const
        {
                return ipEmis() != -1;
        }
        void resetEmis() const
        {
                Emis().ipTran() = m_index;
        }
        bool systemIs(const TransitionList *query) const;
        void check() const
        {
                ASSERT(!hasEmis() || Emis().ipTran() == m_index);
        }
        CollisionProxy Coll() const;

        realnum &WLAng() const;

        realnum EnergyK() const
        {
                return (realnum)T1CM*EnergyWN();
        }
        realnum EnergyErg() const
        {
                return (realnum)ERG1CM*EnergyWN();
        }
        double EnergyRyd() const
        {
                return WAVNRYD*EnergyWN();
        }


        realnum &EnergyWN() const;

        long &ipCont() const;

   
        void Junk( void ) const ;

        void Zero( void ) const ;

        void outline( double nonScatteredFraction, bool lgDoChecks ) const ;

           
        void outline_resonance(  ) const ;
        int &ipLo() const;
        int &ipHi() const;
        void AddHiState() const;
        void AddLoState() const;
};
class TransitionConstProxy
{
public:
        typedef const TransitionListImpl list_type;
        typedef ProxyIterator<TransitionConstProxy,TransitionConstProxy> iterator;
        typedef ProxyIterator<TransitionConstProxy,TransitionConstProxy> const_iterator;  
private:
        friend class  ProxyIterator<TransitionConstProxy,TransitionConstProxy>;
        const list_type *m_list;
        int m_index;
public:
        //TransitionProxy()
        //{
        //      Junk();
        //}

        explicit TransitionConstProxy(): m_list(0), m_index(-1) {}
        explicit TransitionConstProxy(const TransitionListImpl *list, int index) : 
        m_list(list), m_index(index) {}
        void copy(const TransitionConstProxy& other) const;
        qList::iterator Lo() const;
        qList::iterator Hi() const;
        void AddLine2Stack() const;
        EmissionList::reference Emis() const;   
        int ipEmis() const;
        bool associated() const
        {
                return m_list != NULL && m_index >= 0;
        }
        bool hasEmis() const
        {
                return ipEmis() != -1;
        }
        void check() const
        {
                ASSERT(!hasEmis() || Emis().ipTran() == m_index);
        }
        CollisionProxy Coll() const;

        realnum WLAng() const;

        realnum EnergyK() const
        {
                return (realnum)T1CM*EnergyWN();
        }
        realnum EnergyErg() const
        {
                return (realnum)ERG1CM*EnergyWN();
        }
        double EnergyRyd() const
        {
                return WAVNRYD*EnergyWN();
        }


        realnum EnergyWN() const;

        long ipCont() const;

        void outline( double nonScatteredFraction, bool lgDoChecks ) const ;

        void outline_resonance(  ) const ;
        int ipLo() const;
        int ipHi() const;
};


// Structure-of-vectors for transition data
class TransitionListImpl
{
        vector<int> ipHi, ipLo;
        vector<long> ipCont;
        CollisionList Coll;
        vector<realnum> EnergyWN, WLAng;
        // DO NOT IMPLEMENT
        TransitionListImpl(const TransitionListImpl&);
        TransitionListImpl& operator=(const TransitionListImpl&);
public:
        friend class TransitionProxy;
        friend class TransitionConstProxy;
        string chLabel;
        qList *states; // List of individual states
        EmissionList Emis;
        vector<int> ipEmis;
        explicit TransitionListImpl(
                const string &chLabel,
                qList *states) : chLabel(chLabel), states(states), Emis(this)
        {}
        explicit TransitionListImpl(
                const string &chLabel,
                qList *states,
                size_t size) : chLabel(chLabel), states(states), Emis(this)
        {
                resize(size);
        }
        void resize(size_t newsize);
        void reserve(size_t newsize);
        typedef TransitionProxy::iterator iterator;
        typedef TransitionConstProxy::iterator const_iterator;
        typedef TransitionProxy reference;
        reference operator[](size_t i)
        {
                return TransitionProxy(this,i);
        }
        size_t size(void) const
        {
                return ipCont.size();
        }
        void pop_back(void)
        {
                resize(size()-1);
        }
        iterator begin(void)
        {
                return iterator(this,0);
        }
        const_iterator begin(void) const
        {
                return const_iterator(this,0);
        }
        iterator end(void)
        {
                return iterator(this,size());
        }
        const_iterator end(void) const
        {
                return const_iterator(this,size());
        }
        void push_back(const TransitionProxy &tr)
        {
                int newsize=size()+1;
                resize(newsize);
                (*this)[newsize-1].copy(tr);
        }
        const TransitionProxy back(void)
        {
                return *(end()-1);
        }
};

class TransitionList
{
        // Internal vectors all need to be sized consistently (see three
        // functions below)
        count_ptr<TransitionListImpl> p_impl;
public:
        typedef TransitionProxy::iterator iterator;
        typedef TransitionConstProxy::iterator const_iterator;
        explicit TransitionList(const string &chLabel, qList *states, size_t size=0)
                : p_impl(new TransitionListImpl(chLabel, states, size))
        {}
        void resize(size_t newsize)
        {
                p_impl->resize(newsize);
        }
        void reserve(size_t newsize)
        {
                p_impl->reserve(newsize);
        }
        TransitionProxy operator[](size_t i)
        {
                return (*p_impl)[i];
        }
        size_t size(void) const
        {
                return p_impl->size();
        }
        void pop_back(void)
        {
                p_impl->pop_back();
        }
        iterator begin(void)
        {
                return p_impl->begin();
        }
        iterator end(void)
        {
                return p_impl->end();
        }
        void push_back(const TransitionProxy &tr)
        {
                p_impl->push_back(tr);
        }
        const TransitionProxy back(void)
        {
                return p_impl->back();
        }
        string &chLabel()
        {
                return p_impl->chLabel;
        }
        qList *&states()
        {
                return p_impl->states;
        }
        EmissionList &Emis()
        {
                return p_impl->Emis;
        }
        vector<int> &ipEmis()
        {
                return p_impl->ipEmis;
        }
        bool isSame (const TransitionListImpl *other) const
        {
                return p_impl.get_ptr() == other;
        }
};

inline bool TransitionProxy::systemIs(const TransitionList *query) const
{
        return query->isSame(m_list);
}

// Must include all internal vector elements in these three functions 
inline void TransitionListImpl::resize(size_t newsize)
{
        ipLo.resize(newsize);
        ipHi.resize(newsize);
        ipCont.resize(newsize);
        Coll.resize(newsize);
        EnergyWN.resize(newsize);
        WLAng.resize(newsize);
        ipEmis.resize(newsize,-1);
}
inline void TransitionListImpl::reserve(size_t newsize)
{
        ipLo.reserve(newsize);
        ipHi.reserve(newsize);
        ipCont.reserve(newsize);
        Coll.reserve(newsize);
        EnergyWN.reserve(newsize);
        WLAng.reserve(newsize);
        ipEmis.reserve(newsize);
}
inline void TransitionProxy::copy(const TransitionProxy& other) const
{
  m_list->ipLo[m_index] = other.m_list->ipLo[other.m_index];
  m_list->ipHi[m_index] = other.m_list->ipHi[other.m_index];
  m_list->ipCont[m_index] = other.m_list->ipCont[other.m_index];
  m_list->Coll[m_index].copy(other.m_list->Coll[other.m_index]);
  m_list->EnergyWN[m_index] = other.m_list->EnergyWN[other.m_index];
  m_list->WLAng[m_index] = other.m_list->WLAng[other.m_index];
  if (other.m_list->ipEmis[other.m_index] == -1)
  {
          m_list->ipEmis[m_index] = -1;
  }
  else
  {
          ASSERT (m_list->ipEmis[m_index] == -1);
          AddLine2Stack();
          m_list->Emis[m_list->ipEmis[m_index]].copy( 
                  other.m_list->Emis[other.m_list->ipEmis[other.m_index]]);
  }
}
// End of region needing consistency with TransitionListImpl class

// Handle accessors need to see the structure of the TransitionList
inline qList::iterator TransitionProxy::Lo() const
{
        return m_list->states->begin()+m_list->ipLo[m_index];
}
inline qList::iterator TransitionProxy::Hi() const
{
        return m_list->states->begin()+m_list->ipHi[m_index];
}
inline void TransitionProxy::setLo(int ipLo) const
{
        m_list->ipLo[m_index] = ipLo;
}
inline void TransitionProxy::setHi(int ipHi) const
{
        m_list->ipHi[m_index] = ipHi;
}
inline EmissionList::reference TransitionProxy::Emis() const
{
        int ipEmis = m_list->ipEmis[m_index];
        if (ipEmis == -1)
                return DummyEmis;
        else
                return m_list->Emis[ipEmis];
}
inline int& TransitionProxy::ipEmis() const
{
        return m_list->ipEmis[m_index];
}
inline int TransitionConstProxy::ipEmis() const
{
        return m_list->ipEmis[m_index];
}
inline CollisionProxy TransitionProxy::Coll() const
{
        return m_list->Coll[m_index];
}
inline  realnum &TransitionProxy::WLAng() const
{
        return m_list->WLAng[m_index];
}
inline  realnum TransitionConstProxy::WLAng() const
{
        return m_list->WLAng[m_index];
}
inline realnum &TransitionProxy::EnergyWN() const
{
        return m_list->EnergyWN[m_index];
}
inline realnum TransitionConstProxy::EnergyWN() const
{
        return m_list->EnergyWN[m_index];
}
inline long &TransitionProxy::ipCont() const
{
        return m_list->ipCont[m_index];
}
inline long TransitionConstProxy::ipCont() const
{
        return m_list->ipCont[m_index];
}
inline int &TransitionProxy::ipLo() const
{
        return m_list->ipLo[m_index];
}
inline int TransitionConstProxy::ipLo() const
{
        return m_list->ipLo[m_index];
}
inline int &TransitionProxy::ipHi() const
{
        return m_list->ipHi[m_index];
}
inline int TransitionConstProxy::ipHi() const
{
        return m_list->ipHi[m_index];
}

inline TransitionProxy EmissionProxy::Tran() const
{
        TransitionProxy t = TransitionProxy(m_list->m_tlist,ipTran());
        t.check();
        return t;
}
inline TransitionConstProxy EmissionConstProxy::Tran() const
{
        TransitionConstProxy t = TransitionConstProxy(m_list->m_tlist,ipTran());
        t.check();
        return t;
}

void PutLine(const TransitionProxy &t, const char *chComment);

void PutLine(const TransitionProxy &t, const char *chComment, const char *chLabel);

double TexcLine(const TransitionProxy &t);

void DumpLine(const TransitionProxy &t);

double emit_frac(const TransitionProxy &t);

void chIonLbl(char*, const TransitionProxy &t);
void chIonLbl(char *chIonLbl_v, const long& nelem, const long& IonStg);

char* chLineLbl(const TransitionProxy &t);

void PutCS(double cs, 
  const TransitionProxy & t);

void GenerateTransitionConfiguration( const TransitionProxy &t, char *chComment );

double OccupationNumberLine(const TransitionProxy &t);

void PutExtra(double Extra);

void LineConvRate2CS( const TransitionProxy & t , realnum rate );

inline bool lgTauGood( const TransitionProxy& t )
{
        // first iteration only use inward optical depths so scale good
        return ( iteration == 1 || 
                // maser - optical depths also ok (but bizarre) 
                                t.Emis().TauIn() <= 0. || 
                // TauIn < TauTot means outward optical depth is positive, so OK
                                t.Emis().TauIn() < t.Emis().TauTot() );
}

void MakeCS(const TransitionProxy & t );

#endif // _TRANSITION_H_