diff --git a/wrap/main.go b/wrap/main.go index 9cc4562..ceff5cc 100644 --- a/wrap/main.go +++ b/wrap/main.go @@ -165,6 +165,22 @@ type Protocol struct { IPolymorphic map[string]int // polymorphic instance methods } +func (p Protocol) name() string { return p.Name } +func (p Protocol) goName() string { return p.GoName } +func (p Protocol) classMethods() map[string]*Method { return p.ClassMethods } +func (p Protocol) instanceMethods() map[string]*Method { return p.InstanceMethods } +func (p Protocol) cPolymorphic() map[string]int { return p.CPolymorphic } +func (p Protocol) iPolymorphic() map[string]int { return p.IPolymorphic } + +type methodCollection interface { + name() string + goName() string + classMethods() map[string]*Method + instanceMethods() map[string]*Method + cPolymorphic() map[string]int + iPolymorphic() map[string]int +} + //isVoid() returns true if the method has no return value. func (m Method) isVoid() bool { return m.Type.CType() == "void" @@ -274,10 +290,19 @@ type Interface struct { Properties map[string]*Property ClassMethods map[string]*Method InstanceMethods map[string]*Method + CPolymorphic map[string]int // polymorphic class methods + IPolymorphic map[string]int // polymorphic instance methods Protocols []string // Protocols impelemented by this Interface ProcessedInstanceMethods map[string]bool } +func (p Interface) name() string { return p.Name } +func (p Interface) goName() string { return p.GoName } +func (p Interface) classMethods() map[string]*Method { return p.ClassMethods } +func (p Interface) instanceMethods() map[string]*Method { return p.InstanceMethods } +func (p Interface) cPolymorphic() map[string]int { return p.CPolymorphic } +func (p Interface) iPolymorphic() map[string]int { return p.IPolymorphic } + func (w *Wrapper) AddInterface(n *ast.ObjCInterfaceDecl) { //fmt.Printf("ast.ObjCInterfaceDecl: %s\n",n.Name) w.add(n.Name, n.Children()) @@ -350,43 +375,53 @@ func (w *Wrapper) AddProtocol(n *ast.ObjCProtocolDecl) { for _,c := range n.Children() { switch x := c.(type) { case *ast.ObjCMethodDecl: + w.AddMethod(p,x) + } + } + w.Protocols[n.Name] = p +} + +func (w *Wrapper) AddMethod(p methodCollection, x *ast.ObjCMethodDecl) { m := &Method{ Name: x.Name, - Type: types.NewTypeFromString(x.Type,p.Name), - Class: p.Name, - GoClass: p.GoName, + Type: types.NewTypeFromString(x.Type,p.name()), + Class: p.name(), + GoClass: p.goName(), ClassMethod: x.ClassMethod, } //fmt.Printf(" -- Method %s\n",m.Name) var avail bool - m.Parameters, avail = w.GetParms(x,p.Name) + m.Parameters, avail = w.GetParms(x,p.name()) if avail { var mth map[string]*Method var poly map[string]int if m.ClassMethod { - mth = p.ClassMethods - poly = p.CPolymorphic + mth = p.classMethods() + poly = p.cPolymorphic() } else { - mth = p.InstanceMethods - poly = p.IPolymorphic + mth = p.instanceMethods() + poly = p.iPolymorphic() } pname := strings.Title(m.Name) + fmt.Printf("%s: Adding %s (%d)\n",p.name(),m.Name,len(m.Parameters)) if x := poly[m.Name]; x != 0 && len(m.Parameters) > 1 { poly[m.Name] = x + 1 + fmt.Printf(" polymorphic detected (%d)\n",poly[m.Name]) pname = pname + strings.Title(m.Parameters[1].Pname) - if m2 := mth[pname]; m2 != nil { + m.Name = pname + //if m2 := mth[pname]; m2 != nil { + if m2 := mth[m.Name]; m2 != nil { pname2 := pname + strings.Title(m2.Parameters[1].Pname) + m2.Name = pname2 mth[pname2] = m2 delete(mth,m.Name) } } else { poly[m.Name] = 1 } + fmt.Printf("--Method name is %s\n",m.Name) mth[pname] = m } - } - } - w.Protocols[n.Name] = p } //FIXME: copied from nswrap/main.go, should put this in a utils package @@ -451,6 +486,8 @@ func (w *Wrapper) add(name string, ns []ast.Node) { GoName: goname, Properties: map[string]*Property{}, InstanceMethods: map[string]*Method{}, + CPolymorphic: map[string]int{}, + IPolymorphic: map[string]int{}, Protocols: []string{}, ProcessedInstanceMethods: map[string]bool{}, } @@ -464,7 +501,7 @@ func (w *Wrapper) add(name string, ns []ast.Node) { } i.ClassMethods = map[string]*Method{"class": m} } - var avail bool + //var avail bool for _,c := range ns { switch x := c.(type) { case *ast.ObjCPropertyDecl: @@ -479,6 +516,7 @@ func (w *Wrapper) add(name string, ns []ast.Node) { //} case *ast.ObjCMethodDecl: //fmt.Printf("ObjCMethodDecl: %s (%s) %s\n",x.Type,name,x.Name) + /* if name == "NSObject" && x.Name == "initialize" { continue } @@ -497,7 +535,8 @@ func (w *Wrapper) add(name string, ns []ast.Node) { } else { i.InstanceMethods[strings.Title(m.Name)] = m } - } + }*/ + w.AddMethod(i,x) case *ast.ObjCProtocol: //fmt.Printf("ast.ObjCProtocol: %s\n",x.Name) i.Protocols = append(i.Protocols,x.Name)