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

args<-commandArgs(TRUE)
options(digits=22)
library(Matrix)
library(matrixStats)

N = 2
C = 2
Hin = 3
Win = 4

X = matrix(cbind(seq(1,20),seq(1,20),seq(1,8)), nrow=2, ncol=24, byrow=TRUE)
gamma = matrix(c(1,2), byrow=TRUE, nrow=2, ncol=1)
beta = matrix(c(0,1), byrow=TRUE, nrow=2, ncol=1)
ema_mean = matrix(c(4,5), byrow=TRUE, nrow=2, ncol=1)
ema_var = matrix(c(2,3), byrow=TRUE, nrow=2, ncol=1)
mu = 0.95
epsilon = 1e-4

subgrp_means = matrix(colMeans(X), nrow=C, ncol=Hin*Win, byrow=TRUE)
subgrp_vars = matrix(colVars(X) * ((N-1)/N), nrow=C, ncol=Hin*Win, byrow=TRUE)
mean = rowMeans(subgrp_means)  # shape (C, 1)
var = rowMeans(subgrp_vars) + rowVars(subgrp_means)*(((Hin*Win)-1)/(Hin*Win))
ema_mean_upd = mu*ema_mean + (1-mu)*mean
ema_var_upd = mu*ema_var + (1-mu)*var

R = cbind(mean, var, ema_mean_upd, ema_var_upd)

writeMM(as(R, "CsparseMatrix"), paste(args[1], "R", sep=""));