visionpro脚本

#region namespace imports
using System;
using System.Collections;
using System.Drawing;
using System.IO;
using System.Windows.Forms;
using Cognex.VisionPro;
using Cognex.VisionPro.ToolBlock;
using Cognex.VisionPro3D;
using System.Collections.Generic;
using Cognex.GC.DeepLearning;
using ViDi2;
#endregion

public class CogToolBlockAdvancedScript : CogToolBlockAdvancedScriptBase
{
  #region Private Member Variables
  private Cognex.VisionPro.ToolBlock.CogToolBlock mToolBlock;
  #endregion

  /// <summary>
  /// Called when the parent tool is run.
  /// Add code here to customize or replace the normal run behavior.
  /// </summary>
  /// <param name="message">Sets the Message in the tool's RunStatus.</param>
  /// <param name="result">Sets the Result in the tool's RunStatus</param>
  /// <returns>True if the tool should run normally,
  ///          False if GroupRun customizes run behavior</returns>
  public override bool GroupRun(ref string message, ref CogToolResultConstants result)
  {
    // To let the execution stop in this script when a debugger is attached, uncomment the following lines.
    #if DEBUG
    if (System.Diagnostics.Debugger.IsAttached) System.Diagnostics.Debugger.Break();
    #endif

    // Run each tool using the RunTool function
    foreach (ICogTool tool in mToolBlock.Tools)
    {
      //bool isInitializeWork
      if (tool.Name == "InitializeTb")
      {
        CogToolBlock tb_Initial = (CogToolBlock) tool;
        if((bool)tb_Initial.Inputs["isDispose"].Value)
        {
          if(Cognex.GC.DeepLearning.GlobalParameter.control != null)
          {
              Cognex.GC.DeepLearning.Dispose.disposed();
              GlobalParameter.control = null;
          }

          return false;
        }
        if ((bool) tb_Initial.Inputs["isInitializeWorkSpace"].Value)
        {
          //Initialize
          string WorkspacePath = (string) tb_Initial.Inputs["WorkspacePath"].Value;
          string StreamName = (string) tb_Initial.Inputs["StreamName"].Value;
          string ToolName = (string) tb_Initial.Inputs["ToolName"].Value;
          int Gpu_Index = (int) tb_Initial.Inputs["Gpu_Index"].Value;
          Cognex.GC.DeepLearning.Initialize.Initial(WorkspacePath, StreamName, Gpu_Index);
          
          //update parameters
          CogToolBlock tb_Evaluate = mToolBlock.Tools["EvaluateTB"] as CogToolBlock;
          while (tb_Evaluate.Inputs.Count > 0)
          {
            tb_Evaluate.Inputs.RemoveAt(0);
          }
          System.Type type = typeof(Cognex.VisionPro.ICogImage);
          tb_Evaluate.Inputs.Add(new CogToolBlockTerminal("Image", type));

          if (GlobalParameter.stream.Tools[ToolName].Type == ToolType.Blue)
          {
            tb_Evaluate.Inputs.Add(new CogToolBlockTerminal("tool_name", ToolName));
            tb_Evaluate.Inputs.Add(new CogToolBlockTerminal("isGraphicScore", true));
            tb_Evaluate.Inputs.Add(new CogToolBlockTerminal("sampling_Density", -1));
            tb_Evaluate.Inputs.Add(new CogToolBlockTerminal("threshold", -1.0));
            tb_Evaluate.Inputs.Add(new CogToolBlockTerminal("precision", -1.0));
            tb_Evaluate.Inputs.Add(new CogToolBlockTerminal("thickness", 2));
            tb_Evaluate.Inputs.Add(new CogToolBlockTerminal("fontScale", 1));
            tb_Evaluate.Inputs.Add(new CogToolBlockTerminal("isSaveCsvData", true));
            tb_Evaluate.Inputs.Add(new CogToolBlockTerminal("CSVPath", ""));
            tb_Evaluate.Inputs.Add(new CogToolBlockTerminal("isSaveImage", true));
            tb_Evaluate.Inputs.Add(new CogToolBlockTerminal("AnnotationImageSaveFolder", ""));
            tb_Evaluate.Inputs.Add(new CogToolBlockTerminal("AnnotationImageName", ""));

          }
          if (GlobalParameter.stream.Tools[ToolName].Type == ToolType.BlueRead)
          {
            tb_Evaluate.Inputs.Add(new CogToolBlockTerminal("tool_name", ToolName));
            tb_Evaluate.Inputs.Add(new CogToolBlockTerminal("thickness", 2));
            tb_Evaluate.Inputs.Add(new CogToolBlockTerminal("sampling_Density", -1));
            tb_Evaluate.Inputs.Add(new CogToolBlockTerminal("threshold", -1.0));
            tb_Evaluate.Inputs.Add(new CogToolBlockTerminal("isGraphicScore", true));
            tb_Evaluate.Inputs.Add(new CogToolBlockTerminal("fontScale", 1));
            tb_Evaluate.Inputs.Add(new CogToolBlockTerminal("isSaveCsvData", true));
            tb_Evaluate.Inputs.Add(new CogToolBlockTerminal("CSVPath", ""));
            tb_Evaluate.Inputs.Add(new CogToolBlockTerminal("isSaveImage", true));
            tb_Evaluate.Inputs.Add(new CogToolBlockTerminal("AnnotationImageSaveFolder", ""));
            tb_Evaluate.Inputs.Add(new CogToolBlockTerminal("AnnotationImageName", ""));
          }
          if (GlobalParameter.stream.Tools[ToolName].Type == ToolType.Green || GlobalParameter.stream.Tools[ToolName].Type == ToolType.GreenHighDetail)
          {
            tb_Evaluate.Inputs.Add(new CogToolBlockTerminal("tool_name", ToolName));
            tb_Evaluate.Inputs.Add(new CogToolBlockTerminal("threshold", -1.0));
            tb_Evaluate.Inputs.Add(new CogToolBlockTerminal("fontScale", 1));
            tb_Evaluate.Inputs.Add(new CogToolBlockTerminal("thickness", 2));
            tb_Evaluate.Inputs.Add(new CogToolBlockTerminal("isGraphicScore", true));
            tb_Evaluate.Inputs.Add(new CogToolBlockTerminal("isGraphicHeatmap", true));
            tb_Evaluate.Inputs.Add(new CogToolBlockTerminal("CSVPath", ""));
            tb_Evaluate.Inputs.Add(new CogToolBlockTerminal("isSaveCsvData", true));
            tb_Evaluate.Inputs.Add(new CogToolBlockTerminal("isSaveImage", true));
            tb_Evaluate.Inputs.Add(new CogToolBlockTerminal("AnnotationImageSaveFolder", ""));
            tb_Evaluate.Inputs.Add(new CogToolBlockTerminal("AnnotationImageName", ""));
          }
          if (GlobalParameter.stream.Tools[ToolName].Type == ToolType.Red || GlobalParameter.stream.Tools[ToolName].Type == ToolType.RedHighDetail)
          {
            tb_Evaluate.Inputs.Add(new CogToolBlockTerminal("tool_name", ToolName));
            tb_Evaluate.Inputs.Add(new CogToolBlockTerminal("sampling_Density", -1));
            tb_Evaluate.Inputs.Add(new CogToolBlockTerminal("thresholdLow", -1.0));
            tb_Evaluate.Inputs.Add(new CogToolBlockTerminal("thresholdUp", -1.0));
            tb_Evaluate.Inputs.Add(new CogToolBlockTerminal("thickness", 2));
            tb_Evaluate.Inputs.Add(new CogToolBlockTerminal("fontScale", 1));
            tb_Evaluate.Inputs.Add(new CogToolBlockTerminal("isGraphicArea", true));
            tb_Evaluate.Inputs.Add(new CogToolBlockTerminal("isGraphicBoundingRect", true));
            tb_Evaluate.Inputs.Add(new CogToolBlockTerminal("isGraphicContours", true));
            tb_Evaluate.Inputs.Add(new CogToolBlockTerminal("isGraphicOverlayImage", true));
            tb_Evaluate.Inputs.Add(new CogToolBlockTerminal("isSaveCsvData", true));
            tb_Evaluate.Inputs.Add(new CogToolBlockTerminal("CSVPath", ""));
            tb_Evaluate.Inputs.Add(new CogToolBlockTerminal("isSaveImage", true));
            tb_Evaluate.Inputs.Add(new CogToolBlockTerminal("AnnotationImageSaveFolder", ""));
            tb_Evaluate.Inputs.Add(new CogToolBlockTerminal("AnnotationImageName", ""));
          }
        }
      }
      if (tool.Name == "EvaluateTB")
      {
        CogToolBlock tb_Initial = (CogToolBlock) (mToolBlock.Tools["InitializeTb"]);
        ((CogToolBlock) (mToolBlock.Tools["EvaluateTB"])).Inputs["Image"].Value = mToolBlock.Inputs[0].Value;
        if (!(bool) tb_Initial.Inputs["isInitializeWorkSpace"].Value)
        {
          mToolBlock.RunTool(tool, ref message, ref result);
        }
      }
    }
    return false;
  }

  #region When the Current Run Record is Created
  /// <summary>
  /// Called when the current record may have changed and is being reconstructed
  /// </summary>
  /// <param name="currentRecord">
  /// The new currentRecord is available to be initialized or customized.</param>
  public override void ModifyCurrentRunRecord(Cognex.VisionPro.ICogRecord currentRecord)
  {
  }
  #endregion

  #region When the Last Run Record is Created
  /// <summary>
  /// Called when the last run record may have changed and is being reconstructed
  /// </summary>
  /// <param name="lastRecord">
  /// The new last run record is available to be initialized or customized.</param>
  public override void ModifyLastRunRecord(Cognex.VisionPro.ICogRecord lastRecord)
  {
  }
  #endregion

  #region When the Script is Initialized
  /// <summary>
  /// Perform any initialization required by your script here
  /// </summary>
  /// <param name="host">The host tool</param>
  public override void Initialize(Cognex.VisionPro.ToolGroup.CogToolGroup host)
  {
    // DO NOT REMOVE - Call the base class implementation first - DO NOT REMOVE
    base.Initialize(host);


    // Store a local copy of the script host
    this.mToolBlock = ((Cognex.VisionPro.ToolBlock.CogToolBlock) (host));
  }
  #endregion

}

#region namespace imports
using System;
using System.Collections;
using System.Drawing;
using System.IO;
using System.Windows.Forms;
using Cognex.VisionPro;
using Cognex.VisionPro.ToolBlock;
using Cognex.VisionPro3D;
using ViDi2;
using Cognex.GC.DeepLearning;
using System.Collections.Generic;
using Cognex.VisionPro.ImageProcessing;
#endregion

public class CogToolBlockAdvancedScript : CogToolBlockAdvancedScriptBase
{
  #region Private Member Variables
  private Cognex.VisionPro.ToolBlock.CogToolBlock mToolBlock;
  BlueRead read = new BlueRead();
  BlueLocate locate = new BlueLocate();
  GreenClassify green = new GreenClassify();
  RedAnalyze red = new RedAnalyze();
  #endregion

  /// <summary>
  /// Called when the parent tool is run.
  /// Add code here to customize or replace the normal run behavior.
  /// </summary>
  /// <param name="message">Sets the Message in the tool's RunStatus.</param>
  /// <param name="result">Sets the Result in the tool's RunStatus</param>
  /// <returns>True if the tool should run normally,
  ///          False if GroupRun customizes run behavior</returns>
  public override bool GroupRun(ref string message, ref CogToolResultConstants result)
  {
    // To let the execution stop in this script when a debugger is attached, uncomment the following lines.
    #if DEBUG
    if (System.Diagnostics.Debugger.IsAttached) System.Diagnostics.Debugger.Break();
    #endif


    // Run each tool using the RunTool function
    //    foreach(ICogTool tool in mToolBlock.Tools)
    //      mToolBlock.RunTool(tool, ref message, ref result);

    //Init tool
    BlueRead read = new BlueRead();
    BlueLocate locate = new BlueLocate();
    GreenClassify green = new GreenClassify();
    RedAnalyze red = new RedAnalyze();
    
    //image convert
    ICogImage cogImage = mToolBlock.Inputs["Image"].Value as ICogImage;
    IImage iImage = new ViDi2.VisionPro.Image(cogImage);
    Bitmap bitmap = iImage.Bitmap;
    
    //remove allResult
    while(mToolBlock.Outputs.Count > 0)
      mToolBlock.Outputs.RemoveAt(0);
    

    
    //run tools
    if (GlobalParameter.stream.Tools[(string) mToolBlock.Inputs["tool_name"].Value].Type == ToolType.Blue)
    {
      List<Tuple<List<double>, List<ViDi2.Point>, List<ViDi2.Rect>, IImage>> listResult = new List<Tuple<List<double>, List<ViDi2.Point>, List<ViDi2.Rect>, IImage>>();
      locate.EvaluateWithGraphic(iImage,
        (string) mToolBlock.Inputs["tool_name"].Value,
        ref listResult, 
        ref bitmap,
        (bool) mToolBlock.Inputs["isGraphicScore"].Value,
        (int)mToolBlock.Inputs["sampling_Density"].Value,
        (float) mToolBlock.Inputs["threshold"].Value,
        (float) mToolBlock.Inputs["precision"].Value,
        (int)mToolBlock.Inputs["thickness"].Value,
        (int)mToolBlock.Inputs["fontScale"].Value,
        (bool)mToolBlock.Inputs["isSaveCsvData"].Value,
        (string)mToolBlock.Inputs["CSVPath"].Value,
        (bool)mToolBlock.Inputs["isSaveImage"].Value,
        (string)mToolBlock.Inputs["AnnotationImageSaveFolder"].Value,
        (string)mToolBlock.Inputs["AnnotationImageName"].Value
        );
      mToolBlock.Outputs.Add(new CogToolBlockTerminal("ResultData", listResult));

    }
    if (GlobalParameter.stream.Tools[(string) mToolBlock.Inputs["tool_name"].Value].Type == ToolType.BlueRead)
    {

      List<List<Tuple<double, string, List<ViDi2.Point>>>> listResult = new List<List<Tuple<double, string, List<ViDi2.Point>>>>();
      read.EvaluateWithGraphic(iImage,
        (string) mToolBlock.Inputs["tool_name"].Value,
        ref listResult,
        ref bitmap,
        (int)mToolBlock.Inputs["thickness"].Value,
        (int)mToolBlock.Inputs["sampling_Density"].Value,
        (double) mToolBlock.Inputs["threshold"].Value,
        (bool) mToolBlock.Inputs["isGraphicScore"].Value,
        (int)mToolBlock.Inputs["fontScale"].Value,
        (bool)mToolBlock.Inputs["isSaveCsvData"].Value,
       (string)mToolBlock.Inputs["CSVPath"].Value,
       (bool)mToolBlock.Inputs["isSaveImage"].Value,
        (string)mToolBlock.Inputs["AnnotationImageSaveFolder"].Value,
       (string)mToolBlock.Inputs["AnnotationImageName"].Value
        );
      mToolBlock.Outputs.Add(new CogToolBlockTerminal("ResultData", listResult));

            
    }
    if (GlobalParameter.stream.Tools[(string) mToolBlock.Inputs["tool_name"].Value].Type == ToolType.Green || GlobalParameter.stream.Tools[(string) mToolBlock.Inputs["tool_name"].Value].Type == ToolType.GreenHighDetail)
    {

      List<Tuple<double, string, IImage>> listResult = new List<Tuple<double, string, IImage>>();
      green.EvaluateWithGraphicAndResult(iImage,
        (string) mToolBlock.Inputs["tool_name"].Value,
        ref bitmap,
        ref listResult,
        (double) mToolBlock.Inputs["threshold"].Value,
        (int)mToolBlock.Inputs["thickness"].Value,
        (int)mToolBlock.Inputs["fontScale"].Value,
        (bool) mToolBlock.Inputs["isGraphicScore"].Value,
        (bool) mToolBlock.Inputs["isGraphicHeatmap"].Value,
        (string)mToolBlock.Inputs["CSVPath"].Value,
        (bool)mToolBlock.Inputs["isSaveCsvData"].Value,
        (bool)mToolBlock.Inputs["isSaveImage"].Value,
        (string)mToolBlock.Inputs["AnnotationImageSaveFolder"].Value,
        (string)mToolBlock.Inputs["AnnotationImageName"].Value
        );
      mToolBlock.Outputs.Add(new CogToolBlockTerminal("ResultData", listResult));

   
    }
    if (GlobalParameter.stream.Tools[(string) mToolBlock.Inputs["tool_name"].Value].Type == ToolType.Red || GlobalParameter.stream.Tools[(string) mToolBlock.Inputs["tool_name"].Value].Type == ToolType.RedHighDetail)
    {
      //1.view的分数；2.缺陷的中心点；3.缺陷最小外接矩形的四个角点；4.缺陷的面积；5.缺陷的周长；6.缺陷的轮廓点 7.view image; 8.overlay image;
      List<Tuple<double, List<ViDi2.Point>, List<ViDi2.Point[]>, List<double>, List<double>, List<ViDi2.Point[]>, Tuple<IImage, IRedView>>> listResult = new List<Tuple<double, List<ViDi2.Point>, List<ViDi2.Point[]>, List<double>, List<double>, List<ViDi2.Point[]>, Tuple<IImage, IRedView>>>();
      red.EvaluateWithGraphicAndResult(iImage,
        (string) mToolBlock.Inputs["tool_name"].Value,
        ref listResult,
        ref bitmap,
        (int)mToolBlock.Inputs["sampling_Density"].Value,
        (double) mToolBlock.Inputs["thresholdLow"].Value,
        (double) mToolBlock.Inputs["thresholdUp"].Value,
        (int)mToolBlock.Inputs["thickness"].Value,
        (int)mToolBlock.Inputs["fontScale"].Value,
        (bool) mToolBlock.Inputs["isGraphicArea"].Value,
        (bool) mToolBlock.Inputs["isGraphicBoundingRect"].Value,
        (bool) mToolBlock.Inputs["isGraphicContours"].Value,
        (bool) mToolBlock.Inputs["isGraphicOverlayImage"].Value,
        (bool)mToolBlock.Inputs["isSaveCsvData"].Value,
        (string)mToolBlock.Inputs["CSVPath"].Value,
        (bool)mToolBlock.Inputs["isSaveImage"].Value,
        (string)mToolBlock.Inputs["AnnotationImageSaveFolder"].Value,
        (string)mToolBlock.Inputs["AnnotationImageName"].Value
        );
      mToolBlock.Outputs.Add(new CogToolBlockTerminal("ResultData", listResult));
    }
    
    //add RawImage
    mToolBlock.Outputs.Add(new CogToolBlockTerminal("RawImage", cogImage));
    
    //AddResultImg
    Cognex.VisionPro.CogImage24PlanarColor resultImg = new CogImage24PlanarColor(bitmap);
    mToolBlock.Outputs.Add(new CogToolBlockTerminal("ResultImage", resultImg));

    //Display
    CogIPOneImageTool RawDisplay = mToolBlock.Tools["RawDisplay"] as CogIPOneImageTool;
    RawDisplay.InputImage = cogImage;
    RawDisplay.Run();
    
    CogIPOneImageTool ResultDisplay = mToolBlock.Tools["ResultDisplay"] as CogIPOneImageTool;
    ResultDisplay.InputImage = resultImg;
    ResultDisplay.Run();


    return false;
  }

  #region When the Current Run Record is Created
  /// <summary>
  /// Called when the current record may have changed and is being reconstructed
  /// </summary>
  /// <param name="currentRecord">
  /// The new currentRecord is available to be initialized or customized.</param>
  public override void ModifyCurrentRunRecord(Cognex.VisionPro.ICogRecord currentRecord)
  {
  }
  #endregion

  #region When the Last Run Record is Created
  /// <summary>
  /// Called when the last run record may have changed and is being reconstructed
  /// </summary>
  /// <param name="lastRecord">
  /// The new last run record is available to be initialized or customized.</param>
  public override void ModifyLastRunRecord(Cognex.VisionPro.ICogRecord lastRecord)
  {
  }
  #endregion

  #region When the Script is Initialized
  /// <summary>
  /// Perform any initialization required by your script here
  /// </summary>
  /// <param name="host">The host tool</param>
  public override void Initialize(Cognex.VisionPro.ToolGroup.CogToolGroup host)
  {
    // DO NOT REMOVE - Call the base class implementation first - DO NOT REMOVE
    base.Initialize(host);


    // Store a local copy of the script host
    this.mToolBlock = ((Cognex.VisionPro.ToolBlock.CogToolBlock) (host));
    
  }
  #endregion

}