ElementContainerMatrix.hh

/*
$Id: ElementContainerMatrix.hh 2295 2011-08-10 02:57:13Z jisuzuki $
*/


#ifndef ELEMENTCONTAINERMATRIX
#define ELEMENTCONTAINERMATRIX

#include "Header.hh"
#include "NeutronVector.hh"
#include "ElementContainer.hh"
#include "ElementContainerArray.hh"


class ElementContainerMatrix 
  : public NeutronVector< ElementContainerArray, HeaderBase >
{
public:
  using NeutronVector< ElementContainerArray, HeaderBase >::operator();

  ElementContainerMatrix();
  ElementContainerMatrix( HeaderBase pheader );
  ElementContainerMatrix( const NeutronVector< ElementContainerArray, HeaderBase > &ob );

  ElementContainer*      operator()( UInt4 Mindex, UInt4 Aindex );
  vector<Double>*        operator()( UInt4 Mindex, UInt4 Aindex, UInt4 Cindex );
  vector<Double>*        operator()( UInt4 Mindex, UInt4 Aindex, string Key );
  ElementContainerMatrix( const ElementContainerMatrix &ob );
  
  vector<UInt4> SaveToBinFile( string key, string FileName );
  PyObject *SaveToBinFilePy( string key, string FileName );
  void ReadBinFile( string FileName, 
                    vector<UInt4> dimensions, 
                    string key );
  void ReadBinFile( string FileName, PyObject *dimensions, 
                    string key );
  void AppendErr( string Counts, string Error, Double ErrorValue=0.0 );

  void AllSum(ElementContainer &res) {
        for (UInt4 i=0;i<v.size();++i) {
          for (UInt4 j=0;j<v[i]->v.size();++j) {
                res += *v[i]->v[j];
          }
        }
  }

  ElementContainerMatrix Transpose();
  ElementContainerMatrix Mul( Double d );
  ElementContainerMatrix &MulMySelf( Double d );
  ElementContainerMatrix Mul( Double d, Double e );
  ElementContainerMatrix &MulMySelf( Double d, Double e );
  ElementContainerMatrix Mul( pair<Double,Double> &p ){ return Mul(p.first,p.second); }
  ElementContainerMatrix &MulMySelf( pair<Double,Double> &p ){ return MulMySelf(p.first,p.second); }

  ElementContainerMatrix Plus( Double d );
  ElementContainerMatrix &PlusMySelf( Double d );
  ElementContainerMatrix Plus( Double d, Double e );
  ElementContainerMatrix &PlusMySelf( Double d, Double e );
  ElementContainerMatrix Plus( pair<Double,Double> &p ){ return Plus(p.first,p.second); }
  ElementContainerMatrix &PlusMySelf( pair<Double,Double> &p ){ return PlusMySelf(p.first,p.second); }
};


#endif