From 867f4197318fc73e1f0dab07e7ae4f6fe400c7b0 Mon Sep 17 00:00:00 2001
From: graham <g.macpherson@opencfd.co.uk>
Date: Wed, 10 Feb 2010 17:50:31 +0000
Subject: [PATCH] ENH: Distribution. Basing the Distribution class on a List
 instead of a Map.

Resizing by doubling and mapping data when a new entry is added that
is outside of the current bounds.  Trimming the output (normalised or
raw) to only the entries that contain data.

Overriding List clear method to enable the Distribution to be reset,
but retaining the correct number of components.
---
 .../test/Distribution/DistributionTest.C      |   2 +-
 .../Map => Lists}/Distribution/Distribution.C | 272 +++++++++++++-----
 .../Map => Lists}/Distribution/Distribution.H |  30 +-
 .../Distribution/DistributionI.H              |   7 +
 .../oldDistribution/oldDistribution.C         |   0
 .../oldDistribution/oldDistribution.H         |   0
 .../oldDistribution/oldDistributionI.H        |   0
 7 files changed, 226 insertions(+), 85 deletions(-)
 rename src/OpenFOAM/containers/{HashTables/Map => Lists}/Distribution/Distribution.C (62%)
 rename src/OpenFOAM/containers/{HashTables/Map => Lists}/Distribution/Distribution.H (84%)
 rename src/OpenFOAM/containers/{HashTables/Map => Lists}/Distribution/DistributionI.H (92%)
 rename src/OpenFOAM/containers/{HashTables/Map => Lists}/Distribution/oldDistribution/oldDistribution.C (100%)
 rename src/OpenFOAM/containers/{HashTables/Map => Lists}/Distribution/oldDistribution/oldDistribution.H (100%)
 rename src/OpenFOAM/containers/{HashTables/Map => Lists}/Distribution/oldDistribution/oldDistributionI.H (100%)

diff --git a/applications/test/Distribution/DistributionTest.C b/applications/test/Distribution/DistributionTest.C
index 687d5746ac..e59cb1af1e 100644
--- a/applications/test/Distribution/DistributionTest.C
+++ b/applications/test/Distribution/DistributionTest.C
@@ -34,7 +34,7 @@ Description
     normalDistribution(mean, sigma, x) = \
         sqrt(1.0/(2.0*pi*sigma**2))*exp(-(x - mean)**2.0/(2.0*sigma**2))
 
-    plot normalDistribution(8.5, 2.5, x)
+    plot normalDistribution(8.5, 2.5, x), "Distribution_scalar_test_x" w p
     @endverbatim
 
 \*---------------------------------------------------------------------------*/
diff --git a/src/OpenFOAM/containers/HashTables/Map/Distribution/Distribution.C b/src/OpenFOAM/containers/Lists/Distribution/Distribution.C
similarity index 62%
rename from src/OpenFOAM/containers/HashTables/Map/Distribution/Distribution.C
rename to src/OpenFOAM/containers/Lists/Distribution/Distribution.C
index 93edcef5dc..263853428d 100644
--- a/src/OpenFOAM/containers/HashTables/Map/Distribution/Distribution.C
+++ b/src/OpenFOAM/containers/Lists/Distribution/Distribution.C
@@ -26,30 +26,134 @@ License
 
 #include "Distribution.H"
 #include "OFstream.H"
+#include "ListOps.H"
+
+// * * * * * * * * * * * * Private Member Functions  * * * * * * * * * * * * //
+
+template<class Type>
+Foam::label Foam::Distribution<Type>::index
+(
+    direction cmpt,
+    label n
+)
+{
+    List<scalar>& cmptDistribution = (*this)[cmpt];
+
+    if (cmptDistribution.empty())
+    {
+        // Initialise this list with this value
+
+        cmptDistribution.setSize(2, 0.0);
+
+        listStarts_[cmpt] = n;
+
+        return 0;
+    }
+
+    label listIndex = -1;
+
+    label& listStart  = listStarts_[cmpt];
+
+    label testIndex = n - listStart;
+
+    if (testIndex < 0)
+    {
+        // Underflow of this List, storage increase and remapping
+        // required
+
+        List<scalar> newCmptDistribution(2*cmptDistribution.size(), 0.0);
+
+        label sOld = cmptDistribution.size();
+
+        forAll(cmptDistribution, i)
+        {
+            newCmptDistribution[i + sOld] = cmptDistribution[i];
+        }
+
+        cmptDistribution = newCmptDistribution;
+
+        listStart -= sOld;
+
+        // Recursively call this function in case another remap is required.
+        listIndex = index(cmpt, n);
+    }
+    else if (testIndex > cmptDistribution.size() - 1)
+    {
+        // Overflow of this List, storage increase required
+
+        cmptDistribution.setSize(2*cmptDistribution.size(), 0.0);
+
+        // Recursively call this function in case another storage
+        // alteration is required.
+        listIndex = index(cmpt, n);
+    }
+    else
+    {
+        listIndex = n - listStart;
+    }
+
+    return listIndex;
+}
+
+
+template<class Type>
+Foam::Pair<Foam::label> Foam::Distribution<Type>::validLimits
+(
+    direction cmpt
+) const
+{
+    const List<scalar>& cmptDistribution = (*this)[cmpt];
+
+    // limits.first(): lower bound, i.e. the first non-zero entry
+    // limits.second(): upper bound, i.e. the last non-zero entry
+    Pair<label> limits(-1, -1);
+
+    forAll(cmptDistribution, i)
+    {
+        if (cmptDistribution[i] > 0.0)
+        {
+            if (limits.first() == -1)
+            {
+                limits.first() = i;
+                limits.second() = i;
+            }
+            else
+            {
+                limits.second() = i;
+            }
+        }
+    }
+
+    return limits;
+}
+
 
 // * * * * * * * * * * * * * * * * Constructors  * * * * * * * * * * * * * * //
 
 template<class Type>
 Foam::Distribution<Type>::Distribution()
 :
-    List< Map<scalar> >(pTraits<Type>::nComponents),
-    binWidth_(pTraits<Type>::one)
+    List< List<scalar> >(pTraits<Type>::nComponents),
+    binWidth_(pTraits<Type>::one),
+    listStarts_(pTraits<Type>::nComponents, 0)
 {}
 
 
 template<class Type>
 Foam::Distribution<Type>::Distribution(const Type& binWidth)
 :
-    List< Map<scalar> >(pTraits<Type>::nComponents),
-    binWidth_(binWidth)
+    List< List<scalar> >(pTraits<Type>::nComponents),
+    binWidth_(binWidth),
+    listStarts_(pTraits<Type>::nComponents, 0)
 {}
 
 
 template<class Type>
 Foam::Distribution<Type>::Distribution(const Distribution<Type>& d)
 :
-    List< Map<scalar> >(static_cast< const List< Map<scalar> >& >(d)),
-    binWidth_(d.binWidth())
+    List< List<scalar> >(static_cast< const List< List<scalar> >& >(d)),
+    binWidth_(d.binWidth()),
+    listStarts_(d.listStarts())
 {}
 
 
@@ -65,40 +169,53 @@ Foam::Distribution<Type>::~Distribution()
 template<class Type>
 Foam::scalar Foam::Distribution<Type>::totalWeight(direction cmpt) const
 {
-    const Map<scalar>& cmptDistribution = (*this)[cmpt];
+    const List<scalar>& cmptDistribution = (*this)[cmpt];
 
     scalar sumOfWeights = 0.0;
 
-    forAllConstIter(Map<scalar>, cmptDistribution, iter)
+    forAll(cmptDistribution, i)
     {
-        sumOfWeights += iter();
+        sumOfWeights += cmptDistribution[i];
     }
 
     return sumOfWeights;
 }
 
+template<class Type>
+Foam::List<Foam::label> Foam::Distribution<Type>::keys(direction cmpt) const
+{
+    List<label> keys = identity((*this)[cmpt].size());
+
+    forAll(keys, k)
+    {
+        keys[k] += listStarts_[cmpt];
+    }
+
+    return keys;
+}
+
 
 template<class Type>
-inline Type Foam::Distribution<Type>::mean() const
+Type Foam::Distribution<Type>::mean() const
 {
     Type meanValue(pTraits<Type>::zero);
 
     for (direction cmpt = 0; cmpt < pTraits<Type>::nComponents; cmpt++)
     {
-        const Map<scalar>& cmptDistribution = (*this)[cmpt];
+        const List<scalar>& cmptDistribution = (*this)[cmpt];
 
         scalar totalCmptWeight = totalWeight(cmpt);
 
-        List<label> keys = cmptDistribution.sortedToc();
+        List<label> theKeys = keys(cmpt);
 
-        forAll(keys,k)
+        forAll(theKeys, k)
         {
-            label key = keys[k];
+            label key = theKeys[k];
 
             setComponent(meanValue, cmpt) +=
                 (0.5 + scalar(key))
                *component(binWidth_, cmpt)
-               *cmptDistribution[key]
+               *cmptDistribution[k]
                /totalCmptWeight;
         }
     }
@@ -108,7 +225,7 @@ inline Type Foam::Distribution<Type>::mean() const
 
 
 template<class Type>
-inline Type Foam::Distribution<Type>::median()
+Type Foam::Distribution<Type>::median()
 {
     Type medianValue(pTraits<Type>::zero);
 
@@ -145,7 +262,7 @@ inline Type Foam::Distribution<Type>::median()
             }
             else
             {
-                label lastNonZeroIndex = 0;
+                label previousNonZeroIndex = 0;
 
                 scalar cumulative = 0.0;
 
@@ -160,7 +277,7 @@ inline Type Foam::Distribution<Type>::median()
                     {
                         scalar xk = normDist[nD].first();
 
-                        scalar xkm1 = normDist[lastNonZeroIndex].first();
+                        scalar xkm1 = normDist[previousNonZeroIndex].first();
 
                         scalar Sk =
                             cumulative
@@ -178,7 +295,7 @@ inline Type Foam::Distribution<Type>::median()
                         cumulative +=
                             normDist[nD].second()*component(binWidth_, cmpt);
 
-                        lastNonZeroIndex = nD;
+                        previousNonZeroIndex = nD;
                     }
                 }
             }
@@ -197,53 +314,18 @@ void Foam::Distribution<Type>::add
     const Type& weight
 )
 {
+
     for (direction cmpt = 0; cmpt < pTraits<Type>::nComponents; cmpt++)
     {
-        Map<scalar>& cmptDistribution = (*this)[cmpt];
-
-        Map<scalar>::iterator iter(cmptDistribution.begin());
+        List<scalar>& cmptDistribution = (*this)[cmpt];
 
         label n =
             label(component(valueToAdd, cmpt)/component(binWidth_, cmpt))
           - label(neg(component(valueToAdd, cmpt)/component(binWidth_, cmpt)));
 
-        iter = cmptDistribution.find(n);
+        label listIndex = index(cmpt, n);
 
-        if (iter == cmptDistribution.end())
-        {
-            cmptDistribution.insert(n, component(weight, cmpt));
-        }
-        else
-        {
-            cmptDistribution[n] += component(weight, cmpt);
-        }
-    }
-}
-
-
-template<class Type>
-void Foam::Distribution<Type>::insertMissingKeys()
-{
-    for (direction cmpt = 0; cmpt < pTraits<Type>::nComponents; cmpt++)
-    {
-        Map<scalar>& cmptDistribution = (*this)[cmpt];
-
-        Map<scalar>::iterator iter(cmptDistribution.begin());
-
-        List<label> keys = cmptDistribution.sortedToc();
-
-        if (keys.size())
-        {
-            for (label k = keys[0]; k < keys[keys.size()-1]; k++)
-            {
-                iter = cmptDistribution.find(k);
-
-                if (iter == cmptDistribution.end())
-                {
-                    cmptDistribution.insert(k,0);
-                }
-            }
-        }
+        cmptDistribution[listIndex] += component(weight, cmpt);
     }
 }
 
@@ -254,29 +336,39 @@ Distribution<Type>::normalised()
 {
     List< List < Pair<scalar> > > normDistribution(pTraits<Type>::nComponents);
 
-    insertMissingKeys();
-
     for (direction cmpt = 0; cmpt < pTraits<Type>::nComponents; cmpt++)
     {
-        Map<scalar>& cmptDistribution = (*this)[cmpt];
+        List<scalar>& cmptDistribution = (*this)[cmpt];
+
+        if (cmptDistribution.empty())
+        {
+            continue;
+        }
 
         scalar totalCmptWeight = totalWeight(cmpt);
 
-        List<label> keys = cmptDistribution.sortedToc();
+        List<label> cmptKeys = keys(cmpt);
 
         List< Pair<scalar> >& normDist = normDistribution[cmpt];
 
-        normDist.setSize(keys.size());
+        Pair<label> limits = validLimits(cmpt);
 
-        forAll(keys,k)
+        normDist.setSize(limits.second() - limits.first() + 1);
+
+        for
+        (
+            label k = limits.first(), i = 0;
+            k <= limits.second();
+            k++, i++
+        )
         {
-            label key = keys[k];
+            label key = cmptKeys[k];
 
-            normDist[k].first() =
+            normDist[i].first() =
                 (0.5 + scalar(key))*component(binWidth_, cmpt);
 
-            normDist[k].second() =
-                cmptDistribution[key]
+            normDist[i].second() =
+                cmptDistribution[k]
                /totalCmptWeight
                /component(binWidth_, cmpt);
         }
@@ -292,25 +384,35 @@ Distribution<Type>::raw()
 {
     List< List < Pair<scalar> > > rawDistribution(pTraits<Type>::nComponents);
 
-    insertMissingKeys();
-
     for (direction cmpt = 0; cmpt < pTraits<Type>::nComponents; cmpt++)
     {
-        Map<scalar>& cmptDistribution = (*this)[cmpt];
+        List<scalar>& cmptDistribution = (*this)[cmpt];
 
-        List<label> keys = cmptDistribution.sortedToc();
+        if (cmptDistribution.empty())
+        {
+            continue;
+        }
+
+        List<label> cmptKeys = keys(cmpt);
 
         List< Pair<scalar> >& rawDist = rawDistribution[cmpt];
 
-        rawDist.setSize(keys.size());
+        Pair<label> limits = validLimits(cmpt);
 
-        forAll(keys,k)
+        rawDist.setSize(limits.second() - limits.first() + 1);
+
+        for
+        (
+            label k = limits.first(), i = 0;
+            k <= limits.second();
+            k++, i++
+        )
         {
-            label key = keys[k];
+            label key = cmptKeys[k];
 
-            rawDist[k].first() = (0.5 + scalar(key))*component(binWidth_, cmpt);
+            rawDist[i].first() = (0.5 + scalar(key))*component(binWidth_, cmpt);
 
-            rawDist[k].second() = cmptDistribution[key];
+            rawDist[i].second() = cmptDistribution[k];
         }
     }
 
@@ -318,6 +420,18 @@ Distribution<Type>::raw()
 }
 
 
+template<class Type>
+void Foam::Distribution<Type>::clear()
+{
+    for (direction cmpt = 0; cmpt < pTraits<Type>::nComponents; cmpt++)
+    {
+        (*this)[cmpt].clear();
+
+        listStarts_[cmpt] = 0;
+    }
+}
+
+
 template<class Type>
 void Foam::Distribution<Type>::write
 (
@@ -374,7 +488,7 @@ void Foam::Distribution<Type>::operator=
             << abort(FatalError);
     }
 
-    List< Map<scalar> >::operator=(rhs);
+    List< List<scalar> >::operator=(rhs);
 
     binWidth_ = rhs.binWidth();
 }
@@ -390,7 +504,7 @@ Foam::Ostream& Foam::operator<<
 )
 {
     os  << d.binWidth_
-        << static_cast<const List< Map<scalar> >& >(d);
+        << static_cast<const List< List<scalar> >& >(d);
 
     // Check state of Ostream
     os.check
diff --git a/src/OpenFOAM/containers/HashTables/Map/Distribution/Distribution.H b/src/OpenFOAM/containers/Lists/Distribution/Distribution.H
similarity index 84%
rename from src/OpenFOAM/containers/HashTables/Map/Distribution/Distribution.H
rename to src/OpenFOAM/containers/Lists/Distribution/Distribution.H
index 335764968a..5dbdb444c4 100644
--- a/src/OpenFOAM/containers/HashTables/Map/Distribution/Distribution.H
+++ b/src/OpenFOAM/containers/Lists/Distribution/Distribution.H
@@ -39,7 +39,7 @@ SourceFiles
 #ifndef Distribution_H
 #define Distribution_H
 
-#include "Map.H"
+#include "List.H"
 #include "Pair.H"
 
 // * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * //
@@ -60,12 +60,25 @@ Ostream& operator<<(Ostream&, const Distribution<Type>&);
 template<class Type>
 class Distribution
 :
-    public List< Map<scalar> >
+    public List< List<scalar> >
 {
     // Private data
 
+        //- Width of the bin for each component
         Type binWidth_;
 
+        //- The start bin index of each component
+        List<label> listStarts_;
+
+    // Private Member Functions
+
+        //- Return the appropriate List index for the given bin index.
+        //  Resizes the List if required
+        label index(direction cmpt, label n);
+
+        //- Returns the indices of the first and last non-zero entries
+        Pair<label> validLimits(direction cmpt) const;
+
 
 public:
 
@@ -94,6 +107,8 @@ public:
         //  argument
         scalar totalWeight(direction cmpt) const;
 
+        List<label> keys(direction cmpt) const;
+
         Type mean() const;
 
         // From http://mathworld.wolfram.com/StatisticalMedian.html
@@ -108,20 +123,25 @@ public:
             const Type& weight = pTraits<Type>::one
         );
 
-        //- Add the missing keys to the Maps to fill any gaps
-        void insertMissingKeys();
-
         //- Return the normalised distribution and bins
         List< List<Pair<scalar> > > normalised();
 
         //- Return the distribution of the total bin weights
         List< List < Pair<scalar> > > raw();
 
+        //- Resets the Distribution by clearing the stored lists.
+        //  Leaves the same number of them and the same binWidth.
+        void clear();
+
+
     // Access
 
         //- Return the bin width
         inline const Type& binWidth() const;
 
+        //- Return the List start bin indices
+        inline const List<label>& listStarts() const;
+
     // Write
 
         //- Write the distribution to file.  Produces a separate file
diff --git a/src/OpenFOAM/containers/HashTables/Map/Distribution/DistributionI.H b/src/OpenFOAM/containers/Lists/Distribution/DistributionI.H
similarity index 92%
rename from src/OpenFOAM/containers/HashTables/Map/Distribution/DistributionI.H
rename to src/OpenFOAM/containers/Lists/Distribution/DistributionI.H
index 1e4689e5c6..6f37647b3a 100644
--- a/src/OpenFOAM/containers/HashTables/Map/Distribution/DistributionI.H
+++ b/src/OpenFOAM/containers/Lists/Distribution/DistributionI.H
@@ -34,6 +34,13 @@ inline const Type& Foam::Distribution<Type>::binWidth() const
 }
 
 
+template<class Type>
+inline const
+Foam::List<Foam::label>& Foam::Distribution<Type>::listStarts() const
+{
+    return listStarts_;
+}
+
 // * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * //
 
 
diff --git a/src/OpenFOAM/containers/HashTables/Map/Distribution/oldDistribution/oldDistribution.C b/src/OpenFOAM/containers/Lists/Distribution/oldDistribution/oldDistribution.C
similarity index 100%
rename from src/OpenFOAM/containers/HashTables/Map/Distribution/oldDistribution/oldDistribution.C
rename to src/OpenFOAM/containers/Lists/Distribution/oldDistribution/oldDistribution.C
diff --git a/src/OpenFOAM/containers/HashTables/Map/Distribution/oldDistribution/oldDistribution.H b/src/OpenFOAM/containers/Lists/Distribution/oldDistribution/oldDistribution.H
similarity index 100%
rename from src/OpenFOAM/containers/HashTables/Map/Distribution/oldDistribution/oldDistribution.H
rename to src/OpenFOAM/containers/Lists/Distribution/oldDistribution/oldDistribution.H
diff --git a/src/OpenFOAM/containers/HashTables/Map/Distribution/oldDistribution/oldDistributionI.H b/src/OpenFOAM/containers/Lists/Distribution/oldDistribution/oldDistributionI.H
similarity index 100%
rename from src/OpenFOAM/containers/HashTables/Map/Distribution/oldDistribution/oldDistributionI.H
rename to src/OpenFOAM/containers/Lists/Distribution/oldDistribution/oldDistributionI.H