fdm_Z0_calc.h

/*****************************************************************************
*                                                                            *
*  Copyright (C) 2023 Liu An Lin <liuanlin-mx@qq.com>                        *
*                                                                            *
*  Licensed 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                            *
*                                                                            *
*  Unless required by applicable law or agreed to in writing, software       *
*  distributed under the License is distributed on an "AS IS" BASIS,         *
*  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.  *
*  See the License for the specific language governing permissions and       *
*  limitations under the License.                                            *
*                                                                            *
*****************************************************************************/

#ifndef __FDM_Z0_CALC_H__
#define __FDM_Z0_CALC_H__

#include <cstdint>
#include <opencv2/opencv.hpp>
#include <map>
#include "fdm.h"
#include "Z0_calc.h"
class fdm_Z0_calc: public Z0_calc
{
private:
    enum
    {
        FDM_ID_AIR = 0,
        FDM_ID_METAL_GND,
        FDM_ID_METAL_COND1,
        FDM_ID_METAL_COND2,
        FDM_ID_ER,
    };
    
public:
    fdm_Z0_calc();
    virtual ~fdm_Z0_calc();
    
public:
    /* 1个像素代表的长度 */
    virtual void set_precision(float unit = 0.035);
    
    virtual void set_box_size(float w, float h);
    
    virtual std::uint32_t get_type() { return Z0_calc::Z0_CALC_FDM; }
    
    virtual void clean();
    virtual void clean_all();
    /* 坐标是盒子的中心点 */
    virtual void add_ground(float x, float y, float w, float thickness);
    virtual void add_ring_ground(float x, float y, float r, float thickness);
    virtual void add_wire(float x, float y, float w, float thickness, float conductivity);
    virtual void add_ring_wire(float x, float y, float r, float thickness);
    virtual void add_coupler(float x, float y, float w, float thickness, float conductivity);
    virtual void add_elec(float x, float y, float w, float thickness, float er = 4.6);
    virtual void add_ring_elec(float x, float y, float r, float thickness, float er = 4.6);
    virtual bool calc_Z0(float& Zo, float& v, float& c, float& l, float& r, float& g);
    virtual bool calc_coupled_Z0(float& Zodd, float& Zeven, float c_matrix[2][2], float l_matrix[2][2], float r_matrix[2][2], float g_matrix[2][2]);
private:
    std::int32_t _unit2pix(float v) { return round(v * _pix_unit_r);}
    void _draw(float x, float y, float w, float thick, std::uint8_t r, std::uint8_t g, std::uint8_t b);
    void _draw_ring(float x, float y, float radius, float thick, std::uint8_t r, std::uint8_t g, std::uint8_t b);
    float _read_value(const char *str, const char *key);
    bool _is_some(cv::Mat& img1, cv::Mat& img2);
    
    void _init_fdm(fdm& fdm, cv::Mat& img);
    void _calc_Z0(cv::Mat img, float& Z0, float& v, float& c, float& l, float& r, float& g);
private:
    float _pix_unit;
    float _pix_unit_r;
    float _box_w;
    float _box_h;
    float _c_x;
    float _c_y;
    
    cv::Mat _img;
    cv::Mat _last_img;
    std::map<std::uint16_t, std::uint8_t> _er_map;
    std::uint8_t _fdm_er_id;
    
    float _Zo;
    float _c;
    float _l;
    float _v;
    
    float _Zodd;
    float _Zeven;
    
    float _c_matrix[2][2];
    float _l_matrix[2][2];
    
    float _wire_w;
    float _wire_h;
    float _coupler_w;
    float _coupler_h;
    
    float _wire_conductivity;
    float _coupler_conductivity;
};

#endif