Skip to content

Commit 5737b57

Browse files
code cleaning
1 parent bf3b5b9 commit 5737b57

6 files changed

Lines changed: 15 additions & 16 deletions

File tree

include/geode/stochastic/sampling/mcmc/metropolis_hasting_sampler.hpp

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -231,9 +231,12 @@ namespace geode
231231
{
232232
OPENGEODE_ASSERT( proposal.new_object.has_value(),
233233
"[MH] Birth proposal has no new_object" );
234-
const auto delta_log_energy = energy_.delta_log_energy_add( state,
234+
geode::ObjectRef< Type > new_object{
235235
proposal.new_object.value().first,
236-
proposal.new_object.value().second );
236+
proposal.new_object.value().second
237+
};
238+
const auto delta_log_energy =
239+
energy_.delta_log_energy_add( state, new_object );
237240
return accept_or_reject( proposal, state, engine, delta_log_energy,
238241
[]( auto& s, auto& p ) {
239242
s.add_object( std::move( p.new_object.value().first ),

include/geode/stochastic/sampling/mcmc/models/components/energy_term.hpp

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -125,8 +125,7 @@ namespace geode
125125
virtual double total_log( const ObjectSet< Type >& state ) const = 0;
126126

127127
virtual double delta_log_add( const ObjectSet< Type >& state,
128-
const Type& new_object,
129-
const uuid& new_object_subset_id ) const = 0;
128+
const ObjectRef< Type >& new_object ) const = 0;
130129

131130
virtual double delta_log_remove(
132131
const ObjectSet< Type >& state, ObjectId object_id ) const = 0;

include/geode/stochastic/sampling/mcmc/models/components/intensity_term.hpp

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -51,10 +51,9 @@ namespace geode
5151
}
5252

5353
double delta_log_add( const ObjectSet< Type >& /*state*/,
54-
const Type& /*new_object*/,
55-
const uuid& new_object_subset_id ) const override
54+
const ObjectRef< Type >& new_object ) const override
5655
{
57-
if( !this->is_targeted_subset( new_object_subset_id ) )
56+
if( !this->is_targeted_subset( new_object.subset ) )
5857
{
5958
return 0.0;
6059
}

include/geode/stochastic/sampling/mcmc/models/gibbs_energy.hpp

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -79,15 +79,13 @@ namespace geode
7979
}
8080

8181
double delta_log_energy_add( const ObjectSet< Type >& state,
82-
const Type& new_object,
83-
const uuid& new_object_subset_id ) const
82+
const ObjectRef< Type >& new_object ) const
8483
{
8584
double log_energy{ 0.0 };
8685
for( const auto& [id, term] : energy_terms_ )
8786
{
8887
geode_unused( id );
89-
log_energy += term->delta_log_add(
90-
state, new_object, new_object_subset_id );
88+
log_energy += term->delta_log_add( state, new_object );
9189
}
9290
return log_energy;
9391
}

tests/stochastic/sampling/mcmc/models/components/test-intensity-term.cpp

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -52,8 +52,8 @@ void test_normal_positive_intensity( double lambda,
5252
"[test intensity]- total_log wrong value." );
5353

5454
geode::Point2D p3{ { 2., 2. } };
55-
56-
double delta_add = term.delta_log_add( pattern, p3, subset_id );
55+
geode::ObjectRef< geode::Point2D > p_ref{ p3, subset_id };
56+
double delta_add = term.delta_log_add( pattern, p_ref );
5757
OPENGEODE_EXCEPTION( delta_add == neg_log_lambda * 1.,
5858
"[test intensity]- delta_log_add wrong value." );
5959

@@ -81,8 +81,8 @@ void test_normal_zero_intensity( double lambda,
8181
std::isinf( total ), "[test zero intensity]- total_log wrong value." );
8282

8383
geode::Point2D p3{ { 2., 2. } };
84-
85-
double delta_add = term.delta_log_add( pattern, p3, subset_id );
84+
geode::ObjectRef< geode::Point2D > p_ref{ p3, subset_id };
85+
double delta_add = term.delta_log_add( pattern, p_ref );
8686
OPENGEODE_EXCEPTION( std::isinf( delta_add ),
8787
"[test zero intensity]- delta_log_add wrong value." );
8888

tests/stochastic/sampling/mcmc/test-mh-poisson-statistics-and-convergence.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -42,7 +42,7 @@ namespace
4242
geode::ObjectSet< geode::Point2D > state =
4343
mh.initialize_object_set_with_sampling( engine, group_targets );
4444
mh.walk( state, engine, 500 );
45-
constexpr geode::index_t N{ 1000 };
45+
constexpr geode::index_t N{ 5000 };
4646

4747
// Sampling
4848

0 commit comments

Comments
 (0)