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.
203 lines
5.8 KiB
203 lines
5.8 KiB
/* |
|
* 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/xor_combine_engine.h> |
|
#include <thrust/random/detail/random_core_access.h> |
|
|
|
namespace thrust |
|
{ |
|
|
|
namespace random |
|
{ |
|
|
|
template<typename Engine1, size_t s1, |
|
typename Engine2, size_t s2> |
|
xor_combine_engine<Engine1,s1,Engine2,s2> |
|
::xor_combine_engine(void) |
|
:m_b1(),m_b2() |
|
{ |
|
} // end xor_combine_engine::xor_combine_engine() |
|
|
|
template<typename Engine1, size_t s1, |
|
typename Engine2, size_t s2> |
|
xor_combine_engine<Engine1,s1,Engine2,s2> |
|
::xor_combine_engine(const base1_type &urng1, const base2_type &urng2) |
|
:m_b1(urng1),m_b2(urng2) |
|
{ |
|
} // end xor_combine_engine::xor_combine_engine() |
|
|
|
template<typename Engine1, size_t s1, |
|
typename Engine2, size_t s2> |
|
xor_combine_engine<Engine1,s1,Engine2,s2> |
|
::xor_combine_engine(result_type s) |
|
:m_b1(s),m_b2(s) |
|
{ |
|
} // end xor_combine_engine::xor_combine_engine() |
|
|
|
template<typename Engine1, size_t s1, |
|
typename Engine2, size_t s2> |
|
void xor_combine_engine<Engine1,s1,Engine2,s2> |
|
::seed(void) |
|
{ |
|
m_b1.seed(); |
|
m_b2.seed(); |
|
} // end xor_combine_engine::seed() |
|
|
|
template<typename Engine1, size_t s1, |
|
typename Engine2, size_t s2> |
|
void xor_combine_engine<Engine1,s1,Engine2,s2> |
|
::seed(result_type s) |
|
{ |
|
m_b1.seed(s); |
|
m_b2.seed(s); |
|
} // end xor_combine_engine::seed() |
|
|
|
template<typename Engine1, size_t s1, |
|
typename Engine2, size_t s2> |
|
const typename xor_combine_engine<Engine1,s1,Engine2,s2>::base1_type & |
|
xor_combine_engine<Engine1,s1,Engine2,s2> |
|
::base1(void) const |
|
{ |
|
return m_b1; |
|
} // end xor_combine_engine::base1() |
|
|
|
template<typename Engine1, size_t s1, |
|
typename Engine2, size_t s2> |
|
const typename xor_combine_engine<Engine1,s1,Engine2,s2>::base2_type & |
|
xor_combine_engine<Engine1,s1,Engine2,s2> |
|
::base2(void) const |
|
{ |
|
return m_b2; |
|
} // end xor_combine_engine::base2() |
|
|
|
template<typename Engine1, size_t s1, |
|
typename Engine2, size_t s2> |
|
typename xor_combine_engine<Engine1,s1,Engine2,s2>::result_type |
|
xor_combine_engine<Engine1,s1,Engine2,s2> |
|
::operator()(void) |
|
{ |
|
return (result_type(m_b1() - base1_type::min) << shift1) ^ |
|
(result_type(m_b2() - base2_type::min) << shift2); |
|
} // end xor_combine_engine::operator()() |
|
|
|
template<typename Engine1, size_t s1, |
|
typename Engine2, size_t s2> |
|
void xor_combine_engine<Engine1, s1, Engine2, s2> |
|
::discard(unsigned long long z) |
|
{ |
|
for(; z > 0; --z) |
|
{ |
|
this->operator()(); |
|
} // end for |
|
} // end xor_combine_engine::discard() |
|
|
|
|
|
template<typename Engine1, size_t s1, typename Engine2, size_t s2> |
|
template<typename CharT, typename Traits> |
|
std::basic_ostream<CharT,Traits>& xor_combine_engine<Engine1,s1,Engine2,s2> |
|
::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 and fill character |
|
const typename ios_base::fmtflags flags = os.flags(); |
|
const CharT fill = os.fill(); |
|
|
|
const CharT space = os.widen(' '); |
|
os.flags(ios_base::dec | ios_base::fixed | ios_base::left); |
|
os.fill(space); |
|
|
|
// output each base engine in turn |
|
os << base1() << space << base2(); |
|
|
|
// restore old flags and fill character |
|
os.flags(flags); |
|
os.fill(fill); |
|
return os; |
|
} |
|
|
|
|
|
template<typename Engine1, size_t s1, typename Engine2, size_t s2> |
|
template<typename CharT, typename Traits> |
|
std::basic_istream<CharT,Traits>& xor_combine_engine<Engine1,s1,Engine2,s2> |
|
::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 each base engine in turn |
|
is >> m_b1 >> m_b2; |
|
|
|
// restore old flags |
|
is.flags(flags); |
|
return is; |
|
} |
|
|
|
|
|
template<typename Engine1, size_t s1, typename Engine2, size_t s2> |
|
bool xor_combine_engine<Engine1,s1,Engine2,s2> |
|
::equal(const xor_combine_engine<Engine1,s1,Engine2,s2> &rhs) const |
|
{ |
|
return (m_b1 == rhs.m_b1) && (m_b2 == rhs.m_b2); |
|
} |
|
|
|
|
|
template<typename Engine1, size_t s1, typename Engine2, size_t s2, |
|
typename CharT, typename Traits> |
|
std::basic_ostream<CharT,Traits>& |
|
operator<<(std::basic_ostream<CharT,Traits> &os, |
|
const xor_combine_engine<Engine1,s1,Engine2,s2> &e) |
|
{ |
|
return thrust::random::detail::random_core_access::stream_out(os,e); |
|
} |
|
|
|
|
|
template<typename Engine1, size_t s1, typename Engine2, size_t s2, |
|
typename CharT, typename Traits> |
|
std::basic_istream<CharT,Traits>& |
|
operator>>(std::basic_istream<CharT,Traits> &is, |
|
xor_combine_engine<Engine1,s1,Engine2,s2> &e) |
|
{ |
|
return thrust::random::detail::random_core_access::stream_in(is,e); |
|
} |
|
|
|
|
|
template<typename Engine1, size_t s1, typename Engine2, size_t s2> |
|
bool operator==(const xor_combine_engine<Engine1,s1,Engine2,s2> &lhs, |
|
const xor_combine_engine<Engine1,s1,Engine2,s2> &rhs) |
|
{ |
|
return thrust::random::detail::random_core_access::equal(lhs,rhs); |
|
} |
|
|
|
|
|
template<typename Engine1, size_t s1, typename Engine2, size_t s2> |
|
bool operator!=(const xor_combine_engine<Engine1,s1,Engine2,s2> &lhs, |
|
const xor_combine_engine<Engine1,s1,Engine2,s2> &rhs) |
|
{ |
|
return !(lhs == rhs); |
|
} |
|
|
|
|
|
} // end random |
|
|
|
} // end thrust |
|
|
|
|