You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
159 lines
4.6 KiB
159 lines
4.6 KiB
11 years ago
|
/*
|
||
|
* Copyright 2008-2012 NVIDIA Corporation
|
||
|
*
|
||
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
||
|
* you may not use this file except in compliance with the License.
|
||
|
* You may obtain a copy of the License at
|
||
|
*
|
||
|
* http://www.apache.org/licenses/LICENSE-2.0
|
||
|
*
|
||
|
* Unless required by applicable law or agreed to in writing, software
|
||
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
||
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||
|
* See the License for the specific language governing permissions and
|
||
|
* limitations under the License.
|
||
|
*/
|
||
|
|
||
|
#include <thrust/random/linear_feedback_shift_engine.h>
|
||
|
|
||
|
namespace thrust
|
||
|
{
|
||
|
|
||
|
namespace random
|
||
|
{
|
||
|
|
||
|
template<typename UIntType, size_t w, size_t k, size_t q, size_t s>
|
||
|
linear_feedback_shift_engine<UIntType,w,k,q,s>
|
||
|
::linear_feedback_shift_engine(result_type value)
|
||
|
{
|
||
|
seed(value);
|
||
|
} // end linear_feedback_shift_engine::linear_feedback_shift_engine()
|
||
|
|
||
|
template<typename UIntType, size_t w, size_t k, size_t q, size_t s>
|
||
|
void linear_feedback_shift_engine<UIntType,w,k,q,s>
|
||
|
::seed(result_type value)
|
||
|
{
|
||
|
m_value = value;
|
||
|
} // end linear_feedback_shift_engine::seed()
|
||
|
|
||
|
template<typename UIntType, size_t w, size_t k, size_t q, size_t s>
|
||
|
typename linear_feedback_shift_engine<UIntType,w,k,q,s>::result_type
|
||
|
linear_feedback_shift_engine<UIntType,w,k,q,s>
|
||
|
::operator()(void)
|
||
|
{
|
||
|
const UIntType b = (((m_value << q) ^ m_value) & wordmask) >> (k-s);
|
||
|
const UIntType mask = ( (~static_cast<UIntType>(0)) << (w-k) ) & wordmask;
|
||
|
m_value = ((m_value & mask) << s) ^ b;
|
||
|
return m_value;
|
||
|
} // end linear_feedback_shift_engine::operator()()
|
||
|
|
||
|
|
||
|
template<typename UIntType, size_t w, size_t k, size_t q, size_t s>
|
||
|
void linear_feedback_shift_engine<UIntType,w,k,q,s>
|
||
|
::discard(unsigned long long z)
|
||
|
{
|
||
|
for(; z > 0; --z)
|
||
|
{
|
||
|
this->operator()();
|
||
|
} // end for
|
||
|
} // end linear_feedback_shift_engine::discard()
|
||
|
|
||
|
|
||
|
template<typename UIntType, size_t w, size_t k, size_t q, size_t s>
|
||
|
template<typename CharT, typename Traits>
|
||
|
std::basic_ostream<CharT,Traits>& linear_feedback_shift_engine<UIntType,w,k,q,s>
|
||
|
::stream_out(std::basic_ostream<CharT,Traits> &os) const
|
||
|
{
|
||
|
typedef std::basic_ostream<CharT,Traits> ostream_type;
|
||
|
typedef typename ostream_type::ios_base ios_base;
|
||
|
|
||
|
// save old flags & fill character
|
||
|
const typename ios_base::fmtflags flags = os.flags();
|
||
|
const CharT fill = os.fill();
|
||
|
|
||
|
os.flags(ios_base::dec | ios_base::fixed | ios_base::left);
|
||
|
os.fill(os.widen(' '));
|
||
|
|
||
|
// output one word of state
|
||
|
os << m_value;
|
||
|
|
||
|
// restore flags & fill character
|
||
|
os.flags(flags);
|
||
|
os.fill(fill);
|
||
|
|
||
|
return os;
|
||
|
}
|
||
|
|
||
|
|
||
|
template<typename UIntType, size_t w, size_t k, size_t q, size_t s>
|
||
|
template<typename CharT, typename Traits>
|
||
|
std::basic_istream<CharT,Traits>& linear_feedback_shift_engine<UIntType,w,k,q,s>
|
||
|
::stream_in(std::basic_istream<CharT,Traits> &is)
|
||
|
{
|
||
|
typedef std::basic_istream<CharT,Traits> istream_type;
|
||
|
typedef typename istream_type::ios_base ios_base;
|
||
|
|
||
|
// save old flags
|
||
|
const typename ios_base::fmtflags flags = is.flags();
|
||
|
|
||
|
is.flags(ios_base::skipws);
|
||
|
|
||
|
// input one word of state
|
||
|
is >> m_value;
|
||
|
|
||
|
// restore flags
|
||
|
is.flags(flags);
|
||
|
|
||
|
return is;
|
||
|
}
|
||
|
|
||
|
|
||
|
template<typename UIntType, size_t w, size_t k, size_t q, size_t s>
|
||
|
bool linear_feedback_shift_engine<UIntType,w,k,q,s>
|
||
|
::equal(const linear_feedback_shift_engine<UIntType,w,k,q,s> &rhs) const
|
||
|
{
|
||
|
return m_value == rhs.m_value;
|
||
|
}
|
||
|
|
||
|
|
||
|
template<typename UIntType, size_t w, size_t k, size_t q, size_t s>
|
||
|
bool operator==(const linear_feedback_shift_engine<UIntType,w,k,q,s> &lhs,
|
||
|
const linear_feedback_shift_engine<UIntType,w,k,q,s> &rhs)
|
||
|
{
|
||
|
return thrust::random::detail::random_core_access::equal(lhs,rhs);
|
||
|
}
|
||
|
|
||
|
|
||
|
template<typename UIntType, size_t w, size_t k, size_t q, size_t s>
|
||
|
bool operator!=(const linear_feedback_shift_engine<UIntType,w,k,q,s> &lhs,
|
||
|
const linear_feedback_shift_engine<UIntType,w,k,q,s> &rhs)
|
||
|
{
|
||
|
return !(lhs == rhs);
|
||
|
}
|
||
|
|
||
|
|
||
|
template<typename UIntType_, size_t w_, size_t k_, size_t q_, size_t s_,
|
||
|
typename CharT, typename Traits>
|
||
|
std::basic_ostream<CharT,Traits>&
|
||
|
operator<<(std::basic_ostream<CharT,Traits> &os,
|
||
|
const linear_feedback_shift_engine<UIntType_,w_,k_,q_,s_> &e)
|
||
|
{
|
||
|
return thrust::random::detail::random_core_access::stream_out(os,e);
|
||
|
}
|
||
|
|
||
|
|
||
|
template<typename UIntType_, size_t w_, size_t k_, size_t q_, size_t s_,
|
||
|
typename CharT, typename Traits>
|
||
|
std::basic_istream<CharT,Traits>&
|
||
|
operator>>(std::basic_istream<CharT,Traits> &is,
|
||
|
linear_feedback_shift_engine<UIntType_,w_,k_,q_,s_> &e)
|
||
|
{
|
||
|
return thrust::random::detail::random_core_access::stream_in(is,e);
|
||
|
}
|
||
|
|
||
|
|
||
|
} // end random
|
||
|
|
||
|
} // end thrust
|
||
|
|