| /* |
| * Licensed to the Apache Software Foundation (ASF) under one or more |
| * contributor license agreements. See the NOTICE file distributed with |
| * this work for additional information regarding copyright ownership. |
| * The ASF licenses this file to You 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. |
| */ |
| |
| package org.apache.commons.rng.core.source32; |
| |
| import org.apache.commons.rng.core.util.NumberFactory; |
| |
| /** |
| * This abstract class is a base for algorithms from the Xor-Shift-Rotate family of 32-bit |
| * generators with 64-bits of state. |
| * |
| * @see <a href="http://xoshiro.di.unimi.it/">xorshiro / xoroshiro generators</a> |
| * @since 1.3 |
| */ |
| abstract class AbstractXoRoShiRo64 extends IntProvider { |
| /** Size of the state vector. */ |
| private static final int SEED_SIZE = 2; |
| |
| // State is maintained using variables rather than an array for performance |
| |
| /** State 0 of the generator. */ |
| protected int state0; |
| /** State 1 of the generator. */ |
| protected int state1; |
| |
| /** |
| * Creates a new instance. |
| * |
| * @param seed Initial seed. |
| * If the length is larger than 2, only the first 2 elements will |
| * be used; if smaller, the remaining elements will be automatically |
| * set. A seed containing all zeros will create a non-functional generator. |
| */ |
| AbstractXoRoShiRo64(int[] seed) { |
| if (seed.length < SEED_SIZE) { |
| final int[] state = new int[SEED_SIZE]; |
| fillState(state, seed); |
| setState(state); |
| } else { |
| setState(seed); |
| } |
| } |
| |
| /** |
| * Creates a new instance using a 2 element seed. |
| * A seed containing all zeros will create a non-functional generator. |
| * |
| * @param seed0 Initial seed element 0. |
| * @param seed1 Initial seed element 1. |
| */ |
| AbstractXoRoShiRo64(int seed0, int seed1) { |
| state0 = seed0; |
| state1 = seed1; |
| } |
| |
| /** |
| * Copies the state from the array into the generator state. |
| * |
| * @param state the new state |
| */ |
| private void setState(int[] state) { |
| state0 = state[0]; |
| state1 = state[1]; |
| } |
| |
| /** {@inheritDoc} */ |
| @Override |
| protected byte[] getStateInternal() { |
| return composeStateInternal(NumberFactory.makeByteArray(new int[] {state0, state1}), |
| super.getStateInternal()); |
| } |
| |
| /** {@inheritDoc} */ |
| @Override |
| protected void setStateInternal(byte[] s) { |
| final byte[][] c = splitStateInternal(s, SEED_SIZE * 4); |
| |
| setState(NumberFactory.makeIntArray(c[0])); |
| |
| super.setStateInternal(c[1]); |
| } |
| |
| /** {@inheritDoc} */ |
| @Override |
| public int next() { |
| final int result = nextOutput(); |
| |
| final int s0 = state0; |
| int s1 = state1; |
| |
| s1 ^= s0; |
| state0 = Integer.rotateLeft(s0, 26) ^ s1 ^ (s1 << 9); // a, b |
| state1 = Integer.rotateLeft(s1, 13); // c |
| |
| return result; |
| } |
| |
| /** |
| * Use the current state to compute the next output from the generator. |
| * The output function shall vary with respect to different generators. |
| * This method is called from {@link #next()} before the current state is updated. |
| * |
| * @return the next output |
| */ |
| protected abstract int nextOutput(); |
| } |