From 8d4a0718e8aa5b449025aac0343337249ffc737b Mon Sep 17 00:00:00 2001
From: Meng Li <34143965+Meng6@users.noreply.github.com>
Date: Tue, 7 Apr 2020 11:22:04 -0400
Subject: [PATCH] Fix bug of Barnett's location features

---
 src/features/location_barnett/InitializeParams.R | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/features/location_barnett/InitializeParams.R b/src/features/location_barnett/InitializeParams.R
index fb3e0435..1d448329 100644
--- a/src/features/location_barnett/InitializeParams.R
+++ b/src/features/location_barnett/InitializeParams.R
@@ -22,10 +22,10 @@ function(out){
   }
   if(l1+l2==0){phatall=length(ID2)/(length(ID1)+length(ID2))}
   #flight distances
-  fd=apply(out[ID1,],1,function(xx) sqrt((xx[2]-xx[5])^2+(xx[3]-xx[6])^2))
+  fd=apply(matrix(out[ID1,],ncol=7),1,function(xx) sqrt((xx[2]-xx[5])^2+(xx[3]-xx[6])^2))
   
   # flight times: ft
-  ft=apply(out[ID1,],1,function(xx) (xx[7]-xx[4]))
+  ft=apply(matrix(out[ID1,],ncol=7),1,function(xx) (xx[7]-xx[4]))
   fxs=out[ID1,2]
   fys=out[ID1,3]
   # flight angles range [0,2pi]: fa