ComputeSHCoefficientsOfDWIFromSymmetricTensor.cxx

Go to the documentation of this file.

#include "ComputeSHCoefficientsOfDWIFromSymmetricTensorCLP.h"
#include "itkSpecialFunctionGenerator.h"
#include "utlCore.h"

void 
DoMain ( std::vector<double>& e1e2, std::vector<double>& bvec, const int shOrder, std::ostream& os, const bool outputAll, const bool nob )
{
  for ( int i = 0; i < bvec.size(); i += 1 ) 
    {
    double b = bvec[i];
    utl_shared_ptr<utl::NDArray<double,1> > coef = utl::GetSymmetricTensorSHCoef<double>(b, e1e2[0], e1e2[1], shOrder, 0.0, 0.0 );
    if (!nob)
      os << b << " ";
    if (outputAll)
      {
      for ( int i = 0; i < coef->Size(); i += 1 ) 
        os << (*coef)[i] << " ";
      }
    else
      {
      for ( int l = 0; l <= shOrder; l += 2 ) 
        {
        int jj = utl::GetIndexSHj(l, 0);
        os << (*coef)[jj] << " ";
        }
      }
    os << std::endl;
    }
}

int 
main (int argc, char const* argv[])
{
  // GenerateCLP
  PARSE_ARGS;

  utlGlobalException(_EigenValuesArg.isSet() && _FAMDArg.isSet(), "only one of --eigenvalues and --famd can be set");
  utlGlobalException(!_EigenValuesArg.isSet() && !_FAMDArg.isSet(), "need to set one of --eigenvalues and --famd");
  utlGlobalException(_BValues.size()==0, "need to set --bvalues");

  itk::InitializeLUTExp();
  if (_EigenValuesArg.isSet())
    {
    utlGlobalException(_EigenValues.size()!=2, "wrong size in --eigenvalues");
    }
  else
    {
    utlGlobalException(_FAMD.size()!=2, "wrong size in --eigenvalues");
    _EigenValues = utl::GetE1E2FromFAMD(_FAMD[0], _FAMD[1]);
    }

  if (_OutputFileArg.isSet())
    {
    std::ofstream  out; 
    out.open ( _OutputFile.c_str() );
    DoMain(_EigenValues, _BValues, _SHOrder, out, _OutputAllCoefficientsArg.isSet(), _NoBArg.isSet());
    out.close();
    }
  else
    DoMain(_EigenValues, _BValues, _SHOrder, std::cout, _OutputAllCoefficientsArg.isSet(), _NoBArg.isSet());

  return 0;
}