/*
 * This program source code file is part of KiCad, a free EDA CAD application.
 *
 * Copyright (C) 2018 KiCad Developers, see change_log.txt for contributors.
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License
 * as published by the Free Software Foundation; either version 2
 * of the License, or (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, you may find one here:
 * http://www.gnu.org/licenses/old-licenses/gpl-2.0.html
 * or you may search the http://www.gnu.org website for the version 2 license,
 * or you may write to the Free Software Foundation, Inc.,
 * 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA
 */


#ifndef DRC_PROVIDER__H
#define DRC_PROVIDER__H

#include <class_board.h>
#include <class_marker_pcb.h>

#include <functional>
#include <set>

namespace test {

class DRC_ENGINE;

class DRC_TEST_PROVIDER_REGISTRY 
{
    public:
        DRC_TEST_PROVIDER_REGISTRY() {};
        ~DRC_TEST_PROVIDER_REGISTRY() {};

        static DRC_TEST_PROVIDER_REGISTRY& Instance()
        {
            static DRC_TEST_PROVIDER_REGISTRY self;
            return self;
        }

        void RegisterTestProvider(DRC_TEST_PROVIDER* provider) { m_providers.push_back(provider); }
        std::vector<DRC_TEST_PROVIDER*> GetTestProviders() const { return m_providers; }

    private:
        std::vector<DRC_TEST_PROVIDER*> m_providers;

};

template<class T> class DRC_REGISTER_TEST_PROVIDER
{
    public:
        DRC_REGISTER_TEST_PROVIDER()
        {
            T* provider = new T;
            DRC_TEST_PROVIDER_REGISTRY::Instance().RegisterTestProvider( provider );
        }
};

/**
 * DRC_TEST_PROVIDER
 * is a base class that represents a DRC "provider" which runs some DRC functions over a
 * #BOARD and spits out #PCB_MARKERs as needed.
 */
class DRC_TEST_PROVIDER
{
public:
    DRC_TEST_PROVIDER ();
    virtual ~DRC_TEST_PROVIDER() {}

    void SetDRCEngine( DRC_ENGINE *engine )
    {
        m_drcEngine = engine;
    }

    /**
     * Runs this provider against the given PCB with configured options (if any).
     *
     * Note: Board is non-const, as some DRC functions modify the board (e.g. zone fill
     * or polygon coalescing)
     */

    virtual bool Run() = 0;

    virtual void Enable( bool enable );
    virtual bool IsEnabled() const;

    virtual const wxString GetName() const;
    virtual const wxString GetDescription() const;

    virtual void ReportAux( const wxString fmt, ... );
    virtual void Report( std::shared_ptr<DRC_ITEM> item );
    virtual void ReportWithMarker( std::shared_ptr<DRC_ITEM> item, wxPoint aMarkerPos );
    virtual void ReportWithMarker( std::shared_ptr<DRC_ITEM> item, VECTOR2I aMarkerPos );

    virtual void ReportProgress( double aProgress );
    virtual void ReportStage ( const wxString& aStageName, int index, int total );

    virtual std::set<test::DRC_RULE_ID_T> GetMatchingRuleIds() const = 0;

protected:

    int forEachGeometryItem( const std::vector<KICAD_T> aTypes, const LSET aLayers, std::function<int(BOARD_ITEM*)> aFunc );

    virtual void reportRuleStatistics();
    virtual void accountCheck( test::DRC_RULE* ruleToTest );
    virtual bool isErrorLimitExceeded( int error_code );

    EDA_UNITS userUnits() const;
    DRC_ENGINE *m_drcEngine;
    std::unordered_map<test::DRC_RULE*, int> m_stats;
    bool m_enable;
};


};

#endif // DRC_PROVIDER__H