// Copyright 2017 The go-python Authors.  All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.

package slices

import (
	"fmt"
	"math/cmplx"
)

func IntSum(s []int) int {
	sum := 0
	for _, value := range s {
		sum += value
	}
	return sum
}

func CreateSlice() []int {
	return []int{1, 2, 3, 4}
}

type SliceUint8 []uint8
type SliceUint16 []uint16
type SliceUint32 []uint32
type SliceUint64 []uint64

type SliceInt8 []int8
type SliceInt16 []int16
type SliceInt32 []int32
type SliceInt64 []int64

type SliceComplex []complex128

type SliceIface []interface{}

type S struct {
	Name string
}

func CreateSSlice() []*S {
	return []*S{&S{"S0"}, &S{"S1"}, &S{"S2"}}
}

func PrintSSlice(ss []*S) {
	for i, s := range ss {
		fmt.Printf("%d: %v\n", i, s.Name)
	}
}

func PrintS(s *S) {
	fmt.Printf("%v\n", s.Name)
}

func CmplxSqrt(arr SliceComplex) SliceComplex {
	res := make([]complex128, len(arr))
	for i, el := range arr {
		res[i] = cmplx.Sqrt(el)
	}
	return res
}