#-------------------------------------------------------------
#
# 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.
#
#-------------------------------------------------------------

# How to invoke this dml script LineageTrace.dml?
# Assume LR_HOME is set to the home of the dml script
# Assume rows = 20 and cols = 20 for X
# hadoop jar SystemDS.jar -f $LR_HOME/LineageTrace.dml -args "$INPUT_DIR/X" "$OUTPUT_DIR/X" "$OUTPUT_DIR/Y"

# X = rand(rows=20, cols=20);
X = read($1);
max_iteration = 3;
i = 0;

while(i < max_iteration) {
  if (i %% 2 == 1)
    X = X * X * X;
  else
    X = X + 7;
  i = i + 1;
}

Y = t(X) %*% X;

write(X, $2, format="text");
write(Y, $3, format="text");