Changeset 27691

Timestamp:

04/13/23 19:29:17 (2 years ago)

Author:

Mathieu Morlighem

Message:

NEW: added averaging method for inputs

Location:

issm/trunk-jpl/src/c/classes/Inputs

Files:

• Input.h (modified) (1 diff)
• Inputs.cpp (modified) (2 diffs)
• Inputs.h (modified) (1 diff)
• TriaInput.cpp (modified) (1 diff)
• TriaInput.h (modified) (1 diff)

issm/trunk-jpl/src/c/classes/Inputs/Input.h

@@ -48 +48 @@
                 virtual void   Pow(IssmDouble scale_factor){_error_("Not implemented yet");};
                 virtual void   Scale(IssmDouble scale_factor){_error_("Not implemented yet");};
+                virtual void   AverageAndReplace(void){_error_("Not implemented yet");};
 
                 virtual int  GetResultArraySize(void){_error_("Not implemented yet");};

issm/trunk-jpl/src/c/classes/Inputs/Inputs.cpp

@@ -302 +302 @@
 }
 /*}}}*/
-void     Inputs::Shift(int xenum, IssmDouble alpha){/*{{{*/
+void Inputs::Shift(int xenum, IssmDouble alpha){/*{{{*/
 
         _assert_(this);
@@ -314 +314 @@
         /*Shift: */
         this->inputs[index_x]->Shift(alpha);
+}
+/*}}}*/
+void Inputs::AverageAndReplace(int inputenum){/*{{{*/
+
+        _assert_(this);
+
+        /*Get indices from enums*/
+        int index = EnumToIndex(inputenum);
+        if(!this->inputs[index]) _error_("Input "<<EnumToStringx(inputenum)<<" not found");
+
+        this->inputs[index]->AverageAndReplace();
 }
 /*}}}*/

issm/trunk-jpl/src/c/classes/Inputs/Inputs.h

@@ -50 +50 @@
                 void     AXPY(IssmDouble alpha, int xenum, int yenum);
                 void     Shift(int inputenum, IssmDouble alpha);
+                void     AverageAndReplace(int inputenum);
                 void     DeepEcho(void);
                 void     DeepEcho(int enum_in);

issm/trunk-jpl/src/c/classes/Inputs/TriaInput.cpp

@@ -406 +406 @@
 }
 /*}}}*/
+void TriaInput::AverageAndReplace(void){/*{{{*/
+
+        /*Get local sum and local size*/
+        IssmDouble sum  = 0.;
+        for(int i=0;i<this->M*this->N;i++) sum += this->values[i];
+        sum = sum/(this->M*this->N);
+
+        /*Get sum across all procs*/
+        IssmDouble all_sum;
+        ISSM_MPI_Allreduce((void*)&sum,(void*)&all_sum,1,ISSM_MPI_DOUBLE,ISSM_MPI_SUM,IssmComm::GetComm());
+
+        /*Divide by number of procs*/
+        int        num_procs = IssmComm::GetSize();
+        IssmDouble newvalue = all_sum/num_procs;
+
+        /*Now replace existing input*/
+        this->Reset(P0Enum);
+        for(int i=0;i<this->M*this->N;i++) this->values[i] = newvalue;
+}
+/*}}}*/
 
 /*Object functions*/

issm/trunk-jpl/src/c/classes/Inputs/TriaInput.h

@@ -41 +41 @@
                 void AXPY(Input* xinput,IssmDouble scalar);
                 void Shift(IssmDouble scalar);
+                void AverageAndReplace(void);
                 void PointWiseMult(Input* xinput);
                 void Serve(int numindices,int* indices);