/*
 * Copyright (c) 2014 ARM Limited
 * All rights reserved
 *
 * The license below extends only to copyright in the software and shall
 * not be construed as granting a license to any other intellectual
 * property including but not limited to intellectual property relating
 * to a hardware implementation of the functionality of the software
 * licensed hereunder.  You may use the software subject to the license
 * terms below provided that you ensure that this notice is replicated
 * unmodified and in its entirety in all distributions of the software,
 * modified or unmodified, in source code or in binary form.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions are
 * met: redistributions of source code must retain the above copyright
 * notice, this list of conditions and the following disclaimer;
 * redistributions in binary form must reproduce the above copyright
 * notice, this list of conditions and the following disclaimer in the
 * documentation and/or other materials provided with the distribution;
 * neither the name of the copyright holders nor the names of its
 * contributors may be used to endorse or promote products derived from
 * this software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
 * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */

/**
 * @file
 *
 *  C++-only configuration stats handling example
 *
 *  Register with: statistics::registerHandlers(statsReset, statsDump)
 */

#include <iostream>
#include <list>

#include "base/logging.hh"
#include "base/output.hh"
#include "base/statistics.hh"
#include "base/stats/text.hh"
#include "stats.hh"

namespace CxxConfig
{

void statsPrepare()
{
    std::list<statistics::Info *> stats = statistics::statsList();

    /* gather_stats -> prepare */
    for (auto i = stats.begin(); i != stats.end(); ++i){
        statistics::Info *stat = *i;
        statistics::VectorInfo *vector =
            dynamic_cast<statistics::VectorInfo *>(stat);
        if (vector){
            (dynamic_cast<statistics::VectorInfo *>(*i))->prepare();
        }
        else {
            (*i)->prepare();
        }

    }
}

void statsDump()
{
    bool desc = true;
    statistics::Output *output = statistics::initText(filename, desc, true);

    statistics::processDumpQueue();

    std::list<statistics::Info *> stats = statistics::statsList();

    statsEnable();
    statsPrepare();

    output->begin();
    /* gather_stats -> convert_value */
    for (auto i = stats.begin(); i != stats.end(); ++i) {
        statistics::Info *stat = *i;

        const statistics::ScalarInfo *scalar =
            dynamic_cast<statistics::ScalarInfo *>(stat);
        statistics::VectorInfo *vector =
            dynamic_cast<statistics::VectorInfo *>(stat);
        const statistics::Vector2dInfo *vector2d =
            dynamic_cast<statistics::Vector2dInfo *>(vector);
        const statistics::DistInfo *dist =
            dynamic_cast<statistics::DistInfo *>(stat);
        const statistics::VectorDistInfo *vectordist =
            dynamic_cast<statistics::VectorDistInfo *>(stat);
        const statistics::SparseHistInfo *sparse =
            dynamic_cast<statistics::SparseHistInfo *>(stat);
        const statistics::InfoProxy <statistics::Vector2d,
            statistics::Vector2dInfo> *info =
            dynamic_cast<statistics::InfoProxy
            <statistics::Vector2d,statistics::Vector2dInfo>*>(stat);

        if (vector) {
            const statistics::FormulaInfo *formula =
                dynamic_cast<statistics::FormulaInfo *>(vector);
            if (formula){
                output->visit(*formula);
            } else {
                const statistics::VectorInfo *vector1 = vector;
                output->visit(*vector1);
            }
        } else if (vector2d) {
            output->visit(*vector2d);
        } else if (info){
            output->visit(*info);
        } else if (vectordist){
            output->visit(*vectordist);
        } else if (dist) {
            output->visit(*dist);
        } else if (sparse) {
            output->visit(*sparse);
        } else if (scalar) {
            output->visit(*scalar);
        } else {
            warn("Stat not dumped: %s\n", stat->name);
        }
    }
    output->end();
}

void statsReset()
{
    std::cerr << "Stats reset\n";

    statistics::processResetQueue();
}

void statsEnable()
{
    std::list<statistics::Info *> stats = statistics::statsList();

    for (auto i = stats.begin(); i != stats.end(); ++i){
        statistics::Info *stat = *i;
        statistics::VectorInfo *vector =
            dynamic_cast<statistics::VectorInfo *>(stat);
        if (vector){
            (dynamic_cast<statistics::VectorInfo *>(*i))->enable();
        }
        else {
            (*i)->enable();
        }

    }
}

}