SimObjectPopulation.cpp

Go to the documentation of this file.

#include "SimObjectPopulation.h"
#include "SimObject.h"

shared_ptr<SimObjectPopulation> SimObjectPopulation::record( const SimObjectFactory &recFactory, const port_t src_port)
{
        SimObject::ID::Vector rec_id_vec(new vector< SimObject::ID::Packed >(size()));   
        
        for (int i = 0; i < (int)size() ; ++i) {
                (*rec_id_vec)[i] = net->record((*id_vec)[i], src_port, recFactory);
        }
    
    return shared_ptr<SimObjectPopulation>( new SimObjectPopulation(*net, rec_id_vec) );
}
    
shared_ptr<SimObjectPopulation> SimObjectPopulation::record(const SimObjectFactory &recFactory, const string field )
{
        SimObject::ID::Vector rec_id_vec(new vector< SimObject::ID::Packed>(size()));    
        
        for (int i = 0; i < (int)size() ; ++i) {
                (*rec_id_vec)[i] = net->record((*id_vec)[i], field, recFactory);
        }
    
        return shared_ptr<SimObjectPopulation>( new SimObjectPopulation(*net, rec_id_vec) );
}

shared_ptr<SimObjectPopulation> SimObjectPopulation::localSubPopulation()
{
        SimObject::ID::Vector local_vec(new vector< SimObject::ID::Packed>());   
        
        for (int i = 0; i < (int)size() ; ++i) {
                if (SimObject::ID((*id_vec)[i]).node == net->mpi_rank())
                        local_vec->push_back((*id_vec)[i]);
        }
        
        return shared_ptr<SimObjectPopulation>( new SimObjectPopulation(*net, local_vec) );
}

vector<size_t> * SimObjectPopulation::localIndexes()
{
        vector<size_t> * local_idxs = new vector<size_t>;
        for (int i = 0; i < (int)size() ; ++i) {
                if (SimObject::ID((*id_vec)[i]).node == net->mpi_rank())
                        local_idxs->push_back(i);
        }
        return local_idxs;      
}

SimObjectPopulation* SimObjectPopulation::new_subset( vector<size_t> const& subindices ) const {
    size_t s;
    vector<SimObject::ID::Packed> objIDs( subindices.size() );
    for( size_t i=0; i<subindices.size(); i++ ) {
        s = subindices[i];
        objIDs[i] = (*id_vec)[ s ];
    }
    return new SimObjectPopulation( *net, objIDs );
}


bool SimObjectPopulation::setFieldScale(string const & fieldname, double scale)
{
    // loop over local IDs
    for (int i = 0; i < (int)size() ; ++i) {
        SimObject::ID id = SimObject::ID((*id_vec)[i]);

        if (SimObject::ID((*id_vec)[i]).node == net->mpi_rank()) {
            SimObject *obj=net->getObject(id);
            const Field *field=obj->findField(fieldname);
            obj->setScalarField(field, obj->getScalarField(field)*scale);
        }
    }

    return true;
}

---