6

Author: Roushelfy

simulators/6_inv_free/src/NeoHookeanEnergy.cu

@@ -41,10 +41,12 @@ NeoHookeanEnergy<T, dim>::NeoHookeanEnergy(const std::vector<T> &x, const std::v
 {
     pimpl_->device_x.copy_from(x);
     pimpl_->device_e.copy_from(e);
+    pimpl_->device_vol.resize(e.size() / 3);
+    pimpl_->device_IB.resize(e.size() / 3);
     pimpl_->Mu = mu;
     pimpl_->Lambda = lam;
     pimpl_->device_grad.resize(x.size());
-    pimpl_->device_hess.resize_triplets(e.size() * dim * dim * dim * dim);
+    pimpl_->device_hess.resize_triplets(e.size() * 12);
     pimpl_->device_hess.reshape(x.size(), x.size());
     init_vol_IB();
 }
@@ -59,7 +61,7 @@ void NeoHookeanEnergy<T, dim>::init_vol_IB()
         {
             for (int k = 0; k < 2; ++k)
             {
-                TB(k, j) = device_x(device_e(i * 3 + j) * 2 + k) - device_x(device_e(i * 3) * 2 + k);
+                TB(k, j) = device_x(device_e(i * 3 + j+1) * 2 + k) - device_x(device_e(i * 3) * 2 + k);
             }
         }
         device_vol(i) = TB.determinant() / 2;
@@ -94,7 +96,6 @@ T NeoHookeanEnergy<T, dim>::val()
          NeoHookeanEnergyVal(E,  Mu, Lambda,X,device_IB(i),device_vol(i));
          device_val(i) = E; })
         .wait();
-
     return devicesum(device_val);
 }
 

simulators/6_inv_free/src/main.cpp

@@ -5,7 +5,7 @@ int main()
 	float nu = 0.4, E = 1e5;
 	float Mu = E / (2 * (1 + nu)), Lam = E * nu / ((1 + nu) * (1 - 2 * nu));
 	float rho = 1000,
-		  k = 4e4, initial_stretch = 1, n_seg = 8, h = 0.01, side_len = 1, tol = 0.01, mu = 0.11;
+		  k = 2e4, initial_stretch = 1, n_seg = 4, h = 0.01, side_len = 1, tol = 0.01, mu = 0.11;
 	// printf("Running mass-spring simulator with parameters: rho = %f, k = %f, initial_stretch = %f, n_seg = %d, h = %f, side_len = %f, tol = %f\n", rho, k, initial_stretch, n_seg, h, side_len, tol);
 	InvFreeSimulator<float, 2> simulator(rho, side_len, initial_stretch, k, h, tol, mu, Mu, Lam, n_seg);
 	simulator.run();