コード例 #1
0
ファイル: init.go プロジェクト: shenyp09/mx3
// cu.Init(), but error is fatal and does not dump stack.
func tryCuInit() {
	defer func() {
		err := recover()
		util.FatalErr(err, "initialize GPU:")
	}()
	cu.Init(0)
}
コード例 #2
0
ファイル: gui.go プロジェクト: shenyp09/mx3
func gui(w http.ResponseWriter, r *http.Request) {
	// TODO: racy with engine init, mv to package engine
	if guiTempl == nil {
		guiTempl = loadTemplate("gui.html") // TODO: embed.
		guis.Heun = engine.Solver
		guis.Mesh = engine.Mesh()
	}
	util.FatalErr(guiTempl.Execute(w, guis))
}
コード例 #3
0
ファイル: datatable.go プロジェクト: shenyp09/mx3
// make sure tabout is open.
func (t *dataTable) init() {
	if t.Writer == nil {
		f, err := os.OpenFile(OD+t.name+".txt", os.O_WRONLY|os.O_CREATE|os.O_TRUNC, 0666)
		util.FatalErr(err)
		t.Writer = bufio.NewWriter(f)
		fmt.Fprintln(t, "# t(s) mx my mz")
		t.Flush()
	}
}
コード例 #4
0
ファイル: init.go プロジェクト: shenyp09/mx3
func initCpuProf(OD string) {
	if *Flag_cpuprof {
		// start CPU profile to file
		fname := OD + "cpu.pprof"
		f, err := os.Create(fname)
		util.FatalErr(err, "CPU profile")
		err = pprof.StartCPUProfile(f)
		util.FatalErr(err, "CPU profile")
		log.Println("writing CPU profile to", fname)

		// at exit: exec go tool pprof to generate SVG output
		AtExit(func() {
			pprof.StopCPUProfile()
			me := procSelfExe()
			outfile := fname + ".svg"
			saveCmdOutput(outfile, "go", "tool", "pprof", "-svg", me, fname)
		})
	}
}
コード例 #5
0
ファイル: resize.go プロジェクト: shenyp09/mx3
func parseSize(arg string) (size [3]int) {
	words := strings.Split(arg, "x")
	if len(words) != 3 {
		log.Fatal("resize: need N0xN1xN2 argument")
	}
	for i, w := range words {
		v, err := strconv.Atoi(w)
		util.FatalErr(err)
		size[i] = v
	}
	return
}
コード例 #6
0
ファイル: init.go プロジェクト: shenyp09/mx3
// called by init()
func initGPUProf(OD string) {
	if *Flag_gpuprof {
		util.PanicErr(os.Setenv("CUDA_PROFILE", "1"))
		util.PanicErr(os.Setenv("CUDA_PROFILE_CSV", "1"))
		out := OD + "gpuprofile.csv"
		log.Println("writing GPU profile to", out)
		util.PanicErr(os.Setenv("CUDA_PROFILE_LOG", out))
		cfgfile := OD + "cudaprof.cfg"
		util.PanicErr(os.Setenv("CUDA_PROFILE_CONFIG", cfgfile))
		util.FatalErr(ioutil.WriteFile(cfgfile, []byte(CUDA_PROFILE_CONFIG), 0666), "gpuprof")
		//AtExit(cuda.DeviceReset)
	}
}
コード例 #7
0
ファイル: image.go プロジェクト: shenyp09/mx3
func Image(f *data.Slice, fmin, fmax string) *image.NRGBA {
	dim := f.NComp()
	switch dim {
	default:
		log.Fatalf("unsupported number of components: %v", dim)
	case 3:
		return drawVectors(f.Vectors())
	case 1:
		min, max := extrema(f.Host()[0])
		if fmin != "auto" {
			m, err := strconv.ParseFloat(fmin, 32)
			util.FatalErr(err)
			min = float32(m)
		}
		if fmax != "auto" {
			m, err := strconv.ParseFloat(fmax, 32)
			util.FatalErr(err)
			max = float32(m)
		}
		return drawFloats(f.Scalars(), min, max)
	}
	panic("unreachable")
}
コード例 #8
0
ファイル: od.go プロジェクト: shenyp09/mx3
// SetOD sets the output directory where auto-saved files will be stored.
func SetOD(od string, force bool) {
	if OD != "./" {
		log.Fatal("output directory already set to", OD)
	}
	OD = od
	if !strings.HasSuffix(OD, "/") {
		OD += "/"
	}
	log.Println("output directory:", OD)

	{ // make OD
		wd, err := os.Getwd()
		util.FatalErr(err, "create output directory:")
		stat, err2 := os.Stat(wd)
		util.FatalErr(err2, "create output directory:")
		util.LogErr(os.Mkdir(od, stat.Mode())) // already exists is OK
	}

	// fail on non-empty OD
	f, err3 := os.Open(od)
	util.FatalErr(err3, "open output directory:")
	files, _ := f.Readdir(1)
	if !force && len(files) != 0 {
		log.Fatal(od, " not empty, clean it or force with -f")
	}

	// clean output dir
	if len(files) != 0 && OD != "." {
		log.Println("cleaning files in", OD)
		filepath.Walk(OD, func(path string, i os.FileInfo, err error) error {
			if path != OD {
				util.FatalErr(os.RemoveAll(path), "clean output directory:")
			}
			return nil
		})
	}
}
コード例 #9
0
ファイル: server.go プロジェクト: shenyp09/mx3
func GoServe(port string) {

	http.HandleFunc("/cmd/top", Command("top", "-b", "-n", "1"))
	http.HandleFunc("/cmd/uname", Command("uname", "-a"))

	http.HandleFunc("/render/", render)

	http.HandleFunc("/", gui)

	log.Print("serving http://localhost", port, "\n")
	go func() {
		cuda.LockThread()
		util.FatalErr(http.ListenAndServe(port, nil))
	}()
	runtime.Gosched()
}
コード例 #10
0
ファイル: read.go プロジェクト: shenyp09/mx3
func MustReadFile(fname string) (data *Slice, time float64) {
	s, t, err := ReadFile(fname)
	util.FatalErr(err)
	return s, t
}
コード例 #11
0
ファイル: main.go プロジェクト: shenyp09/mx3
func open(fname string) *os.File {
	f, err := os.OpenFile(fname, os.O_WRONLY|os.O_CREATE|os.O_TRUNC, 0666)
	util.FatalErr(err)
	return f
}
コード例 #12
0
ファイル: write.go プロジェクト: shenyp09/mx3
func MustWriteFile(fname string, s *Slice, time float64) {
	err := WriteFile(fname, s, time)
	util.FatalErr(err)
}
コード例 #13
0
ファイル: gui.go プロジェクト: shenyp09/mx3
func loadTemplate(fname string) *template.Template {
	body, err := ioutil.ReadFile(fname)
	util.FatalErr(err)
	return template.Must(template.New(fname).Parse(string(body)))
}