/data/Documents/eclipse/workspace/ColorFull/Col_basis.h

Go to the documentation of this file.

// -*- C++ -*-
/*
 * Col_basis.h
 * Contains the declarations of the class Col_basis, related types and operators.
 * Created on: Aug 9, 2012
 * Author: Malin Sjodahl
 */

#ifndef COLORFULL_Col_basis_h
#define COLORFULL_Col_basis_h


#include "Col_functions.h"


namespace ColorFull {

typedef std::vector<Col_amp> col_basis;

class Col_basis {

public:

        Col_basis(){
                nq=0;
                ng=0;
                trace_basis = false;
                tree_level_gluon_basis = false;
                orthogonal_basis = false;
        }

        virtual ~Col_basis(){}

        int nq;

        int ng;

        col_basis cb;

        Poly_matr P_spm;

        Poly_matr leading_P_spm;

        dmatr d_spm;

        dmatr leading_d_spm;

        Col_functions Col_fun;

        bool  is_Trace_basis() const {return trace_basis;}

        bool  is_Orthogonal_basis() const {return orthogonal_basis;}

        bool  is_Tree_level_gluon_basis() const {return tree_level_gluon_basis;}

        uint size() const {return cb.size();}

        const Col_amp & at( const int i ) const{return cb.at(i);}

        Col_amp & at( const int i ) {return cb.at(i);}

        bool empty() const { return cb.empty(); }

        void clear() { cb.clear(); }

        void append( Col_amp Ca ) { cb.push_back( Ca ); }

        void append( col_basis cb_in );


        /******************** Functions for scalar products **********************/


        void scalar_product_matrix();

        void scalar_product_matrix_num();

        virtual void scalar_product_matrix_no_mem();

        void scalar_product_matrix_num_no_mem();

        void leading_scalar_product_matrix();

        virtual Polynomial scalar_product( const Col_amp & Ca1, const Col_amp & Ca2 );

        virtual cnum scalar_product_num( const Col_amp & Ca1, const Col_amp & Ca2 );

        virtual cnum scalar_product_num( const cvec & v1, const cvec & v2 );

        cnum scalar_product_num_diagonal( const cvec & v1, const cvec & v2 );


        /******************** Functions for reading and writing **********************/

        // Functions for naming files

        std::string basis_file_name() const;

        std::string spm_file_name( const bool leading,  const bool poly) const;

        virtual void read_in_Col_basis( std::string filename );

        virtual void read_in_Col_basis();

        void read_in_d_spm( std::string filename );

        void read_in_d_spm( );

        void read_in_leading_d_spm( std::string filename );

        void read_in_leading_d_spm( );

        void read_in_P_spm( std::string filename );

        void read_in_P_spm( );

        void read_in_leading_P_spm( std::string filename );

        void read_in_leading_P_spm(  );


        // Functions for writing

        virtual void write_out_Col_basis( std::string filename) const;

        virtual void write_out_Col_basis() const;

        void write_out_d_spm( std::string filename ) const;

        void write_out_d_spm( ) const;

        void write_out_P_spm( std::string filename ) const;

        void write_out_P_spm( ) const;

        void write_out_leading_d_spm( std::string filename ) const;

        void write_out_leading_d_spm( ) const;

        void write_out_leading_P_spm( std::string filename ) const;

        void write_out_leading_P_spm( ) const;

        /******************** Other functions **********************/

        void simplify();

        virtual Poly_vec decompose( const Col_amp & Ca );

        Col_amp exchange_gluon( uint vec, int p1, int p2 );

        Poly_matr color_gamma( int p1, int p2 );

        int n_quark_check() const;

        int n_gluon_check() const;

        void rename_indices( Col_str & Cs1, Col_str & Cs2 ) const;

        /******************** Internal functions **********************/

protected:

        bool trace_basis;
        bool tree_level_gluon_basis;
        bool orthogonal_basis;

        virtual void scalar_product_matrix( bool save_P_spm, bool save_d_spm, bool use_mem );

        virtual Polynomial ij_entry( const int i, const int j ) const;

        bool check_diagonal( const dmatr & matr ) const;

        bool check_symmetry( const dmatr & matr ) const;

        void check_spm() const;

        void Col_basis_of_str( std::string str );

        virtual std::ostream&  write_out_Col_basis_to_stream( std::ostream&  out ) const;

        friend std::ostream& operator<<( std::ostream& out, const Col_basis & Cb );

};// end class Col_basis


std::ostream& operator<<( std::ostream& out, const col_basis & cb );


std::ostream& operator<<( std::ostream& out, const Col_basis & Cb );

}// end namespace ColorFull


#endif /* COLBASIS_H_ */