Adding tnlFunctionEnumerator. (bf079ed7) · Commits · TNL / tnl-dev

CMakeLists.txt

+1 −1

Original line number	Diff line number	Diff line
		@@ -85,7 +85,7 @@ if( WITH_CUDA STREQUAL "yes" )
		AddCompilerFlag( "-std=gnu++0x" )
		endif( CUDA_FOUND )
		else( WITH_CUDA STREQUAL "yes" )
		AddCompilerFlag( "-std=gnu++0x" )
		AddCompilerFlag( "-std=gnu++0x -ftree-vectorizer-verbose=1" )
		endif( WITH_CUDA STREQUAL "yes" )

		####

install

+2 −2

Original line number	Diff line number	Diff line
		@@ -2,9 +2,9 @@

		TARGET=TNL
		INSTALL_PREFIX=${HOME}/local
		WITH_CUDA=yes
		WITH_CUDA=no
		TEMPLATE_EXPLICIT_INSTANTIATION=yes
		#VERBOSE="VERBOSE=1"
		VERBOSE="VERBOSE=1"

		CMAKE="cmake"
		CPUS=`grep -c processor /proc/cpuinfo`

src/core/vectors/tnlVectorOperationsHost_impl.h

+1 −1

Original line number	Diff line number	Diff line
		@@ -314,7 +314,7 @@ typename Vector1 :: RealType tnlVectorOperations< tnlHost > :: getScalarProduct(
		Real result = 0;
		const Index n = v1. getSize();
		for( Index i = 0; i < n; i ++ )
		result += v1. getElement( i ) * v2. getElement( i );
		result += v1[ i ] * v2[ i ];
		return result;
		}

src/functions/CMakeLists.txt

+2 −0

Original line number	Diff line number	Diff line
		SET( headers tnlFunctionDiscretizer.h
		tnlFunctionDiscretizer_impl.h
		tnlFunctionEnumerator.h
		tnlFunctionEnumerator_impl.h
		tnlFunctionAdapter.h
		tnlConstantFunction.h
		tnlConstantFunction_impl.h

src/functions/tnlFunctionEnumerator.h

0 → 100644

+177 −0

Original line number	Diff line number	Diff line
		/***************************************************************************
		tnlFunctionEnumerator.h - description
		-------------------
		begin : Mar 5, 2015
		copyright : (C) 2015 by Tomas Oberhuber
		email : tomas.oberhuber@fjfi.cvut.cz
		***************************************************************************/

		/***************************************************************************
		* *
		* This program is free software; you can redistribute it and/or modify *
		* it under the terms of the GNU General Public License as published by *
		* the Free Software Foundation; either version 2 of the License, or *
		* (at your option) any later version. *
		* *
		***************************************************************************/
		#ifndef SRC_FUNCTIONS_TNLFUNCTIONENUMERATOR_H_
		#define SRC_FUNCTIONS_TNLFUNCTIONENUMERATOR_H_

		#include <functions/tnlFunctionAdapter.h>

		template< typename Function,
		typename DofVector >
		class tnlFunctionEnumeratorTraversalUserData
		{
		public:

		typedef DofVector::RealType RealType;

		const RealType *time;

		const Function* function;

		DofVector *u;

		const RealType* functionCoefficient;

		const RealType* dofVectorCoefficient;

		tnlFunctionEnumeratorTraversalUserData( const RealType& time,
		const Function& function,
		DofVector& u,
		const RealType& functionCoefficient,
		const RealType& dofVectorCoefficient )
		: time( &time ),
		function( &function ),
		u( &u ),
		functionCoefficient( &functionCoefficient ),
		dofVectorCoefficient( &dofVectorCoefficient )
		{};
		};


		template< typename Mesh,
		typename Function,
		typename DofVector >
		class tnlFunctionEnumerator
		{
		public:
		typedef Mesh MeshType;
		typedef typename DofVector::RealType RealType;
		typedef typename DofVector::DeviceType DeviceType;
		typedef typename DofVector::IndexType IndexType;
		typedef tnlFunctionEnumeratorTraversalUserData< Function,
		DofVector > TraversalUserData;

		template< int EntityDimensions >
		void enumerate( const MeshType& mesh,
		const Function& function
		DofVector& u,
		const RealType& functionCoefficient = 1.0,
		const RealType& dofVectorCoefficient = 0.0,
		const RealType& time = 0.0 ) const;


		class TraversalEntitiesProcessor
		{
		public:

		template< int EntityDimensions >
		#ifdef HAVE_CUDA
		__host__ __device__
		#endif
		static void processEntity( const MeshType& mesh,
		TraversalUserData& userData,
		const IndexType index )
		{
		typedef tnlFunctionAdapter< MeshType, Function > FunctionAdapter;
		( *userData.u )[ index ] =
		( userData.dofVectorCoefficient ) ( *userData.u )[ index ] +
		( userData.functionCoefficient ) FunctionAdapter::getValue( mesh,
		*userData.function,
		index,
		*userData.time );
		}

		};

		};

		template< int Dimensions,
		typename Real,
		typename Device,
		typename Index,
		typename Function,
		typename DofVector >
		class tnlFunctionEnumerator< tnlGrid< Dimensions, Real, Device, Index >,
		Function,
		DofVector >
		{
		public:

		typedef tnlGrid< Dimensions, Real, Device, Index > MeshType;
		typedef typename MeshType::RealType RealType;
		typedef typename MeshType::DeviceType DeviceType;
		typedef typename MeshType::IndexType IndexType;
		typedef typename MeshType::CoordinatesType CoordinatesType;
		typedef tnlFunctionEnumeratorTraversalUserData< Function,
		DofVector > TraversalUserData;

		template< int EntityDimensions >
		void enumerate( const MeshType& mesh,
		const Function& function,
		DofVector& u,
		const RealType& time = 0.0 ) const;

		class TraversalEntitiesProcessor
		{
		public:

		typedef typename MeshType::VertexType VertexType;

		#ifdef HAVE_CUDA
		__host__ __device__
		#endif
		static void processCell( const MeshType& mesh,
		TraversalUserData& userData,
		const IndexType index,
		const CoordinatesType& coordinates )
		{
		typedef tnlFunctionAdapter< MeshType, Function > FunctionAdapter;
		( *userData.u )[ index ] =
		( userData.dofVectorCoefficient ) ( *userData.u )[ index ] +
		( userData.functionCoefficient ) FunctionAdapter::getValue( mesh,
		*userData.function,
		index,
		coordinates,
		*userData.time );

		}

		#ifdef HAVE_CUDA
		__host__ __device__
		#endif
		static void processFace( const MeshType& mesh,
		TraversalUserData& userData,
		const IndexType index,
		const CoordinatesType& coordinates )
		{
		typedef tnlFunctionAdapter< MeshType, Function > FunctionAdapter;
		( *userData.u )[ index ] =
		( userData.dofVectorCoefficient ) ( *userData.u )[ index ] +
		( userData.functionCoefficient ) FunctionAdapter::getValue( mesh,
		*userData.function,
		index,
		coordinates,
		*userData.time );
		}
		};

		};

		#include <functions/tnlFunctionEnumerator_impl.h>



		#endif /* SRC_FUNCTIONS_TNLFUNCTIONENUMERATOR_H_ */