from layer_naive import MulLayer
from layer_naive import AddLayer

apple = 100
apple_num = 2
orange = 150
orange_num = 3
tax = 1.1

# Layer
mul_apple_layer = MulLayer()
mul_orange_layer = MulLayer()
add_fruit_layer = AddLayer()
mul_tax_layer = MulLayer()

# Forward
apple_price = mul_apple_layer.forward(apple, apple_num)
orange_price = mul_orange_layer.forward(orange, orange_num)
price = add_fruit_layer.forward(apple_price, orange_price)
total_price = mul_tax_layer.forward(price, tax)

# Backward
dtotal_price = 1
dprice, dtax = mul_tax_layer.backward(dtotal_price)
dapple_price, dorange_price = add_fruit_layer.backward(dprice)
dapple, dapple_num = mul_apple_layer.backward(dapple_price)
dorange, dorange_num = mul_orange_layer.backward(dorange_price)

print('apple_price: ', apple_price)
print('orange_price:', orange_price)
print('price:', price)
# coding: utf-8
from layer_naive import MulLayer, AddLayer

apple = 100
apple_num = 2
orange = 150
orange_num = 3
tax = 1.1

# layer
mul_apple_layer = MulLayer()
mul_orange_layer = MulLayer()
add_apple_orange_layer = AddLayer()
mul_tax_layer = MulLayer()

# forward
apple_price = mul_apple_layer.forward(apple, apple_num)  # (1)
orange_price = mul_orange_layer.forward(orange, orange_num)  # (2)
all_price = add_apple_orange_layer.forward(apple_price, orange_price)  # (3)
price = mul_tax_layer.forward(all_price, tax)  # (4)

# backward
dprice = 1
dall_price, dtax = mul_tax_layer.backward(dprice)  # (4)
dapple_price, dorange_price = add_apple_orange_layer.backward(dall_price)  # (3
dorange, dorange_num = mul_orange_layer.backward(dorange_price)  # (2)
dapple, dapple_num = mul_apple_layer.backward(dapple_price)  # (1)

print("price:", int(price))
print("dApple:", dapple)
print("dApple_num:", int(dapple_num))
from layer_naive import MultiLayer
from layer_naive import AddLayer

apple_price = 100
apple_unit = 2
orange_price = 150
orange_unit = 3
tax = 1.1

mul_apple_cost_layer = MultiLayer()
mul_orange_cost_layer = MultiLayer()
mul_cost_layer = AddLayer()
mul_amount_layer = MultiLayer()

apple_cost = mul_apple_cost_layer.forward(apple_price, apple_unit)
orange_cost = mul_orange_cost_layer.forward(orange_price, orange_unit)
cost = mul_cost_layer.forward(apple_cost, orange_cost)
amount = mul_amount_layer.forward(cost, tax)

print(amount)

d_amount = 1
d_cost, d_tax = mul_amount_layer.backward(d_amount)
d_apple_cost, d_orange_cost = mul_cost_layer.backward(d_cost)
d_apple_price, d_apple_unit = mul_apple_cost_layer.backward(d_apple_cost)
d_orange_price, d_orange_unit = mul_orange_cost_layer.backward(d_orange_cost)
print(d_apple_price, d_apple_unit, d_orange_price, d_orange_unit, d_tax)