/* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*- */

/*

 * This file is part of the LibreOffice project.

 *

 * This Source Code Form is subject to the terms of the Mozilla Public

 * License, v. 2.0. If a copy of the MPL was not distributed with this

 * file, You can obtain one at http://mozilla.org/MPL/2.0/.

 *

 * This file incorporates work covered by the following license notice:

 *

 *   Licensed to the Apache Software Foundation (ASF) under one or more

 *   contributor license agreements. See the NOTICE file distributed

 *   with this work for additional information regarding copyright

 *   ownership. The ASF licenses this file to you under the Apache

 *   License, Version 2.0 (the "License"); you may not use this file

 *   except in compliance with the License. You may obtain a copy of

 *   the License at http://www.apache.org/licenses/LICENSE-2.0 .

 */

#include <tools/multisel.hxx>

#include <osl/diagnose.h>

#include <pfuncache.hxx>

#include <printfun.hxx>

#include <docsh.hxx>

#include <markdata.hxx>

#include <prevloc.hxx>

#include <utility>

ScPrintFuncCache::ScPrintFuncCache(ScDocShell& rShell, const ScMarkData& rMark,

                                   ScPrintSelectionStatus aStatus, Size aPrintPageSize, bool bPrintPageLandscape,

                                   bool bUsePrintDialogSetting)

    :

    aSelection(std::move( aStatus )),

    rDocSh( rShell ),

    nTotalPages( 0 ),

    bLocInitialized( false )

{

    //  page count uses the stored cell widths for the printer anyway,

    //  so ScPrintFunc with the document's printer can be used to count

    SfxPrinter* pPrinter = rDocSh.GetPrinter();

    ScRange aRange;

    const ScRange* pSelRange = nullptr;

    if ( rMark.IsMarked() )

    {

        aRange = rMark.GetMarkArea();

        pSelRange = &aRange;

    }

    ScDocument& rDoc = rDocSh.GetDocument();

    SCTAB nTabCount = rDoc.GetTableCount();

    // avoid repeated progress bars if row heights for all sheets are needed

    if ( nTabCount > 1 && rMark.GetSelectCount() == nTabCount )

        rDocSh.UpdatePendingRowHeights( nTabCount-1, true );

    SCTAB nTab;

    for ( nTab=0; nTab<nTabCount; nTab++ )

    {

        tools::Long nAttrPage = nTab > 0 ? nFirstAttr[nTab-1] : 1;

        tools::Long nThisTab = 0;

        if ( rMark.GetTableSelect( nTab ) )

        {

            ScPrintFunc aFunc(rDocSh, pPrinter, nTab, nAttrPage, 0, pSelRange,

                              &aSelection.GetOptions(), nullptr, aPrintPageSize,

                              bPrintPageLandscape, bUsePrintDialogSetting);

            nThisTab = aFunc.GetTotalPages();

            nFirstAttr.push_back( aFunc.GetFirstPageNo() );         // from page style or previous sheet

        }

        else

            nFirstAttr.push_back( nAttrPage );

        nPages.push_back( nThisTab );

        nTotalPages += nThisTab;

    }

}

ScPrintFuncCache::~ScPrintFuncCache()

{

}

void ScPrintFuncCache::InitLocations( const ScMarkData& rMark, OutputDevice* pDev )

{

    if ( bLocInitialized )

        return;                 // initialize only once

    ScRange aRange;

    const ScRange* pSelRange = nullptr;

    if ( rMark.IsMarked() )

    {

        aRange = rMark.GetMarkArea();

        pSelRange = &aRange;

    }

    tools::Long nRenderer = 0;     // 0-based physical page number across sheets

    tools::Long nTabStart = 0;

    ScDocument& rDoc = rDocSh.GetDocument();

    SCTAB nTabCount = rDoc.GetTableCount();

    for (SCTAB nTab : rMark)

    {

        if (nTab >= nTabCount)

            break;

        ScPrintFunc aFunc( pDev, rDocSh, nTab, nFirstAttr[nTab], nTotalPages, pSelRange, &aSelection.GetOptions() );

        aFunc.SetRenderFlag( true );

        tools::Long nDisplayStart = GetDisplayStart( nTab );

        for ( tools::Long nPage=0; nPage<nPages[nTab]; nPage++ )

        {

            Range aPageRange( nRenderer+1, nRenderer+1 );

            MultiSelection aPage( aPageRange );

            aPage.SetTotalRange( Range(0,RANGE_MAX) );

            aPage.Select( aPageRange );

            ScPreviewLocationData aLocData( &rDoc, pDev );

            aFunc.DoPrint( aPage, nTabStart, nDisplayStart, false, &aLocData );

            ScRange aCellRange;

            tools::Rectangle aPixRect;

            if ( aLocData.GetMainCellRange( aCellRange, aPixRect ) )

                aLocations.emplace_back( nRenderer, aCellRange, aPixRect );

            ++nRenderer;

        }

        nTabStart += nPages[nTab];

    }

    bLocInitialized = true;

}

bool ScPrintFuncCache::FindLocation( const ScAddress& rCell, ScPrintPageLocation& rLocation ) const

{

    auto aIter = std::find_if(aLocations.begin(), aLocations.end(),

        [&rCell](const ScPrintPageLocation& rLoc) { return rLoc.aCellRange.Contains(rCell); });

    if (aIter != aLocations.end())

    {

        rLocation = *aIter;

        return true;

    }

    return false;   // not found

}

bool ScPrintFuncCache::IsSameSelection( const ScPrintSelectionStatus& rStatus ) const

{

    return aSelection == rStatus;

}

SCTAB ScPrintFuncCache::GetTabForPage( tools::Long nPage ) const

{

    ScDocument& rDoc = rDocSh.GetDocument();

    SCTAB nTabCount = rDoc.GetTableCount();

    SCTAB nTab = 0;

    while ( nTab < nTabCount && nPage >= nPages[nTab] )

        nPage -= nPages[nTab++];

    if (nTab >= nTabCount)

        nTab = nTabCount - 1;

    return nTab;

}

tools::Long ScPrintFuncCache::GetTabStart( SCTAB nTab ) const

{

    tools::Long nRet = 0;

    const SCTAB maxIndex = std::min(nTab, static_cast<SCTAB>(nPages.size()));

    for ( SCTAB i=0; i<maxIndex; i++ )

        nRet += nPages[i];

    return nRet;

}

tools::Long ScPrintFuncCache::GetDisplayStart( SCTAB nTab ) const

{

    //! merge with lcl_GetDisplayStart in preview?

    tools::Long nDisplayStart = 0;

    ScDocument& rDoc = rDocSh.GetDocument();

    for (SCTAB i=0; i<nTab; i++)

    {

        if ( rDoc.NeedPageResetAfterTab(i) )

            nDisplayStart = 0;

        else

        {

            if ( i < static_cast<SCTAB>(nPages.size()) )

                nDisplayStart += nPages[i];

            else

                OSL_FAIL("nPages out of bounds, FIX IT!");

        }

    }

    return nDisplayStart;

}

/* vim:set shiftwidth=4 softtabstop=4 expandtab: */