Files
armadillo-code/include/armadillo_bits/distr_param.hpp
T
2022-05-26 16:58:34 +10:00

105 lines
1.9 KiB
C++

// SPDX-License-Identifier: Apache-2.0
//
// Copyright 2008-2016 Conrad Sanderson (http://conradsanderson.id.au)
// Copyright 2008-2016 National ICT Australia (NICTA)
//
// 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.
// ------------------------------------------------------------------------
//! \addtogroup distr_param
//! @{
class distr_param
{
public:
uword state;
union
{
int a_int;
double a_double;
};
union
{
int b_int;
double b_double;
};
inline distr_param()
: state(0)
{
}
inline explicit distr_param(const int a, const int b)
: state(1)
, a_int(a)
, b_int(b)
{
}
inline explicit distr_param(const double a, const double b)
: state(2)
, a_double(a)
, b_double(b)
{
}
inline void get_int_vals(int& out_a, int& out_b) const
{
// if state == 0, do nothing
if(state == 1)
{
out_a = a_int;
out_b = b_int;
}
else
if(state == 2)
{
out_a = int(a_double);
out_b = int(b_double);
}
}
inline void get_double_vals(double& out_a, double& out_b) const
{
// if state == 0, do nothing
if(state == 1)
{
out_a = double(a_int);
out_b = double(b_int);
}
else
if(state == 2)
{
out_a = a_double;
out_b = b_double;
}
}
};
//! @}