trueAggregateRollUpHelper.h

Go to the documentation of this file.

/******************************************************************************
** Copyright (c) 2014 MAK Technologies, Inc.
** All rights reserved.
******************************************************************************/



#pragma once

#include "vrfutil/vrfutilDefines.h"
#include <readerWriter/rwProperties.h>
#include <vrfutil/assembliesReaderWriter.h>

#include <boost/function.hpp>

typedef boost::function<bool (const DtReaderWriter* source, DtReaderWriter* destination,
   const DtMergePath& path, const std::map<std::string, std::string>& attributes)> DtRollUpFunction;

class DT_DLL_vrfutil DtTrueAggregateRollUpHelper
{
public:
   typedef std::map<std::string, DtRollUpFunction> RollUpRules;

   DtTrueAggregateRollUpHelper();
   virtual ~DtTrueAggregateRollUpHelper();

   virtual void installCustomRules();

   virtual void addRollUpFunction(const std::string&, DtRollUpFunction);
   const RollUpRules& rollUpRules() const
   {
      return myRollUpRules;
   }

   RollUpRules& rollUpRules()
   {
      return myRollUpRules;
   }

   virtual bool sumFunction(const DtReaderWriter* source, DtReaderWriter* destination, const DtMergePath& path,
      const std::map<std::string, std::string>&);
   virtual bool minimumFunction(const DtReaderWriter* source, DtReaderWriter* destination, const DtMergePath& path,
      const std::map<std::string, std::string>&);
   virtual bool maximumFunction(const DtReaderWriter* source, DtReaderWriter* destination, const DtMergePath& path,
      const std::map<std::string, std::string>&);
   virtual bool averageFunction(const DtReaderWriter* source, DtReaderWriter* destination, const DtMergePath& path,
      const std::map<std::string, std::string>&);
   virtual bool moppLevelFunction(const DtReaderWriter* source, DtReaderWriter* destination, const DtMergePath& path,
      const std::map<std::string, std::string>&);
   virtual bool stringFunction(const DtReaderWriter* source, DtReaderWriter* destination, const DtMergePath& path);


   virtual void rollUp(const std::vector<DtRwProperties*>&, DtRwProperties& result, const std::set<DtString>& ignore = std::set<DtString>()) const;

   virtual void setRollUpRules(const DtAssembliesReaderWriter::RwRootNode&);

   virtual bool loadRollUpRules(const DtFilename&);

   virtual void updateAverages(DtRwProperties&, int totalContributors) const;

   virtual void restore();

protected:

   virtual bool rollUpValues(const DtReaderWriter* source, DtReaderWriter* destination, const DtMergePath& path);

   virtual double averageContributionFrom(const DtReaderWriter* item, const DtMergePath& path,
      const std::map<std::string, std::string>& atr) const;

protected:
   RollUpRules myRollUpRules;
   DtPropertyRollUpFunction myOriginalDefaultRule;
   DtReaderWriterPropertyFactory::RollUpRules myOriginalRollUpRules;

   struct AverageItem
   {
      AverageItem() : total(0) {};
      AverageItem(double t, const DtString& w) : total(t), weight(w) {};
      AverageItem(const AverageItem& orig) : total(orig.total), weight(orig.weight) {};

      double total;
      DtString weight;
   };

   typedef std::map<std::string, AverageItem> AverageItems;

   mutable AverageItems myAverageItems;
   mutable std::set<DtString> myAverageTotals;

   DtAssembliesReaderWriter myAssemblyReaderWriter;
};