// CD2DGradientBrush.h

#pragma once

#include <afxwin.h>
#include <d2d1.h>
#include <vector>

class CD2DGradientBrush
{
public:
    // 公共构造函数
    CD2DGradientBrush(ID2D1RenderTarget* pRenderTarget, const D2D1_POINT_2F& startPoint, const D2D1_POINT_2F& endPoint);

    // 析构函数
    ~CD2DGradientBrush();

    // 受保护方法
protected:
    // 销毁渐变刷资源
    void Destroy();

    // 其他成员函数和数据成员可以在这里添加
    // ...

    // 受保护的渐变停止集合
    std::vector<D2D1_GRADIENT_STOP> m_arGradientStops;

    // 受保护的颜色插值伽马值
    D2D1_GAMMA m_colorInterpolationGamma;

    // 受保护的扩展模式
    D2D1_EXTEND_MODE m_extendMode;

    // 受保护的渐变停止集合指针
    CComPtr<ID2D1GradientStopCollection> m_pGradientStops;

private:
    // 用于存储 Direct2D 渐变刷的私有成员
    CComPtr<ID2D1GradientBrush> m_pGradientBrush;
};

// CD2DGradientBrush.cpp

#include "stdafx.h"
#include "CD2DGradientBrush.h"

CD2DGradientBrush::CD2DGradientBrush(ID2D1RenderTarget* pRenderTarget, const D2D1_POINT_2F& startPoint, const D2D1_POINT_2F& endPoint)
{
    // 初始化渐变停止集合
    D2D1_GRADIENT_STOP gradientStops[] = {
        // ... 添加渐变停止的具体信息 ...
    };

    m_arGradientStops.assign(std::begin(gradientStops), std::end(gradientStops));

    // 初始化颜色插值伽马值
    m_colorInterpolationGamma = D2D1_GAMMA_2_2; // 例如，设置为默认值 D2D1_GAMMA_2_2

    // 初始化扩展模式
    m_extendMode = D2D1_EXTEND_MODE_CLAMP; // 例如，设置为默认值 D2D1_EXTEND_MODE_CLAMP

    // 创建渐变停止集合
    pRenderTarget->CreateGradientStopCollection(
        m_arGradientStops.data(),
        m_arGradientStops.size(),
        &m_pGradientStops
    );

    // 创建线性渐变刷
    pRenderTarget->CreateLinearGradientBrush(
        D2D1::LinearGradientBrushProperties(startPoint, endPoint),
        m_pGradientStops,
        &m_pGradientBrush
    );
}

CD2DGradientBrush::~CD2DGradientBrush()
{
    // 在析构函数中释放渐变刷资源
    Destroy();
    // 可以添加其他需要在对象销毁时释放的资源清理代码
}

void CD2DGradientBrush::Destroy()
{
    // 在销毁函数中释放渐变刷资源
    m_pGradientBrush.Release();

    // 释放渐变停止集合指针
    m_pGradientStops.Release();
    
    // 清理渐变停止集合
    m_arGradientStops.clear();

    // 可以添加其他需要在销毁时释放的资源清理代码
}