/*
 * Copyright (C) 1992-2016 KiCad Developers, see AUTHORS.txt for contributors.
 * Copyright (C) 2018 CERN
 * Author: Maciej Suminski <maciej.suminski@cern.ch>
 *
 * 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 3 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, see <http://www.gnu.org/licenses/>.
 */

#ifndef PCBNEW_PRINTOUT_H
#define PCBNEW_PRINTOUT_H

#include <board_printout.h>
#include <pcb_painter.h>

class BOARD;

struct PCBNEW_PRINTOUT_SETTINGS : BOARD_PRINTOUT_SETTINGS
{
    PCBNEW_PRINTOUT_SETTINGS( const PAGE_INFO& aPageInfo );

    enum DRILL_MARK_SHAPE_T {
        NO_DRILL_SHAPE,
        SMALL_DRILL_SHAPE,
        FULL_DRILL_SHAPE
    } m_drillMarks;             ///< Drill marks shape

    enum PAGINATION_T {
        LAYER_PER_PAGE,
        ALL_LAYERS
    } m_pagination;             ///< Pagination

    bool m_noEdgeLayer;         ///< Disable board outline on each page

    void Load( wxConfigBase* aConfig ) override;
    void Save( wxConfigBase* aConfig ) override;
};


class PCBNEW_PRINTOUT : public BOARD_PRINTOUT
{
public:
    PCBNEW_PRINTOUT( BOARD* aBoard, const PCBNEW_PRINTOUT_SETTINGS& aParams,
            const KIGFX::VIEW* aView, const wxSize& aSheetSize, const wxString& aTitle );

    bool OnPrintPage( int aPage ) override;

protected:
    void setupViewLayers( const std::unique_ptr<KIGFX::VIEW>& aView, const LSET& aLayerSet ) override;

    void setupPainter( const std::unique_ptr<KIGFX::PAINTER>& aPainter ) override;

    void setupGal( KIGFX::GAL* aGal ) override;

    EDA_RECT getBoundingBox() override;

    std::unique_ptr<KIGFX::PAINTER> getPainter( KIGFX::GAL* aGal ) override;

private:
    BOARD* m_board;

    PCBNEW_PRINTOUT_SETTINGS m_pcbnewSettings;
};


namespace KIGFX {
/**
 * Special flavor of PCB_PAINTER that contains
 * modifications to handle printing options.
 */
class PCB_PRINT_PAINTER : public PCB_PAINTER
{
public:
    PCB_PRINT_PAINTER( GAL* aGal )
        : PCB_PAINTER( aGal ), m_drillMarkReal( false ), m_drillMarkSize( 0 )
    {
    }

    /**
     * Set drill marks visibility and options.
     * @param aRealSize when enabled, drill marks represent actual holes. Otherwise aSize
     * parameter is used.
     * @param aSize is drill mark size (internal units), valid only when aRealSize == false.
     */
    void SetDrillMarks( bool aRealSize, unsigned int aSize = 0 )
    {
        m_drillMarkReal = aRealSize;
        m_drillMarkSize = aSize;
    }

protected:
    int getDrillShape( const D_PAD* aPad ) const override;

    VECTOR2D getDrillSize( const D_PAD* aPad ) const override;

    int getDrillSize( const VIA* aVia ) const override;

    ///> Flag deciding whether use the actual hole size or user-specified size for drill marks
    bool m_drillMarkReal;

    ///> User-specified size for drill marks (expressed in internal units)
    int m_drillMarkSize;
};

}; // namespace KIGFX

#endif /* PCBNEW_PRINTOUT_H */