rch_diagram_v2.ipynb

{
  "nbformat": 4,
  "nbformat_minor": 0,
  "metadata": {
    "colab": {
      "name": "rch_diagram_v2.ipynb",
      "provenance": [],
      "collapsed_sections": [],
      "authorship_tag": "ABX9TyO0CubVA/WViZMz3mbT8MxS",
      "include_colab_link": true
    },
    "kernelspec": {
      "name": "python3",
      "display_name": "Python 3"
    }
  },
  "cells": [
    {
      "cell_type": "markdown",
      "metadata": {
        "id": "view-in-github",
        "colab_type": "text"
      },
      "source": [
        "<a href=\"https://colab.research.google.com/github/jeffblackadar/charcoalhearths/blob/master/rch_diagram_v2.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a>"
      ]
    },
    {
      "cell_type": "markdown",
      "metadata": {
        "id": "KpsqePvzYcmx"
      },
      "source": [
        "# Draws a diagram of a Relict Charcol Hearth"
      ]
    },
    {
      "cell_type": "code",
      "metadata": {
        "id": "-zZX04GmY3KW"
      },
      "source": [
        "import cv2\n",
        "import numpy as np\n",
        "from google.colab.patches import cv2_imshow\n",
        "\n",
        "class RCH_segment:\n",
        "    pt_bottom_inside = (0,0)\n",
        "    pt_bottom_outside = (0,0)\n",
        "    pt_top_inside = (0,0)\n",
        "    pt_top_outside = (0,0)    \n",
        "    segment_type = \"\"\n",
        "    fill = False\n",
        "    color = (200,200,200)\n",
        "    def __init__(self,pt_bottom_inside,pt_bottom_outside,pt_top_inside,pt_top_outside,segment_type, fill=False, color=(200,200,200)):\n",
        "        self.pt_bottom_inside = pt_bottom_inside\n",
        "        self.pt_bottom_outside = pt_bottom_outside\n",
        "        self.pt_top_inside = pt_top_inside\n",
        "        self.pt_top_outside = pt_top_outside\n",
        "        self.segment_type = segment_type\n",
        "        self.fill = fill\n",
        "        self.color = color\n",
        "    def get_points(self):\n",
        "        return([self.pt_bottom_inside,self.pt_bottom_outside,self.pt_top_outside,self.pt_top_inside])\n",
        "    def get_type(self):\n",
        "        return(self.segment_type)\n",
        "\n",
        "\n",
        "class RCH_diagram:\n",
        "    # using inches since that is what the source uses, Can convert to metric after  \n",
        "    inch_conp = 2\n",
        "    inch_con_m = 39.37\n",
        "    red = (255, 0, 0)\n",
        "    gray = (200,200,200)\n",
        "    black =  (0, 0, 0)\n",
        "    white =  (255, 255, 255)\n",
        "    lbrown =  (59, 92, 131)\n",
        "    dbrown =  (40, 50, 63)\n",
        "    green =  (40, 200, 40)\n",
        "    height = int(17.5*12)\n",
        "    width = 40*12\n",
        "    origin_pixel_x = width * inch_conp /2\n",
        "    origin_pixel_y = (height * inch_conp - 2)-100\n",
        "    image_background = white\n",
        "    sides = []\n",
        "    sides.append(\"left\")\n",
        "    sides.append(\"right\")\n",
        "    segments = []\n",
        "\n",
        "    rch_image = np.zeros((height * inch_conp, width * inch_conp, 3), np.uint8)\n",
        "    color = tuple(reversed(white))\n",
        "    # Fill image with color\n",
        "    rch_image[:] = color\n",
        "    chimney = [(-4,0),(4,132)]\n",
        "    segments.append(RCH_segment((-5,0),((-15*12),0),(-5,48),((-13*12),48),\"foot_left\",False,black))\n",
        "    segments.append(RCH_segment((5,0),((15*12),0),(5,48),((13*12),48),\"foot_right\",False,black))\n",
        "\n",
        "\n",
        "    segments.append(RCH_segment((-5,49),((-13*12),49),(-5,96),((-9*12),96),\"waist_left\",False,black))\n",
        "    segments.append(RCH_segment((5,49),((13*12),49),(5,96),((9*12),96),\"waist_right\",False,black))\n",
        "\n",
        "    segments.append(RCH_segment((-5,97),((-9*12),97),(-5,120),((-6*12),120),\"shoulder_left\",False,black))\n",
        "    segments.append(RCH_segment((5,97),((9*12),97),(5,120),((6*12),120),\"shoulder_right\",False,black))\n",
        "    #segments.append(RCH_segment((-5,120),((-6*12),120),(-5,132),((-4.5*12),132),\"head_left\",True,lbrown))\n",
        "    #segments.append(RCH_segment((5,120),((6*12),120),(5,132),((4.5*12),132),\"head_right\",True,lbrown))\n",
        "    \n",
        "\n",
        "    segments.append(RCH_segment((-15.25*12,0),((-17.25*12),0),(-13.25*12,48),((-13.75*12),48),\"foot_cover_left\",True,black))\n",
        "    segments.append(RCH_segment((15.25*12,0),((17.25*12),0),(13.25*12,48),((13.75*12),48),\"foot_cover_right\",True,black))\n",
        "    segments.append(RCH_segment((-13.25*12,48),((-13.75*12),48),(-9.25*12,96),((-9.75*12),96),\"waist_cover_left\",True,black))\n",
        "    segments.append(RCH_segment((13.25*12,48),((13.75*12),48),(9.25*12,96),((9.75*12),96),\"waist_cover_right\",True,black))\n",
        "    segments.append(RCH_segment((-9.25*12,96),((-9.75*12),96),(-6.25*12,120),((-6.75*12),120),\"shoulder_cover_left\",True,black))\n",
        "    segments.append(RCH_segment((9.25*12,96),((9.75*12),96),(6.25*12,120),((6.75*12),120),\"shoulder_cover_right\",True,black))\n",
        "    #segments.append(RCH_segment(((-6.25*12),120),((-6.75*12),120),((-3.25*12),138),((-3.75*12),138),\"head_cover_left\",True,black))\n",
        "    #segments.append(RCH_segment(((6.25*12),120),((6.75*12),120),((3.25*12),138),((3.75*12),138),\"head_cover_right\",True,black))\n",
        "    segments.append(RCH_segment((-5.25,127),((-5.65*12),127),(-5.25,138),((-3.75*12),138),\"top_cover_left\",True,black))\n",
        "    segments.append(RCH_segment((5.25,127),((5.65*12),127),(5.25,138),((3.75*12),138),\"top_cover_right\",True,black))\n",
        "\n",
        "    segments.append(RCH_segment((-15*12,0),((-15.25*12),0),(-13*12,48),((-13.25*12),48),\"foot_leaf_cover_left\",True,green))\n",
        "    segments.append(RCH_segment((15*12,0),((15.25*12),0),(13*12,48),((13.25*12),48),\"foot_leaf_cover_right\",True,green))\n",
        "    segments.append(RCH_segment((-13*12,48),((-13.25*12),48),(-9*12,96),((-9.25*12),96),\"waist_leaf_cover_left\",True,green))\n",
        "    segments.append(RCH_segment((13*12,48),((13.25*12),48),(9*12,96),((9.25*12),96),\"waist_leaf_cover_right\",True,green))\n",
        "    segments.append(RCH_segment((-9*12,96),((-9.25*12),96),(-6*12,120),((-6.25*12),120),\"shoulder_leaf_cover_left\",True,green))\n",
        "    segments.append(RCH_segment((9*12,96),((9.25*12),96),(6*12,120),((6.25*12),120),\"shoulder_leaf_cover_right\",True,green))\n",
        "    #segments.append(RCH_segment(((-6*12),120),((-6.25*12),120),(((-4.5*12)),132),(((-4.5*12)-3),132),\"head_leaf_cover_left\",True,green))\n",
        "    #segments.append(RCH_segment(((6*12),120),((6.25*12),120),((4.5*12),132),((4.5*12+3),132),\"head_leaf_cover_right\",True,green))\n",
        "    #segments.append(RCH_segment((-5,132),((-4.50*12),132),(-5,135),((-4.50*12+4.5),135),\"top_leaf_cover_left\",True,green))\n",
        "    #segments.append(RCH_segment((5,132),((4.5*12),132),(5,135),((4.5*12+4.5),135),\"top_leaf_cover_right\",True,green))\n",
        "\n",
        "\n",
        "\n",
        "\n",
        "    segments.append(RCH_segment((-17.25*12,-18),((17.25*12),-18),(-17.25*12,0),((17.25*12),0),\"base\",True,black))\n",
        "\n",
        "    #segments.append(RCH_segment((-16*12,48),((-16*12),96),(-16*12-7,48),((-16*12-7),96),\"billet\",True,lbrown))\n",
        "    segments.append(RCH_segment((-16*12-9,48),((-16*12-9),96),(-16*12-7-9,48),((-16*12-7-9),96),\"billet\",True,dbrown))\n",
        "    \n",
        "    segments.append(RCH_segment((-15*12-15,48),((-15*12-15),48+inch_con_m),(-15*12-1-15,48),((-15*12-1-15),48+inch_con_m),\"m\",True,black))\n",
        "    segments.append(RCH_segment((16*12,48),((16*12),96),(16*12+2,48),((16*12+2),96),\"lap-wood\",True,lbrown))\n",
        "    #segments.append(RCH_segment((16*12-4,48),((16*12-4),96),(16*12+2-4,48),((16*12+2-4),96),\"lap-wood\",True,dbrown))\n",
        "    segments.append(RCH_segment((16*12+6,48),((16*12+6),48+inch_con_m),(16*12+1+6,48),((16*12+1+6),48+inch_con_m),\"m\",True,black))    \n",
        "\n",
        "\n",
        "\n",
        "    def __init__(self):\n",
        "        pass\n",
        "        self.add_layer(4,120,(6*12),12,1.5,\"head\",True,self.lbrown,self.green)\n",
        "        self.add_layer(4,132,(4.5*12),3,2,\"head_cover\",True,self.green,self.green)\n",
        "        self.add_layer(4,135,(4.5*12)-6,12,2,\"head_cap\",True,self.black,self.black)\n",
        "        self.segments.append(RCH_segment((-2*12,125),((-2*12),132),(2*12,125),((2*12),132),\"Bridgen\",True,self.dbrown))\n",
        "    def convert_point(self, point_inches):\n",
        "        p_x = int((point_inches[0] * self.inch_conp) + self.origin_pixel_x)\n",
        "        p_y = int(self.origin_pixel_y - (point_inches[1] * self.inch_conp))\n",
        "        return((p_x,p_y))\n",
        "    def add_layer(self,x_origin, y_origin, width, height, slant, layer_name, layer_fill, layer_color, layer_color_2):\n",
        "        for cn in range(0,2):\n",
        "            if(cn==0):\n",
        "                dir = -1\n",
        "            else:\n",
        "                dir = 1  \n",
        "            #print(dir)  \n",
        "            self.segments.append(RCH_segment((x_origin*dir,y_origin),((width*dir),y_origin),(x_origin*dir,y_origin+height),((width-(height*slant))*dir,y_origin+height),layer_name+\"_\"+self.sides[cn],layer_fill,layer_color))\n",
        "            self.segments.append(RCH_segment((width*dir,y_origin),((width+3)*dir,y_origin),((width-(height*slant))*dir,y_origin+height),(((width-(height*slant))+3)*dir,y_origin+height),layer_name+\"_\"+self.sides[cn]+\"_leaf\",True,layer_color_2))\n",
        "            self.segments.append(RCH_segment(((width+3)*dir,y_origin),((width+9)*dir,y_origin),(((width-(height*slant))+3)*dir,y_origin+height),(((width-(height*slant))+9)*dir,y_origin+height),layer_name+\"_\"+self.sides[cn]+\"_dust\",True,self.black))\n",
        "            #segments.append(RCH_segment((5,120),((6*12),120),(5,132),((4.5*12),132),\"head_right\",True,lbrown))\n",
        "\n",
        "\n",
        "    def convert_rect(self, points_inches):\n",
        "        #print(points_inches[0])\n",
        "        p1 = points_inches[0]\n",
        "        p1px = self.convert_point(p1)\n",
        "        p2 = points_inches[1]\n",
        "        p2px = self.convert_point(p2)\n",
        "        #print(p1px,p2px)\n",
        "        return([p1px,p2px])\n",
        "\n",
        "    def convert_points(self, points_inches):\n",
        "        #print(points_inches[0])\n",
        "        pts_px = []\n",
        "        for p_in in points_inches:\n",
        "            p_px = self.convert_point(p_in)\n",
        "            p_px = [p_px[0],p_px[1]]\n",
        "            pts_px.append(p_px)\n",
        "\n",
        "        #print(pts_px)\n",
        "        pts_px = np.array(pts_px)\n",
        "        return(pts_px)\n",
        "\n",
        "\n",
        "    def draw_rch(self):\n",
        "        chimney_pixels = self.convert_rect(self.chimney)\n",
        "        #print(chimney_pixels[0])\n",
        "        #cv2.rectangle(self.rch_image, chimney_pixels[0], chimney_pixels[1], self.black, 1)\n",
        "        #cv2.rectangle(self.rch_image, (300, 4), (300, 400), self.black, 2)\n",
        "        #Chimney\n",
        "        col = self.lbrown  \n",
        "        cv2.line(self.rch_image,(self.convert_point((5,0))),(self.convert_point((5,48))),col,3)\n",
        "        cv2.line(self.rch_image,(self.convert_point((-5,0))),(self.convert_point((-5,48))),col,3)\n",
        "        cv2.line(self.rch_image,(self.convert_point((5,48))),(self.convert_point((5,96))),col,3)\n",
        "        cv2.line(self.rch_image,(self.convert_point((-5,48))),(self.convert_point((-5,96))),col,3)\n",
        "\n",
        "        for cn in range (0,25):\n",
        "            #if(cn%2==0):\n",
        "            col = self.dbrown  \n",
        "            #else:\n",
        "                \n",
        "            cv2.line(self.rch_image,(self.convert_point((5+3+(cn*7),2))),(self.convert_point((5+3+(cn*6),47))),col,10)\n",
        "            cv2.line(self.rch_image,(self.convert_point((-5-3+(cn*-7),2))),(self.convert_point((-5-3+(cn*-6),47))),col,10)\n",
        "        for cn in range (0,25):            \n",
        "            col = self.lbrown              \n",
        "\n",
        "            cv2.line(self.rch_image,(self.convert_point((5+3+4+(cn*7),1))),(self.convert_point((5+3+2+(cn*6),47))),col,3)\n",
        "            cv2.line(self.rch_image,(self.convert_point((-5-3-4+(cn*-7),1))),(self.convert_point((-5-3-2+(cn*-6),47))),col,3)            \n",
        "        for cn in range (0,18):\n",
        "            col = self.dbrown  \n",
        "            cv2.line(self.rch_image,(self.convert_point((5+3+(cn*8.3),50))),(self.convert_point((5+3+(cn*6),95))),col,10)\n",
        "            cv2.line(self.rch_image,(self.convert_point((-5-3+(cn*-8.3),50))),(self.convert_point((-5-3+(cn*-6),95))),col,10) \n",
        "        for cn in range (0,18):\n",
        "            col = self.lbrown\n",
        "            cv2.line(self.rch_image,(self.convert_point((5+3+4.5+(cn*8.3),49))),(self.convert_point((5+3+2+(cn*6),95))),col,3)\n",
        "            cv2.line(self.rch_image,(self.convert_point((-5-3-4.5+(cn*-8.3),49))),(self.convert_point((-5-3-2+(cn*-6),95))),col,3) \n",
        "\n",
        "\n",
        "        for cn in range (0,24):\n",
        "            #if(cn%2==0):\n",
        "            col = self.dbrown  \n",
        "            #else:\n",
        "            #col = self.dbrown\n",
        "            cv2.line(self.rch_image,(self.convert_point((8+(cn*4.2),97))),(self.convert_point((8+(cn*3),119))),col,10)\n",
        "            cv2.line(self.rch_image,(self.convert_point((-8+(cn*-4.2),97))),(self.convert_point((-8+(cn*-3),119))),col,10) \n",
        "\n",
        "        for cn in range (0,24):\n",
        "            #if(cn%2==0):\n",
        "            col = self.lbrown  \n",
        "            #else:\n",
        "            #col = self.dbrown\n",
        "            cv2.line(self.rch_image,(self.convert_point((4+1.5+(cn*4.2),97))),(self.convert_point((4+1.5+(cn*3),119))),col,3)\n",
        "            cv2.line(self.rch_image,(self.convert_point((-4-1.5+(cn*-4.2),97))),(self.convert_point((-4-1.5+(cn*-3),119))),col,3) \n",
        "\n",
        "\n",
        "\n",
        "        #foot_l_px = self.convert_points(self.foot_left)\n",
        "        #cv2.polylines(self.rch_image,[foot_l_px],True,self.black)\n",
        "        #foot_r_px = self.convert_points(self.foot_right)\n",
        "        #cv2.polylines(self.rch_image,[foot_r_px],True,self.black)\n",
        "        for s in self.segments:\n",
        "            #print(s.get_type(), s.fill)\n",
        "            #print(s.get_points())            \n",
        "            if(s.fill==True):\n",
        "                cv2.fillPoly(self.rch_image,[self.convert_points(s.get_points())],s.color)\n",
        "                \n",
        "            else:\n",
        "                pass  \n",
        "                #cv2.polylines(self.rch_image,[self.convert_points(s.get_points())],True,s.color)\n",
        "        font = cv2.FONT_HERSHEY_SIMPLEX\n",
        "        font_scale = .6\n",
        "        font_color = self.black \n",
        "        font_thickness = 1\n",
        "        cv2.rectangle(self.rch_image, (self.convert_point((-72,-16))), (self.convert_point((200,-6))), self.white, -1)\n",
        "        cv2.putText(self.rch_image, \"Base (Smooth, cleared of vegetation.) ~10m diameter.\" , (self.convert_point((-72,-14))), font, font_scale, font_color, font_thickness, cv2.LINE_AA)\n",
        "        cv2.rectangle(self.rch_image, (self.convert_point((12,2))), (self.convert_point((35,12))), self.white, -1)\n",
        "        cv2.putText(self.rch_image, \"Foot\" , (self.convert_point((12,4))), font, font_scale, font_color, font_thickness, cv2.LINE_AA)\n",
        "        cv2.rectangle(self.rch_image, (self.convert_point((12,50))), (self.convert_point((37,60))), self.white, -1)\n",
        "        cv2.putText(self.rch_image, \"Waist\" , (self.convert_point((12,52))), font, font_scale, font_color, font_thickness, cv2.LINE_AA)\n",
        "        cv2.rectangle(self.rch_image, (self.convert_point((12,98))), (self.convert_point((60,108))), self.white, -1)\n",
        "        cv2.putText(self.rch_image, \"Shoulders\" , (self.convert_point((12,100))), font, font_scale, font_color, font_thickness, cv2.LINE_AA)\n",
        "        cv2.rectangle(self.rch_image, (self.convert_point((30,122))), (self.convert_point((56,132))), self.white, -1)\n",
        "        cv2.putText(self.rch_image, \"Head\" , (self.convert_point((30,124))), font, font_scale, font_color, font_thickness, cv2.LINE_AA)\n",
        "\n",
        "        cv2.rectangle(self.rch_image, (self.convert_point((-90,62))), (self.convert_point((-20,72))), self.white, -1)\n",
        "        cv2.putText(self.rch_image, \"Kindling, chips\" , (self.convert_point((-90,64))), font, font_scale, font_color, font_thickness, cv2.LINE_AA)\n",
        "        cv2.arrowedLine(self.rch_image,(self.convert_point((-20,67))),(self.convert_point((-4,67))),self.white,2,tipLength = 0.4)\n",
        "\n",
        "        cv2.putText(self.rch_image, \"Billet, meter\" , (self.convert_point((-17.5*12,98))), font, font_scale, font_color, font_thickness, cv2.LINE_AA)\n",
        "        cv2.putText(self.rch_image, \"Lap-wood, meter\" , (self.convert_point((10*12,98))), font, font_scale, font_color, font_thickness, cv2.LINE_AA)\n",
        "\n",
        "        \n",
        "        cv2.putText(self.rch_image, \"Chimney\" , (self.convert_point((-24,150))), font, font_scale, font_color, font_thickness, cv2.LINE_AA)\n",
        "        cv2.putText(self.rch_image, \"Bridgen\" , (self.convert_point((-24,126))), font, .5, self.white, font_thickness, cv2.LINE_AA)\n",
        "        cv2.putText(self.rch_image, \"Dust and soil cover\" , (self.convert_point((-13*12,150))), font, font_scale, font_color, font_thickness, cv2.LINE_AA)\n",
        "        cv2.putText(self.rch_image, \"Leaf cover layer\" , (self.convert_point((3*12,150))), font, font_scale, font_color, font_thickness, cv2.LINE_AA)\n",
        "        cv2.arrowedLine(self.rch_image,(self.convert_point((4*12-6,148))),(self.convert_point((4*12-6,136))),self.green,2,tipLength = 0.4)\n",
        "        cv2.arrowedLine(self.rch_image,(self.convert_point((-8*12,148))),(self.convert_point((-8*12,112))),self.black,2,tipLength = 0.2)\n",
        "\n",
        "        \n",
        "        for cn in range(self.origin_pixel_y-226,self.origin_pixel_y,6):\n",
        "        #for cn in range(200,300):\n",
        "            # print(int(self.origin_pixel_x-120)))\n",
        "            self.rch_image[cn:cn+1,int(self.origin_pixel_x-7):int(self.origin_pixel_x+3)] = self.lbrown\n",
        "            self.rch_image[cn:cn+1,int(self.origin_pixel_x+5):int(self.origin_pixel_x+7)] = self.lbrown\n",
        "            self.rch_image[cn+3:cn+4,int(self.origin_pixel_x-7):int(self.origin_pixel_x-5)] = self.lbrown\n",
        "            self.rch_image[cn+3:cn+4,int(self.origin_pixel_x-3):int(self.origin_pixel_x+7)] = self.lbrown\n",
        "\n",
        "            #self.rch_image[cn:cn+1,int(self.origin_pixel_x-8):int(self.origin_pixel_x+8)] = [25, 255, 255]\n",
        "            #self.rch_image[cn:cn+5,int(300-8):int(300+8)] = [25, 255, 255]\n",
        "\n",
        "\n",
        "        return(self.rch_image)\n",
        "rch = RCH_diagram()\n",
        "image = rch.draw_rch()\n",
        "cv2_imshow(image)"
      ],
      "execution_count": null,
      "outputs": []
    }
  ]
}