| /* |
| * 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.sysds.runtime.instructions.cp; |
| |
| import org.apache.sysds.lops.LeftIndex; |
| import org.apache.sysds.lops.RightIndex; |
| import org.apache.sysds.common.Types.DataType; |
| import org.apache.sysds.runtime.DMLRuntimeException; |
| import org.apache.sysds.runtime.controlprogram.context.ExecutionContext; |
| import org.apache.sysds.runtime.instructions.InstructionUtils; |
| import org.apache.sysds.runtime.util.IndexRange; |
| |
| public abstract class IndexingCPInstruction extends UnaryCPInstruction { |
| protected final CPOperand rowLower, rowUpper, colLower, colUpper; |
| |
| protected IndexingCPInstruction(CPOperand in, CPOperand rl, CPOperand ru, CPOperand cl, CPOperand cu, |
| CPOperand out, String opcode, String istr) { |
| super(CPType.MatrixIndexing, null, in, out, opcode, istr); |
| rowLower = rl; |
| rowUpper = ru; |
| colLower = cl; |
| colUpper = cu; |
| } |
| |
| protected IndexingCPInstruction(CPOperand lhsInput, CPOperand rhsInput, CPOperand rl, CPOperand ru, |
| CPOperand cl, CPOperand cu, CPOperand out, String opcode, String istr) { |
| super(CPType.MatrixIndexing, null, lhsInput, rhsInput, out, opcode, istr); |
| rowLower = rl; |
| rowUpper = ru; |
| colLower = cl; |
| colUpper = cu; |
| } |
| |
| protected IndexRange getIndexRange(ExecutionContext ec) { |
| return new IndexRange( //rl, ru, cl, ru |
| (int)(ec.getScalarInput(rowLower).getLongValue()-1), |
| (int)(ec.getScalarInput(rowUpper).getLongValue()-1), |
| (int)(ec.getScalarInput(colLower).getLongValue()-1), |
| (int)(ec.getScalarInput(colUpper).getLongValue()-1)); |
| } |
| |
| public static IndexingCPInstruction parseInstruction ( String str ) { |
| String[] parts = InstructionUtils.getInstructionPartsWithValueType(str); |
| String opcode = parts[0]; |
| |
| if ( opcode.equalsIgnoreCase(RightIndex.OPCODE) ) { |
| if ( parts.length == 7 ) { |
| CPOperand in, rl, ru, cl, cu, out; |
| in = new CPOperand(parts[1]); |
| rl = new CPOperand(parts[2]); |
| ru = new CPOperand(parts[3]); |
| cl = new CPOperand(parts[4]); |
| cu = new CPOperand(parts[5]); |
| out = new CPOperand(parts[6]); |
| if( in.getDataType()==DataType.MATRIX ) |
| return new MatrixIndexingCPInstruction(in, rl, ru, cl, cu, out, opcode, str); |
| else if( in.getDataType() == DataType.FRAME ) |
| return new FrameIndexingCPInstruction(in, rl, ru, cl, cu, out, opcode, str); |
| else if( in.getDataType() == DataType.LIST ) |
| return new ListIndexingCPInstruction(in, rl, ru, cl, cu, out, opcode, str); |
| else |
| throw new DMLRuntimeException("Can index only on matrices, frames, and lists."); |
| } |
| else { |
| throw new DMLRuntimeException("Invalid number of operands in instruction: " + str); |
| } |
| } |
| else if ( opcode.equalsIgnoreCase(LeftIndex.OPCODE)) { |
| if ( parts.length == 8 ) { |
| CPOperand lhsInput, rhsInput, rl, ru, cl, cu, out; |
| lhsInput = new CPOperand(parts[1]); |
| rhsInput = new CPOperand(parts[2]); |
| rl = new CPOperand(parts[3]); |
| ru = new CPOperand(parts[4]); |
| cl = new CPOperand(parts[5]); |
| cu = new CPOperand(parts[6]); |
| out = new CPOperand(parts[7]); |
| if( lhsInput.getDataType()==DataType.MATRIX ) |
| return new MatrixIndexingCPInstruction(lhsInput, rhsInput, rl, ru, cl, cu, out, opcode, str); |
| else if (lhsInput.getDataType() == DataType.FRAME) |
| return new FrameIndexingCPInstruction(lhsInput, rhsInput, rl, ru, cl, cu, out, opcode, str); |
| else if( lhsInput.getDataType() == DataType.LIST ) |
| return new ListIndexingCPInstruction(lhsInput, rhsInput, rl, ru, cl, cu, out, opcode, str); |
| else |
| throw new DMLRuntimeException("Can index only on matrices, frames, and lists."); |
| } |
| else { |
| throw new DMLRuntimeException("Invalid number of operands in instruction: " + str); |
| } |
| } |
| else { |
| throw new DMLRuntimeException("Unknown opcode while parsing a MatrixIndexingCPInstruction: " + str); |
| } |
| } |
| } |