batchatco / go-native-netcdf Goto Github PK

package main

import (
	"fmt"
	"github.com/batchatco/go-native-netcdf/netcdf/api"
	"github.com/batchatco/go-native-netcdf/netcdf/cdf"
	"github.com/samber/lo"
	"runtime"
	"time"
)

func timeNow() string {
	return time.Now().Format("2006-01-02 15:04:05.000")
}
func writeXarray(fn string) {
	cw, err00 := cdf.OpenWriter(fn)
	if !((err00) == (nil)) {
		fmt.Printf("%v cdf.OpenWriter(fn) err00=%v\n", timeNow(), err00)
		panic(err00)
	}
	fmt.Printf("%v define coordinate x \n", timeNow())
	nx := 2
	x := lo.Map[int, uint8](lo.Range(2), func(x int, _ int) uint8 {
		return uint8(x)
	})
	err01 := cw.AddVar("x", api.Variable{x, []string{"x"}, nil})
	if !((err01) == (nil)) {
		fmt.Printf("%v cw.AddVar('x', api.Variable {x, []string {'x'}, nil}) err01=%v\n", timeNow(), err01)
		panic(err01)
	}
	fmt.Printf("%v define coordinate y \n", timeNow())
	ny := 3
	y := lo.Map[int, uint8](lo.Range(3), func(x int, _ int) uint8 {
		return uint8(x)
	})
	err02 := cw.AddVar("y", api.Variable{y, []string{"y"}, nil})
	if !((err02) == (nil)) {
		fmt.Printf("%v cw.AddVar('y', api.Variable {y, []string {'y'}, nil}) err02=%v\n", timeNow(), err02)
		panic(err02)
	}
	fmt.Printf("%v define coordinate z \n", timeNow())
	nz := 4
	z := lo.Map[int, uint8](lo.Range(4), func(x int, _ int) uint8 {
		return uint8(x)
	})
	err03 := cw.AddVar("z", api.Variable{z, []string{"z"}, nil})
	if !((err03) == (nil)) {
		fmt.Printf("%v cw.AddVar('z', api.Variable {z, []string {'z'}, nil}) err03=%v\n", timeNow(), err03)
		panic(err03)
	}
	fmt.Printf("%v fill 3d dataset val \n", timeNow())
	val := make([][][]uint16, nx)
	for i := 0; i < nx; (i)++ {
		val[i] = make([][]uint16, ny)
		for j := 0; j < ny; (j)++ {
			val[i][j] = make([]uint16, nz)
			for k := 0; k < nz; (k)++ {
				val[i][j][k] = uint16((((100) * (i)) + ((10) * (j)) + ((1) * (k))))
			}
		}
	}
	err04 := cw.AddVar("val", api.Variable{val, []string{"x", "y", "z"}, nil})
	if !((err04) == (nil)) {
		fmt.Printf("%v cw.AddVar('val', api.Variable {val, []string {'x', 'y', 'z'}, nil}) err04=%v\n", timeNow(), err04)
		panic(err04)
	}
	defer (func() {
		fmt.Printf("%v close fn=%v\n", timeNow(), fn)
		err05 := cw.Close()
		if !((err05) == (nil)) {
			fmt.Printf("%v cw.Close() err05=%v\n", timeNow(), err05)
			panic(err05)
		}
	})()
}
func readXarray(fn string) {
	nc, err06 := cdf.Open(fn)
	if !((err06) == (nil)) {
		fmt.Printf("%v cdf.Open(fn) err06=%v\n", timeNow(), err06)
		panic(err06)
	}
	defer (func() {
		fmt.Printf("%v close file fn=%v\n", timeNow(), fn)
		nc.Close()
	})()
	fmt.Printf("%v obtain and show variable 'x' from netcdf file \n", timeNow())
	vrX, err07 := nc.GetVariable("x")
	if !((err07) == (nil)) {
		fmt.Printf("%v nc.GetVariable('x') err07=%v\n", timeNow(), err07)
		panic(err07)
	}
	if (vrX) == (nil) {
		fmt.Printf("%v variable x not found \n", timeNow())
		panic("")
	}
	x, hasX := vrX.Values.([]uint8)
	if !(hasX) {
		fmt.Printf("%v variable x conversion to []uint8 failed \n", timeNow())
		panic("")
	}
	fmt.Printf("%v  x[0]=%v x[((len(x))-(1))]=%v len(x)=%v\n", timeNow(), x[0], x[((len(x))-(1))], len(x))
	fmt.Printf("%v obtain and show variable 'y' from netcdf file \n", timeNow())
	vrY, err08 := nc.GetVariable("y")
	if !((err08) == (nil)) {
		fmt.Printf("%v nc.GetVariable('y') err08=%v\n", timeNow(), err08)
		panic(err08)
	}
	if (vrY) == (nil) {
		fmt.Printf("%v variable y not found \n", timeNow())
		panic("")
	}
	y, hasY := vrY.Values.([]uint8)
	if !(hasY) {
		fmt.Printf("%v variable y conversion to []uint8 failed \n", timeNow())
		panic("")
	}
	fmt.Printf("%v  y[0]=%v y[((len(y))-(1))]=%v len(y)=%v\n", timeNow(), y[0], y[((len(y))-(1))], len(y))
	fmt.Printf("%v obtain and show variable 'z' from netcdf file \n", timeNow())
	vrZ, err09 := nc.GetVariable("z")
	if !((err09) == (nil)) {
		fmt.Printf("%v nc.GetVariable('z') err09=%v\n", timeNow(), err09)
		panic(err09)
	}
	if (vrZ) == (nil) {
		fmt.Printf("%v variable z not found \n", timeNow())
		panic("")
	}
	z, hasZ := vrZ.Values.([]uint8)
	if !(hasZ) {
		fmt.Printf("%v variable z conversion to []uint8 failed \n", timeNow())
		panic("")
	}
	fmt.Printf("%v  z[0]=%v z[((len(z))-(1))]=%v len(z)=%v\n", timeNow(), z[0], z[((len(z))-(1))], len(z))
	fmt.Printf("%v obtain and show variable 'val' from netcdf file \n", timeNow())
	vrVal, err10 := nc.GetVariable("val")
	if !((err10) == (nil)) {
		fmt.Printf("%v nc.GetVariable('val') err10=%v\n", timeNow(), err10)
		panic(err10)
	}
	if (vrVal) == (nil) {
		fmt.Printf("%v variable val not found \n", timeNow())
		panic("")
	}
	val, hasVal := vrVal.Values.([][][]uint16)
	if !(hasVal) {
		fmt.Printf("%v variable val conversion to [][][]uint16 failed \n", timeNow())
		panic("")
	}
	fmt.Printf("%v  val[0]=%v val[((len(val))-(1))]=%v len(val)=%v\n", timeNow(), val[0], val[((len(val))-(1))], len(val))
}
func main() {
	fmt.Printf("%v netcdf runtime.Version()=%v\n", timeNow(), runtime.Version())
	fn := "newdata.nc"
	writeXarray(fn)
	readXarray(fn)
}