// register element
func init() {
// information allocator
infogetters["rod"] = func(cellType string, faceConds []*FaceCond) *Info {
// new info
var info Info
// number of nodes in element
nverts := shp.GetNverts(cellType)
// solution variables
ykeys := []string{"ux", "uy"}
if Global.Ndim == 3 {
ykeys = []string{"ux", "uy", "uz"}
}
info.Dofs = make([][]string, nverts)
for m := 0; m < nverts; m++ {
info.Dofs[m] = ykeys
}
// maps
info.Y2F = map[string]string{"ux": "fx", "uy": "fy", "uz": "fz"}
// t1 and t2 variables
info.T2vars = ykeys
return &info
}
// element allocator
eallocators["rod"] = func(cellType string, faceConds []*FaceCond, cid int, edat *inp.ElemData, x [][]float64) Elem {
// basic data
var o Rod
o.Cid = cid
o.X = x
o.Shp = shp.Get(cellType)
ndim := Global.Ndim
o.Nu = ndim * o.Shp.Nverts
var err error
// material model name
matname := edat.Mat
matdata := Global.Sim.Mdb.Get(matname)
if LogErrCond(matdata == nil, "materials database failed on getting %q material\n", matname) {
return nil
}
mdlname := matdata.Model
o.Model = msolid.GetOnedSolid(Global.Sim.Data.FnameKey, matname, mdlname, false)
if LogErrCond(o.Model == nil, "cannot find model named %s\n", mdlname) {
return nil
}
err = o.Model.Init(ndim, matdata.Prms)
if LogErr(err, "Model.Init failed") {
return nil
}
// parameters
for _, p := range matdata.Prms {
switch p.N {
case "A":
o.A = p.V
case "rho":
o.Rho = p.V
}
}
// integration points
var nip int
if s_nip, found := io.Keycode(edat.Extra, "nip"); found {
nip = io.Atoi(s_nip)
}
o.IpsElem, err = shp.GetIps(o.Shp.Type, nip)
if LogErr(err, "GetIps failed") {
return nil
}
nip = len(o.IpsElem)
// scratchpad. computed @ each ip
o.K = la.MatAlloc(o.Nu, o.Nu)
o.M = la.MatAlloc(o.Nu, o.Nu)
o.ue = make([]float64, o.Nu)
o.Rus = make([]float64, o.Nu)
// scratchpad. computed @ each ip
o.grav = make([]float64, ndim)
o.us = make([]float64, ndim)
o.fi = make([]float64, o.Nu)
// return new element
return &o
}
}
// register element
func init() {
// information allocator
infogetters["rod"] = func(sim *inp.Simulation, cell *inp.Cell, edat *inp.ElemData) *Info {
// new info
var info Info
// number of nodes in element
nverts := cell.Shp.Nverts
// solution variables
ykeys := []string{"ux", "uy"}
if sim.Ndim == 3 {
ykeys = []string{"ux", "uy", "uz"}
}
info.Dofs = make([][]string, nverts)
for m := 0; m < nverts; m++ {
info.Dofs[m] = ykeys
}
// maps
info.Y2F = map[string]string{"ux": "fx", "uy": "fy", "uz": "fz"}
// t1 and t2 variables
info.T2vars = ykeys
return &info
}
// element allocator
eallocators["rod"] = func(sim *inp.Simulation, cell *inp.Cell, edat *inp.ElemData, x [][]float64) Elem {
// basic data
var o Rod
o.Cell = cell
o.X = x
o.Ndim = sim.Ndim
o.Nu = o.Ndim * o.Cell.Shp.Nverts
// parameters
matdata := sim.MatParams.Get(edat.Mat)
if matdata == nil {
chk.Panic("cannot get materials data for rod element {tag=%d id=%d material=%q}", cell.Tag, cell.Id, edat.Mat)
}
o.Model = msolid.GetOnedSolid(sim.Key, edat.Mat, matdata.Model, false)
if o.Model == nil {
chk.Panic("cannot get model for rod element {tag=%d id=%d material=%q}", cell.Tag, cell.Id, edat.Mat)
}
err := o.Model.Init(o.Ndim, matdata.Prms)
if err != nil {
chk.Panic("model initialisation failed:\n%v", err)
}
// parameters
for _, p := range matdata.Prms {
switch p.N {
case "A":
o.A = p.V
case "rho":
o.Rho = p.V
}
}
// integration points
o.IpsElem, _, err = o.Cell.Shp.GetIps(edat.Nip, 0)
if err != nil {
chk.Panic("cannot get integration points for rod element {tag=%d id=%d material=%q} with nip=%d", cell.Tag, cell.Id, edat.Mat, edat.Nip)
}
// scratchpad. computed @ each ip
o.K = la.MatAlloc(o.Nu, o.Nu)
o.M = la.MatAlloc(o.Nu, o.Nu)
o.ue = make([]float64, o.Nu)
// scratchpad. computed @ each ip
o.grav = make([]float64, o.Ndim)
o.us = make([]float64, o.Ndim)
o.fi = make([]float64, o.Nu)
// return new element
return &o
}
}