PDFPageOverlayViewProviderを利用したコードにif文を使いたい

Question

実現したいこと

以前こちらでPDFPageOverlayViewProviderのPKTool変更を解決していただきました。
以前の内容
そのコードを使い、if文を利用してbuttonなどの装飾を行いたいのですが、なぜか出来ません。
何がダメなのかも解らず困っています。

発生している問題・分からないこと

buttonやview内に「if文」を入れると、OverlayCoordinatorクラスのsetPen()やsetEraser()が使えなくなってしまいます。
OverlayCoordinatorクラスのpageToViewMapping配列が「if文」を作成すると消えてしまうようです。

環境
macOS Ventura 13.6.6
xcode Version 15.2

エラーメッセージ

エラーメッセージは出ません。

該当のソースコード

import SwiftUI
import PDFKit
import PencilKit

struct PDFUIView: View {
    @State private var isPen = true
    
    var body: some View {
        VStack {
            HStack{
                Button(action: {
                    isPen = true
                }, label: {
                    Text("ペン")
                })
                Button(action: {
                    isPen = false
                }, label: {
                    Text("消しゴム")
                })
                //↓問題のコード
                if isPen == true{
                }else {
                }
                //↑問題のコード
            }
            pdfView(isPen: $isPen)
        }
    }
}

struct pdfView : UIViewRepresentable {
    typealias UIViewType = PDFView
    @Binding var isPen: Bool
    let overlayProvider = OverlayCoordinator()
    func makeUIView(context: Context) -> PDFView {
        let pdfView: PDFView = PDFView()
        if let url =  Bundle.main.url(forResource: "example", withExtension: "pdf") {
            pdfView.pageOverlayViewProvider = overlayProvider
            pdfView.document = PDFDocument(url: url)
            pdfView.autoScales = true
            pdfView.isInMarkupMode = true
        }
        return pdfView
    }
    func updateUIView(_ uiView: PDFView, context: Context) {
        
        if isPen {
            overlayProvider.setPen()
        } else {
            overlayProvider.setEraser()
        }
        
    }
}

class OverlayCoordinator : NSObject,PDFPageOverlayViewProvider {
    var pageToViewMapping = [PDFPage: PKCanvasView]()
    
    func pdfView(_ view: PDFView, overlayViewFor page: PDFPage) -> UIView? {
        var resultView:PKCanvasView? = nil
        if let overlayView = pageToViewMapping[page] {
            resultView = overlayView
        } else {
            let canvasView = PKCanvasView()
            canvasView.drawingPolicy = .anyInput
            canvasView.tool = PKInkingTool(.pen, width: 1)
            canvasView.backgroundColor = .clear
            pageToViewMapping[page] = canvasView
            resultView = canvasView
        }
        return resultView
        
    }
    func setPen() {
        for  i in pageToViewMapping {
            i.value.tool = PKInkingTool(.pen, width: 1)
        }
        print(pageToViewMapping)   //デバッグ用
        
    }
    func setEraser() {
        for  i in pageToViewMapping {
            i.value.tool = PKEraserTool(.bitmap)
        }
        print(pageToViewMapping)   //デバッグ用
    }
    
}

試したこと・調べたこと

• teratailやGoogle等で検索した
• ソースコードを自分なりに変更した
• 知人に聞いた
• その他

上記の詳細・結果

if文を別のところに挿入したり、追加でOverlayCoordinatorとは直接関係のない値を作成しても、「if文」が入ると同様の結果になってしまいました。

@State private var penMode = true

Button(action: {
                    isPen = true
                }, label: {
                    if penMode == true{
                }else {
                }
                })

補足

特になし

Answer 1

ごめんなさい。
前回の回答は不十分な内容だったみたいです。
(UIViewRepresentableは久しぶりだったので、色々忘れてしまいました・・)

OverlayCoordinatorクラスのpageToViewMapping配列が「if文」を作成すると消えてしまうようです。

SwiftUIのビューの仕組みを説明できるほど詳しくないのですが、
OverlayCoordinatorのinitが毎回動いて
pageToViewMappingが初期化されているようですね。
下のようなコードを入れてみたりすると、自分でどこに原因があるか見つけられるかもしれません。

class OverlayCoordinator : NSObject,PDFPageOverlayViewProvider {
    override init() {
        super.init()
        print("init")
    }
    // 省略
}

UIViewRepresentableを使った場合の補足説明は下に引用しておきます。

しかし、このようにstruct内にclassを保持させてやればもっとシンプルにできちゃうのになぜこんな仕組みが用意されているのでしょう？
それはSwiftUIがUIKitのようなclassベースではなくstructベースで、描画されるたびに破棄されて生成されるからです。
structにclassを保持させるパターンだと描画される度にcoordinatorも再生成されます。 makeUIViewの時点でdelegateに渡しているcoordinatorは再描画時には破棄されるので、このコードは再描画されるとdelegateは呼ばれなくなります。
UIViewRepresentableではmakeCoordinatorで生成されたCoordinatorはstructが再描画されても同じCoordinatorが参照できるような仕組みが用意されているわけです。

UIViewRepresentableのCoordinatorはなぜ必要か - The Pragmatic Ball boy

これらを踏まえて、コードを修正しました。

import SwiftUI
import PDFKit
import PencilKit

struct PDFUIView: View {
    @State private var isPen = true
    var body: some View {
        VStack {
            HStack{
                Button(action: {
                    isPen = true
                }, label: {
                    Text("ペン")
                })
                Button(action: {
                    isPen = false
                }, label: {
                    Text("消しゴム")
                })
                //↓問題のコード
                if isPen {
                } else {
                }
                //↑問題のコード
            }
            pdfView(isPen: $isPen)
        }
    }
}

struct pdfView : UIViewRepresentable {
    typealias UIViewType = PDFView
    @Binding var isPen: Bool
//    let overlayProvider = OverlayCoordinator() // 削除
    func makeUIView(context: Context) -> PDFView {
        let pdfView: PDFView = PDFView()
        if let url =  Bundle.main.url(forResource: "example2", withExtension: "pdf") {
            pdfView.pageOverlayViewProvider = context.coordinator // 変更
            pdfView.document = PDFDocument(url: url)
            pdfView.autoScales = true
            pdfView.isInMarkupMode = true
        }
        return pdfView
    }
    func updateUIView(_ uiView: PDFView, context: Context) {
        if isPen {
            context.coordinator.setPen() // 変更
        } else {
            context.coordinator.setEraser() // 変更
        }
    }
    // 追加↓
    func makeCoordinator() -> OverlayCoordinator {
        return OverlayCoordinator()
    }
    // 追加↑
}

class OverlayCoordinator: NSObject, PDFPageOverlayViewProvider {
    var pageToViewMapping = [PDFPage: PKCanvasView]()
    func pdfView(_ view: PDFView, overlayViewFor page: PDFPage) -> UIView? {
        var resultView:PKCanvasView? = nil
        if let overlayView = pageToViewMapping[page] {
            resultView = overlayView
        } else {
            let canvasView = PKCanvasView()
            canvasView.drawingPolicy = .anyInput
            canvasView.tool = PKInkingTool(.pen, width: 1)
            canvasView.backgroundColor = .clear
            pageToViewMapping[page] = canvasView
            resultView = canvasView
        }
        return resultView
    }
    func setPen() {
        for i in pageToViewMapping {
            i.value.tool = PKInkingTool(.pen, width: 1)
        }
    }
    func setEraser() {
        for i in pageToViewMapping {
            i.value.tool = PKEraserTool(.bitmap)
        }
    }
}

SwiftUIのTutorialsにもmakeCoordinator()を使うことが記載されていましたね・・

Section 4
Add a custom page control
Step 3
Create a nested Coordinator type in PageControl, and add a makeCoordinator() method to create and return a new coordinator.

Interfacing with UIKit — SwiftUI Tutorials | Apple Developer Documentation

Comments

[Gameno]

回答ありがとうございます。
試したところ問題が解決しました！
ベストアンサーに選ばせていただきました。

PDFPageOverlayViewProviderを調べたときに、なぜ必要なのかわからず後回しにしていた問題が、こんな形で影響してくるとは思ってもいませんでした。勉強不足を痛感しました。
本当にありがとうございました。