/** * Copyright (C) 2017 Jan Schäfer (jansch@users.sourceforge.net) * * 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. */ package org.jskat.ai.nn.input; import org.jskat.data.Trick; import org.jskat.player.ImmutablePlayerKnowledge; import org.jskat.util.Card; public class CurrentTrickStrategy extends AbstractInputStrategy { @Override public int getNeuronCount() { return 3 * 32; } @Override public double[] getNetworkInput(ImmutablePlayerKnowledge knowledge, Card cardToPlay) { double[] result = getEmptyInputs(); Trick trick = knowledge.getCurrentTrick(); setTrickCardInputs(result, trick); return result; } protected final void setTrickCardInputs(double[] result, Trick trick) { // set already played cards if (trick.getFirstCard() != null) { result[getNetworkInputIndex(trick.getFirstCard())] = ON; } if (trick.getSecondCard() != null) { result[32 + getNetworkInputIndex(trick.getSecondCard())] = ON; } if (trick.getThirdCard() != null) { result[64 + getNetworkInputIndex(trick.getThirdCard())] = ON; } } }