Single limit/tolerance option now working

infocalc.py

@@ -37,9 +37,9 @@ parser_sim.add_argument('mean', type=float,
 parser_sim.add_argument('stddev', type=float,
                     help="sample standard deviation")
 ## General Arguments
-parser.add_argument('minvalue', type=float,
+parser.add_argument('--lowerbound', type=float,
                     help='Tolerance low limit')
-parser.add_argument('maxvalue', type=float,
+parser.add_argument('--upperbound', type=float,
                     help='Tolerance high limit')
 parser.add_argument('--normalizey', action="store_true",
                     help='Set y-axis to normalized probability density')
@@ -81,10 +81,7 @@ fh.setFormatter(spamformatter)
 logger.addHandler(ch)
 logger.addHandler(fh)
 
-logger.info("Creating infocalc log file %s", logpath)
-
-lowerbound = args.minvalue
-upperbound = args.maxvalue
+logger.debug("Creating infocalc log file %s", logpath)
 
 # seed values for variable scoping
 mean = 0
@@ -105,8 +102,19 @@ elif args.mode == "SIM":
     stddev = args.stddev
     samplesize = args.samplesize
 
+# time to deal with the bounds
+    
 # Delta Degrees of Freedom: ddof=0 for population, ddof=1 for sample std dev
-prob = norm.cdf(upperbound, mean, stddev) - norm.cdf(lowerbound, mean, stddev)
+prob = 0
+if args.upperbound and args.lowerbound:
+    prob = norm.cdf(args.upperbound, mean, stddev) - norm.cdf(args.lowerbound, mean, stddev)
+elif args.upperbound:
+    prob = norm.cdf(args.upperbound, mean, stddev)
+elif args.lowerbound:
+    prob = 1 - norm.cdf(args.lowerbound, mean, stddev)
+else:
+    prob = 1# no bounds set!
+    ##TODO!!!!
 #print("probability: %f", prob)
 info = -np.emath.log2(prob)
 #print("information content: %f bits", info)
@@ -127,20 +135,37 @@ if args.graphinfo:#put info on corner of graph
     # place a text box in upper left in axes coords
     ax.text(0.05, 0.95, textstr, transform=ax.transAxes, fontsize=args.fontsize,
             verticalalignment='top', bbox=props)
+xgraphlimits = {"min": mean-3*stddev, "max": mean+3*stddev}
+if args.lowerbound and xgraphlimits["min"] > args.lowerbound:
+    xgraphlimits["min"] = args.lowerbound
+if args.upperbound and xgraphlimits["max"] < args.upperbound:
+    xgraphlimits["max"] = args.upperbound
 
-x = np.linspace(mean-3*stddev, mean+3*stddev, 500)
+x = np.linspace(xgraphlimits["min"], xgraphlimits["max"], 500)
 y = norm.pdf(x, loc=mean, scale=stddev)
 if args.normalizey:
     y = y * stddev#rescale back to unity area
 plt.axvline(x=mean, color="green", linestyle="dashed", label="mean")
-plt.axvline(lowerbound, color="red")
-plt.axvline(upperbound, color="red")
+if args.lowerbound:
+    plt.axvline(args.lowerbound, color="red")
+if args.upperbound:
+    plt.axvline(args.upperbound, color="red")
 
 plt.plot(x, y, 'b-', label='Normal distribution')
 #yt = scipy.stats.t.pdf(x, len(data)-1, mean, stddev)
 #plt.plot(x, yt, 'g-', label='T Distribution')
-coloredregion = (x >= lowerbound) & ( x <= upperbound ) #select x values
+
+# Filter for which region to fill
+coloredregion = x#default fill all
+if args.lowerbound and args.upperbound:
+    coloredregion = (x >= args.lowerbound) & ( x <= args.upperbound )
+elif args.upperbound:
+    coloredregion =  x <= args.upperbound 
+elif args.lowerbound:
+    coloredregion = x >= args.lowerbound
+    
 plt.fill_between(x, 0, y, where=coloredregion, color="grey", alpha=0.5, label="Design range",)
+    
 if args.xlabel:
     plt.xlabel(args.xlabel)
 plt.ylabel('Probability density')
@@ -150,6 +175,7 @@ if args.legend:
 top = plt.ylim()[1]
 
 if args.outfile:
+    logger.info(f"Graph output to {args.outfile}")
     plt.savefig(args.outfile,bbox_inches='tight')
 else:
     plt.show()