OilfieldToolsNet/LBPM
4
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

enable grid reader for color simulation

Browse Source
This commit is contained in:
JamesEMcclure 2019-11-15 10:37:26 -05:00
parent a750533a7f
commit 9efd6461a0
4 changed files with 27 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
common/Domain.cpp
View File

@ -329,7 +329,6 @@ void Domain::Decomp(std::string Filename)
char *SegData = NULL;
if (RANK==0){
printf("Input media: %s\n",Filename.c_str());
printf("Relabeling %lu values\n",ReadValues.size());
for (int idx=0; idx<ReadValues.size(); idx++){
@ -485,6 +484,7 @@ void Domain::Decomp(std::string Filename)
else{
// solid checkers
SegData[k*global_Nx*global_Ny+j*global_Nx+i] = 0;
}
}
}

1
common/Domain.h
View File

@ -16,7 +16,6 @@
#include "common/Communication.h"
#include "common/Database.h"
class Domain;
template<class TYPE> class PatchData;

21
common/ReadMicroCT.cpp
View File

@ -70,6 +70,8 @@ Array<uint8_t> readMicroCT( const Database& domain, MPI_Comm comm )
auto n = domain.getVector<int>( "n" );
int rank = comm_rank(MPI_COMM_WORLD);
auto nproc = domain.getVector<int>( "nproc" );
auto ReadValues = domain.getVector<int>( "ReadValues" );
auto WriteValues = domain.getVector<int>( "WriteValues" );
RankInfoStruct rankInfo( rank, nproc[0], nproc[1], nproc[2] );
// Determine the largest file number to get
@ -94,6 +96,25 @@ Array<uint8_t> readMicroCT( const Database& domain, MPI_Comm comm )
}
data = readMicroCT( filename );
}
// Relabel the data
for (int k = 0; k<Nfz; k++){
for (int j = 0; j<Nfy; j++){
for (int i = 0; i<Nfx; i++){
//n = k*Nfx*Nfy + j*Nfx + i;
//char locval = loc_id[n];
char locval = data(i,j,k);
for (int idx=0; idx<ReadValues.size(); idx++){
signed char oldvalue=ReadValues[idx];
signed char newvalue=WriteValues[idx];
if (locval == oldvalue){
data(i,j,k) = newvalue;
idx = ReadValues.size();
}
}
}
}
}
// Redistribute the data
data = redistribute( srcRankInfo, data, rankInfo, { n[0], n[1], n[2] }, comm );

5
models/ColorModel.cpp
View File

@ -4,6 +4,7 @@ color lattice boltzmann model
#include "models/ColorModel.h"
#include "analysis/distance.h"
#include "analysis/morphology.h"
#include "common/ReadMicroCT.h"
#include <stdlib.h>
#include <time.h>
@ -189,6 +190,10 @@ void ScaLBL_ColorModel::ReadInput(){
Mask->Decomp(first_image);
IMAGE_INDEX++;
}
else if (domain_db->keyExists( "GridFile" )){
auto input_id = readMicroCT( *domain_db, MPI_COMM_WORLD );
for (int i=0; i<Nx*Ny*Nz; i++) Mask->id[i] = input_id(i);
}
else if (domain_db->keyExists( "Filename" )){
auto Filename = domain_db->getScalar<std::string>( "Filename" );
Mask->Decomp(Filename);