allow some more positions for return

src/col/vct/col/ast/statement/exceptional/ReturnImpl.scala

@@ -1,18 +1,19 @@
 package vct.col.ast.statement.exceptional
 
-import vct.col.ast.{AbstractMethod, Constructor, Return, Void}
+import vct.col.ast._
 import vct.col.print.{Ctx, Doc, Empty, Text}
 import vct.col.ast.ops.ReturnOps
 import vct.col.check.{CheckContext, CheckError, ReturnOutsideMethod}
 
 trait ReturnImpl[G] extends ExceptionalStatementImpl[G] with ReturnOps[G] { this: Return[G] =>
-  override def check(context: CheckContext[G]): Seq[CheckError] =
-    super.check(context) ++ (context.currentApplicable match {
-      case None => Seq(ReturnOutsideMethod(this))
-      case Some(_: Constructor[G]) => Seq(ReturnOutsideMethod(this))
-      case Some(_: AbstractMethod[G]) => Nil
-      case Some(_) => Seq(ReturnOutsideMethod(this))
-    })
+  override def check(context: CheckContext[G]): Seq[CheckError] = {
+    val app = context.declarationStack.collectFirst {
+      case _: Procedure[G] | _: InstanceMethod[G] | _: InstanceOperatorMethod[G] => ()
+      case _: JavaMethod[G] | _: CFunctionDefinition[G] | _: CPPFunctionDefinition[G] => ()
+    }
+    val wrongReturn = if(app.isEmpty) Seq(ReturnOutsideMethod(this)) else Nil
+    super.check(context) ++ wrongReturn
+  }
 
   override def layout(implicit ctx: Ctx): Doc =
     Text("return") <> (if(result == Void[G]()) Text(";") else Empty <+> result <> ";")