{-# OPTIONS_GHC  -fno-warn-orphans #-}
{-# LANGUAGE QuasiQuotes #-}
{-# LANGUAGE FlexibleInstances #-}
{-# LANGUAGE TypeOperators #-}
{-# LANGUAGE DataKinds #-}
{-# LANGUAGE ScopedTypeVariables #-}

module AddrOfRegression where

import Ivory.Language

[ivory|
struct param_info
  { param_requested :: Stored Uint8
  ; other           :: Stored Uint16
  }
|]

test1 :: ModuleDef
test1 = do
  defMemArea param_info_area
  incl t1
  where
  param_info_area :: MemArea (Array 512 (Struct "param_info"))
  param_info_area = area "g_param_info" Nothing

  param_info_ref :: Ref Global (Array 512 (Struct "param_info"))
  param_info_ref = addrOf param_info_area

  t1 :: Def ('[]:->())
  t1 = proc "t1" $ body $ do
    arrayMap $ \ix ->
      store ((param_info_ref ! ix) ~> param_requested) 1

test1_noarray :: ModuleDef
test1_noarray = do
  defMemArea param_info_area
  incl t1
  where
  param_info_area :: MemArea (Struct "param_info")
  param_info_area = area "single_param_info" Nothing

  param_info_ref :: Ref Global (Struct "param_info")
  param_info_ref = addrOf param_info_area

  t1 :: Def ('[]:->())
  t1 = proc "t1_noarray" $ body $ do
    store (param_info_ref ~> param_requested) 1

test2 :: ModuleDef
test2 = do
  defMemArea atom_array_area
  incl t2
  where
  atom_array_area :: MemArea (Array 512 (Stored IFloat))
  atom_array_area = area "atom_array" Nothing

  atom_array_ref :: Ref Global (Array 512 (Stored IFloat))
  atom_array_ref = addrOf atom_array_area

  t2 :: Def ('[]:->())
  t2 = proc "t2" $ body $ do
    arrayMap $ \ix ->
      store (atom_array_ref ! ix) 1

test3 :: ModuleDef
test3 = do
  incl t3
  where
  t3 :: Def ('[]:->())
  t3 = proc "t3" $ body $ do
    (stack_array :: Ref (Stack s) (Array 512 (Stored IFloat))) <- local izero
    arrayMap $ \ix ->
      store (stack_array ! ix) 1


cmodule :: Module
cmodule = package "AddrOfRegression" $ do
  defStruct (Proxy :: Proxy "param_info")
  test1
  test1_noarray
  test2
  test3